new: invoices command

Author: Andrei

bun.lockb

@@ -20,6 +20,7 @@
     "fs-extra": "^11.2.0",
     "inquirer": "^12.0.0",
     "minimist": "^1.2.8",
-    "open": "^10.1.0"
+    "open": "^10.1.0",
+    "puppeteer": "^23.5.3"
   }
 }

package.json

@@ -3,7 +3,7 @@ import { Command } from "commander";
 import logger from "./utils/logger";
 import Invoice from "./utils/invoice";
 import minimist from "minimist"
-import { initConfig, openConfigDirectory, openConfigurationFile, timeout } from "./utils";
+import { initConfig, openConfigDirectory, openConfigurationFile, openInvoicesDirectory } from "./utils";
 import * as os from "os"
 import inquirer from "inquirer";
 
@@ -84,6 +84,11 @@ const commands = [
     description: "Open the configuration directory",
     action: openConfigDirectory
   },
+  {
+    name: "invoices",
+    description: "Open the configuration directory",
+    action: openInvoicesDirectory
+  },
   {
     name: "defaults",
     description: "Setup your default details for creating invoices.",

src/index.ts

@@ -2,21 +2,25 @@ import logger from "./logger"
 import path from "path"
 import fs from "fs-extra"
 import { load, type CheerioAPI } from "cheerio"
+import { DEFAULT_PATH } from ".."
+import { deepFindByKey } from "."
+import puppeteer from 'puppeteer'
+
 
 class HTML {
-  private requiredIds: string[] = ["wow"]
+  private requiredIds: string[] = ["createdAt", "fullName", "sortCode", "bankNum"]
 
   create(path: string) {
     return this.parser(path)
   }
 
   async parser(filePath: string): Promise<boolean> {
     try {
-      if (path.extname(filePath) !== '.html') 
+      if (path.extname(filePath) !== '.html')
         throw new Error('file type')
 
       const fileExists = await fs.pathExists(filePath);
-      if (!fileExists) 
+      if (!fileExists)
         throw new Error('file non-existant')
 
       const htmlContent = await fs.readFile(filePath, 'utf-8');
@@ -34,17 +38,43 @@ class HTML {
         logger.debug().success('All required IDs are present in the HTML.');
       }
 
-      return true 
+      return true
     } catch (error) {
       logger.debug().error(`An error occured: ${error}`)
       return false
     }
   }
 
   async editor($: CheerioAPI) {
+    const config = fs.readJsonSync(DEFAULT_PATH)
+    for (const id of this.requiredIds) {
+      const value = deepFindByKey(config, id)
+      if (!value) {
+        logger.error(`No value was found for ${id}`);
+        process.exit(0)
+      }
+
+      $(`#${id}`).text(value)
+    }
 
+    const html = $.html()
+    await this.convertToPdf(html, config.invoices_path)
   }
 
+  async convertToPdf(html: string, outputPath: string) {
+    const browser = await puppeteer.launch()
+    const page = await browser.newPage()
+
+    await page.setContent(html, { waitUntil: 'networkidle0' })
+
+    await page.pdf({
+      path: outputPath,
+      format: "A4",
+      printBackground: true
+    })
+
+    await browser.close()
+  }
 }
 
 export default HTML

src/utils/html.ts

@@ -24,6 +24,18 @@ export async function openConfigurationFile() {
   })
 }
 
+export async function openInvoicesDirectory() {
+  const config = fs.readJsonSync(DEFAULT_PATH)
+  try {
+    await open(config.invoices_path);
+    logger.info(`Opened configuration directory: ${config.invoices_path}`);
+  } catch (error) {
+    logger.error("Error opening configuration directory")
+  }
+
+  process.exit(0)
+}
+
 export async function openConfigDirectory() {
   const configDir = path.dirname(DEFAULT_PATH);
   try {
@@ -105,4 +117,29 @@ export function configSetup(configPath: string) {
   return fs.existsSync(configPath);
 }
 
+export function deepFindByKey(obj: any, keyToFind: string) {
+  if (typeof obj !== 'object' || obj === null) {
+    return undefined;
+  }
+
+  // Check if the current object contains the key
+  if (obj.hasOwnProperty(keyToFind)) {
+    return obj[keyToFind];
+  }
+
+  // Recursively search in each nested object
+  for (const key in obj) {
+    if (obj.hasOwnProperty(key)) {
+      const value = obj[key];
+
+      if (typeof value === 'object') {
+        const result : any = deepFindByKey(value, keyToFind);
+        if (result !== undefined) {
+          return result;
+        }
+      }
+    }
+  }
 
+  return undefined;
+}

src/utils/index.ts

@@ -9,28 +9,28 @@ interface DefaultOption<T extends 'input' | 'number' = 'input'> {
   type: T
   message: string
   validate: (value: T extends 'number' ? number : string) => void
-} 
+}
 
 interface DefaultOptionPrompts {
   sortCode: DefaultOption
-  bankNum: DefaultOption<"number"> 
+  bankNum: DefaultOption<"number">
   fullName: DefaultOption
 }
 
 class Invoice {
 
-  defaultOptionPrompts : DefaultOptionPrompts = {
-    sortCode:  {
+  defaultOptionPrompts: DefaultOptionPrompts = {
+    sortCode: {
       type: "input",
       message: "Enter your sort code",
       validate: val => val.length == 8 && val.includes('-') && !isNaN(parseInt(val.replaceAll('-', '')))
     },
-    bankNum:  {
+    bankNum: {
       type: "number",
       message: "Enter your bank account number",
       validate: num => num.toString().length == 8
     },
-    fullName:  {
+    fullName: {
       type: "input",
       message: "Enter your fullname",
       validate: str => str.includes(' ') && str.length > 5
@@ -130,7 +130,7 @@ class Invoice {
   }
 
   async askForDefault(key: keyof DefaultOptionPrompts): Promise<string> {
-      // @ts-ignore
+    // @ts-ignore
     const value = await inquirer.prompt([
       {
         name: key,
@@ -145,32 +145,36 @@ class Invoice {
 
 
   async setupDefaultValues(args?: string[]) {
-
     logger.debug().info(`Args: ${JSON.stringify(args)}`)
 
     const config = fs.readJsonSync(DEFAULT_PATH)
 
     if (args?.length) {
-      args.forEach(async arg => {
-        const key = Object.keys(this.defaultOptionPrompts).find(o => o.toLowerCase() == arg)
-        if (!key) return logger.error(`Can't edit a value that doesn't exist: ${arg}`)
-
-        logger.info(key)
+      for (const arg of args) {
+        const key = Object.keys(this.defaultOptionPrompts).find(o => o.toLowerCase() === arg)
+        if (!key) {
+          logger.error(`Can't edit a value that doesn't exist: ${arg}`)
+          continue
+        }
 
         config['default_values'] = {
+          ...config['default_values'],  // Preserve existing default values
           [key]: await this.askForDefault(key as keyof DefaultOptionPrompts)
         }
-      })
-    }
-    else {
+      }
+
+      fs.writeJsonSync(DEFAULT_PATH, config, { spaces: 2 })
+      process.exit(0)
+    } else {
       config['default_values'] = {
         fullName: await this.askForDefault('fullName'),
         bankNum: await this.askForDefault('bankNum'),
-        sortCode: await this.askForDefault('sortCode') 
+        sortCode: await this.askForDefault('sortCode')
       }
-    }
 
-    fs.writeJsonSync(DEFAULT_PATH, config, { spaces: 2 })
+      fs.writeJsonSync(DEFAULT_PATH, config, { spaces: 2 })
+      process.exit(0)
+    }
   }
 }