Add create storage endpoint

Author: kossnocorp

.gitignore

@@ -1,6 +1,5 @@
 # Node.js
 node_modules
-yarn-error.log
 
 /lib
 /secrets
@@ -12,4 +11,7 @@ yarn-error.log
 
 # Test projects
 /test/*/build
-/test/*/.firebaserc
+/test/*/.firebaserc
+
+# Firebase
+firebase-debug.log

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@backupfire/firebase",
-  "version": "1.3.0-beta.1",
+  "version": "1.3.0-beta.6",
   "description": "Backup Fire Firebase agent",
   "keywords": [
     "backup Firebase database",

src/storage/index.ts

@@ -1,7 +1,7 @@
 import asyncMiddleware from '../_lib/asyncMiddleware'
 import { Storage as CloudStorage, Bucket } from '@google-cloud/storage'
 
-export type StorageOptions = {
+export interface StorageOptions {
   bucketsAllowlist?: string[]
 }
 
@@ -14,34 +14,72 @@ export function storageListMiddleware({ bucketsAllowlist }: StorageOptions) {
   })
 }
 
+interface StorageRetentionData {
+  removeOldBackups: boolean
+  keepBackupsValue: number
+  keepBackupsUnit: KeepBackupsUnit
+}
+
+type KeepBackupsUnit = 'years' | 'months' | 'days'
+
+export interface CreateStorageRequestBody
+  extends Partial<StorageRetentionData> {
+  storageId: string
+  location: string
+  storageClass?: 'standard' | 'nearline' | 'coldline' | 'archive'
+}
+
 export function createStorageMiddleware({ bucketsAllowlist }: StorageOptions) {
+  const storage = new CloudStorage()
+
   return asyncMiddleware(async (request, response) => {
-    // TODO: Validate options
+    const body = request.body as CreateStorageRequestBody
+
+    const [bucket] = await storage.createBucket(body.storageId, {
+      [body.storageClass || 'nearline']: true,
+      location: body.location
+    })
+
+    if (
+      body.removeOldBackups &&
+      body.keepBackupsValue &&
+      body.keepBackupsUnit
+    ) {
+      const deleteAge = keepBackupInDays(
+        body.keepBackupsValue,
+        body.keepBackupsUnit
+      )
 
-    // await backupUsers(options)
+      if (body.removeOldBackups) {
+        await bucket.addLifecycleRule(
+          {
+            action: { type: 'Delete' },
+            condition: { age: deleteAge }
+          },
+          { append: false }
+        )
+      } else {
+        await bucket.setMetadata({ lifecycle: null })
+      }
+    }
 
-    response.send([])
+    const [bucketData] = await bucket.get()
+    response.send(bucketAsStorage(bucketData))
   })
 }
 
-export type UpdateStorageOptions = {
-  bucketsAllowlist?: string[]
+export interface PasswordedStorageOptions extends StorageOptions {
   adminPassword: string
 }
 
-type UpdateStorageRequestBody = {
+interface UpdateStorageRequestBody extends StorageRetentionData {
   password: string
-  removeOldBackups: boolean
-  keepBackupsValue: number
-  keepBackupsUnit: KeepBackupsUnit
 }
 
-type KeepBackupsUnit = 'years' | 'months' | 'days'
-
 export function updateStorageMiddleware({
   bucketsAllowlist,
   adminPassword
-}: UpdateStorageOptions) {
+}: PasswordedStorageOptions) {
   return asyncMiddleware(async (request, response) => {
     // TODO: Validate options
     const storageId = request.params.storageId as string
@@ -73,15 +111,10 @@ export function updateStorageMiddleware({
     }
 
     const [bucketData] = await bucket.get()
-
     response.send(bucketAsStorage(bucketData))
   })
 }
 
-export interface ListFilesOptions {
-  bucketsAllowlist?: string[]
-}
-
 export interface ListFilesRequestQuery {
   path: string
   maxResults: string | undefined
@@ -90,12 +123,12 @@ export interface ListFilesRequestQuery {
   prefix: string | undefined
 }
 
-export function listFilesMiddleware({ bucketsAllowlist }: ListFilesOptions) {
+export function listFilesMiddleware({ bucketsAllowlist }: StorageOptions) {
   const storage = new CloudStorage()
 
   return asyncMiddleware(async (request, response) => {
     const storageId = request.params.storageId as string
-    const query = request.body as ListFilesRequestQuery
+    const query = (request.query as unknown) as ListFilesRequestQuery
 
     const bucket = storage.bucket(storageId)
     const [files] = await bucket.getFiles({
@@ -163,7 +196,7 @@ export function keepBackupInDays(value: number, unit: KeepBackupsUnit) {
   }
 }
 
-export type Storage = {
+export interface Storage {
   name: string
   id: string
   location: StorageLocation
@@ -175,7 +208,7 @@ export type Storage = {
   lifecycle: LifecycleRule[]
 }
 
-type LifecycleRule = {
+interface LifecycleRule {
   action: { type: string; storageClass?: string } | string
   condition: { [key: string]: boolean | Date | number | string }
   storageClass?: string