@difizen/magent-au/src/knowledge/knowledge-manager.ts

import { Fetcher } from '@difizen/magent-core';
import { inject, prop, singleton } from '@difizen/mana-app';

import {
  KnowledgeModelFactory,
  type KnowledgeModel,
  type KnowledgeModelOption,
} from './protocol.js';

@singleton()
export class KnowledgeManager {
  protected cache: Map<string, KnowledgeModel> = new Map<string, KnowledgeModel>();
  @inject(KnowledgeModelFactory) factory: KnowledgeModelFactory;
  @inject(Fetcher) fetcher: Fetcher;

  @prop()
  knowledgeOptions: KnowledgeModelOption[] = [];

  getAll = async (): Promise<KnowledgeModelOption[]> => {
    const defaultValue: KnowledgeModelOption[] = [];
    const res = await this.fetcher.get<KnowledgeModelOption[]>(`/api/v1/knowledge`);
    if (res.status === 200) {
      return res.data;
    }
    return defaultValue;
  };

  createKnowledge = async (option: KnowledgeModelOption): Promise<string> => {
    const res = await this.fetcher.post<string>(`/api/v1/knowledge`, option);
    return res.data;
  };

  updateKnowledge = async (option: KnowledgeModelOption): Promise<string> => {
    const res = await this.fetcher.put<string>(`/api/v1/knowledge/${option.id}`, {
      nickname: option.nickname,
      description: option.description,
    });
    return res.data;
  };

  deleteKnowledge = async (knowledge_id: string): Promise<boolean> => {
    const res = await this.fetcher.delete<string>(`/api/v1/knowledge/${knowledge_id}`);
    return Boolean(res.data);
  };

  getOrCreate = (option: KnowledgeModelOption): KnowledgeModel => {
    const exist = this.cache.get(option.id);
    if (exist) {
      return exist;
    }
    const tool = this.factory(option);
    this.cache.set(tool.id, tool);
    return tool;
  };
}