1import { GeneratedImage, type Image, WebImage } from './image.js';
2 
3function decode_html(s: string | null | undefined): string | null | undefined {
4  if (s == null) return s;
5  return s
6    .replace(/&/g, '&')
7    .replace(/&lt;/g, '<')
8    .replace(/&gt;/g, '>')
9    .replace(/&quot;/g, '"')
10    .replace(/&apos;/g, "'")
11    .replace(/&#39;/g, "'")
12    .replace(/&nbsp;/g, ' ')
13    .replace(/&#x([0-9a-fA-F]+);/g, (_, hex) => String.fromCodePoint(parseInt(hex, 16)))
14    .replace(/&#(\d+);/g, (_, dec) => String.fromCodePoint(parseInt(dec, 10)));
15}
16 
17export class Candidate {
18  public rcid: string;
19  public text: string;
20  public thoughts: string | null;
21  public web_images: WebImage[];
22  public generated_images: GeneratedImage[];
23 
24  constructor(params: {
25    rcid: string;
26    text: string;
27    thoughts?: string | null;
28    web_images?: WebImage[];
29    generated_images?: GeneratedImage[];
30  }) {
31    this.rcid = params.rcid;
32    this.text = decode_html(params.text) ?? '';
33    this.thoughts = decode_html(params.thoughts) ?? null;
34    this.web_images = params.web_images ?? [];
35    this.generated_images = params.generated_images ?? [];
36  }
37 
38  toString(): string {
39    return this.text;
40  }
41 
42  get images(): Image[] {
43    return [...this.web_images, ...this.generated_images];
44  }
45}
46