feat: allow config to be string or number

unjs · Dec 4, 2020 · 13f4275 · 13f4275
1 parent 526ad3e
commit 13f4275
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -10,6 +10,8 @@ export interface GetPortOptions {
   memoName: string
 }
 
+export type GetPortInput = Partial<GetPortOptions> | number | string
+
 const defaults = {
   name: 'default',
   random: false,
@@ -18,7 +20,11 @@ const defaults = {
   memoName: 'port'
 }
 
-export async function getPort (config?: Partial<GetPortOptions>): Promise<number> {
+export async function getPort (config?: GetPortInput): Promise<number> {
+  if (typeof config === 'number' || typeof config === 'string') {
+    config = { port: parseInt(config + '') }
+  }
+
   const options = { ...defaults, ...config } as GetPortOptions
 
   const portsToCheck: number[] = []