comfy-pr 0.2.24 → 0.2.25

package/src/createIssueComment.ts

@@ -1,29 +1,29 @@
-import DIE from "@snomiao/die";
-import { gh } from "./gh";
-import { ghUser } from "./ghUser";
-import { parseIssueUrl } from "./parseIssueUrl";
-
-if (import.meta.main) {
-  const url = "https://github.com/snomiao/ComfyNode-Registry-test/pull/1";
-  const body = "Hello World @snomiao";
-  const result = await createIssueComment(url, body, ghUser.login);
-  console.log(result.comment.html_url);
-  console.log(result.comments.map((e) => e.html_url));
-}
-
-export async function createIssueComment(issueUrl: string, body: string, by: string) {
-  const comments = (
-    await gh.issues.listComments({
-      ...parseIssueUrl(issueUrl),
-    })
-  ).data;
-  const result = await (async function () {
-    const commentExisted = comments.find((e) => e.body === body);
-    if (commentExisted) return { comment: commentExisted, comments };
-    if (by !== ghUser.login) DIE("Fails to creating issue: user not match");
-    const comment = (await gh.issues.createComment({ ...parseIssueUrl(issueUrl), body })).data;
-    console.log("comment created: " + comment.html_url);
-    return { comment, comments: [...comments, comment] };
-  })();
-  return result;
-}
+import DIE from "@snomiao/die";
+import { gh } from "./gh";
+import { ghUser } from "./ghUser";
+import { parseIssueUrl } from "./parseIssueUrl";
+
+if (import.meta.main) {
+  const url = "https://github.com/snomiao/ComfyNode-Registry-test/pull/1";
+  const body = "Hello World @snomiao";
+  const result = await createIssueComment(url, body, ghUser.login);
+  console.log(result.comment.html_url);
+  console.log(result.comments.map((e) => e.html_url));
+}
+
+export async function createIssueComment(issueUrl: string, body: string, by: string) {
+  const comments = (
+    await gh.issues.listComments({
+      ...parseIssueUrl(issueUrl),
+    })
+  ).data;
+  const result = await (async function () {
+    const commentExisted = comments.find((e) => e.body === body);
+    if (commentExisted) return { comment: commentExisted, comments };
+    if (by !== ghUser.login) DIE("Fails to creating issue: user not match");
+    const comment = (await gh.issues.createComment({ ...parseIssueUrl(issueUrl), body })).data;
+    console.log("comment created: " + comment.html_url);
+    return { comment, comments: [...comments, comment] };
+  })();
+  return result;
+}

package/src/fetchCMNodes.ts

@@ -1,22 +1,22 @@
-import { fetchJson } from "./utils/fetchJson";
-
-if (import.meta.main) {
-  console.log(await fetchCMNodes());
-}
-export async function fetchCMNodes() {
-  const customNodeListSource =
-    process.env.CUSTOM_LIST_SOURCE ||
-    "https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json";
-  const nodeList = (await fetchJson(customNodeListSource)) as {
-    custom_nodes: {
-      author: "Dr.Lt.Data" | string;
-      title: "ComfyUI-Manager" | string;
-      id: "manager" | string;
-      reference: "https://github.com/ltdrdata/ComfyUI-Manager" | string;
-      files: ["https://github.com/ltdrdata/ComfyUI-Manager"] | string[];
-      install_type: "git-clone" | string;
-      description: "ComfyUI-Manager itself is also a custom node." | string;
-    }[];
-  };
-  return nodeList.custom_nodes;
-}
+import { fetchJson } from "./utils/fetchJson";
+
+if (import.meta.main) {
+  console.log(await fetchCMNodes());
+}
+export async function fetchCMNodes() {
+  const customNodeListSource =
+    process.env.CUSTOM_LIST_SOURCE ||
+    "https://raw.githubusercontent.com/ltdrdata/ComfyUI-Manager/main/custom-node-list.json";
+  const nodeList = (await fetchJson(customNodeListSource)) as {
+    custom_nodes: {
+      author: "Dr.Lt.Data" | string;
+      title: "ComfyUI-Manager" | string;
+      id: "manager" | string;
+      reference: "https://github.com/ltdrdata/ComfyUI-Manager" | string;
+      files: ["https://github.com/ltdrdata/ComfyUI-Manager"] | string[];
+      install_type: "git-clone" | string;
+      description: "ComfyUI-Manager itself is also a custom node." | string;
+    }[];
+  };
+  return nodeList.custom_nodes;
+}

package/src/fetchComfyRegistryNodes.ts

@@ -1,11 +1,11 @@
-import type { mockPublishedNodes } from "../mocks/mockPublishedNodes";
-import DIE from "@snomiao/die";
-import { fetchJson } from "./utils/fetchJson";
-
-export async function fetchCRNodes() {
-  const r = (await fetchJson<typeof mockPublishedNodes>(
-    "https://api.comfy.org/nodes?page=1&limit=99999999"
-  )) as typeof mockPublishedNodes;
-  r.totalPages === 1 || DIE("FAIL TO FETCH ALL NODES");
-  return r.nodes;
-}
+import type { mockPublishedNodes } from "../mocks/mockPublishedNodes";
+import DIE from "@snomiao/die";
+import { fetchJson } from "./utils/fetchJson";
+
+export async function fetchCRNodes() {
+  const r = (await fetchJson<typeof mockPublishedNodes>(
+    "https://api.comfy.org/nodes?page=1&limit=99999999"
+  )) as typeof mockPublishedNodes;
+  r.totalPages === 1 || DIE("FAIL TO FETCH ALL NODES");
+  return r.nodes;
+}

package/src/fetchCurrentGeoInfo.ts

@@ -1,6 +1,6 @@
-import { fetchJson } from "./utils/fetchJson";
-export async function fetchCurrentGeoInfo() {
-  const { query, city, iat, lon, countryCode, region, regionName } = (await fetchJson("http://ip-api.com/json")) as any;
-  const geo = { ip: query, city, iat, lon, countryCode, region, regionName };
-  return geo;
-}
+import { fetchJson } from "./utils/fetchJson";
+export async function fetchCurrentGeoInfo() {
+  const { query, city, iat, lon, countryCode, region, regionName } = (await fetchJson("http://ip-api.com/json")) as any;
+  const geo = { ip: query, city, iat, lon, countryCode, region, regionName };
+  return geo;
+}

package/src/fetchRelatedPullWithComments.ts

@@ -1,15 +1,15 @@
-import pMap from "p-map";
-import { fetchIssueComments } from "./gh/fetchIssueComments";
-import { matchRelatedPulls } from "./matchRelatedPulls";
-import type { GithubPullParsed } from "./parsePullsState";
-import { summaryLastPullComment } from "./summaryLastPullComment";
-/** @deprecated */
-export async function fetchRelatedPullWithComments(repository: string, pulls: GithubPullParsed[]) {
-  const relatedPulls = await matchRelatedPulls(pulls);
-  const relatedPullsWithComment = await pMap(relatedPulls, async (data) => {
-    const comments = await fetchIssueComments(repository, data.pull);
-    const lastText = summaryLastPullComment(comments);
-    return { ...data, comments, lastText };
-  });
-  return relatedPullsWithComment;
-}
+import pMap from "p-map";
+import { fetchIssueComments } from "./gh/fetchIssueComments";
+import { matchRelatedPulls } from "./matchRelatedPulls";
+import type { GithubPullParsed } from "./parsePullsState";
+import { summaryLastPullComment } from "./summaryLastPullComment";
+/** @deprecated */
+export async function fetchRelatedPullWithComments(repository: string, pulls: GithubPullParsed[]) {
+  const relatedPulls = await matchRelatedPulls(pulls);
+  const relatedPullsWithComment = await pMap(relatedPulls, async (data) => {
+    const comments = await fetchIssueComments(repository, data.pull);
+    const lastText = summaryLastPullComment(comments);
+    return { ...data, comments, lastText };
+  });
+  return relatedPullsWithComment;
+}

package/src/fetchRepoDescriptionMap.ts

@@ -1,15 +1,15 @@
-import { fromPairs } from "rambda";
-import { fetchCMNodes } from "./fetchCMNodes";
-
-
-export async function fetchRepoDescriptionMap() {
-  const nodeList = await fetchCMNodes();
-  const repoDescriptionMap = fromPairs(
-    nodeList.map((e) => [e.reference, e.description])
-  );
-  repoDescriptionMap["https://github.com/snomiao/ComfyNode-Registry-test"] =
-    "ComfyNode-Registry-test-description";
-
-  console.log("Fetched " + nodeList.length + " CustomNode descriptions");
-  return repoDescriptionMap;
-}
+import { fromPairs } from "rambda";
+import { fetchCMNodes } from "./fetchCMNodes";
+
+
+export async function fetchRepoDescriptionMap() {
+  const nodeList = await fetchCMNodes();
+  const repoDescriptionMap = fromPairs(
+    nodeList.map((e) => [e.reference, e.description])
+  );
+  repoDescriptionMap["https://github.com/snomiao/ComfyNode-Registry-test"] =
+    "ComfyNode-Registry-test-description";
+
+  console.log("Fetched " + nodeList.length + " CustomNode descriptions");
+  return repoDescriptionMap;
+}

package/src/followRuleSchema.ts

@@ -1,76 +1,78 @@
-import { $before, $fresh, $stale } from "@/packages/mongodb-pipeline-ts/$fresh";
-import { tryCatch } from "rambda";
-import { z } from "zod";
-
-const znot_include = z.object({ "not-include": z.string() }).transform((x) => ({ $nin: x["not-include"] }));
-const mString = z.string().transform((x) =>
-  tryCatch<string, string | RegExp>(
-    (x) => new RegExp(x) as RegExp,
-    (x) => x as string | RegExp,
-  )(x),
-);
-const zbefore = z
-  .object({ $before: z.coerce.date().or(z.string()).or(z.number()) })
-  .transform((x) => $before(x.$before));
-const zfresh = z.object({ $fresh: z.coerce.date().or(z.string()).or(z.number()) }).transform((x) => $fresh(x.$fresh));
-const mstale = z.object({ $stale: z.coerce.date().or(z.string()).or(z.number()) }).transform((x) => $stale(x.$stale));
-const mDate = z.date().or(zbefore).or(zfresh).or(mstale);
-const mAny = z.number().or(znot_include).or(mString).or(zbefore).or(zfresh);
-const mNumber = z
-  .number()
-  .or(z.object({ $gt: z.number() }))
-  .or(z.object({ $lt: z.number() }))
-  .or(z.object({ $eq: z.number() }))
-  .or(z.object({ $ne: z.number() }))
-  .or(z.object({ $gte: z.number() }))
-  .or(z.object({ $lte: z.number() }));
-export const zAddCommentAction = z
-  .object({
-    by: z.string(),
-    body: z.string(),
-  })
-  .strict();
-const zFollowUpRule = z.object({
-  name: z.string(),
-  $match: z
-    .object({
-      actived_at: mDate,
-      email: mString,
-      comments: mNumber,
-      created_at: mDate,
-      comments_author: mString,
-      head: mString,
-      lastwords: mString,
-      on_registry: z.boolean(),
-      nickName: mString,
-      ownername: mString,
-      state: z.enum(["OPEN", "CLOSED", "MERGED"]),
-      updated_at: mDate,
-      url: mString,
-    })
-    .strict()
-    .partial(),
-  action: z
-    .object({
-      "add-comment": zAddCommentAction,
-      "send-email": z
-        .object({
-          provider: z.string(),
-          from: z.string(),
-          to: z.string(),
-          subject: z.string(),
-          body: z.string(),
-        })
-        .strict(), // WARN: not implementd
-      "update-issue": z
-        .object({
-          tags: mAny,
-        })
-        .strict(),
-      // close: z.any(),
-    })
-    .strict()
-    .partial(),
-});
-// zPullsStatus
-export const zFollowUpRules = zFollowUpRule.array();
+import { $before, $fresh, $stale } from "@/packages/mongodb-pipeline-ts/$fresh";
+import { tryCatch } from "rambda";
+import { z } from "zod";
+
+const znot_include = z.object({ "not-include": z.string() }).transform((x) => ({ $nin: x["not-include"] }));
+const mString = z.string().transform((x) =>
+  tryCatch<string, string | RegExp>(
+    (x) => new RegExp(x) as RegExp,
+    (x) => x as string | RegExp,
+  )(x),
+);
+const zbefore = z
+  .object({ $before: z.coerce.date().or(z.string()).or(z.number()) })
+  .transform((x) => $before(x.$before));
+const zfresh = z.object({ $fresh: z.coerce.date().or(z.string()).or(z.number()) }).transform((x) => $fresh(x.$fresh));
+const mstale = z.object({ $stale: z.coerce.date().or(z.string()).or(z.number()) }).transform((x) => $stale(x.$stale));
+const mDate = z.date().or(zbefore).or(zfresh).or(mstale);
+const mAny = z.number().or(znot_include).or(mString).or(zbefore).or(zfresh);
+const mNumber = z
+  .number()
+  .or(z.object({ $gt: z.number() }))
+  .or(z.object({ $lt: z.number() }))
+  .or(z.object({ $eq: z.number() }))
+  .or(z.object({ $ne: z.number() }))
+  .or(z.object({ $gte: z.number() }))
+  .or(z.object({ $lte: z.number() }));
+export const zAddCommentAction = z
+  .object({
+    by: z.string(),
+    body: z.string(),
+  })
+  .strict();
+export const zSendEmailAction = z
+  .object({
+    provider: z.enum(["google"]),
+    name: z.string(),
+    from: z.string(),
+    to: z.string(),
+    subject: z.string(),
+    body: z.string(),
+  })
+  .strict();
+const zFollowUpRule = z.object({
+  name: z.string(),
+  $match: z
+    .object({
+      actived_at: mDate,
+      email: mString,
+      comments: mNumber,
+      created_at: mDate,
+      comments_author: mString,
+      head: mString,
+      lastwords: mString,
+      on_registry: z.boolean(),
+      nickName: mString,
+      ownername: mString,
+      state: z.enum(["OPEN", "CLOSED", "MERGED"]),
+      updated_at: mDate,
+      url: mString,
+    })
+    .strict()
+    .partial(),
+  action: z
+    .object({
+      "add-comment": zAddCommentAction,
+      "send-email": zSendEmailAction, // WARN: not implementd
+      "update-issue": z
+        .object({
+          tags: mAny,
+        })
+        .strict(),
+      // close: z.any(),
+    })
+    .strict()
+    .partial(),
+});
+// zPullsStatus
+export const zFollowUpRules = zFollowUpRule.array();

package/src/getActivatedShell.ts

@@ -1,18 +1,18 @@
-import { $ as zx } from "zx";
-import { getActivateCMD } from "./cli/getActivateCMD";
-
-if (import.meta.main) {
-  // await checkComfyActivated();
-  zx.verbose = true;
-  const $ = getActivatedShell();
-  const p = await $`comfy-cli --version`;
-  console.log(p.stdout);
-  // zx({ prefix:  })`comfy-cli --help`;
-}
-
-export function getActivatedShell() {
-  const activate = getActivateCMD();
-  return zx({
-    prefix: `echo Comfy CLI version: $(comfy-cli --version) || (apt-get install -y python3-venv && python -m venv .venv && ${activate} && pip install comfy-cli); `,
-  });
-}
+import { $ as zx } from "zx";
+import { getActivateCMD } from "./cli/getActivateCMD";
+
+if (import.meta.main) {
+  // await checkComfyActivated();
+  zx.verbose = true;
+  const $ = getActivatedShell();
+  const p = await $`comfy-cli --version`;
+  console.log(p.stdout);
+  // zx({ prefix:  })`comfy-cli --help`;
+}
+
+export function getActivatedShell() {
+  const activate = getActivateCMD();
+  return zx({
+    prefix: `echo Comfy CLI version: $(comfy-cli --version) || (apt-get install -y python3-venv && python -m venv .venv && ${activate} && pip install comfy-cli); `,
+  });
+}

package/src/getBranchWorkingDir.ts

@@ -1,16 +1,16 @@
-import { rm } from "fs/promises";
-import { getRepoWorkingDir } from "./getRepoWorkingDir";
-import { parseUrlRepoOwner } from "./parseOwnerRepo";
-
-export async function getBranchWorkingDir(
-  upstreamUrl: string,
-  forkUrl: string,
-  branch: string
-) {
-  const src = parseUrlRepoOwner(upstreamUrl);
-  const dir = getRepoWorkingDir(forkUrl);
-  const packageName = src.repo;
-  const cwd = `${dir}/${branch}/${packageName}`;
-  await rm(cwd, { recursive: true }).catch(() => null);
-  return cwd;
-}
+import { rm } from "fs/promises";
+import { getRepoWorkingDir } from "./getRepoWorkingDir";
+import { parseUrlRepoOwner } from "./parseOwnerRepo";
+
+export async function getBranchWorkingDir(
+  upstreamUrl: string,
+  forkUrl: string,
+  branch: string
+) {
+  const src = parseUrlRepoOwner(upstreamUrl);
+  const dir = getRepoWorkingDir(forkUrl);
+  const packageName = src.repo;
+  const cwd = `${dir}/${branch}/${packageName}`;
+  await rm(cwd, { recursive: true }).catch(() => null);
+  return cwd;
+}

package/src/getRepoWorkingDir.ts

@@ -1,6 +1,6 @@
-import { parseUrlRepoOwner } from "./parseOwnerRepo";
-
-export function getRepoWorkingDir(forkUrl: string) {
-  return `prs/${parseUrlRepoOwner(forkUrl).repo}`;
-}
+import { parseUrlRepoOwner } from "./parseOwnerRepo";
+
+export function getRepoWorkingDir(forkUrl: string) {
+  return `prs/${parseUrlRepoOwner(forkUrl).repo}`;
+}
   

package/src/ghUser.ts

@@ -1,6 +1,6 @@
-import { gh } from "./gh";
-
-export const ghUser = (await gh.users.getAuthenticated()).data;
-
-console.log("Fetch Current Github User...");
-console.log(`Current Github User: ${ghUser.login} <${ghUser.email}>`);
+import { gh } from "./gh";
+
+export const ghUser = (await gh.users.getAuthenticated()).data;
+
+console.log("Fetch Current Github User...");
+console.log(`Current Github User: ${ghUser.login} <${ghUser.email}>`);

package/src/index.ts

@@ -1,22 +1,24 @@
-import "dotenv/config";
-import { checkComfyActivated } from "./checkComfyActivated";
-import { initializeFollowRules } from "./initializeFollowRules";
-import { updateAuthors } from "./updateAuthors";
-import { updateCNRepos } from "./updateCNRepos";
-import { runFollowRuleSet } from "./updateFollowRuleSet";
-import { updateSlackMessages } from "./updateSlackMessages";
-import { tLog } from "./utils/tLog";
-
-if (import.meta.main) {
-  await Promise.all([
-    // try send msgs that didn't send in last run
-    updateSlackMessages(),
-    checkComfyActivated(), // needed if make pr
-    updateCNRepos(),
-    updateAuthors(),
-  ]);
-  await initializeFollowRules();
-  await tLog("runFollowRuleSet", runFollowRuleSet);
-  console.log("All done");
-  process.exit(0);
-}
+import "dotenv/config";
+import { checkComfyActivated } from "./checkComfyActivated";
+import { updateEmailTasks } from "./EmailTasks";
+import { initializeFollowRules } from "./initializeFollowRules";
+import { updateAuthors } from "./updateAuthors";
+import { updateCNRepos } from "./updateCNRepos";
+import { runFollowRuleSet } from "./updateFollowRuleSet";
+import { updateSlackMessages } from "./updateSlackMessages";
+import { tLog } from "./utils/tLog";
+
+if (import.meta.main) {
+  await Promise.all([
+    // try send msgs that didn't send in last run
+    updateSlackMessages(),
+    checkComfyActivated(), // needed if make pr
+    updateCNRepos(),
+    updateAuthors(),
+    updateEmailTasks(),
+  ]);
+  await initializeFollowRules();
+  await tLog("runFollowRuleSet", runFollowRuleSet);
+  console.log("All done");
+  process.exit(0);
+}

package/src/initializeFollowRules.ts

@@ -1,26 +1,26 @@
-import DIE from "@snomiao/die";
-import { readFile } from "fs/promises";
-import { FollowRuleSets } from "./FollowRules";
-
-if (import.meta.main) {
-  const followRules = await initializeFollowRules();
-  const ruleset = (await FollowRuleSets.findOne({ name: "default" })) ?? DIE("default ruleset not found");
-  if (!ruleset?.enabled) DIE("default ruleset not enabled");
-
-  console.log("initialized follow rules");
-}
-/**
- *
- * @author: snomiao <snomiao@gmail.com>
- */
-export async function initializeFollowRules() {
-  return await updateFollowRules("default", await readFile("./templates/follow-rules.yaml", "utf8"));
-}
-
-export async function updateFollowRules(name: string, rawRuleYaml: string) {
-  return (await FollowRuleSets.findOneAndUpdate(
-    { name },
-    { $setOnInsert: { yaml: rawRuleYaml, enabled: false } },
-    { upsert: true, returnDocument: "after" },
-  ))!;
-}
+import DIE from "@snomiao/die";
+import { readFile } from "fs/promises";
+import { FollowRuleSets } from "./FollowRules";
+
+if (import.meta.main) {
+  const followRules = await initializeFollowRules();
+  const ruleset = (await FollowRuleSets.findOne({ name: "default" })) ?? DIE("default ruleset not found");
+  if (!ruleset?.enabled) DIE("default ruleset not enabled");
+
+  console.log("initialized follow rules");
+}
+/**
+ *
+ * @author: snomiao <snomiao@gmail.com>
+ */
+export async function initializeFollowRules() {
+  return await updateFollowRules("default", await readFile("./templates/follow-rules.yaml", "utf8"));
+}
+
+export async function updateFollowRules(name: string, rawRuleYaml: string) {
+  return (await FollowRuleSets.findOneAndUpdate(
+    { name },
+    { $setOnInsert: { yaml: rawRuleYaml, enabled: false } },
+    { upsert: true, returnDocument: "after" },
+  ))!;
+}