Add support for CSV input file

Author: titouanmathis

bin/test-redirection.js

@@ -4,7 +4,7 @@ import { resolve } from 'path';
 import chalk from 'chalk';
 import { cac } from 'cac';
 import run from '../src/index.js';
-import { importJson } from '../src/utils.js';
+import { importJson, importCsv } from '../src/utils.js';
 
 const pkg = importJson('../package.json', import.meta.url);
 const PKG_NAME = pkg.name;
@@ -29,6 +29,8 @@ cli
 		default: 100,
 	})
 	.option('--ignore-query-parameters', 'Ignore query parameters in the final URL.')
+	.option('-p, --parser [parser]', 'Define which parser should be used: json or csv.')
+	.option('--csv-delimiter [delimiter]', 'Define the delimiter of the input CSV file, can be a string or a RegExp.')
 	.action((configPath, options) => {
 		const resolvedConfigPath = resolve(process.cwd(), configPath);
 
@@ -39,7 +41,16 @@ cli
 			process.exit(1);
 		}
 
-		const config = importJson(resolvedConfigPath, import.meta.url);
+		let config;
+
+		if (options.parser === 'csv') {
+			config = importCsv(resolvedConfigPath, import.meta.url, {
+				delimiter: new RegExp(options.csvDelimiter)
+			});
+		} else {
+			config = importJson(resolvedConfigPath, import.meta.url);
+		}
+
 		run(config, options);
 	});
 

src/utils.js

@@ -1,5 +1,15 @@
 import { readFileSync } from 'fs';
 
+/**
+ * Import a file.
+ * @param {string} path The path to the file.
+ * @param {string} base The base path from which the import should be resolved.
+ * @returns {string}
+ */
+export function importFile(path, base) {
+  return readFileSync(new URL(path, base)).toString();
+}
+
 /**
  * Import a JSON file.
  *
@@ -8,7 +18,35 @@ import { readFileSync } from 'fs';
  * @return {Object}
  */
 export function importJson(path, base) {
-	return JSON.parse(readFileSync(new URL(path, base)).toString());
+  return JSON.parse(importFile(path, base));
+}
+
+/**
+ * Parse a CSV file.
+ * @param {string} csv
+ * @returns {Array<any>}
+ * @param {{ delimiter?: string|RegExp }} [options]
+ * @returns {Array<{ from: string, to: string }>}
+ */
+export function csvToJson(csv, { delimiter = ' ' } = {}) {
+  return csv.split('\n').filter(Boolean).map((line) => {
+    const [from, to] = line.split(delimiter);
+    return {
+      from: from ?? '',
+      to: to ?? '',
+    };
+  });
+}
+
+/**
+ * Import a CSV file as a config object.
+ * @param {string} path
+ * @param {string} base
+ * @param {{ delimiter?: string|RegExp }} [options]
+ * @returns {Array<{ from: string, to: string }>}
+ */
+export function importCsv(path, base, options) {
+  return csvToJson(importFile(path, base), options);
 }
 
 /**
@@ -18,7 +56,7 @@ export function importJson(path, base) {
  * @return {Promise<number>}
  */
 export async function wait(delay = 0) {
-	return new Promise((resolve) => {
-		setTimeout(() => resolve(delay), delay);
-	});
-}
+  return new Promise((resolve) => {
+    setTimeout(() => resolve(delay), delay);
+  });
+}