119 lines
4 KiB
JavaScript
119 lines
4 KiB
JavaScript
const inquirer = require('inquirer');
|
|
const fs = require('fs');
|
|
const path = require('path');
|
|
const { root, src, thoughts: thoughtsDir } = require('../constants');
|
|
const axios = require('axios');
|
|
const cheerio = require('cheerio');
|
|
const chalk = require('chalk');
|
|
|
|
process.argv.shift();
|
|
process.argv.shift();
|
|
|
|
const readmeContent = fs.readFileSync(path.join(root, 'readme.md')).toString();
|
|
|
|
const problems = readmeContent.match(/^- \[.] (?:\[(.*)\]|(.*))/gm).map(res => {
|
|
const sanitised = res.substring(8).replace('[', '').replace(']', '');
|
|
return sanitised.match(/[0-9]* - (.*)/)[1];
|
|
});
|
|
|
|
const questions = [];
|
|
|
|
if (isNaN(process.argv[0]))
|
|
questions.push({
|
|
name: 'problemNumber',
|
|
message: 'Which problem would you like to solve?',
|
|
type: 'number',
|
|
validate: input => {
|
|
input = parseInt(input);
|
|
|
|
let alreadyGenerated = false;
|
|
fs.readdirSync(src).forEach(file => {
|
|
if (file.startsWith(input)) alreadyGenerated = true;
|
|
});
|
|
|
|
if (alreadyGenerated) return 'Please choose a problem you have not already completed!';
|
|
else return true;
|
|
}
|
|
});
|
|
|
|
inquirer
|
|
.prompt([
|
|
...questions,
|
|
{
|
|
name: 'thoughts',
|
|
message: 'Should I generate a thoughts document for you?',
|
|
type: 'confirm',
|
|
default: false
|
|
}
|
|
])
|
|
.then(({ problemNumber, thoughts }) => {
|
|
if (!problemNumber) problemNumber = parseInt(process.argv[0]);
|
|
const fileName = `${problemNumber} - ${problems[problemNumber - 1]}`;
|
|
|
|
// Fetch the problem data off of projecteuler.net
|
|
axios
|
|
.get(`https://projecteuler.net/problem=${problemNumber}`)
|
|
.then(({ data }) => {
|
|
const $ = cheerio.load(data);
|
|
const problemContent = $('.problem_content')
|
|
.text()
|
|
.trim()
|
|
.split('\n')
|
|
.map(r => `// ${r}`)
|
|
.join('\n');
|
|
|
|
// Generate the source file
|
|
fs.writeFileSync(
|
|
path.join(src, `${fileName}.ts`),
|
|
`${problemContent}
|
|
export = {};
|
|
|
|
// Output
|
|
console.log();`
|
|
);
|
|
|
|
// Generate the thoughts file
|
|
if (thoughts)
|
|
fs.writeFileSync(
|
|
path.join(thoughtsDir, `${fileName}.md`),
|
|
`<div align="center">
|
|
|
|
### ${problems[problemNumber - 1]}
|
|
</div>`
|
|
);
|
|
|
|
// Check it off in the readme
|
|
const regex = new RegExp(` \\[.\\] ${problemNumber} - .*`);
|
|
const match = readmeContent.match(regex);
|
|
|
|
let newLine = ` [x] [${match[0]
|
|
.replace('[ ]', '')
|
|
.trim()}](src/${encodeURIComponent(fileName)}.ts)`;
|
|
|
|
if (thoughts)
|
|
newLine += `\n - [Thoughts](thoughts/${encodeURIComponent(fileName)}.md)`;
|
|
|
|
const newContent = readmeContent.replace(regex, newLine);
|
|
|
|
fs.writeFileSync(path.join(root, 'readme.md'), newContent);
|
|
|
|
// If the problem's ID is greater than 100, add it to the gitignore
|
|
if (problemNumber > 100) {
|
|
const gitignorePath = path.join(root, '.gitignore');
|
|
const gitignoreContent = fs.readFileSync(gitignorePath).toString();
|
|
let newContent = `${gitignoreContent}\nsrc/${fileName}.ts`;
|
|
if (thoughts) newContent += `\nthoughts/${fileName}.md`;
|
|
|
|
fs.writeFileSync(gitignorePath, newContent);
|
|
}
|
|
})
|
|
.catch(() =>
|
|
console.error(
|
|
chalk.red(
|
|
chalk.bold(
|
|
'There was an error generating files for that challenge - please ensure that it exists on Project Euler!'
|
|
)
|
|
)
|
|
)
|
|
);
|
|
});
|