Added file tree functionality

This commit is contained in:
sw5678 2023-12-13 09:19:16 +00:00
parent 6ed9d4554a
commit 76ba630d59
3 changed files with 123 additions and 1 deletions

View file

@ -294,7 +294,8 @@
"Escape string",
"Unescape string",
"Pseudo-Random Number Generator",
"Sleep"
"Sleep",
"File Tree"
]
},
{

View file

@ -0,0 +1,100 @@
/**
* @author sw5678
* @copyright Crown Copyright 2016
* @license Apache-2.0
*/
import Operation from "../Operation.mjs";
import Utils from "../Utils.mjs";
import {INPUT_DELIM_OPTIONS} from "../lib/Delim.mjs";
/**
* Unique operation
*/
class FileTree extends Operation {
/**
* Unique constructor
*/
constructor() {
super();
this.name = "File Tree";
this.module = "Default";
this.description = "Creates file tree from list of file paths (Similar too tree linux command)";
this.inputType = "string";
this.outputType = "string";
this.args = [
{
name: "File Path Delimiter",
type: "binaryString",
value: "/"
},
{
name: "Delimiter",
type: "option",
value: INPUT_DELIM_OPTIONS
}
];
}
/**
* @param {string} input
* @param {Object[]} args
* @returns {string}
*/
run(input, args) {
// Set up arrow and pipe for nice output display
const ARROW = '|---';
const PIPE = '| ';
// Get args from input
const file_delim = args[0];
const entry_delim = Utils.charRep(args[1]);
// Store path to print
let completed_list = [];
let print_list = [];
// Loop through all entries
const file_paths = input.split(entry_delim).unique().sort();
for (var i = 0; i < file_paths.length; i++)
{
// Split by file delimiter
let path = file_paths[i].split(file_delim);
if (path[0] == '')
{
path = path.slice(1, path.length);
}
for (var j = 0; j < path.length; j++)
{
let print_line;
let key;
if (j == 0)
{
print_line = path[j];
key = path[j];
}
else
{
print_line = PIPE.repeat(j-1) + ARROW + path[j]
key = path.slice(0, j+1).join("/");
}
// Check to see we have already added that path
if (!completed_list.includes(key))
{
completed_list.push(key);
print_list.push(print_line);
}
}
}
return print_list.join("\n");
}
}
export default FileTree;

View file

@ -0,0 +1,21 @@
/**
* @author sw5678
* @copyright Crown Copyright 2023
* @license Apache-2.0
*/
import TestRegister from "../../lib/TestRegister.mjs";
TestRegister.addTests([
{
"name": "Swap Case: basic example",
"input": "/test_dir1/test_file1.txt\n/test_dir1/test_file2.txt\n/test_dir2/test_file1.txt",
"expectedOutput": "test_dir1\n|---test_file1.txt\n|---test_file2.txt\ntest_dir2\n|---test_file1.txt",
"recipeConfig": [
{
"op": "File Tree",
"args": [
],
},
],
}
]);