import { Storage } from '@google-cloud/storage';
import { tool } from 'ai';
import { z } from 'zod';
import { generateUUID } from '../utils';

const storage = new Storage();
const bucketName = process.env.GOOGLE_STORAGE_BUCKET || 'prodeen-tmp';

export const fetchFile = () => {
  return tool({
    description: 'Fetch a file based on it's name',
    inputSchema: z.object({
      name: z.string().describe('Name o the remote file')
    }),
    execute: async ({ name }) => {
      // Logic of fetching the file
      // ...
      // Download saved file from remote storage and convert to base64
      const [downloadedBuffer] = await file.download();
      const base64 = downloadedBuffer.toString('base64');

      return {
        name: displayName,
        type: 'pdf',
        data: base64,
        text: '',
      };
    },
    toModelOutput: (result) => {
      const { type, data, text } = result as {
        type: string;
        data: string;
        text: string;
      };
      switch (type) {
        case 'text':
          return { type: 'text', value: text };
        case 'pdf':
          return {
            type: 'content',
            value: [
              {
                type: 'media',
                data,
                mediaType: 'application/pdf',
              },
            ],
          };
        default:
          return { type: 'error-text', value: 'Unsupported file type' };
      }
    },
  });
};