added initial code

1 files changed, 155 insertions, 0 deletions

diff --git a/animegrep.sh b/animegrep.sh
new file mode 100755
index 0000000..5c7297a
--- /dev/null
+++ b/animegrep.sh
@@ -0,0 +1,155 @@
+#!/bin/bash
+
+# check for mkvextract
+command -v mkvextract >/dev/null 2>&1 || { echo >&2 "Requires mkvextract. Aborting."; exit; }
+
+POSITIONAL=();
+while [[ $# -gt 0 ]]
+do
+	key="$1";
+	case $key in
+		-d|--directory)
+			DIRECTORY="$2";
+			shift;
+			shift;
+			;;
+		-f|--file)
+			FILE="$2";
+			shift;
+			shift;
+			;;
+		-t|--track)
+			TRACK="$2";
+			shift;
+			shift;
+			;;
+		-w|--word)
+			WORD="$2";
+			shift;
+			shift;
+			;;
+		-m|--merge)
+			MERGE=1;
+			shift;
+			shift;
+			;;
+	esac
+done
+set -- "{$POSITIONAL[@]}";
+
+# check if out directory exists
+# TODO handle case out exists but other subdirs don't
+if [ ! -d "out" ]; then
+	echo "making out directory..";
+	mkdir out;
+	mkdir out/clips;
+fi
+
+if ! [[ -z $FILE || -z $DIRECTORY ]]; then
+	echo "You have both -d and -f set.. choose one, not both..";
+	exit;
+fi
+
+if [[ -z $FILE && -z $DIRECTORY ]]; then
+	echo "no file or directory provided (use -f [file] or -d [dir])..";
+	exit;
+fi
+
+if [ -z "$TRACK" ]; then
+	#assume subtitle track = 2
+	echo "track not set (use -t [truck number]). You kind find the subtitle track number using 'mkvinfo [file]'..";
+	exit;
+fi
+
+if [ -z "$WORD" ]; then
+	echo "no word set, using default word (use -w [word] to set the word)..";
+	WORD="idiot"; # baka~
+fi
+
+if ! [ -z "$FILE" ]; then
+	#echo "using single file..";
+	SINGLEFILE=true;
+fi
+
+if ! [ -z "$DIRECTORY" ]; then
+	#echo "using directory";
+	SINGLEFILE=false;
+fi
+
+# we're done the boring stuff
+
+x=0;
+
+function getsubs
+{
+	# $1 = file
+	#echo "getting subs for "$1"";
+	rm out/subs.srt;
+	CFILE=$(basename "$1" .mkv);
+	#echo "CFILE is "$CFILE"";
+	#echo "extracting subs from $1 track ${TRACK} to out/subs.srt";
+	mkvextract tracks "$1" ${TRACK}:out/subs.srt;
+	grepsubs "$1" "$CFILE";
+}
+
+function grepsubs
+{
+	# $1 = file $2 = CFILE
+	while read -r line; do
+		((x++));
+		#echo "line in grep ${line}";
+		#parselinie $x "$1" $line;
+		parseline "$1" "$line" "$2";
+	done < <(grep "$WORD" out/subs.srt);
+}
+
+function parseline
+{
+
+	# $1 = file $2 = line $3 = CFILE
+	IFS=',' read -r -a array <<< "$2";
+	echo "start time: ${array[1]}";
+	echo "end time: ${array[2]}";
+	cutvideo "$1" "${array[1]}" "${array[2]}" "$3";
+}
+
+function cutvideo
+{
+	# $1 = file $2 = start time $3 = end time $4 = CFILE
+	ffmpeg -i "$1" -ss 0"$2" -to 0"$3" -async 1 -c:v libx264 -preset ultrafast out/clips/clip_"$x"_"$4".mkv < /dev/null
+}
+
+function merge
+{
+	echo "merging mkv's..";
+	str=();
+	x=0;
+	for f in out/clips/*.mkv; do
+		if [ $x == 0 ]; then
+			str+=""$f"";
+		fi
+		if [ $x != 0 ]; then
+			str+=" +"$f"";
+		fi
+		((x++));
+	done;
+	mkvmerge $str -o out/out.mkv
+}
+
+# getsubs function will run the chain that does the whole process
+
+if ! [ -z "$MERGE" ]; then
+	merge;
+	exit;
+fi
+
+if [ $SINGLEFILE == true ]; then
+	echo "doing one file";
+	getsubs "$FILE";
+fi
+
+if [ $SINGLEFILE == false ]; then
+	for f in "$DIRECTORY"/*.mkv; do
+		getsubs "$f";
+	done;
+fi