1export class Gem {
2  constructor(
3    public id: string,
4    public name: string,
5    public description: string | null,
6    public prompt: string | null,
7    public predefined: boolean,
8  ) {}
9 
10  toString(): string {
11    return `Gem(id='${this.id}', name='${this.name}', description='${this.description}', prompt='${this.prompt}', predefined=${this.predefined})`;
12  }
13}
14 
15export class GemJar implements Iterable<Gem> {
16  private m = new Map<string, Gem>();
17 
18  constructor(entries?: Iterable<[string, Gem]>) {
19    if (entries) for (const [id, gem] of entries) this.m.set(id, gem);
20  }
21 
22  [Symbol.iterator](): Iterator<Gem> {
23    return this.m.values();
24  }
25 
26  entries(): IterableIterator<[string, Gem]> {
27    return this.m.entries();
28  }
29 
30  values(): IterableIterator<Gem> {
31    return this.m.values();
32  }
33 
34  has(id: string): boolean {
35    return this.m.has(id);
36  }
37 
38  set(id: string, gem: Gem): this {
39    this.m.set(id, gem);
40    return this;
41  }
42 
43  get(id?: string | null, name?: string | null, def: Gem | null = null): Gem | null {
44    if (id == null && name == null) {
45      throw new Error('At least one of gem id or name must be provided.');
46    }
47 
48    if (id != null) {
49      const g = this.m.get(id) ?? null;
50      if (!g) return def;
51      if (name != null) return g.name === name ? g : def;
52      return g;
53    }
54 
55    if (name != null) {
56      for (const g of this.m.values()) {
57        if (g.name === name) return g;
58      }
59      return def;
60    }
61 
62    return def;
63  }
64 
65  filter(predefined: boolean | null = null, name: string | null = null): GemJar {
66    const out: [string, Gem][] = [];
67    for (const [id, gem] of this.m.entries()) {
68      if (predefined != null && gem.predefined !== predefined) continue;
69      if (name != null && gem.name !== name) continue;
70      out.push([id, gem]);
71    }
72    return new GemJar(out);
73  }
74}
75 
76