From 604678b29ab2d0bb0bbe9ffbeb544cef8a4a5caf Mon Sep 17 00:00:00 2001
From: Xmader <xmader@outlook.com>
Date: Thu, 26 Nov 2020 16:53:01 -0500
Subject: [PATCH] feat(cli): ask for the page url or path to local file

---
 src/cli.ts | 36 +++++++++++++++++++-----------------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/src/cli.ts b/src/cli.ts
index 7387d9f..b5dc098 100644
--- a/src/cli.ts
+++ b/src/cli.ts
@@ -18,7 +18,7 @@ const SCORE_URL_PREFIX = 'https://musescore.com/'
 const EXT = '.mscz'
 
 interface Params {
-  url: string;
+  fileInit: string;
   confirmed: boolean;
   part: number;
   types: number[];
@@ -26,25 +26,27 @@ interface Params {
 }
 
 void (async () => {
-  const fileInit: string | undefined = process.argv[2]
-  const isLocalFile = fileInit?.endsWith(EXT) && fs.existsSync(fileInit)
-
   let scoreinfo: ScoreInfo
-  if (!isLocalFile) {
-    // ask for the page url
-    const { url } = await inquirer.prompt<Params>({
-      type: 'input',
-      name: 'url',
-      message: 'Score URL:',
-      suffix: ` (starts with "${SCORE_URL_PREFIX}")\n `,
-      validate (input: string) {
-        return input && input.startsWith(SCORE_URL_PREFIX)
-      },
-      default: fileInit,
-    })
+  // ask for the page url or path to local file
+  const { fileInit } = await inquirer.prompt<Params>({
+    type: 'input',
+    name: 'fileInit',
+    message: 'Score URL or path to local MSCZ file:',
+    suffix: `\n  (starts with "${SCORE_URL_PREFIX}" or local filepath ends with "${EXT}")\n `,
+    validate (input: string) {
+      return input &&
+        (
+          input.startsWith(SCORE_URL_PREFIX) ||
+          (input.endsWith(EXT) && fs.statSync(input).isFile())
+        )
+    },
+    default: process.argv[2],
+  })
 
+  const isLocalFile = fileInit.endsWith(EXT)
+  if (!isLocalFile) {
     // request scoreinfo
-    scoreinfo = await ScoreInfoHtml.request(url)
+    scoreinfo = await ScoreInfoHtml.request(fileInit)
 
     // confirmation
     const { confirmed } = await inquirer.prompt<Params>({