1import { GRPC } from '../constants.js';
2import { APIError } from '../exceptions.js';
3import { Gem, GemJar, RPCData } from '../types/index.js';
4import { logger } from '../utils/logger.js';
5import { extract_json_from_response, get_nested_value } from '../utils/parsing.js';
6 
7export abstract class GemMixin {
8  protected _gems: GemJar | null = null;
9 
10  protected abstract _run<T>(fn: () => Promise<T>, retry: number): Promise<T>;
11  protected abstract _batch_execute(payloads: RPCData[], opts?: RequestInit): Promise<Response>;
12  protected abstract close(delay?: number): Promise<void>;
13 
14  get gems(): GemJar {
15    if (this._gems == null) {
16      throw new Error(
17        'Gems not fetched yet. Call `GeminiClient.fetch_gems()` method to fetch gems from gemini.google.com.',
18      );
19    }
20    return this._gems;
21  }
22 
23  async fetch_gems(include_hidden: boolean = false, opts?: RequestInit): Promise<GemJar> {
24    return await this._run(async () => {
25      const res = await this._batch_execute(
26        [
27          new RPCData(GRPC.LIST_GEMS, include_hidden ? '[4]' : '[3]', 'system'),
28          new RPCData(GRPC.LIST_GEMS, '[2]', 'custom'),
29        ],
30        opts,
31      );
32 
33      let response_json: unknown;
34      try {
35        response_json = extract_json_from_response(await res.text());
36        if (!Array.isArray(response_json)) throw new Error('Invalid response');
37      } catch {
38        await this.close();
39        throw new APIError('Failed to fetch gems. Invalid response data received. Client will try to re-initialize on next request.');
40      }
41 
42      let predefined: unknown[] = [];
43      let custom: unknown[] = [];
44 
45      try {
46        for (const part of response_json as unknown[]) {
47          if (!Array.isArray(part)) continue;
48          const ident = part[part.length - 1];
49          const body = get_nested_value<string | null>(part, [2], null);
50          if (!body) continue;
51 
52          if (ident === 'system') {
53            const parsed = JSON.parse(body) as unknown[];
54            predefined = (Array.isArray(parsed) ? (parsed[2] as unknown[]) : []) ?? [];
55          } else if (ident === 'custom') {
56            const parsed = JSON.parse(body) as unknown[] | null;
57            if (parsed) custom = (parsed[2] as unknown[]) ?? [];
58          }
59        }
60 
61        if (predefined.length === 0 && custom.length === 0) throw new Error('No gems');
62      } catch {
63        await this.close();
64        logger.debug('Invalid response while parsing gems');
65        throw new APIError('Failed to fetch gems. Invalid response data received. Client will try to re-initialize on next request.');
66      }
67 
68      const entries: [string, Gem][] = [];
69 
70      for (const gem of predefined) {
71        if (!Array.isArray(gem)) continue;
72        const id = String(get_nested_value(gem, [0], ''));
73        if (!id) continue;
74        entries.push([
75          id,
76          new Gem(
77            id,
78            String(get_nested_value(gem, [1, 0], '')),
79            get_nested_value<string | null>(gem, [1, 1], null),
80            get_nested_value<string | null>(gem, [2, 0], null),
81            true,
82          ),
83        ]);
84      }
85 
86      for (const gem of custom) {
87        if (!Array.isArray(gem)) continue;
88        const id = String(get_nested_value(gem, [0], ''));
89        if (!id) continue;
90        entries.push([
91          id,
92          new Gem(
93            id,
94            String(get_nested_value(gem, [1, 0], '')),
95            get_nested_value<string | null>(gem, [1, 1], null),
96            get_nested_value<string | null>(gem, [2, 0], null),
97            false,
98          ),
99        ]);
100      }
101 
102      this._gems = new GemJar(entries);
103      return this._gems;
104    }, 2);
105  }
106 
107  async create_gem(name: string, prompt: string, description: string = ''): Promise<Gem> {
108    return await this._run(async () => {
109      const payload = JSON.stringify([
110        [
111          name,
112          description,
113          prompt,
114          null,
115          null,
116          null,
117          null,
118          null,
119          0,
120          null,
121          1,
122          null,
123          null,
124          null,
125          [],
126        ],
127      ]);
128 
129      const res = await this._batch_execute([new RPCData(GRPC.CREATE_GEM, payload)]);
130      try {
131        const response_json = extract_json_from_response(await res.text()) as unknown[];
132        const gem_id = JSON.parse(String((response_json[0] as unknown[])[2]))[0] as string;
133        return new Gem(gem_id, name, description, prompt, false);
134      } catch {
135        await this.close();
136        throw new APIError('Failed to create gem. Invalid response data received. Client will try to re-initialize on next request.');
137      }
138    }, 2);
139  }
140 
141  async update_gem(gem: Gem | string, name: string, prompt: string, description: string = ''): Promise<Gem> {
142    return await this._run(async () => {
143      const gem_id = typeof gem === 'string' ? gem : gem.id;
144      const payload = JSON.stringify([
145        gem_id,
146        [
147          name,
148          description,
149          prompt,
150          null,
151          null,
152          null,
153          null,
154          null,
155          0,
156          null,
157          1,
158          null,
159          null,
160          null,
161          [],
162          0,
163        ],
164      ]);
165 
166      await this._batch_execute([new RPCData(GRPC.UPDATE_GEM, payload)]);
167      return new Gem(gem_id, name, description, prompt, false);
168    }, 2);
169  }
170 
171  async delete_gem(gem: Gem | string, opts?: RequestInit): Promise<void> {
172    return await this._run(async () => {
173      const gem_id = typeof gem === 'string' ? gem : gem.id;
174      const payload = JSON.stringify([gem_id]);
175      await this._batch_execute([new RPCData(GRPC.DELETE_GEM, payload)], opts);
176    }, 2);
177  }
178}
179 
180