feat(plugin): 支持自定义域名 (#11)

Co-authored-by: chenqun1 <[email protected]>

packages/plugin/jest.config.js

@@ -0,0 +1,10 @@
+const path = require('path')
+
+module.exports = {
+  rootDir: path.resolve(__dirname, './'),
+  transform: {
+    '^.+\\.[t|j]sx?$': 'ts-jest',
+  },
+  moduleFileExtensions: ['js', 'ts'],
+  testEnvironment: 'node'
+}

packages/plugin/package.json

@@ -30,6 +30,11 @@
     "debug": "^4.3.1"
   },
   "devDependencies": {
-    "rimraf": "^3.0.2"
+    "@types/jest": "^26.0.24",
+    "jest": "^27.0.6",
+    "rimraf": "^3.0.2",
+    "ts-jest": "^27.0.3",
+    "typescript": "^4.3.5",
+    "vite": "^2.4.3"
   }
 }

packages/plugin/src/__tests__/index.ts

@@ -0,0 +1,38 @@
+import ViteCertPlugin from '../index'
+import Mkcert from '../mkcert/index'
+
+
+const install = jest.fn()
+
+jest.mock('../mkcert/index', () => {
+    return class {
+      static create() {
+        return new this()
+      }
+      install(...args: any[]) {
+        install(...args)
+      }
+      public async init() {}
+    }
+})
+
+describe('custom hostnames', () => {
+  test('should install with custom hostnames', async () => {
+
+    const plugin = ViteCertPlugin({
+      hostnames: ['www.test.com']
+    })
+
+    await (plugin as any).config({
+      server: {
+        https: true
+      }
+    })
+
+    expect(install).toHaveBeenCalledTimes(1)
+
+    const args = install.mock.calls[0][0]
+    expect(JSON.stringify(args)).toMatch('www.test.com')
+    // expect((args as string[]).includes('www.test.com')).toBeTruthy()
+  })
+})

packages/plugin/src/index.ts

@@ -27,7 +27,8 @@ const plugin = (options?: ViteCertificateOptions): Plugin => {
 
       await mkcert.init()
 
-      const certificate = await mkcert.install(['localhost', ...ips])
+      const hostnames = Array.from(new Set(['localhost', ...ips, ...(options?.hostnames || [])]))
+      const certificate = await mkcert.install(hostnames)
 
       return {
         server: {

packages/plugin/src/mkcert/index.ts

@@ -34,6 +34,13 @@ export type MkcertOptions = {
    * @default none
    */
   mkcertPath?: string
+
+  /**
+   * Custom hostnames
+   *
+   * @default []
+   */
+  hostnames?: string[]
 }
 
 export type MkcertProps = MkcertOptions & {