@kevisual/cnb 0.0.56 → 0.0.57

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/cnb",
-  "version": "0.0.56",
+  "version": "0.0.57",
   "description": "",
   "main": "index.js",
   "basename": "/root/cnb",

package/src/index.ts

@@ -8,6 +8,7 @@ import { Issue } from "./issue/index.ts";
 import { Mission } from "./mission/index.ts";
 import { AiBase } from "./ai/index.ts";
 import { RepoLabel, IssueLabel } from "./labels/index.ts";
+import { RegistryPackage, PackageManagement } from "./package/index.ts";
 
 type CNBOptions = CNBCoreOptions<{
 }>;
@@ -25,6 +26,10 @@ export class CNB extends CNBCore {
     repoLabel: RepoLabel;
     issueLabel: IssueLabel;
   };
+  packages!: {
+    registry: RegistryPackage;
+    package: PackageManagement;
+  };
   constructor(options: CNBOptions) {
     super({ ...options, token: options.token, cookie: options.cookie, cnb: options.cnb });
     this.init(options);
@@ -47,6 +52,10 @@ export class CNB extends CNBCore {
       repoLabel: new RepoLabel(options),
       issueLabel: new IssueLabel(options),
     };
+    this.packages = {
+      registry: new RegistryPackage(options),
+      package: new PackageManagement(options),
+    };
   }
   setToken(token: string) {
     this.token = token;
@@ -58,6 +67,8 @@ export class CNB extends CNBCore {
     this.mission.token = token;
     this.labels.repoLabel.token = token;
     this.labels.issueLabel.token = token;
+    this.packages.registry.token = token;
+    this.packages.package.token = token;
   }
   setCookie(cookie: string) {
     this.cookie = cookie;
@@ -69,6 +80,8 @@ export class CNB extends CNBCore {
     this.mission.cookie = cookie;
     this.labels.repoLabel.cookie = cookie;
     this.labels.issueLabel.cookie = cookie;
+    this.packages.registry.cookie = cookie;
+    this.packages.package.cookie = cookie;
   }
   getCNBVersion = getCNBVersion
 }
@@ -95,4 +108,5 @@ type VersionInfo = {
 }
 
 export * from './issue/npc/env.ts'
-export * from './labels/index.ts'
+export * from './labels/index.ts'
+export * from './package/index.ts'

package/src/issue/index.ts

@@ -63,8 +63,11 @@ export type IssueItem = {
   number: string;
   priority: string;
   started_at: string;
-  state: string;
-  state_reason: string;
+  state: 'open' | 'closed';
+  /**
+   * state_reason must be one of completed, not_planned, reopened
+   */
+  state_reason: 'completed' | 'not_planned' | 'reopened';
   title: string;
   updated_at: string;
 };

package/src/knowledge/index.ts

@@ -1,5 +1,6 @@
 import { CNBCore, CNBCoreOptions, Result } from "../cnb-core.ts";
 
+// https://cnb.cool/cnb/plugins/cnbcool/knowledge-base/-/blob/main/src/api.py
 export class KnowledgeBase extends CNBCore {
   constructor(options: CNBCoreOptions) {
     super(options);
@@ -12,10 +13,7 @@ export class KnowledgeBase extends CNBCore {
     metadata_filtering_conditions?: MetadataFilteringConditions
   }): Promise<Result<QueryRag[]>> {
     const url = `/${repo}/-/knowledge/base/query`;
-    let postData = {
-      query: data.query,
-    };
-    return this.post({ url, data: postData });
+    return this.post({ url, data });
   }
   getEmbeddingModels(repo: string): Promise<Result<Array<{ name: string, dimensions: number }>>> {
     const url = `/${repo}/-/knowledge/embedding/models`;
@@ -31,6 +29,95 @@ export class KnowledgeBase extends CNBCore {
     const url = `/${repo}/-/knowledge/base`;
     return this.request({ url, method: 'DELETE' });
   }
+
+  /**
+   * 未暴露
+   * @param repo 
+   * @param data 
+   * @returns 
+   */
+  createKnowledgeBase(repo: string, data: {
+    embedding_model_name: string;
+    include: string;
+    exclude: string;
+    issue_sync_enabled?: boolean;
+    processing?: {
+      chunk_size: number;
+      chunk_overlap: number;
+      text_separator: string;
+    };
+    issue?: {
+      labels: string;
+      state: string;
+    };
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/base`;
+    return this.post({ url, data });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param data 
+   * @returns
+   */
+  updateKnowledgeBase(repo: string, data: {
+    last_commit_sha?: string;
+    issue_last_sync_time?: string;
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/base`;
+    return this.put({ url, data });
+  }
+
+  /**
+   * 未暴露
+   * @param repo 
+   * @param text 
+   * @returns 
+   */
+  getEmbedding(repo: string, text: string): Promise<Result<{ embedding: number[] }>> {
+    const url = `/${repo}/-/knowledge/embedding`;
+    return this.post({ url, data: { text } });
+  }
+  /**
+   *  未暴露
+   * @param repo 
+   * @param chunksData 
+   * @returns 
+   */
+  addDocument(repo: string, chunksData: {
+    path: string;
+    chunks: Array<{
+      content: string;
+      hash: string;
+      offset: number;
+      size: number;
+    }>;
+  }): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/documents/upsert-document-with-chunks`;
+    return this.post({ url, data: chunksData });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param paths 
+   * @returns 
+   */
+
+  deleteDocument(repo: string, paths: string[]): Promise<Result<any>> {
+    const url = `/${repo}/-/knowledge/documents`;
+    return this.delete({ url, data: { paths } });
+  }
+  /**
+   * 未暴露
+   * @param repo 
+   * @param page 
+   * @param page_size 
+   * @returns 
+   */
+  listDocument(repo: string, page: number = 1, page_size: number = 50): Promise<Result<any[]>> {
+    const url = `/${repo}/-/knowledge/documents`;
+    return this.get({ url, params: { page, page_size } });
+  }
 }
 
 type MetadataFilteringConditions = {
@@ -42,7 +129,7 @@ type MetadataFilteringConditions = {
      */
     value?: string
   }>
-  logical_operator?: 'adn' | 'or'
+  logical_operator?: 'and' | 'or'
 }
 
 type QueryRag = {

package/src/package/index.ts

@@ -0,0 +1,2 @@
+export * from './registry.ts';
+export * from './package.ts';

package/src/package/package.ts

@@ -0,0 +1,134 @@
+import { CNBCore, CNBCoreOptions, Result } from "../cnb-core.ts";
+import type { PackageType, Package, ListPackagesParams } from "./registry.ts";
+
+/**
+ * 制品管理
+ */
+export class PackageManagement extends CNBCore {
+  constructor(options: CNBCoreOptions) {
+    super(options);
+  }
+
+  /**
+   * 查询制品列表
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param params 查询参数
+   */
+  list(
+    slug: string,
+    type: PackageType,
+    params?: ListPackagesParams & { page?: number; page_size?: number }
+  ): Promise<Result<Package[]>> {
+    const url = `/${slug}/-/packages`;
+    return this.get({
+      url,
+      params: {
+        type,
+        ...params,
+      },
+      headers: {
+        Accept: "application/vnd.cnb.api+json",
+      },
+    });
+  }
+
+  /**
+   * 获取制品详情
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param name 制品名称
+   */
+  getOne(
+    slug: string,
+    type: PackageType,
+    name: string
+  ): Promise<Result<any>> {
+    const url = `/${slug}/-/packages/${type}/${encodeURIComponent(name)}`;
+    return this.get({
+      url,
+      headers: {
+        Accept: "application/vnd.cnb.api+json",
+      },
+    });
+  }
+
+  /**
+   * 删除制品
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param name 制品名称
+   */
+  remove(
+    slug: string,
+    type: PackageType,
+    name: string
+  ): Promise<Result<void>> {
+    const url = `/${slug}/-/packages/${type}/${encodeURIComponent(name)}`;
+    return this.delete({ url });
+  }
+
+  /**
+   * 获取制品标签详情
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param name 制品名称
+   * @param tag 标签名称
+   */
+  getTag(
+    slug: string,
+    type: PackageType,
+    name: string,
+    tag: string
+  ): Promise<Result<any>> {
+    const url = `/${slug}/-/packages/${type}/${encodeURIComponent(name)}/tags/${encodeURIComponent(tag)}`;
+    return this.get({
+      url,
+      headers: {
+        Accept: "application/vnd.cnb.api+json",
+      },
+    });
+  }
+
+  /**
+   * 删除制品标签
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param name 制品名称
+   * @param tag 标签名称
+   */
+  removeTag(
+    slug: string,
+    type: PackageType,
+    name: string,
+    tag: string
+  ): Promise<Result<void>> {
+    const url = `/${slug}/-/packages/${type}/${encodeURIComponent(name)}/tags/${encodeURIComponent(tag)}`;
+    return this.delete({ url });
+  }
+
+  /**
+   * 获取制品标签列表
+   * @param slug 资源路径
+   * @param type 制品类型
+   * @param name 制品名称
+   * @param params 查询参数
+   */
+  listTags(
+    slug: string,
+    type: PackageType,
+    name: string,
+    params?: { page?: number; page_size?: number }
+  ): Promise<Result<any>> {
+    const url = `/${slug}/-/packages/${type}/${encodeURIComponent(name)}/tags`;
+    return this.get({
+      url,
+      params,
+      headers: {
+        Accept: "application/vnd.cnb.api+json",
+      },
+    });
+  }
+}
+
+export type { PackageType, Package, ListPackagesParams };

package/src/package/registry.ts

@@ -0,0 +1,145 @@
+import { CNBCore, CNBCoreOptions, Result } from "../cnb-core.ts";
+
+/**
+ * 制品类型
+ */
+export type PackageType =
+  | "all"
+  | "docker"
+  | "helm"
+  | "docker-model"
+  | "maven"
+  | "npm"
+  | "ohpm"
+  | "pypi"
+  | "nuget"
+  | "composer"
+  | "conan"
+  | "cargo";
+
+/**
+ * 制品库可见性类型
+ */
+export type VisibilityType = "private" | "public";
+
+/**
+ * 制品库排序类型
+ */
+export type PackageOrderingType = "pull_count" | "last_push_at" | "name_ascend" | "name_descend";
+
+/**
+ * 最后推送者信息
+ */
+export type LastPusher = {
+  is_frozen: boolean;
+  is_lock: boolean;
+  name: string;
+  username: string;
+};
+
+/**
+ * 制品基本信息
+ */
+export type Package = {
+  count: number;
+  description: string;
+  labels: string[];
+  last_artifact_name: string;
+  last_pusher: LastPusher;
+  name: string;
+  package: string;
+  package_type: PackageType;
+  pull_count: number;
+  recent_pull_count: number;
+};
+
+/**
+ * 制品库信息
+ */
+export type Registry = {
+  created_at: string;
+  description: string;
+  name: string;
+  owner: string;
+  type: PackageType;
+  visibility: VisibilityType;
+};
+
+/**
+ * 制品库列表查询参数
+ */
+export type ListGroupRegistriesParams = {
+  /** 页码 */
+  page?: number;
+  /** 每页数量 */
+  page_size?: number;
+  /** 制品仓库类型 */
+  registry_type?: "npm" | "maven" | "ohpm";
+  /** 制品仓库可见性类型 */
+  filter_type?: VisibilityType;
+  /** 排序字段 */
+  order_by?: "created_at" | "name";
+};
+
+/**
+ * 制品列表查询参数
+ */
+export type ListPackagesParams = {
+  /** 顺序类型 */
+  ordering?: PackageOrderingType;
+  /** 关键字，搜索制品名称 */
+  name?: string;
+};
+
+/**
+ * 制品库管理
+ */
+export class RegistryPackage extends CNBCore {
+  constructor(options: CNBCoreOptions) {
+    super(options);
+  }
+
+  /**
+   * 查询组织下面用户有权限查看到的制品仓库
+   * @param slug 组织 slug
+   * @param params 查询参数
+   */
+  listGroupRegistries(
+    slug: string,
+    params?: ListGroupRegistriesParams
+  ): Promise<Result<Registry[]>> {
+    const url = `/${slug}/-/registries`;
+    return this.get({
+      url,
+      params,
+      headers: {
+        Accept: "application/vnd.cnb.api+json",
+      },
+    });
+  }
+
+  /**
+   * 设置制品库可见性
+   * @param registry 制品库路径
+   * @param data 可见性设置
+   */
+  setVisibility(
+    registry: string,
+    data: { visibility: VisibilityType }
+  ): Promise<Result<void>> {
+    const url = `/${registry}/-/settings/set_visibility`;
+    return this.post({
+      url,
+      data,
+    });
+  }
+
+  /**
+   * 删除制品库
+   * @param registry 制品库路径
+   */
+  remove(registry: string): Promise<Result<void>> {
+    const url = `/${registry}`;
+    return this.delete({ url });
+  }
+}