[New] Add transform func to middle to set args.repo before parse

bin/can-merge

@@ -22,6 +22,12 @@ const {
 	NODE_ENV,
 } = process.env;
 
+const setRepository = (argv) => {
+	const args = argv;
+	args.repo = getRepo(argv.remote);
+	return argv;
+};
+
 const args = Yargs
 	.usage('Usage: can-merge -p <pr> [-r <repo>]')
 	.help()
@@ -60,21 +66,20 @@ const args = Yargs
 		},
 	})
 	.check((argv) => {
-		if (!argv.repo && !getRepo()) {
-			throw new Error('Could not infer any git repository from directory. \
-			\nAre you sure you are in a git repository? Otherwise, please specify --repo');
-		}
-		if (argv.remote && !getRepo(argv.remote)) {
-			throw new Error(`Could not find remote git repo with remote name ${argv.remote}`);
+		if (!getRepo(argv.remote)) {
+			if (!argv.repo) {
+				throw new Error('Could not infer any git repository from directory. \
+				\nAre you sure you are in a git repository? Otherwise, please specify --repo');
+			}
+			if (argv.remote) {
+				throw new Error(`Could not find remote git repo with remote name ${argv.remote}`);
+			}
 		}
 		return true;
 	})
+	.middleware(setRepository)
 	.parse();
 
-if (args.remote) {
-	args.repo = getRepo(args.remote);
-}
-
 const token = args.token || GITHUB_TOKEN || GH_TOKEN;
 
 runQuery(args.pr, args.repo, args.sha, token).then((response) => {