feat: Now it will fetches the latest git commit hash when you pick a custom nodes

This commit is contained in:
BennyKok 2024-01-18 01:00:13 +08:00
parent 56eb91d102
commit 1ae295b806
2 changed files with 68 additions and 3 deletions

View File

@ -343,7 +343,7 @@ export function MachineList({
: undefined
}
title="New Machine"
description="Add custom Comfyui machines to your account."
description="Add custom ComfyUI machines to your account."
serverAction={addCustomMachine}
formSchema={addCustomMachineSchema}
fieldConfig={{

View File

@ -27,7 +27,9 @@ import { cn } from "@/lib/utils";
import { findAllDeployments } from "@/server/curdDeploments";
import { Check, ChevronsUpDown } from "lucide-react";
import * as React from "react";
import { toast } from "sonner";
import useSWR from "swr";
import { z } from "zod";
export function SnapshotPickerView({
field,
@ -161,6 +163,24 @@ type CustomNodeList = {
}[];
};
const RepoSchema = z.object({
default_branch: z.string(),
});
const BranchInfoSchema = z.object({
commit: z.object({
sha: z.string(),
}),
});
function extractRepoName(repoUrl: string) {
const url = new URL(repoUrl);
const pathParts = url.pathname.split("/");
const repoName = pathParts[2].replace(".git", "");
const author = pathParts[1];
return `${author}/${repoName}`;
}
function CustomNodesSelector({
field,
}: Pick<AutoFormInputComponentProps, "field">) {
@ -216,7 +236,52 @@ function CustomNodesSelector({
<CommandItem
key={index}
value={framework.reference}
onSelect={(currentValue) => {
onSelect={async (currentValue) => {
const repoName = extractRepoName(currentValue);
console.log(repoName);
const id = toast.loading(`Fetching repo info...`);
// toast.info("hi");
const repo = await fetch(
`https://api.github.com/repos/${repoName}`
)
.then((x) => x.json())
.then((x) => {
console.log(x);
return x;
})
.then((x) => RepoSchema.parse(x))
.catch((e) => {
console.error(e);
toast.dismiss(id);
toast.error(`Failed to fetch repo info ${e.message}`);
return null;
});
if (!repo) return;
const branch = repo.default_branch;
const branchInfo = await fetch(
`https://api.github.com/repos/${repoName}/branches/${branch}`
)
.then((x) => x.json())
.then((x) => BranchInfoSchema.parse(x))
.catch((e) => {
console.error(e);
toast.dismiss(id);
toast.error(
`Failed to fetch branch info ${e.message}`
);
return null;
});
toast.dismiss(id);
if (!branchInfo) return;
let nodeList: Record<
string,
{
@ -233,7 +298,7 @@ function CustomNodesSelector({
} else {
nodeList = {
[currentValue]: {
hash: "latest",
hash: branchInfo?.commit.sha,
disabled: false,
},
...x,