feat: added more drive info to drive query

powerhouse-inc · Sep 12, 2024 · 328a9fa · 328a9fa
1 parent 69d30b0
commit 328a9fa
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 4 deletions.
diff --git a/api/src/graphql/server/generated/drive/nexus.ts b/api/src/graphql/server/generated/drive/nexus.ts
@@ -292,10 +292,12 @@ export interface NexusGenObjects {
     type: string; // String!
   }
   DocumentDriveState: { // root type
+    availableOffline: boolean; // Boolean!
     icon?: string | null; // String
     id: string; // ID!
     name: string; // String!
     nodes: Array<NexusGenRootTypes['Node'] | null>; // [Node]!
+    sharingType?: string | null; // String
     slug?: string | null; // String
   }
   FixedIncome: { // root type
@@ -771,10 +773,12 @@ export interface NexusGenFieldTypes {
     type: string; // String!
   }
   DocumentDriveState: { // field return type
+    availableOffline: boolean; // Boolean!
     icon: string | null; // String
     id: string; // ID!
     name: string; // String!
     nodes: Array<NexusGenRootTypes['Node'] | null>; // [Node]!
+    sharingType: string | null; // String
     slug: string | null; // String
   }
   FixedIncome: { // field return type
@@ -1280,10 +1284,12 @@ export interface NexusGenFieldTypeNames {
     type: 'String'
   }
   DocumentDriveState: { // field return type name
+    availableOffline: 'Boolean'
     icon: 'String'
     id: 'ID'
     name: 'String'
     nodes: 'Node'
+    sharingType: 'String'
     slug: 'String'
   }
   FixedIncome: { // field return type name

diff --git a/api/src/graphql/server/generated/drive/schema.graphql b/api/src/graphql/server/generated/drive/schema.graphql
@@ -204,10 +204,12 @@ type DefaultOperation implements IOperation {
 }
 
 type DocumentDriveState {
+  availableOffline: Boolean!
   icon: String
   id: ID!
   name: String!
   nodes: [Node]!
+  sharingType: String
   slug: String
 }
 

diff --git a/api/src/graphql/server/generated/index/nexus.ts b/api/src/graphql/server/generated/index/nexus.ts
@@ -107,10 +107,12 @@ export interface NexusGenObjects {
     sharingType?: string | null; // String
   }
   DocumentDriveState: { // root type
+    availableOffline: boolean; // Boolean!
     icon?: string | null; // String
     id: string; // ID!
     name: string; // String!
     nodes: Array<NexusGenRootTypes['Node'] | null>; // [Node]!
+    sharingType?: string | null; // String
     slug?: string | null; // String
   }
   Mutation: {};
@@ -184,10 +186,12 @@ export interface NexusGenFieldTypes {
     sharingType: string | null; // String
   }
   DocumentDriveState: { // field return type
+    availableOffline: boolean; // Boolean!
     icon: string | null; // String
     id: string; // ID!
     name: string; // String!
     nodes: Array<NexusGenRootTypes['Node'] | null>; // [Node]!
+    sharingType: string | null; // String
     slug: string | null; // String
   }
   Mutation: { // field return type
@@ -270,10 +274,12 @@ export interface NexusGenFieldTypeNames {
     sharingType: 'String'
   }
   DocumentDriveState: { // field return type name
+    availableOffline: 'Boolean'
     icon: 'String'
     id: 'ID'
     name: 'String'
     nodes: 'Node'
+    sharingType: 'String'
     slug: 'String'
   }
   Mutation: { // field return type name

diff --git a/api/src/graphql/server/generated/index/schema.graphql b/api/src/graphql/server/generated/index/schema.graphql
@@ -46,10 +46,12 @@ input DocumentDriveLocalStateInput {
 }
 
 type DocumentDriveState {
+  availableOffline: Boolean!
   icon: String
   id: ID!
   name: String!
   nodes: [Node]!
+  sharingType: String
   slug: String
 }
 

diff --git a/api/src/modules/document-drive/drive-resolver.ts b/api/src/modules/document-drive/drive-resolver.ts
@@ -37,6 +37,7 @@ export const Node = objectType({
   }
 });
 
+
 export const DocumentDriveStateObject = objectType({
   name: 'DocumentDriveState',
   definition(t) {
@@ -45,6 +46,8 @@ export const DocumentDriveStateObject = objectType({
     t.nonNull.list.field('nodes', { type: Node });
     t.string('icon');
     t.string('slug');
+    t.string("sharingType");
+    t.nonNull.boolean("availableOffline");
   }
 });
 
@@ -318,10 +321,10 @@ export const getDrive = queryField('drive', {
   type: DocumentDriveStateObject,
   resolve: async (_parent, _args, ctx: Context) => {
     try {
-      const drive = (await ctx.prisma.document.getDrive(
+      const {global, local} = (await ctx.prisma.document.getDrive(
         ctx.driveId ?? '1'
-      )) as DocumentDriveState;
-      return drive;
+      )) ;
+      return {...global, ...local, sharingType: local.sharingType?.toUpperCase() ?? null};
     } catch (e: any) {
       throw new DocumentDriveError({
         code: 500,

diff --git a/api/src/modules/document/model.ts b/api/src/modules/document/model.ts
@@ -120,7 +120,7 @@ export function getDocumentDriveCRUD(prisma: Prisma.TransactionClient) {
     getDrive: async (id: string) => {
       try {
         const { state } = await driveServer.getDrive(id);
-        return state.global;
+        return state;
       } catch (e) {
         logger.error(e);
         throw new Error('Drive not found');