Files
CloudDeploy/prepdb.sh

110 lines
4.7 KiB
Bash

# sqlite db replace function
replace_in_sqlite_db() {
local DB_PATH="$1"
local OLD_STRING="$2"
local NEW_STRING="$3"
# Check if the database file exists
if [ ! -f "$DB_PATH" ]; then
echo "Error: Database file '$DB_PATH' does not exist."
return 1
fi
# Check if jq is installed
if ! command -v jq &> /dev/null; then
echo "Error: 'jq' is not installed. Please install it first (e.g., 'sudo apt install jq')."
return 1
fi
echo "Starting replacement of '$OLD_STRING' with '$NEW_STRING' in all string-based columns in '$DB_PATH'..."
# Escape single quotes in the strings for SQL
local OLD_STRING_ESC=$(echo "$OLD_STRING" | sed "s/'/''/g")
local NEW_STRING_ESC=$(echo "$NEW_STRING" | sed "s/'/''/g")
# Get a list of all tables in the database
local TABLES
TABLES=$(sqlite3 "$DB_PATH" ".tables")
# Loop through each table
for TABLE in $TABLES; do
echo "Processing table: $TABLE"
# Get a list of all columns in the table
sqlite3 "$DB_PATH" "PRAGMA table_info($TABLE);" | awk -F'|' '{print $2","$3}' | while IFS=, read -r COLUMN COLUMN_TYPE; do
# Skip empty lines
if [ -z "$COLUMN" ]; then
continue
fi
# Check if the column is a string-based type (TEXT, VARCHAR, CHAR, CLOB, etc.)
if [[ "$COLUMN_TYPE" =~ [Tt][Ee][Xx][Tt]|[Vv][Aa][Rr][Cc][Hh][Aa][Rr]|[Cc][Hh][Aa][Rr]|[Cc][Ll][Oo][Bb]|[Ss][Tt][Rr][Ii][Nn][Gg] ]]; then
echo " Processing column: $COLUMN (type: $COLUMN_TYPE)"
# Check if the column contains JSON data
local IS_JSON=$(sqlite3 "$DB_PATH" "
SELECT COUNT(*) FROM $TABLE
WHERE json_valid($COLUMN) = 1 LIMIT 1;
")
if [ "$IS_JSON" -gt 0 ]; then
echo " Detected JSON data in column $COLUMN. Processing with jq..."
# Create a temporary table to store updated JSON
sqlite3 "$DB_PATH" "
CREATE TABLE IF NOT EXISTS temp_$TABLE (rowid INTEGER PRIMARY KEY, new_$COLUMN TEXT);
"
# Export JSON data, modify with jq, and store in temp table
sqlite3 "$DB_PATH" "SELECT rowid, $COLUMN FROM $TABLE WHERE json_valid($COLUMN) = 1;" | while read -r ROWID JSON_DATA; do
# Skip the header line (rowid|column)
if [ "$ROWID" != "rowid" ]; then
# Replace the string in the JSON using jq
local MODIFIED_JSON
MODIFIED_JSON=$(echo "$JSON_DATA" | jq --arg old_str "$OLD_STRING" --arg new_str "$NEW_STRING" '
walk(if type == "string" then gsub($old_str; $new_str) else . end)
')
# Insert the modified JSON into the temp table
sqlite3 "$DB_PATH" "
INSERT INTO temp_$TABLE (rowid, new_$COLUMN) VALUES ($ROWID, '$MODIFIED_JSON');
"
fi
done
# Update the original table with the modified JSON
sqlite3 "$DB_PATH" "
UPDATE $TABLE
SET $COLUMN = (SELECT new_$COLUMN FROM temp_$TABLE WHERE temp_$TABLE.rowid = $TABLE.rowid)
WHERE rowid IN (SELECT rowid FROM temp_$TABLE);
"
# Drop the temporary table
sqlite3 "$DB_PATH" "DROP TABLE temp_$TABLE;"
else
echo " Column $COLUMN is not JSON. Using replace()."
# Use replace() for non-JSON strings
sqlite3 "$DB_PATH" "
UPDATE $TABLE
SET $COLUMN = replace($COLUMN, '$OLD_STRING_ESC', '$NEW_STRING_ESC')
WHERE $COLUMN LIKE '%$OLD_STRING_ESC%';
"
fi
fi
done
done
echo "Replacement completed for all string-based columns (including JSON) in '$DB_PATH'."
}
#set script directory
scriptdir="$(dirname "$(realpath "$0")")"
#example usage: replace_in_sqlite_db "database.sqlite" "Europe/Amsterdam" "UTC"
replace_in_sqlite_db "$scriptdir/stacks/npm/data/database.sqlite" "sdgserver.online" "?domain?"
replace_in_sqlite_db "$scriptdir/stacks/npm/data/database.sqlite" "192.168.2.132" "?localip?"
replace_in_sqlite_db "$scriptdir/stacks/npm/data/database.sqlite" "z5fGWz2i0q" "?adminpass?"
replace_in_sqlite_db "$scriptdir/stacks/npm/data/database.sqlite" "0.0.0.0/0" "?localip?"