src/store/virtual-machines/virtual-machines-actions.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { Dispatch } from "redux";
   6 import { RootState } from 'store/store';
   7 import { ServiceRepository } from "services/services";
   8 import { navigateToUserVirtualMachines, navigateToAdminVirtualMachines, navigateToRootProject } from "store/navigation/navigation-action";
   9 import { bindDataExplorerActions } from 'store/data-explorer/data-explorer-action';
  10 import { formatDate } from "common/formatters";
  11 import { unionize, ofType, UnionOf } from "common/unionize";
  12 import { VirtualMachineLogins } from 'models/virtual-machines';
  13 import { FilterBuilder } from "services/api/filter-builder";
  14 import { ListResults } from "services/common-service/common-service";
  15 import { dialogActions } from 'store/dialog/dialog-actions';
  16 import { snackbarActions, SnackbarKind } from 'store/snackbar/snackbar-actions';
  17 import { PermissionLevel } from "models/permission";
  18 import { deleteResources, updateResources } from 'store/resources/resources-actions';
  19 import { Participant } from "views-components/sharing-dialog/participant-select";
  20 import { initialize, reset } from "redux-form";
  21 import { getUserDisplayName, UserResource } from "models/user";
  22
  23 export const virtualMachinesActions = unionize({
  24     SET_REQUESTED_DATE: ofType<string>(),
  25     SET_VIRTUAL_MACHINES: ofType<ListResults<any>>(),
  26     SET_LOGINS: ofType<VirtualMachineLogins>(),
  27     SET_LINKS: ofType<ListResults<any>>()
  28 });
  29
  30 export type VirtualMachineActions = UnionOf<typeof virtualMachinesActions>;
  31
  32 export const VIRTUAL_MACHINES_PANEL = 'virtualMachinesPanel';
  33 export const VIRTUAL_MACHINE_ATTRIBUTES_DIALOG = 'virtualMachineAttributesDialog';
  34 export const VIRTUAL_MACHINE_REMOVE_DIALOG = 'virtualMachineRemoveDialog';
  35 export const VIRTUAL_MACHINE_ADD_LOGIN_DIALOG = 'virtualMachineAddLoginDialog';
  36 export const VIRTUAL_MACHINE_ADD_LOGIN_FORM = 'virtualMachineAddLoginForm';
  37 export const VIRTUAL_MACHINE_REMOVE_LOGIN_DIALOG = 'virtualMachineRemoveLoginDialog';
  38
  39 export const VIRTUAL_MACHINE_UPDATE_LOGIN_UUID_FIELD = 'uuid';
  40 export const VIRTUAL_MACHINE_ADD_LOGIN_VM_FIELD = 'vmUuid';
  41 export const VIRTUAL_MACHINE_ADD_LOGIN_USER_FIELD = 'user';
  42 export const VIRTUAL_MACHINE_ADD_LOGIN_GROUPS_FIELD = 'groups';
  43 export const VIRTUAL_MACHINE_ADD_LOGIN_EXCLUDE = 'excludedPerticipants';
  44
  45 export const openUserVirtualMachines = () =>
  46     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
  47         dispatch<any>(navigateToUserVirtualMachines);
  48     };
  49
  50 export const openAdminVirtualMachines = () =>
  51     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
  52         const user = getState().auth.user;
  53         if (user && user.isAdmin) {
  54             dispatch<any>(navigateToAdminVirtualMachines);
  55         } else {
  56             dispatch<any>(navigateToRootProject);
  57             dispatch(snackbarActions.OPEN_SNACKBAR({ message: "You don't have permissions to view this page", hideDuration: 2000, kind: SnackbarKind.ERROR }));
  58         }
  59     };
  60
  61 export const openVirtualMachineAttributes = (uuid: string) =>
  62     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
  63         const virtualMachineData = getState().virtualMachines.virtualMachines.items.find(it => it.uuid === uuid);
  64         dispatch(dialogActions.OPEN_DIALOG({ id: VIRTUAL_MACHINE_ATTRIBUTES_DIALOG, data: { virtualMachineData } }));
  65     };
  66
  67 const loadRequestedDate = () =>
  68     (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
  69         const date = services.virtualMachineService.getRequestedDate();
  70         dispatch(virtualMachinesActions.SET_REQUESTED_DATE(date));
  71     };
  72
  73 export const loadVirtualMachinesAdminData = () =>
  74     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
  75         dispatch<any>(loadRequestedDate());
  76
  77         const virtualMachines = await services.virtualMachineService.list();
  78         dispatch(updateResources(virtualMachines.items));
  79         dispatch(virtualMachinesActions.SET_VIRTUAL_MACHINES(virtualMachines));
  80
  81
  82         const logins = await services.permissionService.list({
  83             filters: new FilterBuilder()
  84             .addIn('head_uuid', virtualMachines.items.map(item => item.uuid))
  85             .addEqual('name', PermissionLevel.CAN_LOGIN)
  86             .getFilters(),
  87             limit: 1000
  88         });
  89         dispatch(updateResources(logins.items));
  90         dispatch(virtualMachinesActions.SET_LINKS(logins));
  91
  92         const users = await services.userService.list({
  93             filters: new FilterBuilder()
  94             .addIn('uuid', logins.items.map(item => item.tailUuid))
  95             .getFilters(),
  96             count: "none", // Necessary for federated queries
  97             limit: 1000
  98         });
  99         dispatch(updateResources(users.items));
 100
 101         const getAllLogins = await services.virtualMachineService.getAllLogins();
 102         dispatch(virtualMachinesActions.SET_LOGINS(getAllLogins));
 103     };
 104
 105 export const loadVirtualMachinesUserData = () =>
 106     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 107         dispatch<any>(loadRequestedDate());
 108         const virtualMachines = await services.virtualMachineService.list();
 109         const virtualMachinesUuids = virtualMachines.items.map(it => it.uuid);
 110         const links = await services.linkService.list({
 111             filters: new FilterBuilder()
 112                 .addIn("head_uuid", virtualMachinesUuids)
 113                 .getFilters()
 114         });
 115         dispatch(virtualMachinesActions.SET_VIRTUAL_MACHINES(virtualMachines));
 116         dispatch(virtualMachinesActions.SET_LINKS(links));
 117     };
 118
 119 export const openAddVirtualMachineLoginDialog = (vmUuid: string) =>
 120     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 121         // Get login permissions of vm
 122         const virtualMachines = await services.virtualMachineService.list();
 123         dispatch(updateResources(virtualMachines.items));
 124         const logins = await services.permissionService.list({
 125             filters: new FilterBuilder()
 126             .addIn('head_uuid', virtualMachines.items.map(item => item.uuid))
 127             .addEqual('name', PermissionLevel.CAN_LOGIN)
 128             .getFilters()
 129         });
 130         dispatch(updateResources(logins.items));
 131
 132         dispatch(initialize(VIRTUAL_MACHINE_ADD_LOGIN_FORM, {
 133                 [VIRTUAL_MACHINE_ADD_LOGIN_VM_FIELD]: vmUuid,
 134                 [VIRTUAL_MACHINE_ADD_LOGIN_GROUPS_FIELD]: [],
 135             }));
 136         dispatch(dialogActions.OPEN_DIALOG( {id: VIRTUAL_MACHINE_ADD_LOGIN_DIALOG, data: {excludedParticipants: logins.items.map(it => it.tailUuid)}} ));
 137     }
 138
 139 export const openEditVirtualMachineLoginDialog = (permissionUuid: string) =>
 140     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 141         const login = await services.permissionService.get(permissionUuid);
 142         const user = await services.userService.get(login.tailUuid);
 143         dispatch(initialize(VIRTUAL_MACHINE_ADD_LOGIN_FORM, {
 144                 [VIRTUAL_MACHINE_UPDATE_LOGIN_UUID_FIELD]: permissionUuid,
 145                 [VIRTUAL_MACHINE_ADD_LOGIN_USER_FIELD]: {name: getUserDisplayName(user, true, true), uuid: login.tailUuid},
 146                 [VIRTUAL_MACHINE_ADD_LOGIN_GROUPS_FIELD]: login.properties.groups,
 147             }));
 148         dispatch(dialogActions.OPEN_DIALOG( {id: VIRTUAL_MACHINE_ADD_LOGIN_DIALOG, data: {updating: true}} ));
 149     }
 150
 151 export interface AddLoginFormData {
 152     [VIRTUAL_MACHINE_UPDATE_LOGIN_UUID_FIELD]: string;
 153     [VIRTUAL_MACHINE_ADD_LOGIN_VM_FIELD]: string;
 154     [VIRTUAL_MACHINE_ADD_LOGIN_USER_FIELD]: Participant;
 155     [VIRTUAL_MACHINE_ADD_LOGIN_GROUPS_FIELD]: string[];
 156 }
 157
 158
 159 export const addUpdateVirtualMachineLogin = ({uuid, vmUuid, user, groups}: AddLoginFormData) =>
 160     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 161         let userResource: UserResource | undefined = undefined;
 162         try {
 163             // Get user
 164             userResource = await services.userService.get(user.uuid, false);
 165         } catch (e) {
 166                 dispatch(snackbarActions.OPEN_SNACKBAR({ message: "Failed to get user details.", hideDuration: 2000, kind: SnackbarKind.ERROR }));
 167                 return;
 168         }
 169         try {
 170             if (uuid) {
 171                 const permission = await services.permissionService.update(uuid, {
 172                     tailUuid: userResource.uuid,
 173                     name: PermissionLevel.CAN_LOGIN,
 174                     properties: {
 175                         username: userResource.username,
 176                         groups,
 177                     }
 178                 });
 179                 dispatch(updateResources([permission]));
 180             } else {
 181                 const permission = await services.permissionService.create({
 182                     headUuid: vmUuid,
 183                     tailUuid: userResource.uuid,
 184                     name: PermissionLevel.CAN_LOGIN,
 185                     properties: {
 186                         username: userResource.username,
 187                         groups,
 188                     }
 189                 });
 190                 dispatch(updateResources([permission]));
 191             }
 192
 193             dispatch(reset(VIRTUAL_MACHINE_ADD_LOGIN_FORM));
 194             dispatch(dialogActions.CLOSE_DIALOG({ id: VIRTUAL_MACHINE_ADD_LOGIN_DIALOG }));
 195             dispatch<any>(loadVirtualMachinesAdminData());
 196
 197             dispatch(snackbarActions.OPEN_SNACKBAR({
 198                 message: `Permission updated`,
 199                 kind: SnackbarKind.SUCCESS
 200             }));
 201         } catch (e) {
 202             dispatch(snackbarActions.OPEN_SNACKBAR({ message: e.message, hideDuration: 2000, kind: SnackbarKind.ERROR }));
 203         }
 204     };
 205
 206 export const openRemoveVirtualMachineLoginDialog = (uuid: string) =>
 207     (dispatch: Dispatch, getState: () => RootState) => {
 208         dispatch(dialogActions.OPEN_DIALOG({
 209             id: VIRTUAL_MACHINE_REMOVE_LOGIN_DIALOG,
 210             data: {
 211                 title: 'Remove login permission',
 212                 text: 'Are you sure you want to remove this permission?',
 213                 confirmButtonLabel: 'Remove',
 214                 uuid
 215             }
 216         }));
 217     };
 218
 219 export const removeVirtualMachineLogin = (uuid: string) =>
 220     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 221         try {
 222             await services.permissionService.delete(uuid);
 223             dispatch<any>(deleteResources([uuid]));
 224
 225             dispatch<any>(loadVirtualMachinesAdminData());
 226
 227             dispatch(snackbarActions.OPEN_SNACKBAR({
 228                 message: `Login permission removed`,
 229                 kind: SnackbarKind.SUCCESS
 230             }));
 231         } catch (e) {
 232             dispatch(snackbarActions.OPEN_SNACKBAR({ message: e.message, hideDuration: 2000, kind: SnackbarKind.ERROR }));
 233         }
 234     };
 235
 236 export const saveRequestedDate = () =>
 237     (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 238         const date = formatDate((new Date()).toISOString());
 239         services.virtualMachineService.saveRequestedDate(date);
 240         dispatch<any>(loadRequestedDate());
 241     };
 242
 243 export const openRemoveVirtualMachineDialog = (uuid: string) =>
 244     (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 245         dispatch(dialogActions.OPEN_DIALOG({
 246             id: VIRTUAL_MACHINE_REMOVE_DIALOG,
 247             data: {
 248                 title: 'Remove virtual machine',
 249                 text: 'Are you sure you want to remove this virtual machine?',
 250                 confirmButtonLabel: 'Remove',
 251                 uuid
 252             }
 253         }));
 254     };
 255
 256 export const removeVirtualMachine = (uuid: string) =>
 257     async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
 258         dispatch(snackbarActions.OPEN_SNACKBAR({ message: 'Removing ...', kind: SnackbarKind.INFO }));
 259         await services.virtualMachineService.delete(uuid);
 260         dispatch(snackbarActions.OPEN_SNACKBAR({ message: 'Removed.', hideDuration: 2000, kind: SnackbarKind.SUCCESS }));
 261         dispatch<any>(loadVirtualMachinesAdminData());
 262     };
 263
 264 const virtualMachinesBindedActions = bindDataExplorerActions(VIRTUAL_MACHINES_PANEL);
 265
 266 export const loadVirtualMachinesPanel = () =>
 267     (dispatch: Dispatch) => {
 268         dispatch(virtualMachinesBindedActions.REQUEST_ITEMS());
 269     };