feat(file): add google drive support

Author: Efk3

src/app/file/actions/file.action.ts

@@ -25,6 +25,8 @@ export enum FileActionTypes {
   RemoveFile = '[FILE] Remove File',
   RemoveFileSuccess = '[FILE] Remove File Success',
   RemoveFileFail = '[FILE] Remove File Fail',
+
+  Clear = '[FILE] Clear File Store',
 }
 
 /**
@@ -54,16 +56,19 @@ export class GetFilesSuccess implements Action {
  */
 export class GetFile implements Action {
   public readonly type = FileActionTypes.GetFile;
+
   constructor(public payload: BudgetFile) {}
 }
 
 export class GetFileSuccess implements Action {
   public readonly type = FileActionTypes.GetFileSuccess;
+
   constructor(public payload: { file: BudgetFile; content: string }) {}
 }
 
 export class GetFileFail implements Action {
   public readonly type = FileActionTypes.GetFileFail;
+
   constructor(public payload: string) {}
 }
 
@@ -151,6 +156,13 @@ export class RemoveFileFail implements Action {
   constructor(public payload: BudgetFile) {}
 }
 
+export class Clear implements Action {
+  public readonly type = FileActionTypes.Clear;
+  public payload: any;
+
+  constructor() {}
+}
+
 export type FileActions =
   | GetFiles
   | GetFilesSuccess
@@ -169,4 +181,5 @@ export type FileActions =
   | RenameFileFail
   | RemoveFile
   | RemoveFileSuccess
-  | RemoveFileFail;
+  | RemoveFileFail
+  | Clear;

src/app/file/reducers/file.reducer.ts

@@ -73,6 +73,10 @@ export function reducer(state = initialState, action: FileActions): FileState {
       );
     }
 
+    case FileActionTypes.Clear: {
+      return initialState;
+    }
+
     default: {
       return state;
     }

src/app/file/services/dropbox-file-handler.ts

@@ -1,6 +1,8 @@
 import { HttpClient } from '@angular/common/http';
 import { Store } from '@ngrx/store';
 import { Dropbox } from 'dropbox';
+import { Observable } from 'rxjs/Observable';
+import { of } from 'rxjs/observable/of';
 import {
   CreateFileFail,
   CreateFileSuccess,
@@ -21,6 +23,7 @@ import { FileHandler } from './file-handler.interface';
 import FileMetadataReference = DropboxTypes.files.FileMetadataReference;
 
 export class DropboxFileHandler implements FileHandler {
+  public ready: Observable<boolean> = of(true);
   private client: Dropbox;
 
   constructor(private http: HttpClient, private fileStore: Store<FileState>, accessToken: string) {

src/app/file/services/file-handler.interface.ts

@@ -1,6 +1,9 @@
+import { Observable } from 'rxjs/Observable';
 import { BudgetFile } from '../models/budget-file.model';
 
 export interface FileHandler {
+  ready: Observable<boolean>;
+
   list();
 
   get(file: BudgetFile);

src/app/file/services/file.service.ts

@@ -2,15 +2,18 @@ import { HttpClient } from '@angular/common/http';
 import { Injectable } from '@angular/core';
 import { Actions, Effect } from '@ngrx/effects';
 import { Store } from '@ngrx/store';
-import { tap } from 'rxjs/operators';
+import { of } from 'rxjs/observable/of';
+import { switchMap, takeWhile, tap } from 'rxjs/operators';
+import { AuthActionTypes } from '../../auth/actions/auth';
 import { AuthenticationType } from '../../auth/models/authentication-type.model';
 import { getAuthenticationInfo } from '../../auth/reducers/auth.selector';
 import { AuthState } from '../../auth/reducers/auth.state';
-import { CreateFile, FileActionTypes, GetFile, GetFiles, RemoveFile, RenameFile, SaveFile } from '../actions/file.action';
+import { Clear, CreateFile, FileActionTypes, GetFile, RemoveFile, RenameFile, SaveFile } from '../actions/file.action';
 import { BudgetFile } from '../models/budget-file.model';
 import { FileState } from '../reducers/file.state';
 import { DropboxFileHandler } from './dropbox-file-handler';
 import { FileHandler } from './file-handler.interface';
+import { GoogleFileHandler } from './google-file-handler';
 
 @Injectable()
 export class FileService {
@@ -36,7 +39,13 @@ export class FileService {
     .ofType<RenameFile>(FileActionTypes.RenameFile)
     .pipe(tap(action => this.rename(action.payload.file, action.payload.newName)));
 
+  @Effect()
+  private logoutEffect = this.actions
+    .ofType(AuthActionTypes.LogoutSuccess, AuthActionTypes.LogoutFail)
+    .pipe(switchMap(() => of(new Clear())));
+
   private fileHandler: FileHandler;
+  private ready: boolean = false;
 
   constructor(
     private authStore: Store<AuthState>,
@@ -46,61 +55,76 @@ export class FileService {
   ) {
     this.authStore.select(getAuthenticationInfo).subscribe(authenticationInfo => {
       if (authenticationInfo.accessToken) {
-        if (authenticationInfo.type === AuthenticationType.DROPBOX) {
-          this.fileHandler = new DropboxFileHandler(this.http, this.fileStore, authenticationInfo.accessToken);
+        switch (authenticationInfo.type) {
+          case AuthenticationType.DROPBOX:
+            this.fileHandler = new DropboxFileHandler(this.http, this.fileStore, authenticationInfo.accessToken);
+            break;
+          case AuthenticationType.GOOGLE:
+            this.fileHandler = new GoogleFileHandler(this.http, this.fileStore, authenticationInfo.accessToken);
+            break;
+        }
+
+        if (this.fileHandler) {
+          this.fileHandler.ready.pipe(takeWhile(ready => !ready)).subscribe(null, null, () => {
+            this.ready = true;
+            this.fileHandler.list();
+          });
         }
-        this.fileStore.dispatch(new GetFiles());
       } else {
         this.fileHandler = undefined;
       }
     });
   }
 
   private list() {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.list();
   }
 
   private get(file: BudgetFile) {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.get(file);
   }
 
   private create(name: string) {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.create(name);
   }
 
   private save(file: BudgetFile, newContent: string) {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.save(file, newContent);
   }
 
   private remove(file: BudgetFile) {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.remove(file);
   }
 
   private rename(file: BudgetFile, newName: string) {
-    if (!this.fileHandler) {
+    if (!this.isFileHandlerReady()) {
       return;
     }
 
     this.fileHandler.rename(file, newName);
   }
+
+  private isFileHandlerReady() {
+    return this.fileHandler && this.ready;
+  }
 }

src/app/file/services/google-file-handler.ts

@@ -0,0 +1,158 @@
+import { HttpClient, HttpHeaders } from '@angular/common/http';
+import { Store } from '@ngrx/store';
+import { BehaviorSubject } from 'rxjs/BehaviorSubject';
+import { Observable } from 'rxjs/Observable';
+import {
+  CreateFileFail,
+  CreateFileSuccess,
+  GetFileFail,
+  GetFilesSuccess,
+  GetFileSuccess,
+  RemoveFileFail,
+  RemoveFileSuccess,
+  RenameFileFail,
+  RenameFileSuccess,
+  SaveFileFail,
+  SaveFileSuccess,
+} from '../actions/file.action';
+import { BudgetFile } from '../models/budget-file.model';
+import { FileState } from '../reducers/file.state';
+import { FileHandler } from './file-handler.interface';
+
+export class GoogleFileHandler implements FileHandler {
+  public ready: Observable<boolean>;
+  private readonly directoryName: string = 'K3 Budget';
+  private directoryId: string;
+  private readySubject = new BehaviorSubject<boolean>(false);
+
+  constructor(private http: HttpClient, private fileStore: Store<FileState>, private accessToken: string) {
+    this.ready = this.readySubject.asObservable();
+    this.getDirectoryId();
+  }
+
+  public async list() {
+    try {
+      const result = await this.http
+        .get<any>(`https://www.googleapis.com/drive/v3/files`, {
+          params: { pageSize: '999', fields: 'files(id, name)', q: `"${this.directoryId}" in parents` },
+          headers: this.getHeader(),
+        })
+        .toPromise();
+
+      this.fileStore.dispatch(
+        new GetFilesSuccess({
+          files: result.files.map(entry => {
+            return { id: entry.id, name: entry.name };
+          }),
+        })
+      );
+    } catch (err) {
+      // @TODO: handle error
+    }
+  }
+
+  public async get(file: BudgetFile) {
+    try {
+      const content = await this.http
+        .get(`https://www.googleapis.com/drive/v3/files/${file.id}`, {
+          responseType: 'text',
+          params: { alt: 'media' },
+          headers: this.getHeader(),
+        })
+        .toPromise();
+
+      this.fileStore.dispatch(new GetFileSuccess({ file, content }));
+    } catch (err) {
+      this.fileStore.dispatch(new GetFileFail('Can not download file'));
+    }
+  }
+
+  public async create(name: string) {
+    try {
+      const file = await this.http
+        .post<any>(
+          `https://www.googleapis.com/drive/v3/files`,
+          { name, parents: [this.directoryId], fields: 'id' },
+          { headers: this.getHeader() }
+        )
+        .toPromise();
+
+      this.fileStore.dispatch(new CreateFileSuccess(new BudgetFile(file.id, name)));
+    } catch (err) {
+      this.fileStore.dispatch(new CreateFileFail(name));
+    }
+  }
+
+  public async remove(file: BudgetFile) {
+    try {
+      await this.http.delete(`https://www.googleapis.com/drive/v3/files/${file.id}`, { headers: this.getHeader() }).toPromise();
+
+      this.fileStore.dispatch(new RemoveFileSuccess(file));
+    } catch (err) {
+      this.fileStore.dispatch(new RemoveFileFail(file));
+    }
+  }
+
+  public async rename(file: BudgetFile, newName: string) {
+    try {
+      await this.http
+        .patch(`https://www.googleapis.com/drive/v3/files/${file.id}`, { name: newName }, { headers: this.getHeader() })
+        .toPromise();
+
+      this.fileStore.dispatch(new RenameFileSuccess({ file, newName }));
+    } catch (err) {
+      this.fileStore.dispatch(new RenameFileFail({ file, newName }));
+    }
+  }
+
+  public async save(file: BudgetFile, newContent: string) {
+    try {
+      await this.http
+        .patch(`https://www.googleapis.com/upload/drive/v3/files/${file.id}`, newContent, {
+          params: { uploadType: 'media' },
+          headers: this.getHeader(),
+        })
+        .toPromise();
+
+      this.fileStore.dispatch(new SaveFileSuccess(file));
+    } catch (err) {
+      this.fileStore.dispatch(new SaveFileFail(file));
+    }
+  }
+
+  private async getDirectoryId() {
+    try {
+      // check budget directory
+      const result = await this.http
+        .get<any>(`https://www.googleapis.com/drive/v3/files`, {
+          params: { pageSize: '1', fields: 'files(id)', q: `name = "${this.directoryName}" and trashed = false` },
+          headers: this.getHeader(),
+        })
+        .toPromise();
+
+      if (result.files.length === 1) {
+        // found
+        this.directoryId = result.files[0].id;
+      } else {
+        // directory does not exist, create it
+        const createDirectory = await this.http
+          .post<any>(
+            `https://www.googleapis.com/drive/v3/files`,
+            { name: this.directoryName, mimeType: 'application/vnd.google-apps.folder', fields: 'id' },
+            { headers: this.getHeader() }
+          )
+          .toPromise();
+
+        this.directoryId = createDirectory.id;
+      }
+
+      this.readySubject.next(true);
+    } catch (err) {
+      // @TODO: handle error
+    }
+  }
+
+  private getHeader() {
+    return new HttpHeaders().set('Authorization', 'Bearer ' + this.accessToken);
+  }
+}