@weirdfingers/boards 0.5.2 → 0.6.0

package/dist/index.mjs

@@ -574,6 +574,47 @@ var GENERATION_FRAGMENT = gql`
     completedAt
   }
 `;
+var ARTIFACT_LINEAGE_FRAGMENT = gql`
+  fragment ArtifactLineageFragment on ArtifactLineage {
+    generationId
+    role
+    artifactType
+  }
+`;
+var ANCESTRY_NODE_FRAGMENT = gql`
+  ${GENERATION_FRAGMENT}
+  fragment AncestryNodeFragment on AncestryNode {
+    depth
+    role
+    generation {
+      ...GenerationFragment
+    }
+    parents {
+      depth
+      role
+      generation {
+        ...GenerationFragment
+      }
+    }
+  }
+`;
+var DESCENDANT_NODE_FRAGMENT = gql`
+  ${GENERATION_FRAGMENT}
+  fragment DescendantNodeFragment on DescendantNode {
+    depth
+    role
+    generation {
+      ...GenerationFragment
+    }
+    children {
+      depth
+      role
+      generation {
+        ...GenerationFragment
+      }
+    }
+  }
+`;
 var GET_CURRENT_USER = gql`
   ${USER_FRAGMENT}
   query GetCurrentUser {
@@ -663,6 +704,36 @@ var GET_GENERATION = gql`
     }
   }
 `;
+var GET_ANCESTRY = gql`
+  ${ANCESTRY_NODE_FRAGMENT}
+  query GetAncestry($id: UUID!, $maxDepth: Int = 25) {
+    generation(id: $id) {
+      ancestry(maxDepth: $maxDepth) {
+        ...AncestryNodeFragment
+      }
+    }
+  }
+`;
+var GET_DESCENDANTS = gql`
+  ${DESCENDANT_NODE_FRAGMENT}
+  query GetDescendants($id: UUID!, $maxDepth: Int = 25) {
+    generation(id: $id) {
+      descendants(maxDepth: $maxDepth) {
+        ...DescendantNodeFragment
+      }
+    }
+  }
+`;
+var GET_INPUT_ARTIFACTS = gql`
+  ${ARTIFACT_LINEAGE_FRAGMENT}
+  query GetInputArtifacts($id: UUID!) {
+    generation(id: $id) {
+      inputArtifacts {
+        ...ArtifactLineageFragment
+      }
+    }
+  }
+`;
 var CREATE_BOARD = gql`
   ${BOARD_FRAGMENT}
   ${USER_FRAGMENT}
@@ -744,6 +815,14 @@ var RETRY_GENERATION = gql`
     }
   }
 `;
+var UPLOAD_ARTIFACT_FROM_URL = gql`
+  ${GENERATION_FRAGMENT}
+  mutation UploadArtifactFromUrl($input: UploadArtifactInput!) {
+    uploadArtifact(input: $input) {
+      ...GenerationFragment
+    }
+  }
+`;
 var BoardRole = /* @__PURE__ */ ((BoardRole2) => {
   BoardRole2["VIEWER"] = "VIEWER";
   BoardRole2["EDITOR"] = "EDITOR";
@@ -758,14 +837,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
@@ -1264,6 +1343,198 @@ function useGenerators(options = {}) {
   };
 }
 
+// src/hooks/useUpload.ts
+import { useCallback as useCallback6, useState as useState5 } from "react";
+import { useMutation as useMutation4 } from "urql";
+function useUpload() {
+  const [isUploading, setIsUploading] = useState5(false);
+  const [progress, setProgress] = useState5(0);
+  const [error, setError] = useState5(null);
+  const { apiUrl } = useApiConfig();
+  const auth = useAuth();
+  const [, uploadFromUrlMutation] = useMutation4(UPLOAD_ARTIFACT_FROM_URL);
+  const upload = useCallback6(
+    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
+import { useMemo as useMemo5 } from "react";
+import { useQuery as useQuery4 } from "urql";
+function useAncestry(generationId, options = {}) {
+  const { maxDepth = 25, pause = false } = options;
+  const [{ data, fetching, error }] = useQuery4({
+    query: GET_ANCESTRY,
+    variables: { id: generationId, maxDepth },
+    pause
+  });
+  const ancestry = useMemo5(
+    () => 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 }] = useQuery4({
+    query: GET_DESCENDANTS,
+    variables: { id: generationId, maxDepth },
+    pause
+  });
+  const descendants = useMemo5(
+    () => 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 }] = useQuery4({
+    query: GET_INPUT_ARTIFACTS,
+    variables: { id: generationId },
+    pause
+  });
+  const inputArtifacts = useMemo5(
+    () => 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
 import { Provider as UrqlProvider } from "urql";
 import { jsx as jsx4 } from "react/jsx-runtime";
@@ -1296,6 +1567,8 @@ function BoardsProvider({
 var VERSION = "0.1.0";
 export {
   ADD_BOARD_MEMBER,
+  ANCESTRY_NODE_FRAGMENT,
+  ARTIFACT_LINEAGE_FRAGMENT,
   ArtifactType,
   AuthProvider,
   BOARD_FRAGMENT,
@@ -1306,13 +1579,17 @@ export {
   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,
@@ -1320,6 +1597,7 @@ export {
   RETRY_GENERATION,
   UPDATE_BOARD,
   UPDATE_BOARD_MEMBER_ROLE,
+  UPLOAD_ARTIFACT_FROM_URL,
   USER_FRAGMENT,
   VERSION,
   createGraphQLClient,
@@ -1328,13 +1606,18 @@ export {
   parseArtifactSlot,
   parseGeneratorSchema,
   parseSettingsField,
+  useAncestry,
   useApiConfig,
   useAuth,
   useAuthOptional,
   useBoard,
   useBoards,
+  useDescendants,
   useGeneration,
   useGeneratorSelection,
-  useGenerators
+  useGenerators,
+  useInputArtifacts,
+  useLineage,
+  useUpload
 };
 //# sourceMappingURL=index.mjs.map