import { createSDK, handleError } from './utils.js';
import { ArtifactType, NotebookLMLanguage } from '../src/types/artifact.js';
import * as readline from 'readline';

/**
 * Example: Creating artifacts with language support
 * 
 * This example demonstrates:
 * 1. Setting the notebook's default output language
 * 2. Creating artifacts that use the notebook's default language
 * 3. Creating artifacts with explicit language overrides
 * 
 * The notebook's default language is used for all artifacts unless
 * a specific language is provided in the customization options.
 */

async function promptUser(question: string): Promise<string> {
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
  });

  return new Promise((resolve) => {
    rl.question(question, (answer) => {
      rl.close();
      resolve(answer.trim());
    });
  });
}

async function main() {
  const sdk = await createSDK({ debug: false });

  try {
    await sdk.connect(); // Initialize SDK with authentication

    const notebookId = process.env.NOTEBOOK_ID || 'your-notebook-id';
    
    // Optional: Set the notebook's default output language
    // This will be used as the default for all artifacts unless overridden
    const setDefaultLanguage = process.env.SET_DEFAULT_LANGUAGE === 'true';
    if (setDefaultLanguage) {
      const defaultLang = process.env.DEFAULT_LANGUAGE || NotebookLMLanguage.ENGLISH;
      console.log(`\n=== Setting Notebook Default Language ===\n`);
      console.log(`Setting default language to: ${defaultLang}`);
      await sdk.notebookLanguage.set(notebookId, defaultLang);
      console.log(`✅ Default language set to ${defaultLang}\n`);
      
      // Verify the language was set
      const currentLang = await sdk.notebookLanguage.get(notebookId);
      console.log(`Current notebook language: ${currentLang}\n`);
    } else {
      // Get current default language
      const currentLang = await sdk.notebookLanguage.get(notebookId);
      console.log(`\n=== Notebook Default Language ===\n`);
      console.log(`Current default language: ${currentLang}`);
      console.log(`(All artifacts will use this language unless overridden)\n`);
    }
    
    // List sources first
    console.log('=== Listing Sources ===\n');
    const sources = await sdk.sources.list(notebookId);
    
    if (sources.length === 0) {
      console.error('❌ No sources found in notebook. Please add sources before creating artifacts.');
      process.exit(1);
    }
    
    console.log(`Found ${sources.length} source(s):\n`);
    sources.forEach((source, index) => {
      const statusText = source.status === 2 ? 'READY' : source.status === 1 ? 'PROCESSING' : source.status === 3 ? 'FAILED' : 'UNKNOWN';
      console.log(`  ${index + 1}. [${source.sourceId}] ${source.title || 'Untitled'} (${source.type}) - ${statusText}`);
    });
    console.log();
    
    // Warn if any sources are not ready
    const notReadySources = sources.filter(s => s.status !== 2);
    if (notReadySources.length > 0) {
      console.warn(`⚠️  Warning: ${notReadySources.length} source(s) are not ready (status != READY).`);
      console.warn('   Artifact creation may fail if sources are still processing.\n');
    }
    
    // Ask user to select sources
    const selection = await promptUser(
      `Select sources to use (enter numbers separated by commas, e.g., "1,2,3" or "all" for all sources): `
    );
    
    let sourceIds: string[] = [];
    
    if (selection.toLowerCase() === 'all' || selection === '') {
      // Use ALL sources
      sourceIds = sources.map(s => s.sourceId);
      console.log(`Using ALL ${sourceIds.length} source(s)\n`);
    } else {
      // Parse user selection
      const selectedIndices = selection
        .split(',')
        .map(s => parseInt(s.trim(), 10))
        .filter(n => !isNaN(n) && n >= 1 && n <= sources.length);
      
      if (selectedIndices.length === 0) {
        console.error('❌ Invalid selection. Please enter valid source numbers or "all".');
        process.exit(1);
      }
      
      sourceIds = selectedIndices.map(index => sources[index - 1].sourceId);
      console.log(`Using ${sourceIds.length} selected source(s): ${selectedIndices.join(', ')}\n`);
    }

    console.log('=== Creating Artifacts ===\n');
    console.log('Note: Artifacts will use the notebook\'s default language unless explicitly specified.\n');
    console.log('Language support: Audio, Video, Report, Infographics, Slide Decks\n');

    // // 1. Create Quiz
    // console.log('1. Creating Quiz...');
    // // Note: Quiz language is set via instructions, not customization.language
    // const quiz = await sdk.artifacts.create(notebookId, ArtifactType.QUIZ, {
    //   title: 'Chapter 1 Quiz',
    //   instructions: 'Create questions covering key concepts from the sources',
    //   customization: {
    //     numberOfQuestions: 2, // 1=Fewer, 2=Standard, 3=More
    //     difficulty: 2, // 1=Easy, 2=Medium, 3=Hard
    //     // language is not supported for Quiz - use instructions instead
    //   },
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    // });
    // console.log(`   Created: ${quiz.title}`);
    // console.log(`   ID: ${quiz.artifactId}`);
    // console.log(`   State: ${quiz.state}\n`);

    // 2. Create Flashcards
    console.log('2. Creating Flashcards...');
    // Note: Flashcard language is set via instructions, not customization.language
    const flashcards = await sdk.artifacts.create(notebookId, ArtifactType.FLASHCARDS, {
      title: 'Key Terms Flashcards',
      customization: {
        numberOfCards: 1, // 1=Fewer, 2=Standard/More (API only accepts 1 or 2; 3 is auto-mapped to 2)
        difficulty: 1, // 1=Easy, 2=Medium, 3=Hard
        // language is not supported for Flashcards - use instructions instead
      },
      sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    });
    console.log(`   Created: ${flashcards.title}`);
    console.log(`   ID: ${flashcards.artifactId}`);
    console.log(`   State: ${flashcards.state}\n`);

    // // 3. Create Slide Deck
    // console.log('3. Creating Slide Deck...');
    // // Slide decks support language customization - will use notebook default if not specified
    // const slides = await sdk.artifacts.create(notebookId, ArtifactType.SLIDE_DECK, {
    //   title: 'Presentation Slides',
    //   instructions: 'Create a comprehensive presentation',
    //   customization: {
    //     format: 2, // 2=Presenter slides, 3=Detailed deck
    //     length: 2, // 1=Short, 2=Default, 3=Long
    //     language: NotebookLMLanguage.ENGLISH, // Optional: overrides notebook default
    //     // If language is omitted, will use notebook's default language
    //   },
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    // });
    // console.log(`   Created: ${slides.title}`);
    // console.log(`   ID: ${slides.artifactId}`);
    // console.log(`   State: ${slides.state}\n`);

    // // 4. Create Infographic
    // console.log('4. Creating Infographic...');
    // // Infographics support language customization - will use notebook default if not specified
    // const infographic = await sdk.artifacts.create(notebookId, ArtifactType.INFOGRAPHIC, {
    //   title: 'Data Visualization',
    //   customization: {
    //     orientation: 1, // 1=Landscape, 2=Portrait, 3=Square
    //     levelOfDetail: 2, // 1=Concise, 2=Standard, 3=Detailed
    //     language: NotebookLMLanguage.ENGLISH, // Optional: overrides notebook default
    //     // If language is omitted, will use notebook's default language
    //   },
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    // });
    // console.log(`   Created: ${infographic.title}`);
    // console.log(`   ID: ${infographic.artifactId}`);
    // console.log(`   State: ${infographic.state}\n`);

    // // 5. Create Audio Overview
    // console.log('5. Creating Audio Overview...');
    // // Audio supports language customization - will use notebook default if not specified
    // const audio = await sdk.artifacts.create(notebookId, ArtifactType.AUDIO, {
    //   title: 'Audio Summary',
    //   instructions: 'Create a narrative summary',
    //   customization: {
    //     format: 0, // 0=Deep dive, 1=Brief, 2=Critique, 3=Debate
    //     length: 2, // 1=Short, 2=Default, 3=Long
    //     language: NotebookLMLanguage.ENGLISH, // Optional: overrides notebook default
    //     // If language is omitted, will use notebook's default language
    //   },
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    // });
    // console.log(`   Created: ${audio.title}`);
    // console.log(`   ID: ${audio.artifactId}`);
    // console.log(`   State: ${audio.state}\n`);

    // // 6. Create Video Overview
    // console.log('6. Creating Video Overview...');
    // // Video supports language customization - will use notebook default if not specified
    // const video = await sdk.artifacts.create(notebookId, ArtifactType.VIDEO, {
    //   title: 'Explainer Video',
    //   instructions: 'Create an explainer video covering the main topics',
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    //   customization: {
    //     format: 1, // 1=Explainer, 2=Brief
    //     visualStyle: 0, // 0=Auto-select, 1=Custom, 2=Classic, 3=Whiteboard, 4=Kawaii, 5=Anime, 6=Watercolour, 7=Anime (alt), 8=Retro print, 9=Heritage, 10=Paper-craft
    //     focus: 'Key concepts and main findings',
    //     language: NotebookLMLanguage.ENGLISH, // Optional: overrides notebook default
    //     // If language is omitted, will use notebook's default language
    //   },
    // });
    // console.log(`   Created: ${video.title}`);
    // console.log(`   ID: ${video.artifactId}`);
    // console.log(`   State: ${video.state}\n`);

    // // 7. Create Report
    // console.log('7. Creating Report...');
    // // Reports support language - will use notebook default if not specified
    // const report = await sdk.artifacts.create(notebookId, ArtifactType.REPORT, {
    //   title: 'Research Report',
    //   instructions: 'Create a comprehensive research report',
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    //   // Report language uses notebook's default language automatically
    // });
    // console.log(`   Created: ${report.title}`);
    // console.log(`   ID: ${report.artifactId}`);
    // console.log(`   State: ${report.state}\n`);

    // // 8. Create Mind Map
    // console.log('8. Creating Mind Map...');
    // const mindMap = await sdk.artifacts.create(notebookId, ArtifactType.MIND_MAP, {
    //   title: 'Concept Map',
    //   instructions: 'Create a visual mind map of key concepts',
    //   sourceIds: sourceIds, // Optional: specify sources to use (omits to use all)
    // });
    // console.log(`   Created: ${mindMap.title}`);
    // console.log(`   ID: ${mindMap.artifactId}`);
    // console.log(`   State: ${mindMap.state}\n`);

    // console.log('=== All Artifacts Created ===');
    // console.log('\nNote: Artifacts are created asynchronously.');
    // console.log('Check the state field - when state is READY, use artifacts.get() to fetch full content.');
  } catch (error) {
    handleError(error, 'Failed to create artifacts');
  }

  sdk.dispose();
  process.exit(0);
}

main().catch(console.error);