update(user/bkots):Remove unnecessary must()
[BK-2020-03.git] / unitproc / bkt-prune_path_rootside
... / ...
CommitLineData
1#!/bin/bash
2
3prune_path_rootside() {
4 # Desc: Prunes a path from the root-side to a specified prune level.
5 # Input: arg1 str path
6 # arg2 int prune level (0-indexed)
7 # Depends: GNU sed 4.8
8 # Version: 0.0.1
9 local path="$1";
10 local prune_level="$2";
11
12 # Check for absolute or relative path
13 if [[ "$path" =~ ^/ ]]; then
14 flag_root=true;
15 # Remove initial /
16 path="$(echo "$path" | sed -e 's:^/::' )";
17 else
18 flag_root=false;
19 fi;
20
21 # Save path as array with `/` as element delimiter
22 local IFS='/';
23 read -ra parts <<< "$path";
24
25 # Assemble pruned path from prune_level
26 local pruned_path="";
27 for (( i=prune_level; i<${#parts[@]}; i++ )); do
28 pruned_path+="${parts[i]}/";
29 done;
30
31 # Trim trailing `/` delimiter
32 pruned_path=$(echo "$pruned_path" | sed 's:/*$::');
33
34 # Restore initial / if appropriate
35 if [[ "$flag_root" == "true" ]] && [[ "$prune_level" -eq 0 ]]; then
36 pruned_path=/"$pruned_path";
37 fi;
38
39 # Output pruned path
40 echo "$pruned_path";
41 #declare -p path prune_level parts pruned_path && printf "========\n"; # debug
42 return 0;
43}; # prune path rootside to int specified level
44
45printf "========Test 1========\n";
46prune_path_rootside "foo/bar/baz" 0;
47prune_path_rootside "foo/bar/baz" 1;
48prune_path_rootside "foo/bar/baz" 2;
49prune_path_rootside "foo/bar/baz" 3;
50printf "========Test 2========\n";
51prune_path_rootside "/foo/bar/baz" 0;
52prune_path_rootside "/foo/bar/baz" 1;
53prune_path_rootside "/foo/bar/baz" 2;
54prune_path_rootside "/foo/bar/baz" 3;