Add new features
New command-line options: import, autoimport, init Improve adding of tags and files
This commit is contained in:
parent
c71d3af0a2
commit
e6e13c10b0
SSH Key Fingerprint:
SHA256:vMuSwvwAvcT6htVAioMP7rzzwMQNi3roESyhv+nAxeg
205
tag.sh
205
tag.sh
|
|
@ -2,6 +2,10 @@
|
||||||
set -ueo pipefail
|
set -ueo pipefail
|
||||||
# usage:
|
# usage:
|
||||||
# - init - create an empty database
|
# - init - create an empty database
|
||||||
|
# - import - import $FILE or import $DIR, both can be multiple values
|
||||||
|
# - autoimport ($REGEX|video) $FOLDER - will import all files in $FOLDER with a $REGEX filter,
|
||||||
|
# or one of the preset filters:
|
||||||
|
# - video
|
||||||
# - add (tag|filename) $VALUE - adds a new tag or filename of $VALUE
|
# - add (tag|filename) $VALUE - adds a new tag or filename of $VALUE
|
||||||
# - list (tags|files) - lists all the values in a table
|
# - list (tags|files) - lists all the values in a table
|
||||||
# - bytag $TAG - list all files by tag label
|
# - bytag $TAG - list all files by tag label
|
||||||
|
|
@ -10,6 +14,11 @@ set -ueo pipefail
|
||||||
|
|
||||||
DB_FILE="tags.db"
|
DB_FILE="tags.db"
|
||||||
DB_SCHEMA="database.sql"
|
DB_SCHEMA="database.sql"
|
||||||
|
declare -A FILTERS
|
||||||
|
FILTERS[video]="avi|flv|mkv|mov|mp4|mpg|ogv|webm|wmv"
|
||||||
|
# LIST_RECURSIVE_FILES_CMD="find ${FOLDER} -type f -regextype posix-extended -iregex \"${REGEX}\""
|
||||||
|
|
||||||
|
[ "${DEBUG:-0}" = "1" ] && set -x
|
||||||
|
|
||||||
fail() {
|
fail() {
|
||||||
echo "$1" >&2;
|
echo "$1" >&2;
|
||||||
|
|
@ -20,8 +29,23 @@ log() {
|
||||||
echo "$1"
|
echo "$1"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
init() {
|
||||||
|
[ -f "$DB_FILE" ] && fail "Database file \"$DB_FILE\" already exists. Aborting."
|
||||||
|
cat "$DB_SCHEMA" | sqlite3 "$DB_FILE"
|
||||||
|
log "Empty database file \"$DB_FILE\" created."
|
||||||
|
}
|
||||||
|
|
||||||
|
enumerate_positional_parameters() {
|
||||||
|
log "Positional parameters in $FUNCNAME:"
|
||||||
|
CNT=0
|
||||||
|
for VALUE in "$@"; do
|
||||||
|
let CNT++ || true
|
||||||
|
echo "${CNT}: $VALUE"
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
listtags() {
|
listtags() {
|
||||||
# $FILENAME $LIMIT
|
# $FILENAME $LIMIT=ALL
|
||||||
[ -z "$1" ] && fail "No filename supplied."
|
[ -z "$1" ] && fail "No filename supplied."
|
||||||
FILENAME="$1"
|
FILENAME="$1"
|
||||||
shift
|
shift
|
||||||
|
|
@ -30,77 +54,136 @@ listtags() {
|
||||||
ADDITIONAL_QUERY=""
|
ADDITIONAL_QUERY=""
|
||||||
[ $LIMIT -gt 0 ] && ADDITIONAL_QUERY="LIMIT $LIMIT"
|
[ $LIMIT -gt 0 ] && ADDITIONAL_QUERY="LIMIT $LIMIT"
|
||||||
|
|
||||||
sqlite3 -table "$DB_FILE" \
|
sqlite_query \
|
||||||
"SELECT filename, label from files \
|
"SELECT filename, label from files \
|
||||||
INNER JOIN tags_ties ON tags_ties.fid = files.fid \
|
INNER JOIN tags_ties ON tags_ties.id = files.id \
|
||||||
INNER JOIN tags ON tags.tid = tags_ties.tid \
|
INNER JOIN tags ON tags.id = tags_ties.id \
|
||||||
WHERE filename = \"$FILENAME\"\
|
WHERE filename = \"$FILENAME\"\
|
||||||
$ADDITIONAL_QUERY"
|
$ADDITIONAL_QUERY"
|
||||||
}
|
}
|
||||||
|
|
||||||
if [[ "$1" = "init" ]]; then
|
add() {
|
||||||
[ -f "$DB_FILE" ] && fail "Database file \"$DB_FILE\" already exists. Aborting."
|
# $TYPE(tag|path|hash) $VALUE1..$VALUEN
|
||||||
cat "$DB_SCHEMA" | sqlite3 "$DB_FILE"
|
[ "$1" = "tag" ] && shift && add_tag "$@" && exit 0
|
||||||
log "Empty database file \"$DB_FILE\" created."
|
[ "$1" = "path" ] && shift && add_path "$@" && exit 0
|
||||||
|
[ "$1" = "hash" ] && fail "Adding hashes is not implemented yet." && TABLE="hashes" && COLUMN="md5,filename"
|
||||||
exit 0
|
exit 0
|
||||||
fi
|
}
|
||||||
|
|
||||||
if [[ "$1" = "add" ]]; then
|
sqlite_query() {
|
||||||
shift
|
# $QUERY
|
||||||
[ "$1" = "tag" ] && TABLE="tags" && COLUMN="label"
|
sqlite3 -table "$DB_FILE" "$1"
|
||||||
[ "$1" = "file" ] && TABLE="files" && COLUMN="filename"
|
}
|
||||||
# TODO: refactor the add function to allow adding hashes
|
|
||||||
# [ "$1" = "hash" ] && TABLE="hashes" && COLUMN="md5,filename"
|
sqlite_lastrows() {
|
||||||
shift
|
# $TABLE $LIMIT
|
||||||
[ -z "$1" ] && fail "No value supplied."
|
sqlite_query "SELECT * FROM \"$1\" ORDER BY id DESC LIMIT ${2}"
|
||||||
COUNTER=0
|
}
|
||||||
while true; do
|
|
||||||
NEW_TAG="$1"
|
sqlite_insert() {
|
||||||
|
# $TABLE $COLUMN $VALUES
|
||||||
|
# $VALUES can be comma-delimited
|
||||||
|
[ -z "$1" ] && fail "No table specified."
|
||||||
|
TABLE="$1"
|
||||||
|
[ -z "$2" ] && fail "No column(s) supplied."
|
||||||
|
COLUMN="$2"
|
||||||
|
[ -z "$3" ] && fail "No column values supplied."
|
||||||
|
shift 2
|
||||||
|
VALUES="(\"$1\")"
|
||||||
|
if [[ ! -z "${2:-}" ]]; then
|
||||||
|
while true; do
|
||||||
|
shift
|
||||||
|
[ -z "${1:-}" ] && break
|
||||||
|
VALUES+=",(\"$1\")"
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
QUERY="INSERT INTO $TABLE ($COLUMN) VALUES ${VALUES} RETURNING *;"
|
||||||
|
sqlite_query "$QUERY"
|
||||||
|
}
|
||||||
|
|
||||||
|
add_tag() {
|
||||||
|
# $TAG1...$TAGN
|
||||||
|
TABLE="tags"
|
||||||
|
COLUMN="label"
|
||||||
|
sqlite_insert "$TABLE" "$COLUMN" "$@"
|
||||||
|
}
|
||||||
|
|
||||||
|
add_path() {
|
||||||
|
# $PATH1..$PATHN
|
||||||
|
TABLE="files"
|
||||||
|
COLUMN="filename"
|
||||||
|
sqlite_insert "$TABLE" "$COLUMN" "$@"
|
||||||
|
}
|
||||||
|
|
||||||
|
add_path_auto() {
|
||||||
|
# $REGEX $FOLDER
|
||||||
|
find "${2}" -type f -regextype posix-extended -iregex "$1" -exec "$0" import "{}" +
|
||||||
|
}
|
||||||
|
|
||||||
|
main() {
|
||||||
|
if [[ "$1" = "init" ]]; then
|
||||||
|
init
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$1" = "import" ]]; then
|
||||||
shift
|
shift
|
||||||
sqlite3 -table "$DB_FILE" "INSERT INTO $TABLE ($COLUMN) VALUES (\"${NEW_TAG}\")"
|
add "path" "$@"
|
||||||
COUNTER=$((COUNTER++))
|
fi
|
||||||
[ -z "${1:-}" ] && break
|
|
||||||
done
|
|
||||||
sqlite3 -table "$DB_FILE" "SELECT * FROM \"$TABLE\" LIMIT $COUNTER"
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$1" = "list" ]]; then
|
if [[ "$1" = "autoimport" ]]; then
|
||||||
[ -z "$2" ] && fail "No table supplied."
|
|
||||||
TABLE_NAME="$2"
|
|
||||||
sqlite3 -table "$DB_FILE" "SELECT * FROM \"${TABLE_NAME}\""
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$1" = "bytag" ]]; then
|
|
||||||
[ -z "$2" ] && fail "No tag supplied."
|
|
||||||
TAG_NAME="$2"
|
|
||||||
sqlite3 -table "$DB_FILE" "SELECT filename FROM files WHERE fid = (SELECT fid FROM tags_ties WHERE tid = (SELECT tid FROM tags WHERE label = \"${TAG_NAME}\"))"
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$1" = "tag" ]]; then
|
|
||||||
shift
|
|
||||||
[ -z "$1" ] && fail "No filename supplied."
|
|
||||||
FILENAME="$1"
|
|
||||||
shift
|
|
||||||
[ -z "$1" ] && fail "No tag supplied."
|
|
||||||
COUNTER=0
|
|
||||||
while true; do
|
|
||||||
LABEL="$1"
|
|
||||||
shift
|
shift
|
||||||
sqlite3 -table "$DB_FILE" "INSERT INTO tags_ties (fid, tid) VALUES ((SELECT fid FROM files WHERE filename = \"$FILENAME\"),(SELECT tid FROM tags WHERE label = \"$LABEL\"))"
|
if [[ "${1:-}" =~ "${!FILTERS[@]}" ]]; then
|
||||||
COUNTER=$((COUNTER++))
|
add_path_auto ".*(${FILTERS[$1]})" "$2"
|
||||||
[ -z "${1:-}" ] && break
|
else
|
||||||
done
|
add_path_auto "$1" "$2"
|
||||||
listtags "$FILENAME" "$COUNTER"
|
fi
|
||||||
exit 0
|
fi
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$1" = "listtags" ]]; then
|
if [[ "$1" = "add" ]]; then
|
||||||
listtags "$2"
|
shift
|
||||||
exit 0
|
add "$@"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
if [[ "$1" = "list" ]]; then
|
||||||
|
[ -z "$2" ] && fail "No table supplied."
|
||||||
|
TABLE_NAME="$2"
|
||||||
|
sqlite3 -table "$DB_FILE" "SELECT * FROM \"${TABLE_NAME}\""
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$1" = "bytag" ]]; then
|
||||||
|
[ -z "$2" ] && fail "No tag supplied."
|
||||||
|
TAG_NAME="$2"
|
||||||
|
sqlite3 -table "$DB_FILE" "SELECT filename FROM files WHERE id = (SELECT id FROM tags_ties WHERE id = (SELECT id FROM tags WHERE label = \"${TAG_NAME}\"))"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$1" = "tag" ]]; then
|
||||||
|
shift
|
||||||
|
[ -z "$1" ] && fail "No filename supplied."
|
||||||
|
FILENAME="$1"
|
||||||
|
shift
|
||||||
|
[ -z "$1" ] && fail "No tag supplied."
|
||||||
|
COUNTER=0
|
||||||
|
while true; do
|
||||||
|
LABEL="$1"
|
||||||
|
shift
|
||||||
|
sqlite3 -table "$DB_FILE" "INSERT INTO tags_ties (id, id) VALUES ((SELECT id FROM files WHERE filename = \"$FILENAME\"),(SELECT id FROM tags WHERE label = \"$LABEL\"))"
|
||||||
|
COUNTER=$((COUNTER++))
|
||||||
|
[ -z "${1:-}" ] && break
|
||||||
|
done
|
||||||
|
listtags "$FILENAME" "$COUNTER"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$1" = "listtags" ]]; then
|
||||||
|
listtags "$2"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
main "$@"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue