@weirdfingers/boards 0.5.3 → 0.6.1

package/dist/index.js

@@ -21,6 +21,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   ADD_BOARD_MEMBER: () => ADD_BOARD_MEMBER,
+  ANCESTRY_NODE_FRAGMENT: () => ANCESTRY_NODE_FRAGMENT,
+  ARTIFACT_LINEAGE_FRAGMENT: () => ARTIFACT_LINEAGE_FRAGMENT,
   ArtifactType: () => ArtifactType,
   AuthProvider: () => AuthProvider,
   BOARD_FRAGMENT: () => BOARD_FRAGMENT,
@@ -31,13 +33,17 @@ __export(index_exports, {
   CREATE_BOARD: () => CREATE_BOARD,
   CREATE_GENERATION: () => CREATE_GENERATION,
   DELETE_BOARD: () => DELETE_BOARD,
+  DESCENDANT_NODE_FRAGMENT: () => DESCENDANT_NODE_FRAGMENT,
   GENERATION_FRAGMENT: () => GENERATION_FRAGMENT,
+  GET_ANCESTRY: () => GET_ANCESTRY,
   GET_BOARD: () => GET_BOARD,
   GET_BOARDS: () => GET_BOARDS,
   GET_CURRENT_USER: () => GET_CURRENT_USER,
+  GET_DESCENDANTS: () => GET_DESCENDANTS,
   GET_GENERATION: () => GET_GENERATION,
   GET_GENERATIONS: () => GET_GENERATIONS,
   GET_GENERATORS: () => GET_GENERATORS,
+  GET_INPUT_ARTIFACTS: () => GET_INPUT_ARTIFACTS,
   GenerationStatus: () => GenerationStatus,
   GeneratorSelectionProvider: () => GeneratorSelectionProvider,
   NoAuthProvider: () => NoAuthProvider,
@@ -45,6 +51,7 @@ __export(index_exports, {
   RETRY_GENERATION: () => RETRY_GENERATION,
   UPDATE_BOARD: () => UPDATE_BOARD,
   UPDATE_BOARD_MEMBER_ROLE: () => UPDATE_BOARD_MEMBER_ROLE,
+  UPLOAD_ARTIFACT_FROM_URL: () => UPLOAD_ARTIFACT_FROM_URL,
   USER_FRAGMENT: () => USER_FRAGMENT,
   VERSION: () => VERSION,
   createGraphQLClient: () => createGraphQLClient,
@@ -53,14 +60,19 @@ __export(index_exports, {
   parseArtifactSlot: () => parseArtifactSlot,
   parseGeneratorSchema: () => parseGeneratorSchema,
   parseSettingsField: () => parseSettingsField,
+  useAncestry: () => useAncestry,
   useApiConfig: () => useApiConfig,
   useAuth: () => useAuth,
   useAuthOptional: () => useAuthOptional,
   useBoard: () => useBoard,
   useBoards: () => useBoards,
+  useDescendants: () => useDescendants,
   useGeneration: () => useGeneration,
   useGeneratorSelection: () => useGeneratorSelection,
-  useGenerators: () => useGenerators
+  useGenerators: () => useGenerators,
+  useInputArtifacts: () => useInputArtifacts,
+  useLineage: () => useLineage,
+  useUpload: () => useUpload
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -634,6 +646,47 @@ var GENERATION_FRAGMENT = import_urql2.gql`
     completedAt
   }
 `;
+var ARTIFACT_LINEAGE_FRAGMENT = import_urql2.gql`
+  fragment ArtifactLineageFragment on ArtifactLineage {
+    generationId
+    role
+    artifactType
+  }
+`;
+var ANCESTRY_NODE_FRAGMENT = import_urql2.gql`
+  ${GENERATION_FRAGMENT}
+  fragment AncestryNodeFragment on AncestryNode {
+    depth
+    role
+    generation {
+      ...GenerationFragment
+    }
+    parents {
+      depth
+      role
+      generation {
+        ...GenerationFragment
+      }
+    }
+  }
+`;
+var DESCENDANT_NODE_FRAGMENT = import_urql2.gql`
+  ${GENERATION_FRAGMENT}
+  fragment DescendantNodeFragment on DescendantNode {
+    depth
+    role
+    generation {
+      ...GenerationFragment
+    }
+    children {
+      depth
+      role
+      generation {
+        ...GenerationFragment
+      }
+    }
+  }
+`;
 var GET_CURRENT_USER = import_urql2.gql`
   ${USER_FRAGMENT}
   query GetCurrentUser {
@@ -723,6 +776,36 @@ var GET_GENERATION = import_urql2.gql`
     }
   }
 `;
+var GET_ANCESTRY = import_urql2.gql`
+  ${ANCESTRY_NODE_FRAGMENT}
+  query GetAncestry($id: UUID!, $maxDepth: Int = 25) {
+    generation(id: $id) {
+      ancestry(maxDepth: $maxDepth) {
+        ...AncestryNodeFragment
+      }
+    }
+  }
+`;
+var GET_DESCENDANTS = import_urql2.gql`
+  ${DESCENDANT_NODE_FRAGMENT}
+  query GetDescendants($id: UUID!, $maxDepth: Int = 25) {
+    generation(id: $id) {
+      descendants(maxDepth: $maxDepth) {
+        ...DescendantNodeFragment
+      }
+    }
+  }
+`;
+var GET_INPUT_ARTIFACTS = import_urql2.gql`
+  ${ARTIFACT_LINEAGE_FRAGMENT}
+  query GetInputArtifacts($id: UUID!) {
+    generation(id: $id) {
+      inputArtifacts {
+        ...ArtifactLineageFragment
+      }
+    }
+  }
+`;
 var CREATE_BOARD = import_urql2.gql`
   ${BOARD_FRAGMENT}
   ${USER_FRAGMENT}
@@ -804,6 +887,14 @@ var RETRY_GENERATION = import_urql2.gql`
     }
   }
 `;
+var UPLOAD_ARTIFACT_FROM_URL = import_urql2.gql`
+  ${GENERATION_FRAGMENT}
+  mutation UploadArtifactFromUrl($input: UploadArtifactInput!) {
+    uploadArtifact(input: $input) {
+      ...GenerationFragment
+    }
+  }
+`;
 var BoardRole = /* @__PURE__ */ ((BoardRole2) => {
   BoardRole2["VIEWER"] = "VIEWER";
   BoardRole2["EDITOR"] = "EDITOR";
@@ -818,14 +909,14 @@ var GenerationStatus = /* @__PURE__ */ ((GenerationStatus2) => {
   GenerationStatus2["CANCELLED"] = "CANCELLED";
   return GenerationStatus2;
 })(GenerationStatus || {});
-var ArtifactType = /* @__PURE__ */ ((ArtifactType4) => {
-  ArtifactType4["IMAGE"] = "image";
-  ArtifactType4["VIDEO"] = "video";
-  ArtifactType4["AUDIO"] = "audio";
-  ArtifactType4["TEXT"] = "text";
-  ArtifactType4["LORA"] = "lora";
-  ArtifactType4["MODEL"] = "model";
-  return ArtifactType4;
+var ArtifactType = /* @__PURE__ */ ((ArtifactType5) => {
+  ArtifactType5["IMAGE"] = "IMAGE";
+  ArtifactType5["VIDEO"] = "VIDEO";
+  ArtifactType5["AUDIO"] = "AUDIO";
+  ArtifactType5["TEXT"] = "TEXT";
+  ArtifactType5["LORA"] = "LORA";
+  ArtifactType5["MODEL"] = "MODEL";
+  return ArtifactType5;
 })(ArtifactType || {});
 
 // src/hooks/useBoards.ts
@@ -1324,8 +1415,200 @@ function useGenerators(options = {}) {
   };
 }
 
-// src/providers/BoardsProvider.tsx
+// src/hooks/useUpload.ts
+var import_react8 = require("react");
 var import_urql7 = require("urql");
+function useUpload() {
+  const [isUploading, setIsUploading] = (0, import_react8.useState)(false);
+  const [progress, setProgress] = (0, import_react8.useState)(0);
+  const [error, setError] = (0, import_react8.useState)(null);
+  const { apiUrl } = useApiConfig();
+  const auth = useAuth();
+  const [, uploadFromUrlMutation] = (0, import_urql7.useMutation)(UPLOAD_ARTIFACT_FROM_URL);
+  const upload = (0, import_react8.useCallback)(
+    async (request) => {
+      setError(null);
+      setProgress(0);
+      setIsUploading(true);
+      try {
+        if (typeof request.source === "string") {
+          const result2 = await uploadFromUrlMutation({
+            input: {
+              boardId: request.boardId,
+              artifactType: request.artifactType,
+              fileUrl: request.source,
+              userDescription: request.userDescription,
+              parentGenerationId: request.parentGenerationId
+            }
+          });
+          if (result2.error) {
+            throw new Error(result2.error.message);
+          }
+          if (!result2.data?.uploadArtifact) {
+            throw new Error("Upload failed");
+          }
+          setProgress(100);
+          setIsUploading(false);
+          return {
+            id: result2.data.uploadArtifact.id,
+            storageUrl: result2.data.uploadArtifact.storageUrl,
+            thumbnailUrl: result2.data.uploadArtifact.thumbnailUrl,
+            status: "completed",
+            artifactType: result2.data.uploadArtifact.artifactType,
+            generatorName: result2.data.uploadArtifact.generatorName
+          };
+        }
+        const formData = new FormData();
+        formData.append("board_id", request.boardId);
+        formData.append("artifact_type", request.artifactType.toLowerCase());
+        formData.append("file", request.source);
+        if (request.userDescription) {
+          formData.append("user_description", request.userDescription);
+        }
+        if (request.parentGenerationId) {
+          formData.append("parent_generation_id", request.parentGenerationId);
+        }
+        const token = await auth.getToken();
+        const headers = {};
+        if (token) {
+          headers.Authorization = `Bearer ${token}`;
+        }
+        const result = await new Promise((resolve, reject) => {
+          const xhr = new XMLHttpRequest();
+          xhr.upload.addEventListener("progress", (e) => {
+            if (e.lengthComputable) {
+              const percentComplete = e.loaded / e.total * 100;
+              setProgress(percentComplete);
+            }
+          });
+          xhr.addEventListener("load", () => {
+            if (xhr.status === 200) {
+              try {
+                const data = JSON.parse(xhr.responseText);
+                resolve({
+                  id: data.id,
+                  storageUrl: data.storageUrl,
+                  thumbnailUrl: data.thumbnailUrl,
+                  status: "completed",
+                  artifactType: data.artifactType,
+                  generatorName: data.generatorName
+                });
+              } catch (err) {
+                reject(new Error("Failed to parse response"));
+              }
+            } else {
+              try {
+                const errorData = JSON.parse(xhr.responseText);
+                reject(new Error(errorData.detail || `Upload failed: ${xhr.statusText}`));
+              } catch {
+                reject(new Error(`Upload failed: ${xhr.statusText}`));
+              }
+            }
+          });
+          xhr.addEventListener("error", () => {
+            reject(new Error("Upload failed: Network error"));
+          });
+          xhr.addEventListener("abort", () => {
+            reject(new Error("Upload cancelled"));
+          });
+          xhr.open("POST", `${apiUrl}/api/uploads/artifact`);
+          Object.entries(headers).forEach(([key, value]) => {
+            xhr.setRequestHeader(key, value);
+          });
+          xhr.send(formData);
+        });
+        setProgress(100);
+        setIsUploading(false);
+        return result;
+      } catch (err) {
+        const uploadError = err instanceof Error ? err : new Error("Upload failed");
+        setError(uploadError);
+        setIsUploading(false);
+        setProgress(0);
+        throw uploadError;
+      }
+    },
+    [uploadFromUrlMutation, apiUrl, auth]
+  );
+  return {
+    upload,
+    isUploading,
+    progress,
+    error
+  };
+}
+
+// src/hooks/useLineage.ts
+var import_react9 = require("react");
+var import_urql8 = require("urql");
+function useAncestry(generationId, options = {}) {
+  const { maxDepth = 25, pause = false } = options;
+  const [{ data, fetching, error }] = (0, import_urql8.useQuery)({
+    query: GET_ANCESTRY,
+    variables: { id: generationId, maxDepth },
+    pause
+  });
+  const ancestry = (0, import_react9.useMemo)(
+    () => data?.generation?.ancestry || null,
+    [data?.generation?.ancestry]
+  );
+  return {
+    ancestry,
+    loading: fetching,
+    error: error ? new Error(error.message) : null
+  };
+}
+function useDescendants(generationId, options = {}) {
+  const { maxDepth = 25, pause = false } = options;
+  const [{ data, fetching, error }] = (0, import_urql8.useQuery)({
+    query: GET_DESCENDANTS,
+    variables: { id: generationId, maxDepth },
+    pause
+  });
+  const descendants = (0, import_react9.useMemo)(
+    () => data?.generation?.descendants || null,
+    [data?.generation?.descendants]
+  );
+  return {
+    descendants,
+    loading: fetching,
+    error: error ? new Error(error.message) : null
+  };
+}
+function useInputArtifacts(generationId, options = {}) {
+  const { pause = false } = options;
+  const [{ data, fetching, error }] = (0, import_urql8.useQuery)({
+    query: GET_INPUT_ARTIFACTS,
+    variables: { id: generationId },
+    pause
+  });
+  const inputArtifacts = (0, import_react9.useMemo)(
+    () => data?.generation?.inputArtifacts || [],
+    [data?.generation?.inputArtifacts]
+  );
+  return {
+    inputArtifacts,
+    loading: fetching,
+    error: error ? new Error(error.message) : null
+  };
+}
+function useLineage(generationId, options = {}) {
+  const ancestryResult = useAncestry(generationId, options);
+  const descendantsResult = useDescendants(generationId, options);
+  const inputArtifactsResult = useInputArtifacts(generationId, options);
+  const loading = ancestryResult.loading || descendantsResult.loading || inputArtifactsResult.loading;
+  const error = ancestryResult.error || descendantsResult.error || inputArtifactsResult.error;
+  return {
+    ancestry: ancestryResult.ancestry,
+    descendants: descendantsResult.descendants,
+    inputArtifacts: inputArtifactsResult.inputArtifacts,
+    loading,
+    error
+  };
+}
+
+// src/providers/BoardsProvider.tsx
+var import_urql9 = require("urql");
 var import_jsx_runtime4 = require("react/jsx-runtime");
 function BoardsProvider({
   children,
@@ -1349,7 +1632,7 @@ function BoardsProvider({
     },
     tenantId
   });
-  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(AuthProvider, { provider: authProvider, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ApiConfigProvider, { config: apiConfig, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(import_urql7.Provider, { value: client, children }) }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(AuthProvider, { provider: authProvider, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(ApiConfigProvider, { config: apiConfig, children: /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(import_urql9.Provider, { value: client, children }) }) });
 }
 
 // src/index.ts
@@ -1357,6 +1640,8 @@ var VERSION = "0.1.0";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ADD_BOARD_MEMBER,
+  ANCESTRY_NODE_FRAGMENT,
+  ARTIFACT_LINEAGE_FRAGMENT,
   ArtifactType,
   AuthProvider,
   BOARD_FRAGMENT,
@@ -1367,13 +1652,17 @@ var VERSION = "0.1.0";
   CREATE_BOARD,
   CREATE_GENERATION,
   DELETE_BOARD,
+  DESCENDANT_NODE_FRAGMENT,
   GENERATION_FRAGMENT,
+  GET_ANCESTRY,
   GET_BOARD,
   GET_BOARDS,
   GET_CURRENT_USER,
+  GET_DESCENDANTS,
   GET_GENERATION,
   GET_GENERATIONS,
   GET_GENERATORS,
+  GET_INPUT_ARTIFACTS,
   GenerationStatus,
   GeneratorSelectionProvider,
   NoAuthProvider,
@@ -1381,6 +1670,7 @@ var VERSION = "0.1.0";
   RETRY_GENERATION,
   UPDATE_BOARD,
   UPDATE_BOARD_MEMBER_ROLE,
+  UPLOAD_ARTIFACT_FROM_URL,
   USER_FRAGMENT,
   VERSION,
   createGraphQLClient,
@@ -1389,13 +1679,18 @@ var VERSION = "0.1.0";
   parseArtifactSlot,
   parseGeneratorSchema,
   parseSettingsField,
+  useAncestry,
   useApiConfig,
   useAuth,
   useAuthOptional,
   useBoard,
   useBoards,
+  useDescendants,
   useGeneration,
   useGeneratorSelection,
-  useGenerators
+  useGenerators,
+  useInputArtifacts,
+  useLineage,
+  useUpload
 });
 //# sourceMappingURL=index.js.map