Merge pull request #4002 from uselagoon/get-file-backup-resolvers

feat: get backup by backup id and file by file id resolver

Author: rocketeerbkw

services/api/src/resolvers.js

@@ -98,6 +98,7 @@ const {
   uploadFilesForTask,
   deleteFilesForTask,
   getDownloadLink,
+  getDownloadLinkByTaskFileId,
 } = require('./resources/file/resolvers');
 
 const {
@@ -283,6 +284,7 @@ const {
   addRestore,
   getRestoreByBackupId,
   updateRestore,
+  getBackupDownloadLinkByBackupId,
   backupSubscriber,
   getRestoreLocation,
 } = require('./resources/backup/resolvers');
@@ -683,7 +685,9 @@ async function getResolvers() {
     checkBulkImportProjectsAndGroupsToOrganization,
     allPlatformUsers: getAllPlatformUsers,
     getAuditLogs,
-    listAllRetentionPolicies
+    listAllRetentionPolicies,
+    getBackupDownloadLinkByBackupId,
+    getDownloadLinkByTaskFileId,
   },
   Mutation: {
     addProblem,

services/api/src/resources/backup/resolvers.ts

@@ -498,6 +498,31 @@ export const updateRestore: ResolverFn = async (
   return restoreData;
 };
 
+export const getBackupDownloadLinkByBackupId: ResolverFn = async (
+  root,
+  { backupId },
+  { sqlClientPool, hasPermission, userActivityLogger, adminScopes }
+) => {
+  const perms = await query(sqlClientPool, Sql.selectPermsForBackup(backupId));
+
+  await hasPermission('backup', 'view', {
+    project: R.path(['0', 'pid'], perms)
+  });
+
+  const rows = await query(
+    sqlClientPool,
+    Sql.selectRestoreByBackupId(backupId)
+  );
+  const row = R.prop(0, rows)
+
+  if (!row || row.restoreLocation == null) {
+    throw new Error(`no restore file available`);
+  }
+  const backupData = R.prop(0, rows);
+  const [restLoc, restSize] = await getRestoreLocation(backupId, backupData.restoreLocation, sqlClientPool, userActivityLogger);
+  return restLoc;
+}
+
 export const getRestoreByBackupId: ResolverFn = async (
   { backupId },
   args,

services/api/src/resources/file/resolvers.ts

@@ -16,15 +16,10 @@ const { Upload } = require('@aws-sdk/lib-storage');
 const isGCS = process.env.S3_FILES_GCS || 'false'
 const bucket = process.env.S3_FILES_BUCKET || 'lagoon-files'
 
-
-export const getDownloadLink: ResolverFn = async ({ s3Key }, input, { userActivityLogger }) => {
-  const command = new GetObjectCommand({
-    Bucket: bucket,
-    Key: s3Key,
-  });
+async function generateDownloadLink(s3Key: string, userActivityLogger?: Function ) {
+  const command = new GetObjectCommand({ Bucket: bucket, Key: s3Key });
 
   if (typeof userActivityLogger === 'function') {
-
     const auditLog: AuditLog = {
       resource: {
         type: AuditType.FILE,
@@ -40,12 +35,33 @@ export const getDownloadLink: ResolverFn = async ({ s3Key }, input, { userActivi
       }
     });
   }
-
-  return getSignedUrl(s3Client, command, {
+  const signedUrl = await getSignedUrl(s3Client, command, {
     expiresIn: s3Config.signedLinkExpiration,
   });
+  return signedUrl;
+}
+
+export const getDownloadLink: ResolverFn = async ({ s3Key }, input, { userActivityLogger }) => {
+  return generateDownloadLink(s3Key, userActivityLogger);
 };
 
+export const getDownloadLinkByTaskFileId: ResolverFn = async (
+  root,
+  { taskId, fileId },
+  { sqlClientPool, hasPermission, userActivityLogger }
+) => {
+  const rowsPerms = await query(sqlClientPool, taskSql.selectPermsForTask(taskId));
+
+  await hasPermission('task', 'view', {
+    project: R.path(['0', 'pid'], rowsPerms)
+  });
+
+  const rows = await query(sqlClientPool, Sql.selectTaskFileById(taskId, fileId));
+
+  const downloadUrl = await generateDownloadLink(rows[0].s3Key, userActivityLogger)
+  return downloadUrl;
+}
+
 export const getFilesByTaskId: ResolverFn = async (
   { id: tid },
   _args,

services/api/src/resources/file/sql.ts

@@ -8,6 +8,13 @@ export const Sql = {
       .andWhere('s3_file.deleted', '0000-00-00 00:00:00')
       .orderBy('s3_file.created', 'desc')
       .toString(),
+  selectTaskFileById: (tid: number, fid: number) =>
+    knex('task_file')
+      .join('s3_file', 'task_file.fid', '=', 's3_file.id')
+      .where('task_file.tid', tid)
+      .andWhere('task_file.fid', fid)
+      .andWhere('s3_file.deleted', '0000-00-00 00:00:00')
+      .toString(),
   insertFile: ({
     id,
     filename,

services/api/src/typeDefs.js

@@ -1522,6 +1522,8 @@ const typeDefs = gql`
     allPlatformUsers(id: String, email: String, gitlabId: Int, role: PlatformRole): [User]
     getAuditLogs(input: AuditLogInput): [AuditLog]
     listAllRetentionPolicies(name: String, type: RetentionPolicyType): [RetentionPolicy]
+    getBackupDownloadLinkByBackupId(backupId: String!): String
+    getDownloadLinkByTaskFileId(taskId: Int!, fileId: Int!): String
   }
 
   type ProjectGroupsToOrganization {