[BK-2020-03.git] / user / bkfeh

#!/usr/bin/env bash
# Desc: Wrapper for feh that accepts directory paths via posargs or stdin lines.
# Version: 0.3.0
# Ref/Attrib: [1] Tange, Ole. GNU Parallel with Bash Array. 2019-03-24. https://unix.stackexchange.com/a/508365/411854
# Depends: GNU Parallel, GNU Bash v5.1.16, feh 3.6.3, GNU Coreutils 8.32 (b2sum)

#===Declare local functions===
yell() { echo "$0: $*" >&2; } # print script path and all args to stderr
die() { yell "$*"; exit 111; } # same as yell() but non-zero exit status
must() { "$@" || die "cannot $*"; } # runs args as command, reports args if command fails
checkapp() {
    # Desc: If arg is a command, save result in assoc array 'appRollCall'
    # Usage: checkapp arg1 arg2 arg3 ...
    # Version: 0.1.1
    # Input: global assoc. array 'appRollCall'
    # Output: adds/updates key(value) to global assoc array 'appRollCall'
    # Depends: bash 5.0.3
    local returnState    

    #===Process Args===
    for arg in "$@"; do
	if command -v "$arg" 1>/dev/null 2>&1; then # Check if arg is a valid command
	    appRollCall[$arg]="true";
	    if ! [ "$returnState" = "false" ]; then returnState="true"; fi;
	else
	    appRollCall[$arg]="false"; returnState="false";
	fi;
    done;

    #===Determine function return code===
    if [ "$returnState" = "true" ]; then
	return 0;
    else
	return 1;
    fi;
} # Check that app exists
checkfile() {
    # Desc: If arg is a file path, save result in assoc array 'fileRollCall'
    # Usage: checkfile arg1 arg2 arg3 ...
    # Version: 0.1.1
    # Input: global assoc. array 'fileRollCall'
    # Output: adds/updates key(value) to global assoc array 'fileRollCall';
    # Output: returns 0 if app found, 1 otherwise
    # Depends: bash 5.0.3
    local returnState

    #===Process Args===
    for arg in "$@"; do
	if [ -f "$arg" ]; then
	    fileRollCall["$arg"]="true";
	    if ! [ "$returnState" = "false" ]; then returnState="true"; fi;
	else
	    fileRollCall["$arg"]="false"; returnState="false";
	fi;
    done;
    
    #===Determine function return code===
    if [ "$returnState" = "true" ]; then
	return 0;
    else
	return 1;
    fi;
} # Check that file exists
checkdir() {
    # Desc: If arg is a dir path, save result in assoc array 'dirRollCall'
    # Usage: checkdir arg1 arg2 arg3 ...
    # Version 0.1.1
    # Input: global assoc. array 'dirRollCall'
    # Output: adds/updates key(value) to global assoc array 'dirRollCall';
    # Output: returns 0 if app found, 1 otherwise
    # Depends: Bash 5.0.3
    local returnState

    #===Process Args===
    for arg in "$@"; do
	if [ -d "$arg" ]; then
	    dirRollCall["$arg"]="true";
	    if ! [ "$returnState" = "false" ]; then returnState="true"; fi
	else
	    dirRollCall["$arg"]="false"; returnState="false";
	fi
    done
    
    #===Determine function return code===
    if [ "$returnState" = "true" ]; then
	return 0;
    else
	return 1;
    fi
} # Check that dir exists
displayMissing() {
    # Desc: Displays missing apps, files, and dirs
    # Usage: displayMissing
    # Version 0.1.1
    # Input: associative arrays: appRollCall, fileRollCall, dirRollCall
    # Output: stderr: messages indicating missing apps, file, or dirs
    # Depends: bash 5, checkAppFileDir()
    local missingApps value appMissing missingFiles fileMissing
    local missingDirs dirMissing
    
    #==BEGIN Display errors==
    #===BEGIN Display Missing Apps===
    missingApps="Missing apps  :";
    #for key in "${!appRollCall[@]}"; do echo "DEBUG:$key => ${appRollCall[$key]}"; done
    for key in "${!appRollCall[@]}"; do
	value="${appRollCall[$key]}";
	if [ "$value" = "false" ]; then
	    #echo "DEBUG:Missing apps: $key => $value";
	    missingApps="$missingApps""$key ";
	    appMissing="true";
	fi;
    done;
    if [ "$appMissing" = "true" ]; then  # Only indicate if an app is missing.
	echo "$missingApps" 1>&2;
    fi;
    unset value;
    #===END Display Missing Apps===

    #===BEGIN Display Missing Files===
    missingFiles="Missing files:";
    #for key in "${!fileRollCall[@]}"; do echo "DEBUG:$key => ${fileRollCall[$key]}"; done
    for key in "${!fileRollCall[@]}"; do
	value="${fileRollCall[$key]}";
	if [ "$value" = "false" ]; then
	    #echo "DEBUG:Missing files: $key => $value";
	    missingFiles="$missingFiles""$key ";
	    fileMissing="true";
	fi;
    done;
    if [ "$fileMissing" = "true" ]; then  # Only indicate if an app is missing.
	echo "$missingFiles" 1>&2;
    fi;
    unset value;
    #===END Display Missing Files===

    #===BEGIN Display Missing Directories===
    missingDirs="Missing dirs:";
    #for key in "${!dirRollCall[@]}"; do echo "DEBUG:$key => ${dirRollCall[$key]}"; done
    for key in "${!dirRollCall[@]}"; do
	value="${dirRollCall[$key]}";
	if [ "$value" = "false" ]; then
	    #echo "DEBUG:Missing dirs: $key => $value";
	    missingDirs="$missingDirs""$key ";
	    dirMissing="true";
	fi;
    done;
    if [ "$dirMissing" = "true" ]; then  # Only indicate if an dir is missing.
	echo "$missingDirs" 1>&2;
    fi;
    unset value;
    #===END Display Missing Directories===

    #==END Display errors==
} # Display missing apps, files, dirs
check_depends() {
    if ! checkapp feh parallel bkshuf b2sum; then
        displayMissing;
        die "FATAL:Missing apps.";
    fi;
    return 1;
}; # check dependencies
checkInt() {
    # Desc: Checks if arg is integer
    # Usage: checkInt arg
    # Input: arg: integer
    # Output: - return code 0 (if arg is integer)
    #         - return code 1 (if arg is not integer)
    # Example: if ! checkInt $arg; then echo "not int"; fi;
    # Version: 0.0.1
    local returnState

    #===Process Arg===
    if [[ $# -ne 1 ]]; then
	die "ERROR:Invalid number of arguments:$#";
    fi;
    
    RETEST1='^[0-9]+$'; # Regular Expression to test
    if [[ ! $1 =~ $RETEST1 ]] ; then
	returnState="false";
    else
	returnState="true";
    fi;

    #===Determine function return code===
    if [ "$returnState" = "true" ]; then
	return 0;
    else
	return 1;
    fi;
} # Checks if arg is integer
read_stdin() {
    # Desc: Consumes stdin; outputs as stdout lines
    # Input: stdin (consumes)
    # Output: stdout (newline delimited)
    # Example: printf "foo\nbar\n" | read_stdin
    # Depends: GNU bash (version 5.1.16)
    # Version: 0.0.1
    local input_stdin output;

    # Store stdin
    if [[ -p /dev/stdin ]]; then
        input_stdin="$(cat -)";
    fi; 
    
    # Store as output array elements
    ## Read in stdin
    if [[ -n $input_stdin ]]; then
        while read -r line; do
            output+=("$line");
        done < <(printf "%s\n" "$input_stdin");
    fi;

    # Print to stdout
    printf "%s\n" "${output[@]}";
}; # read stdin to stdout lines
read_psarg() {
    # Desc: Reads arguments; outputs as stdout lines
    # Input: args
    # Output: stdout (newline delimited)
    # Example: read_psarg "$@"
    # Depends: GNU bash (version 5.1.16)
    # Version: 0.0.1
    local input_psarg output;
    
    # Store arguments
    if [[ $# -gt 0 ]]; then
        input_psarg="$*";
    fi;
    
    # Store as output array elements
    ## Read in positional arguments
    if [[ -n $input_psarg ]]; then
        for arg in "$@"; do
            output+=("$arg");
        done;
    fi;

    # Print to stdout
    printf "%s\n" "${output[@]}";
}; # read positional argument to stdout lines
find_flist() {
    # Desc: print file list to stdout via `find` using script parameters
    # Input: arg1: path to dir
    #        var: find_depth
    #        var: pattern_find_iregex
    #        var: find_size
    if [[ ! -d "$1" ]]; then return 1; fi;
    must find "$1" -maxdepth "$fdepth" -type f -iregex "$firegex" -size +"$fsize";
}; # print file list to stdout from dir with script parameters
save_sample() {
    # Usage: save_sample arg1
    # Input: arg1   list_paths        (list of files to take samples from)
    #        envvar BKFEH_SAMPLE_DIR  (environment variable set outside of this script)
    #        envvar BKFEH_SAMPLE_SIZE (space limit for sample dir files)
    # Depends: GNU Parallel, GNU find, GNU Coreutils 8.32 (cut, find, du)
    #   BK-2020-03: bkshuf (0.0.1), yell()
    local list_paths
    sample_count="100"; # max number of images to put in sample dir
    sample_max_space="10000000"; # max bytes to put in sample dir

    # Load environment variables if set
    if [[ ! -v BKFEH_SAMPLE_DIR ]]; then return 0; fi; # return early if environment var not set.
    if [[ -v BKFEH_SAMPLE_SIZE ]] && checkInt "$BKFEH_SAMPLE_SIZE"; then
        sample_max_space="$BKFEH_SAMPLE_SIZE";
    fi;
    if [[ -v BKFEH_SAMPLE_COUNT ]] && checkInt "$BKFEH_SAMPLE_COUNT"; then
        sample_count="$BKFEH_SAMPLE_COUNT";
    fi;
    
    if [[ -n "$1" ]]; then
        list_paths="$1"; # newline-delimited list of file paths to sample from
    else
        yell "ERROR:NO paths available to sample.";
    fi;
    
    if [[ -d "$BKFEH_SAMPLE_DIR" ]]; then
        #sample_dir="$BKFEH_SAMPLE_DIR";
        yell "STATUS:Environment variable BKFEH_SAMPLE_DIR set. Clearing and saving samples...";
        
        ## clear previous sample
        count_prev_samples="$(find "$BKFEH_SAMPLE_DIR" -maxdepth 1 -type f | wc -l)";
        yell "STATUS:Deleting $count_prev_samples previous samples...";
        find "$BKFEH_SAMPLE_DIR" -maxdepth 1 -type f -exec rm '{}' \; ;
        
        ## save random sample
        yell "STATUS:Saving random sample of size $sample_count to $BKFEH_SAMPLE_DIR...";
        list_paths_sample="$(echo "$list_paths" | bkshuf "$sample_count" | head -n"$sample_count")";
        n_samp=0; # init sample file counter
        sample_log="$BKFEH_SAMPLE_DIR"/paths.txt;
        printf "%s,%s,%s\n" "n_samp" "file_hash" "file_path" >> "$sample_log";
        while read -r line; do
            if [[ -z "$line" ]]; then continue; fi;
            ### check size limit
            sample_act_space="$(du -bd1 "$BKFEH_SAMPLE_DIR" | cut -f1 )"; # actual used space
            cand_space="$(du -bd1 "$line" | cut -f1 )"; # size of candidate file to add
            sample_req_space="$((sample_act_space + cand_space))";

            ### Customize file names
            n_samp_w="$(printf "%s" "$sample_count" | wc -c)";
            n_samp_fmt="%0""$n_samp_w""d";
            n_samp_dd="$(printf "$n_samp_fmt" "$n_samp")"; # sample number fixed-width
            file_path="$line";
            #file_dir="$(dirname "$line")";
            file_name="$(basename "$line")";
            file_hash="$(b2sum -l32 "$line" | awk '{print $1}')"; # use file hash to avoid clobbering
            file_ext="${file_name##*.}";
            file_name="${file_name%.*}";
            file_shortname="${file_name:0:32}";
            file_name_new="$n_samp_dd"_"$file_hash".."$file_shortname"."$file_ext";
            file_path_new="$BKFEH_SAMPLE_DIR"/"$file_name_new"            
            if [[ "$sample_req_space" -lt "$sample_max_space" ]]; then
                #### add file to sample dir
                must cp -n "$file_path" "$file_path_new";
                #### note path in sample dir log
                printf "%s,%s,%s\n" "$n_samp_dd" "$file_hash" "$file_path" \
                       >> "$sample_log";
            fi;
            ((n_samp++));
        done < <( echo "$list_paths_sample" );
    else
        yell "ERROR:Does not exist: $BKFEH_SAMPLE_DIR";
    fi;    
}; # save sample of files

main() {
    # Depends: read_stdin_psarg() v0.0.1, check_depends()
    local re_dotfile;
    declare -a dirs_stdin dirs_psarg;
    declare -a paths_images;
    declare list_paths_images;
    check_depends;

    #Populate dirs_stdin and dirs_psarg arrays
    ## Read stdin as lines
    re_dotfile="^\."; # first char is a dot
    while read -r line; do
        line="$(readlink -e "$line")";
        # Check if dir
        if [[ ! -d "$line" ]]; then
            echo "ERROR:Not a dir:$line" 1>&2;
            continue;
        fi;
        dir_name="$(basename "$line")";
        # Exclude dotdirs
        if [[ "$dir_name" =~ $re_dotfile ]]; then
            echo "ERROR:Is a dotdir:$line" 1>&2;
            continue
        fi;
        dirs_stdin+=("$line");
    done < <(read_stdin);
    ## Read positional arguments as lines
    re_dotfile="^\."; # first char is a dot
    while read -r line; do
        line="$(readlink -e "$line")";
        # Check if dir
        if [[ ! -d "$line" ]]; then
            echo "ERROR:Not a dir:$line" 1>&2;
            continue;
        fi;
        dir_name="$(basename "$line")";
        # Exclude dotdirs
        if [[ "$dir_name" =~ $re_dotfile ]]; then
            echo "ERROR:Is a dotdir:$line" 1>&2;
            continue
        fi;
        dirs_psarg+=("$line");
    done < <(read_psarg "$@");
    
    # Catch all arrays empty
    if [[ "${#dirs_stdin[@]}" -le 0 ]] && [[ "${#dirs_psarg[@]}" -le 0 ]]; then
        die "FATAL:No valid directories provided.";
    fi;
    
    # Generate file list
    # Find settings
    firegex=".+\(jpg\|jpeg\|gif\|png\|webm\)$"; # update according to `find . -type f | grep -Eo "\.([[:alnum:]])+$" | sort -u`
    fsize="10k"; # default: minimum "10k"
    export firegex fsize ; # export for parallel
    ## Call find_filelist() in parallel for positional argument input
    if [[ "${#dirs_psarg[@]}" -gt 0 ]]; then
        fdepth=10; export fdepth; # 10 for dirs from positional arguments
        paths_images+=("$( parallel find_flist {} "$fdepth" "$firegex" "$fsize" ::: "${dirs_psarg[@]}" )"); # See [1]
    fi;
    ## Call find_filelist() in parallel for stdin input
    if [[ "${#dirs_stdin[@]}" -gt 0 ]]; then
        fdepth=1; export fdepth; # 1 for dirs from stdin
        paths_images+=("$( parallel find_flist {} "$fdepth" "$firegex" "$fsize" ::: "${dirs_stdin[@]}" )"); # See [1]
    fi;
    
    # Convert paths_images array into file list
    for i in "${!paths_images[@]}"; do
        list_paths_images="$(printf "%s\n%s" "${paths_images[$i]}" "$list_paths_images")";

        # Get stats
        file_count="$(wc -l < <(echo -n "$list_paths_images"))";
        echo "$DEBUG:file_count:$file_count"
    done;
    
    # Sort, remove duplicate paths
    list_paths_images="$(echo "$list_paths_images" | sort -u | tr -s '\n')";

    # Remove paths with dotfiles
    list_paths_images="$(echo "$list_paths_images" | grep -viE "/\." )";

    # Write
    list_paths_images_tmp="/dev/shm/$(date +%Y%m%dT%H%M%S.%N%z)"..feh_paths.txt;
    must echo -n "$list_paths_images" > "$list_paths_images_tmp";

    # Print stats
    yell "STATUS:Built file list in $SECONDS seconds.";

    # Run feh with filelist
    feh --full-screen --auto-zoom --draw-filename --filelist "$list_paths_images_tmp" && \
        must rm "$list_paths_images_tmp" &

    # Save sample to path in env. var. BKFEH_SAMPLE_DIR if set
    save_sample "$list_paths_images";
    
};
export -f yell die must read_stdin read_psarg find_flist;
#==END Define local functions==

main "$@";


# Author: Steven Baltakatei Sandoval
# License: GPLv3+
Commit	Line	Data
9eaab35f	1	#!/usr/bin/env bash
acaa0f3d	2	# Desc: Wrapper for feh that accepts directory paths via posargs or stdin lines.
82fabd02	3	# Version: 0.3.0
9eaab35f	4	# Ref/Attrib: [1] Tange, Ole. GNU Parallel with Bash Array. 2019-03-24. https://unix.stackexchange.com/a/508365/411854
82fabd02	5	# Depends: GNU Parallel, GNU Bash v5.1.16, feh 3.6.3, GNU Coreutils 8.32 (b2sum)
9eaab35f SBS	6
	7	#===Declare local functions===
	8	yell() { echo "$0: $*" >&2; } # print script path and all args to stderr
	9	die() { yell "$*"; exit 111; } # same as yell() but non-zero exit status
	10	must() { "$@" \|\| die "cannot $*"; } # runs args as command, reports args if command fails
	11	checkapp() {
	12	# Desc: If arg is a command, save result in assoc array 'appRollCall'
	13	# Usage: checkapp arg1 arg2 arg3 ...
	14	# Version: 0.1.1
	15	# Input: global assoc. array 'appRollCall'
	16	# Output: adds/updates key(value) to global assoc array 'appRollCall'
	17	# Depends: bash 5.0.3
	18	local returnState
	19
	20	#===Process Args===
	21	for arg in "$@"; do
	22	if command -v "$arg" 1>/dev/null 2>&1; then # Check if arg is a valid command
	23	appRollCall[$arg]="true";
	24	if ! [ "$returnState" = "false" ]; then returnState="true"; fi;
	25	else
	26	appRollCall[$arg]="false"; returnState="false";
	27	fi;
	28	done;
	29
	30	#===Determine function return code===
	31	if [ "$returnState" = "true" ]; then
	32	return 0;
	33	else
	34	return 1;
	35	fi;
	36	} # Check that app exists
	37	checkfile() {
	38	# Desc: If arg is a file path, save result in assoc array 'fileRollCall'
	39	# Usage: checkfile arg1 arg2 arg3 ...
	40	# Version: 0.1.1
	41	# Input: global assoc. array 'fileRollCall'
	42	# Output: adds/updates key(value) to global assoc array 'fileRollCall';
	43	# Output: returns 0 if app found, 1 otherwise
	44	# Depends: bash 5.0.3
	45	local returnState
	46
	47	#===Process Args===
	48	for arg in "$@"; do
	49	if [ -f "$arg" ]; then
	50	fileRollCall["$arg"]="true";
	51	if ! [ "$returnState" = "false" ]; then returnState="true"; fi;
	52	else
	53	fileRollCall["$arg"]="false"; returnState="false";
	54	fi;
	55	done;
	56
	57	#===Determine function return code===
	58	if [ "$returnState" = "true" ]; then
	59	return 0;
	60	else
	61	return 1;
	62	fi;
	63	} # Check that file exists
	64	checkdir() {
	65	# Desc: If arg is a dir path, save result in assoc array 'dirRollCall'
	66	# Usage: checkdir arg1 arg2 arg3 ...
	67	# Version 0.1.1
	68	# Input: global assoc. array 'dirRollCall'
	69	# Output: adds/updates key(value) to global assoc array 'dirRollCall';
70	# Output: returns 0 if app found, 1 otherwise
71	# Depends: Bash 5.0.3
72	local returnState
73
74	#===Process Args===
75	for arg in "$@"; do
76	if [ -d "$arg" ]; then
77	dirRollCall["$arg"]="true";
78	if ! [ "$returnState" = "false" ]; then returnState="true"; fi
79	else
80	dirRollCall["$arg"]="false"; returnState="false";
81	fi
82	done
83
84	#===Determine function return code===
85	if [ "$returnState" = "true" ]; then
86	return 0;
87	else
88	return 1;
89	fi
90	} # Check that dir exists
91	displayMissing() {
92	# Desc: Displays missing apps, files, and dirs
93	# Usage: displayMissing
94	# Version 0.1.1
95	# Input: associative arrays: appRollCall, fileRollCall, dirRollCall
96	# Output: stderr: messages indicating missing apps, file, or dirs
97	# Depends: bash 5, checkAppFileDir()
98	local missingApps value appMissing missingFiles fileMissing
99	local missingDirs dirMissing
100
101	#==BEGIN Display errors==
102	#===BEGIN Display Missing Apps===
103	missingApps="Missing apps :";
104	#for key in "${!appRollCall[@]}"; do echo "DEBUG:$key => ${appRollCall[$key]}"; done
105	for key in "${!appRollCall[@]}"; do
106	value="${appRollCall[$key]}";
107	if [ "$value" = "false" ]; then
108	#echo "DEBUG:Missing apps: $key => $value";
109	missingApps="$missingApps""$key ";
110	appMissing="true";
111	fi;
112	done;
113	if [ "$appMissing" = "true" ]; then # Only indicate if an app is missing.
114	echo "$missingApps" 1>&2;
115	fi;
116	unset value;
117	#===END Display Missing Apps===
118
119	#===BEGIN Display Missing Files===
120	missingFiles="Missing files:";
121	#for key in "${!fileRollCall[@]}"; do echo "DEBUG:$key => ${fileRollCall[$key]}"; done
122	for key in "${!fileRollCall[@]}"; do
123	value="${fileRollCall[$key]}";
124	if [ "$value" = "false" ]; then
125	#echo "DEBUG:Missing files: $key => $value";
126	missingFiles="$missingFiles""$key ";
127	fileMissing="true";
128	fi;
129	done;
130	if [ "$fileMissing" = "true" ]; then # Only indicate if an app is missing.
131	echo "$missingFiles" 1>&2;
132	fi;
133	unset value;
134	#===END Display Missing Files===
135
136	#===BEGIN Display Missing Directories===
137	missingDirs="Missing dirs:";
138	#for key in "${!dirRollCall[@]}"; do echo "DEBUG:$key => ${dirRollCall[$key]}"; done
139	for key in "${!dirRollCall[@]}"; do
140	value="${dirRollCall[$key]}";
141	if [ "$value" = "false" ]; then
142	#echo "DEBUG:Missing dirs: $key => $value";
143	missingDirs="$missingDirs""$key ";
144	dirMissing="true";
145	fi;
146	done;
147	if [ "$dirMissing" = "true" ]; then # Only indicate if an dir is missing.
148	echo "$missingDirs" 1>&2;
149	fi;
150	unset value;
151	#===END Display Missing Directories===
152
153	#==END Display errors==
154	} # Display missing apps, files, dirs
155	check_depends() {
82fabd02 SBS	156	if ! checkapp feh parallel bkshuf b2sum; then
	157	displayMissing;
	158	die "FATAL:Missing apps.";
	159	fi;
9eaab35f SBS	160	return 1;
9eaab35f SBS	161	}; # check dependencies
acaa0f3d SBS	162	checkInt() {
	163	# Desc: Checks if arg is integer
	164	# Usage: checkInt arg
	165	# Input: arg: integer
	166	# Output: - return code 0 (if arg is integer)
	167	# - return code 1 (if arg is not integer)
	168	# Example: if ! checkInt $arg; then echo "not int"; fi;
	169	# Version: 0.0.1
	170	local returnState
	171
	172	#===Process Arg===
	173	if [[ $# -ne 1 ]]; then
	174	die "ERROR:Invalid number of arguments:$#";
	175	fi;
	176
	177	RETEST1='^[0-9]+$'; # Regular Expression to test
	178	if [[ ! $1 =~ $RETEST1 ]] ; then
	179	returnState="false";
	180	else
	181	returnState="true";
	182	fi;
	183
	184	#===Determine function return code===
	185	if [ "$returnState" = "true" ]; then
	186	return 0;
	187	else
	188	return 1;
	189	fi;
	190	} # Checks if arg is integer
335b71e1 SBS	191	read_stdin() {
335b71e1 SBS	192	# Desc: Consumes stdin; outputs as stdout lines
9eaab35f	193	# Input: stdin (consumes)
9eaab35f	194	# Output: stdout (newline delimited)
335b71e1	195	# Example: printf "foo\nbar\n" \| read_stdin
9eaab35f	196	# Depends: GNU bash (version 5.1.16)
335b71e1 SBS	197	# Version: 0.0.1
335b71e1 SBS	198	local input_stdin output;
9eaab35f SBS	199
	200	# Store stdin
	201	if [[ -p /dev/stdin ]]; then
	202	input_stdin="$(cat -)";
335b71e1	203	fi;
966a8d11	204
335b71e1	205	# Store as output array elements
9eaab35f SBS	206	## Read in stdin
	207	if [[ -n $input_stdin ]]; then
	208	while read -r line; do
	209	output+=("$line");
	210	done < <(printf "%s\n" "$input_stdin");
	211	fi;
335b71e1 SBS	212
	213	# Print to stdout
	214	printf "%s\n" "${output[@]}";
	215	}; # read stdin to stdout lines
	216	read_psarg() {
	217	# Desc: Reads arguments; outputs as stdout lines
	218	# Input: args
	219	# Output: stdout (newline delimited)
	220	# Example: read_psarg "$@"
	221	# Depends: GNU bash (version 5.1.16)
	222	# Version: 0.0.1
	223	local input_psarg output;
	224
	225	# Store arguments
	226	if [[ $# -gt 0 ]]; then
	227	input_psarg="$*";
	228	fi;
	229
	230	# Store as output array elements
9eaab35f SBS	231	## Read in positional arguments
	232	if [[ -n $input_psarg ]]; then
	233	for arg in "$@"; do
	234	output+=("$arg");
	235	done;
	236	fi;
	237
	238	# Print to stdout
	239	printf "%s\n" "${output[@]}";
335b71e1 SBS	240	}; # read positional argument to stdout lines
335b71e1 SBS	241	find_flist() {
9eaab35f SBS	242	# Desc: print file list to stdout via `find` using script parameters
	243	# Input: arg1: path to dir
	244	# var: find_depth
	245	# var: pattern_find_iregex
	246	# var: find_size
	247	if [[ ! -d "$1" ]]; then return 1; fi;
335b71e1	248	must find "$1" -maxdepth "$fdepth" -type f -iregex "$firegex" -size +"$fsize";
9eaab35f	249	}; # print file list to stdout from dir with script parameters
4db78240 SBS	250	save_sample() {
	251	# Usage: save_sample arg1
	252	# Input: arg1 list_paths (list of files to take samples from)
	253	# envvar BKFEH_SAMPLE_DIR (environment variable set outside of this script)
acaa0f3d	254	# envvar BKFEH_SAMPLE_SIZE (space limit for sample dir files)
82fabd02 SBS	255	# Depends: GNU Parallel, GNU find, GNU Coreutils 8.32 (cut, find, du)
82fabd02 SBS	256	# BK-2020-03: bkshuf (0.0.1), yell()
4db78240	257	local list_paths
acaa0f3d SBS	258	sample_count="100"; # max number of images to put in sample dir
	259	sample_max_space="10000000"; # max bytes to put in sample dir
	260
	261	# Load environment variables if set
4db78240	262	if [[ ! -v BKFEH_SAMPLE_DIR ]]; then return 0; fi; # return early if environment var not set.
acaa0f3d SBS	263	if [[ -v BKFEH_SAMPLE_SIZE ]] && checkInt "$BKFEH_SAMPLE_SIZE"; then
	264	sample_max_space="$BKFEH_SAMPLE_SIZE";
	265	fi;
	266	if [[ -v BKFEH_SAMPLE_COUNT ]] && checkInt "$BKFEH_SAMPLE_COUNT"; then
	267	sample_count="$BKFEH_SAMPLE_COUNT";
	268	fi;
4db78240	269
82fabd02	270	if [[ -n "$1" ]]; then
4db78240 SBS	271	list_paths="$1"; # newline-delimited list of file paths to sample from
	272	else
	273	yell "ERROR:NO paths available to sample.";
	274	fi;
	275
	276	if [[ -d "$BKFEH_SAMPLE_DIR" ]]; then
82fabd02	277	#sample_dir="$BKFEH_SAMPLE_DIR";
4db78240 SBS	278	yell "STATUS:Environment variable BKFEH_SAMPLE_DIR set. Clearing and saving samples...";
	279
	280	## clear previous sample
82fabd02 SBS	281	count_prev_samples="$(find "$BKFEH_SAMPLE_DIR" -maxdepth 1 -type f \| wc -l)";
82fabd02 SBS	282	yell "STATUS:Deleting $count_prev_samples previous samples...";
4db78240 SBS	283	find "$BKFEH_SAMPLE_DIR" -maxdepth 1 -type f -exec rm '{}' \; ;
	284
	285	## save random sample
	286	yell "STATUS:Saving random sample of size $sample_count to $BKFEH_SAMPLE_DIR...";
82fabd02 SBS	287	list_paths_sample="$(echo "$list_paths" \| bkshuf "$sample_count" \| head -n"$sample_count")";
	288	n_samp=0; # init sample file counter
	289	sample_log="$BKFEH_SAMPLE_DIR"/paths.txt;
	290	printf "%s,%s,%s\n" "n_samp" "file_hash" "file_path" >> "$sample_log";
4db78240 SBS	291	while read -r line; do
	292	if [[ -z "$line" ]]; then continue; fi;
	293	### check size limit
	294	sample_act_space="$(du -bd1 "$BKFEH_SAMPLE_DIR" \| cut -f1 )"; # actual used space
	295	cand_space="$(du -bd1 "$line" \| cut -f1 )"; # size of candidate file to add
	296	sample_req_space="$((sample_act_space + cand_space))";
82fabd02 SBS	297
	298	### Customize file names
	299	n_samp_w="$(printf "%s" "$sample_count" \| wc -c)";
	300	n_samp_fmt="%0""$n_samp_w""d";
	301	n_samp_dd="$(printf "$n_samp_fmt" "$n_samp")"; # sample number fixed-width
	302	file_path="$line";
	303	#file_dir="$(dirname "$line")";
	304	file_name="$(basename "$line")";
	305	file_hash="$(b2sum -l32 "$line" \| awk '{print $1}')"; # use file hash to avoid clobbering
	306	file_ext="${file_name##*.}";
	307	file_name="${file_name%.*}";
	308	file_shortname="${file_name:0:32}";
	309	file_name_new="$n_samp_dd"_"$file_hash".."$file_shortname"."$file_ext";
	310	file_path_new="$BKFEH_SAMPLE_DIR"/"$file_name_new"
4db78240 SBS	311	if [[ "$sample_req_space" -lt "$sample_max_space" ]]; then
4db78240 SBS	312	#### add file to sample dir
82fabd02 SBS	313	must cp -n "$file_path" "$file_path_new";
	314	#### note path in sample dir log
	315	printf "%s,%s,%s\n" "$n_samp_dd" "$file_hash" "$file_path" \
	316	>> "$sample_log";
4db78240	317	fi;
82fabd02	318	((n_samp++));
4db78240 SBS	319	done < <( echo "$list_paths_sample" );
	320	else
	321	yell "ERROR:Does not exist: $BKFEH_SAMPLE_DIR";
	322	fi;
	323	}; # save sample of files
	324
9eaab35f SBS	325	main() {
9eaab35f SBS	326	# Depends: read_stdin_psarg() v0.0.1, check_depends()
966a8d11	327	local re_dotfile;
335b71e1	328	declare -a dirs_stdin dirs_psarg;
9eaab35f SBS	329	declare -a paths_images;
	330	declare list_paths_images;
	331	check_depends;
	332
335b71e1 SBS	333	#Populate dirs_stdin and dirs_psarg arrays
335b71e1 SBS	334	## Read stdin as lines
966a8d11	335	re_dotfile="^\."; # first char is a dot
9eaab35f	336	while read -r line; do
4db78240	337	line="$(readlink -e "$line")";
966a8d11	338	# Check if dir
9eaab35f SBS	339	if [[ ! -d "$line" ]]; then
	340	echo "ERROR:Not a dir:$line" 1>&2;
	341	continue;
	342	fi;
966a8d11 SBS	343	dir_name="$(basename "$line")";
	344	# Exclude dotdirs
	345	if [[ "$dir_name" =~ $re_dotfile ]]; then
	346	echo "ERROR:Is a dotdir:$line" 1>&2;
	347	continue
	348	fi;
335b71e1 SBS	349	dirs_stdin+=("$line");
	350	done < <(read_stdin);
	351	## Read positional arguments as lines
	352	re_dotfile="^\."; # first char is a dot
	353	while read -r line; do
4db78240	354	line="$(readlink -e "$line")";
335b71e1 SBS	355	# Check if dir
	356	if [[ ! -d "$line" ]]; then
	357	echo "ERROR:Not a dir:$line" 1>&2;
	358	continue;
	359	fi;
	360	dir_name="$(basename "$line")";
	361	# Exclude dotdirs
	362	if [[ "$dir_name" =~ $re_dotfile ]]; then
	363	echo "ERROR:Is a dotdir:$line" 1>&2;
	364	continue
	365	fi;
	366	dirs_psarg+=("$line");
	367	done < <(read_psarg "$@");
	368
	369	# Catch all arrays empty
	370	if [[ "${#dirs_stdin[@]}" -le 0 ]] && [[ "${#dirs_psarg[@]}" -le 0 ]]; then
	371	die "FATAL:No valid directories provided.";
	372	fi;
9eaab35f SBS	373
9eaab35f SBS	374	# Generate file list
335b71e1 SBS	375	# Find settings
	376	firegex=".+\(jpg\\|jpeg\\|gif\\|png\\|webm\)$"; # update according to `find . -type f \| grep -Eo "\.([[:alnum:]])+$" \| sort -u`
	377	fsize="10k"; # default: minimum "10k"
	378	export firegex fsize ; # export for parallel
	379	## Call find_filelist() in parallel for positional argument input
	380	if [[ "${#dirs_psarg[@]}" -gt 0 ]]; then
	381	fdepth=10; export fdepth; # 10 for dirs from positional arguments
	382	paths_images+=("$( parallel find_flist {} "$fdepth" "$firegex" "$fsize" ::: "${dirs_psarg[@]}" )"); # See [1]
	383	fi;
	384	## Call find_filelist() in parallel for stdin input
	385	if [[ "${#dirs_stdin[@]}" -gt 0 ]]; then
75e92939	386	fdepth=1; export fdepth; # 1 for dirs from stdin
335b71e1 SBS	387	paths_images+=("$( parallel find_flist {} "$fdepth" "$firegex" "$fsize" ::: "${dirs_stdin[@]}" )"); # See [1]
	388	fi;
	389
9eaab35f SBS	390	# Convert paths_images array into file list
	391	for i in "${!paths_images[@]}"; do
	392	list_paths_images="$(printf "%s\n%s" "${paths_images[$i]}" "$list_paths_images")";
	393
	394	# Get stats
	395	file_count="$(wc -l < <(echo -n "$list_paths_images"))";
	396	echo "$DEBUG:file_count:$file_count"
	397	done;
966a8d11 SBS	398
966a8d11 SBS	399	# Sort, remove duplicate paths
9eaab35f SBS	400	list_paths_images="$(echo "$list_paths_images" \| sort -u \| tr -s '\n')";
9eaab35f SBS	401
335b71e1 SBS	402	# Remove paths with dotfiles
	403	list_paths_images="$(echo "$list_paths_images" \| grep -viE "/\." )";
	404
9eaab35f SBS	405	# Write
	406	list_paths_images_tmp="/dev/shm/$(date +%Y%m%dT%H%M%S.%N%z)"..feh_paths.txt;
	407	must echo -n "$list_paths_images" > "$list_paths_images_tmp";
	408
	409	# Print stats
	410	yell "STATUS:Built file list in $SECONDS seconds.";
	411
	412	# Run feh with filelist
acaa0f3d SBS	413	feh --full-screen --auto-zoom --draw-filename --filelist "$list_paths_images_tmp" && \
acaa0f3d SBS	414	must rm "$list_paths_images_tmp" &
9eaab35f	415
acaa0f3d SBS	416	# Save sample to path in env. var. BKFEH_SAMPLE_DIR if set
	417	save_sample "$list_paths_images";
	418
9eaab35f	419	};
335b71e1	420	export -f yell die must read_stdin read_psarg find_flist;
9eaab35f SBS	421	#==END Define local functions==
	422
	423	main "$@";
	424
75e92939	425
9eaab35f SBS	426	# Author: Steven Baltakatei Sandoval
9eaab35f SBS	427	# License: GPLv3+