import { Networking } from '../../../../Network';
import { BaseTool } from './BaseTool';
import { v4 as uuidv4 } from 'uuid';

export class WebsiteInfoScraperTool extends BaseTool<{ url: string }> {
    private _addLinkedUrlDoc: (url: string, id: string) => void;

    constructor(addLinkedUrlDoc: (url: string, id: string) => void) {
        super(
            'websiteInfoScraper',
            'Scrape detailed information from a specific website identified as the most relevant',
            {
                url: {
                    type: 'string',
                    description: 'The URL of the website to scrape',
                    required: true,
                },
            },
            'Provide the URL of the website that you have identified as the most relevant from the previous search. This tool will scrape and process detailed information from that specific website. It will also create a document from the scraped content for future reference.',
            'Returns the full HTML content from the provided URL and creates a document from the content for further analysis.'
        );
        this._addLinkedUrlDoc = addLinkedUrlDoc;
    }

    async execute(args: { url: string }): Promise<any> {
        try {
            const { website_image_base64 } = await Networking.PostToServer('/scrapeWebsite', { url: args.url });
            const id = uuidv4();
            this._addLinkedUrlDoc(args.url, id);
            return [
                { type: 'text', text: `<chunk chunk_id=${id} chunk_type=url> ` },
                {
                    type: 'image_url',
                    image_url: {
                        url: `data:image/jpeg;base64,${website_image_base64}`,
                    },
                },
                { type: 'text', text: `</chunk>\n` },
            ];
        } catch (error) {
            return [{ type: 'text', text: 'An error occurred while scraping the website.' }];
        }
    }
}