feat: add filter method for problems

Author: gengjiawen

util/leetcode.test.ts

@@ -1,5 +1,6 @@
 import { extract_rust_solution, underline } from './utils'
-import { sequelize } from './crawl_leetcode'
+import { Problem, sequelize } from './crawl_leetcode'
+import { generatePool, JSconfig } from './pools_utils'
 
 test('test underline filename', () => {
   const a = underline('minimum-swaps-to-group-all-1s-together')
@@ -36,3 +37,16 @@ test('leetcode statistics', async () => {
   )
   console.log(`paid count`, paidCount[0])
 })
+
+test('generate special', async () => {
+  const r = await Problem.findAll({
+    where: {
+      frontend_id: 141,
+    },
+  })
+
+  await generatePool({
+    problems: r,
+    config: JSconfig,
+  })
+})

util/pools_utils.ts

@@ -27,7 +27,7 @@ export const JSconfig: CodeConfig = {
   getCode: (c) => c.find((a) => a.value === 'javascript').defaultCode,
   ext: 'js',
   comment: '//',
-  fileNameStyle: (s) => s.replace(/[\s|?]/g, ''),
+  fileNameStyle: (s, i) => `_${i.toString().padStart(4, '0')}_${underline(s)}`,
 }
 
 export const Goconfig: CodeConfig = {
@@ -131,8 +131,3 @@ export async function generatePool({
     })
   })
 }
-
-console.log('generate pools finished')
-let test = false
-process.argv.includes('--test') && (test = true)
-// generatePool({ config: Rust, test: test }).catch(console.error)