@powerhousedao/codegen 4.1.0-dev.9 → 5.0.0-staging.1

package/dist/src/codegen/.hygen/templates/powerhouse/generate-document-model-subgraph/resolvers.esm.t

@@ -2,49 +2,55 @@
 to: "<%= rootDir %>/<%= h.changeCase.param(subgraph) %>/resolvers.ts"
 force: true
 ---
-/* eslint-disable @typescript-eslint/no-unsafe-argument */
-/* eslint-disable @typescript-eslint/no-unsafe-assignment */
-/* eslint-disable @typescript-eslint/no-unsafe-member-access */
 import { type Subgraph } from "@powerhousedao/reactor-api";
 import { addFile } from "document-drive";
-import { actions } from "../../document-models/<%- h.changeCase.param(documentType) %>/index.js";
-import { generateId } from "document-model";
+import { actions <% modules.forEach(module => { %><% module.operations.forEach(op => { %>, type <%- h.changeCase.pascal(op.name) %>Input<%_ })}); %>, type <%- h.changeCase.pascal(documentType) %>Document } from "../../document-models/<%- h.changeCase.param(documentType) %>/index.js";
+import { setName } from "document-model";
 
-const DEFAULT_DRIVE_ID = "powerhouse";
-
-export const getResolvers = (subgraph: Subgraph): Record<string, any> => {
+export const getResolvers = (subgraph: Subgraph) => {
   const reactor = subgraph.reactor;
 
   return ({
     Query: {
-      <%- h.changeCase.pascal(documentType) %>: async (_: any, args: any, ctx: any) => {
+      <%- h.changeCase.pascal(documentType) %>: async () => {
         return {
-          getDocument: async (args: any) => {
-            const driveId: string = args.driveId || DEFAULT_DRIVE_ID;
-            const docId: string = args.docId || "";
-            const doc = await reactor.getDocument(driveId, docId);
-             return {
+          getDocument: async (args: { docId: string, driveId: string }) => {
+            const { docId, driveId } = args;
+
+            if(!docId) {
+              throw new Error("Document id is required");
+            }
+            
+            if(driveId) {
+              const docIds = await reactor.getDocuments(driveId);
+              if(!docIds.includes(docId)) {
+                throw new Error(`Document with id ${docId} is not part of ${driveId}`);
+              }
+            }
+
+            const doc = await reactor.getDocument<<%- h.changeCase.pascal(documentType) %>Document>(docId);
+            return {
               driveId: driveId,
               ...doc,
               ...doc.header,
               state: doc.state.global,
               stateJSON: doc.state.global,
-              revision: doc.header.revision["global"] ?? 0,
+              revision: doc.header?.revision?.global ?? 0,
             };
           },
-          getDocuments: async (args: any) => {
-            const driveId: string = args.driveId || DEFAULT_DRIVE_ID;
+          getDocuments: async (args: { driveId: string }) => {
+            const { driveId } = args;
             const docsIds = await reactor.getDocuments(driveId);
             const docs = await Promise.all(
               docsIds.map(async (docId) => {
-                const doc = await reactor.getDocument(driveId, docId);
+                const doc = await reactor.getDocument<<%- h.changeCase.pascal(documentType) %>Document>(docId);
                 return {
                   driveId: driveId,
                   ...doc,
                   ...doc.header,
                   state: doc.state.global,
                   stateJSON: doc.state.global,
-                  revision: doc.header.revision["global"] ?? 0,
+                  revision: doc.header?.revision?.global ?? 0,
                 };
               }),
             );
@@ -57,47 +63,50 @@ export const getResolvers = (subgraph: Subgraph): Record<string, any> => {
       },
     },
     Mutation: {
-
-      <%- h.changeCase.pascal(documentType) %>_createDocument: async (_: any, args: any) => {
-        const driveId: string = args.driveId || DEFAULT_DRIVE_ID;
-        const docId = generateId();
+      <%- h.changeCase.pascal(documentType) %>_createDocument: async (_: unknown, args: { name: string, driveId?: string }) => {
+        const { driveId, name } = args;
+        const document = await reactor.addDocument("<%- documentTypeId %>");
         
-        await reactor.addDriveAction(driveId, addFile({
-          id: docId,
-          name: args.name,
-          documentType: "<%- documentTypeId %>",
-          synchronizationUnits:[
-            { 
-              branch: "main", 
-              scope: "global", 
-              syncId: generateId(), 
-            },
-            { 
-              branch: "main", 
-              scope: "local", 
-              syncId: generateId(), 
-            }
-          ],
-        }));
+        if(driveId) {
+          await reactor.addAction(
+            driveId,
+            addFile({
+              name,
+              id: document.header.id,
+              documentType: "<%- documentTypeId %>",
+            }),
+          );
+        }
+
+        if(name) {
+          await reactor.addAction(
+            document.header.id,
+            setName(name),
+          );
+        }
 
-        return docId;
+        return document.header.id;
       },
 
 <% modules.forEach(module => { _%>
 <% module.operations.forEach(op => { _%>
-        <%- h.changeCase.pascal(documentType) + '_' + h.changeCase.camel(op.name) 
-        %>: async (_: any, args: any) => {
-            const driveId: string = args.driveId || DEFAULT_DRIVE_ID;
-            const docId: string = args.docId || "";
-            const doc = await reactor.getDocument(driveId, docId);
+        <%- h.changeCase.pascal(documentType) + '_' + h.changeCase.camel(op.name) %>: async (_: unknown, args: { docId: string, input: <%- h.changeCase.pascal(op.name) %>Input}) => {
+            const { docId, input } = args;
+            const doc = await reactor.getDocument<<%- h.changeCase.pascal(documentType) %>Document>(docId);
+            if(!doc) {
+              throw new Error("Document not found");
+            }
 
-            await reactor.addAction(
-                driveId,
+            const result = await reactor.addAction(
                 docId,
-                actions.<%- h.changeCase.camel(op.name) %>({...args.input})
+                actions.<%- h.changeCase.camel(op.name) %>(input)
             );
             
-            return (doc.header.revision["global"] ?? 0) + 1;
+            if(result.status !== "SUCCESS") {
+              throw new Error(result.error?.message ?? "Failed to <%- h.changeCase.camel(op.name) %>");
+            }
+
+            return true;
         },
 
 <%_ })}); %>

package/dist/src/codegen/.hygen/templates/powerhouse/generate-document-model-subgraph/schema.esm.t

@@ -17,8 +17,8 @@ Queries: <%= h.changeCase.pascal(documentType) %>
 """
 
 type <%- h.changeCase.pascal(documentType) %>Queries {
-    getDocument(driveId: String, docId: PHID): <%- h.changeCase.pascal(documentType) %>
-    getDocuments(driveId: String): [<%- h.changeCase.pascal(documentType) %>!]
+    getDocument(docId: PHID!, driveId: PHID): <%- h.changeCase.pascal(documentType) %>
+    getDocuments(driveId: String!): [<%- h.changeCase.pascal(documentType) %>!]
 }
 
 type Query {
@@ -30,7 +30,7 @@ Mutations: <%= h.changeCase.pascal(documentType) %>
 """
 type Mutation {
 
-    <%- h.changeCase.pascal(documentType) %>_createDocument(driveId:String, name:String): String
+    <%- h.changeCase.pascal(documentType) %>_createDocument(name:String!, driveId:String): String
 
 <% modules.forEach(module => { _%>
 <% module.operations.forEach(op => { _%>

package/dist/src/codegen/.hygen/templates/powerhouse/generate-drive-editor/components/CreateDocument.esm.t

@@ -2,49 +2,63 @@
 to: "<%= rootDir %>/<%= h.changeCase.param(name) %>/components/CreateDocument.tsx"
 unless_exists: true
 ---
+import {
+  addDocument,
+  useDocumentModelModules,
+  useSelectedDriveId,
+  useSelectedFolder,
+  type VetraDocumentModelModule,
+} from "@powerhousedao/reactor-browser";
 import { Button } from "@powerhousedao/design-system";
-import { type DocumentModelModule } from "document-model";
 
-interface CreateDocumentProps {
-  documentModels?: DocumentModelModule[];
-  createDocument: (doc: DocumentModelModule) => void;
-}
+/**
+ * Document creation UI component.
+ * Displays available document types as clickable buttons.
+ */
+export const CreateDocument = () => {
+  const selectedDriveId = useSelectedDriveId();
+  const selectedFolder = useSelectedFolder();
+  const documentModelModules = useDocumentModelModules();
 
-function getDocumentSpec(doc: DocumentModelModule) {
-  if ("documentModelState" in doc) {
-    return doc.documentModelState as DocumentModelModule["documentModel"];
+  async function handleAddDocument(module: VetraDocumentModelModule) {
+    if (!selectedDriveId) {
+      return;
+    }
+    await addDocument(
+      selectedDriveId,
+      `New ${module.documentModel.name} document`,
+      module.documentModel.id,
+      selectedFolder?.id,
+    );
   }
 
-  return doc.documentModel;
-}
-
-export const CreateDocument: React.FC<CreateDocumentProps> = ({
-  documentModels,
-  createDocument,
-}) => {
   return (
     <div className="px-6">
+      {/* Customize section title here */}
       <h3 className="mb-3 mt-4 text-sm font-bold text-gray-600">
         New document
       </h3>
+      {/* Customize layout by changing flex-wrap, gap, or grid layout */}
       <div className="flex w-full flex-wrap gap-4">
-        {documentModels?.map((doc) => {
-          const spec = getDocumentSpec(doc);
+        {documentModelModules?.map((documentModelModule) => {
           return (
             <Button
-              key={spec.id}
-              color="light"
+              key={documentModelModule.documentModel.id}
+              color="light" // Customize button appearance
               size="small"
               className="cursor-pointer"
-              title={spec.name}
-              aria-description={spec.description}
-              onClick={() => createDocument(doc)}
+              title={documentModelModule.documentModel.name}
+              aria-description={documentModelModule.documentModel.description}
+              onClick={() => handleAddDocument(documentModelModule)}
             >
-              <span className="text-sm">{spec.name}</span>
+              {/* Customize document type display format */}
+              <span className="text-sm">
+                {documentModelModule.documentModel.name}
+              </span>
             </Button>
           );
         })}
       </div>
     </div>
   );
-}; 
+};