github_mirrors/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter.git synced 2025-02-02 14:18:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

dev: interactive prompt for file limit

Browse Source
This commit is contained in:
KernelDeimos 2024-12-02 11:09:50 -05:00
parent ac44aaf393
commit aadb261b29
1 changed files with 28 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tools/comment-writer/main.js
View File

@ -524,10 +524,11 @@ const main = async () => {
excludes: FILE_EXCLUDES,
}, rootpath);
let i = 0;
let i = 0, limit = undefined;
for await ( const value of walk_iter ) {
if ( limit !== undefined && i >= limit ) break;
i++;
if ( i == 12 ) process.exit(0);
// Exit after processing 12 files
if ( value.is_dir ) {
console.log('directory:', value.path);
@ -550,7 +551,7 @@ const main = async () => {
console.log('File was already commented by AI; skipping...');
continue;
}
}
} else metadata = {};
let refs = null;
// Check if there are any references in the metadata
@ -589,16 +590,18 @@ const main = async () => {
}
}
const action = await enq.prompt({
const action = limit === undefined ? await enq.prompt({
type: 'select',
name: 'action',
message: 'Select action:',
choices: [
'generate',
'skip',
'all',
'limit',
'exit',
]
})
}) : 'generate';
// const action = 'generate';
// Check if user wants to exit the program
@ -611,6 +614,26 @@ const main = async () => {
continue;
}
if ( action.action === 'limit' ) {
limit = await enq.prompt({
type: 'input',
name: 'limit',
message: 'Enter limit:'
});
i = 1;
limit = Number(limit.limit);
}
if ( action.action === 'all' ) {
limit = await enq.prompt({
type: 'input',
name: 'limit',
message: 'Enter limit:'
});
i = 1;
limit = Number(limit.limit);
}
const { definitions } = js_processor.process(lines);
const key_places = [];
key_places.push({