import { promises, existsSync, mkdirSync } from 'fs'
import { join } from 'path'
import { Problem } from './CodeforcesScraper'

const { writeFile } = promises

export default async function(problem: Problem) {
    try {
        const cwd = process.cwd()

        const outputPath = join(cwd, problem.problemSetId.toString())

        if(!existsSync(outputPath))
            mkdirSync(outputPath, { recursive: true })

        for(let i = 0; i < problem.examples.length; ++i) {
            const outputFileWithoutExtension = join(problem.difficulty + '-' + (i + 1))

            const qualifiedInput = join(outputPath, outputFileWithoutExtension + '.in')
            await writeFile(qualifiedInput, problem.examples[i].input, 'utf8')
            
            if(problem.examples[i].output !== null) {
                const qualifiedOutput = join(outputPath, outputFileWithoutExtension + '.out')
                await writeFile(qualifiedOutput, problem.examples[i].output as string, 'utf8')
            }
        }
    }
    catch (e) {
        console.error(e)
    }
}