import * as plugins from './mod.plugins.js';
import type { IPlannedChange } from './interfaces.format.js';
import { logger } from '../gitzone.logging.js';

export class DiffReporter {
  private diffs: Map<string, string> = new Map();
  
  async generateDiff(filePath: string, oldContent: string, newContent: string): Promise<string> {
    const diff = plugins.smartdiff.createDiff(oldContent, newContent);
    this.diffs.set(filePath, diff);
    return diff;
  }
  
  async generateDiffForChange(change: IPlannedChange): Promise<string | null> {
    if (change.type !== 'modify') {
      return null;
    }
    
    try {
      const exists = await plugins.smartfile.fs.fileExists(change.path);
      if (!exists) {
        return null;
      }
      
      const currentContent = await plugins.smartfile.fs.toStringSync(change.path);
      
      // For planned changes, we need the new content
      if (!change.content) {
        return null;
      }
      
      return await this.generateDiff(change.path, currentContent, change.content);
    } catch (error) {
      logger.log('error', `Failed to generate diff for ${change.path}: ${error.message}`);
      return null;
    }
  }
  
  displayDiff(filePath: string, diff?: string): void {
    const diffToShow = diff || this.diffs.get(filePath);
    
    if (!diffToShow) {
      logger.log('warn', `No diff available for ${filePath}`);
      return;
    }
    
    console.log(`\n${this.formatDiffHeader(filePath)}`);
    console.log(this.colorDiff(diffToShow));
    console.log('━'.repeat(50));
  }
  
  displayAllDiffs(): void {
    if (this.diffs.size === 0) {
      logger.log('info', 'No diffs to display');
      return;
    }
    
    console.log('\nFile Changes:');
    console.log('═'.repeat(50));
    
    for (const [filePath, diff] of this.diffs) {
      this.displayDiff(filePath, diff);
    }
  }
  
  private formatDiffHeader(filePath: string): string {
    return `📄 ${filePath}`;
  }
  
  private colorDiff(diff: string): string {
    const lines = diff.split('\n');
    const coloredLines = lines.map(line => {
      if (line.startsWith('+') && !line.startsWith('+++')) {
        return `\x1b[32m${line}\x1b[0m`; // Green for additions
      } else if (line.startsWith('-') && !line.startsWith('---')) {
        return `\x1b[31m${line}\x1b[0m`; // Red for deletions
      } else if (line.startsWith('@')) {
        return `\x1b[36m${line}\x1b[0m`; // Cyan for line numbers
      } else {
        return line;
      }
    });
    
    return coloredLines.join('\n');
  }
  
  async saveDiffReport(outputPath: string): Promise<void> {
    const report = {
      timestamp: new Date().toISOString(),
      totalFiles: this.diffs.size,
      diffs: Array.from(this.diffs.entries()).map(([path, diff]) => ({
        path,
        diff
      }))
    };
    
    await plugins.smartfile.memory.toFs(JSON.stringify(report, null, 2), outputPath);
    logger.log('info', `Diff report saved to ${outputPath}`);
  }
  
  hasAnyDiffs(): boolean {
    return this.diffs.size > 0;
  }
  
  getDiffCount(): number {
    return this.diffs.size;
  }
}