feat: 增加截图新配置

2024-06-10 10:46:16 +08:00 · 2024-06-10 10:46:16 +08:00 · 03ff09289f
parent 841c31d4ec
commit 03ff09289f
2 changed files with 11 additions and 2 deletions
--- a/example/index.tsx
+++ b/example/index.tsx
@ -13,19 +13,21 @@ export class Image {
   * @returns
   */
  getHelloComponent(uid: number, data: DataType) {
-    // 生成 html 地址
+    // 生成 html 地址 或 html字符串
    const Address = Com.create(<Hello data={data} />, {
      /**
       * 注意，不设置json_dir时，
       * html_head路径应该是../public/output.css
       * 且html_head默认值路径也是../public/output.css
-       * 因此，不增加其他head的话，html_head和join_dir都可以胜利
+       * 因此，不增加其他head的话，html_head和join_dir都可以省略
       * { html_name: `${uid}.html`}
       */
      html_head: `<link rel="stylesheet" href="../../public/output.css"></link>`,
      // html/hello/uid.html
      join_dir: 'hello',
      html_name: `${uid}.html`
+      // 不生成文件，返回的将是html字符串
+      // file_create:false
    })
    return Pup.render(Address)
  }
--- a/src/utils/component.ts
+++ b/src/utils/component.ts
@ -8,6 +8,7 @@ type ComponentCreateOpsionType = {
  html_name?: string
  join_dir?: string
  html_body?: string
+  file_create?: boolean
 }

 /**
@ -39,6 +40,12 @@ export class Component {
    const head = `<head>${options?.html_head ?? Link}</head>`
    const body = `<body> ${str} ${options?.html_body ?? ''}</body>`
    const html = `${DOCTYPE}<html>${head}${body}</html>`
+    if (
+      typeof options?.file_create == 'boolean' &&
+      options?.file_create == false
+    ) {
+      return html
+    }
    writeFileSync(address, html)
    return address
  }