src/services/auth-service/auth-service.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { User } from "~/models/user";
   6 import { AxiosInstance } from "axios";
   7 import { ApiActions, ProgressFn } from "~/services/api/api-actions";
   8 import * as uuid from "uuid/v4";
   9
  10 export const API_TOKEN_KEY = 'apiToken';
  11 export const USER_EMAIL_KEY = 'userEmail';
  12 export const USER_FIRST_NAME_KEY = 'userFirstName';
  13 export const USER_LAST_NAME_KEY = 'userLastName';
  14 export const USER_UUID_KEY = 'userUuid';
  15 export const USER_OWNER_UUID_KEY = 'userOwnerUuid';
  16 export const USER_IS_ADMIN = 'userIsAdmin';
  17
  18 export interface UserDetailsResponse {
  19     email: string;
  20     first_name: string;
  21     last_name: string;
  22     uuid: string;
  23     owner_uuid: string;
  24     is_admin: boolean;
  25 }
  26
  27 export class AuthService {
  28
  29     constructor(
  30         protected apiClient: AxiosInstance,
  31         protected baseUrl: string,
  32         protected actions: ApiActions) { }
  33
  34     public saveApiToken(token: string) {
  35         localStorage.setItem(API_TOKEN_KEY, token);
  36     }
  37
  38     public removeApiToken() {
  39         localStorage.removeItem(API_TOKEN_KEY);
  40     }
  41
  42     public getApiToken() {
  43         return localStorage.getItem(API_TOKEN_KEY) || undefined;
  44     }
  45
  46     public getUuid() {
  47         return localStorage.getItem(USER_UUID_KEY) || undefined;
  48     }
  49
  50     public getOwnerUuid() {
  51         return localStorage.getItem(USER_OWNER_UUID_KEY) || undefined;
  52     }
  53
  54     public getUser(): User | undefined {
  55         const email = localStorage.getItem(USER_EMAIL_KEY);
  56         const firstName = localStorage.getItem(USER_FIRST_NAME_KEY);
  57         const lastName = localStorage.getItem(USER_LAST_NAME_KEY);
  58         const uuid = localStorage.getItem(USER_UUID_KEY);
  59         const ownerUuid = localStorage.getItem(USER_OWNER_UUID_KEY);
  60         const isAdmin = Boolean(localStorage.getItem(USER_IS_ADMIN));
  61
  62         return email && firstName && lastName && uuid && ownerUuid && isAdmin
  63             ? { email, firstName, lastName, uuid, ownerUuid, isAdmin }
  64             : undefined;
  65     }
  66
  67     public saveUser(user: User) {
  68         localStorage.setItem(USER_EMAIL_KEY, user.email);
  69         localStorage.setItem(USER_FIRST_NAME_KEY, user.firstName);
  70         localStorage.setItem(USER_LAST_NAME_KEY, user.lastName);
  71         localStorage.setItem(USER_UUID_KEY, user.uuid);
  72         localStorage.setItem(USER_OWNER_UUID_KEY, user.ownerUuid);
  73         localStorage.setItem(USER_IS_ADMIN, JSON.stringify(user.isAdmin));
  74     }
  75
  76     public removeUser() {
  77         localStorage.removeItem(USER_EMAIL_KEY);
  78         localStorage.removeItem(USER_FIRST_NAME_KEY);
  79         localStorage.removeItem(USER_LAST_NAME_KEY);
  80         localStorage.removeItem(USER_UUID_KEY);
  81         localStorage.removeItem(USER_OWNER_UUID_KEY);
  82     }
  83
  84     public login() {
  85         const currentUrl = `${window.location.protocol}//${window.location.host}/token`;
  86         window.location.assign(`${this.baseUrl || ""}/login?return_to=${currentUrl}`);
  87     }
  88
  89     public logout() {
  90         const currentUrl = `${window.location.protocol}//${window.location.host}`;
  91         window.location.assign(`${this.baseUrl || ""}/logout?return_to=${currentUrl}`);
  92     }
  93
  94     public getUserDetails = (): Promise<User> => {
  95         const reqId = uuid();
  96         this.actions.progressFn(reqId, true);
  97         return this.apiClient
  98             .get<UserDetailsResponse>('/users/current')
  99             .then(resp => {
 100                 this.actions.progressFn(reqId, false);
 101                 return {
 102                     email: resp.data.email,
 103                     firstName: resp.data.first_name,
 104                     lastName: resp.data.last_name,
 105                     uuid: resp.data.uuid,
 106                     ownerUuid: resp.data.owner_uuid,
 107                     isAdmin: resp.data.is_admin
 108                 };
 109             })
 110             .catch(e => {
 111                 this.actions.progressFn(reqId, false);
 112                 this.actions.errorFn(reqId, e);
 113                 throw e;
 114             });
 115     }
 116
 117     public getRootUuid() {
 118         const uuid = this.getOwnerUuid();
 119         const uuidParts = uuid ? uuid.split('-') : [];
 120         return uuidParts.length > 1 ? `${uuidParts[0]}-${uuidParts[1]}` : undefined;
 121     }
 122 }