Upload entrypoint.sh

entrypoint.sh

@@ -0,0 +1,336 @@
+#!/bin/sh
+set -e
+
+CONFIG_FILE="${APP_HOME}/config.yaml"
+
+# Priority 1: Use USERNAME/PASSWORD if both are provided
+if [ -n "${USERNAME}" ] && [ -n "${PASSWORD}" ]; then
+  echo "--- Basic auth enabled: Creating config.yaml with provided credentials. ---"
+  
+  cat <<EOT > ${CONFIG_FILE}
+dataRoot: ./data
+listen: true
+listenAddress:
+  ipv4: 0.0.0.0
+  ipv6: '[::]'
+protocol:
+    ipv4: true
+    ipv6: false
+dnsPreferIPv6: false
+autorunHostname: "auto"
+port: 8000
+autorunPortOverride: -1
+ssl:
+  enabled: false
+  certPath: "./certs/cert.pem"
+  keyPath: "./certs/privkey.pem"
+whitelistMode: false
+enableForwardedWhitelist: false
+whitelist:
+  - ::1
+  - 127.0.0.1
+whitelistDockerHosts: true
+basicAuthMode: true
+basicAuthUser:
+  username: "${USERNAME}"
+  password: "${PASSWORD}"
+enableCorsProxy: false
+requestProxy:
+  enabled: false
+  url: "socks5://username:password@example.com:1080"
+  bypass:
+    - localhost
+    - 127.0.0.1
+enableUserAccounts: false
+enableDiscreetLogin: false
+autheliaAuth: false
+perUserBasicAuth: false
+sessionTimeout: -1
+disableCsrfProtection: false
+securityOverride: false
+logging:
+  enableAccessLog: true
+  minLogLevel: 0
+rateLimiting:
+  preferRealIpHeader: false
+autorun: false
+avoidLocalhost: false
+backups:
+  common:
+    numberOfBackups: 50
+  chat:
+    enabled: true
+    checkIntegrity: true
+    maxTotalBackups: -1
+    throttleInterval: 10000
+thumbnails:
+  enabled: true
+  format: "jpg"
+  quality: 95
+  dimensions: { 'bg': [160, 90], 'avatar': [96, 144] }
+performance:
+  lazyLoadCharacters: false
+  memoryCacheCapacity: '100mb'
+  useDiskCache: true
+allowKeysExposure: true
+skipContentCheck: false
+whitelistImportDomains:
+  - localhost
+  - cdn.discordapp.com
+  - files.catbox.moe
+  - raw.githubusercontent.com
+requestOverrides: []
+extensions:
+  enabled: true
+  autoUpdate: false
+  models:
+    autoDownload: true
+    classification: Cohee/distilbert-base-uncased-go-emotions-onnx
+    captioning: Xenova/vit-gpt2-image-captioning
+    embedding: Cohee/jina-embeddings-v2-base-en
+    speechToText: Xenova/whisper-small
+    textToSpeech: Xenova/speecht5_tts
+enableDownloadableTokenizers: true
+promptPlaceholder: "[Start a new chat]"
+openai:
+  randomizeUserId: false
+  captionSystemPrompt: ""
+deepl:
+  formality: default
+mistral:
+  enablePrefix: false
+ollama:
+  keepAlive: -1
+  batchSize: -1
+claude:
+  enableSystemPromptCache: false
+  cachingAtDepth: -1
+enableServerPlugins: true
+enableServerPluginsAutoUpdate: false
+EOT
+
+# Priority 2: Use CONFIG_YAML if provided (and username/password are not)
+elif [ -n "${CONFIG_YAML}" ]; then
+  echo "--- Found CONFIG_YAML, creating config.yaml from environment variable. ---"
+  printf '%s\n' "${CONFIG_YAML}" > ${CONFIG_FILE}
+
+# Priority 3: No config provided, let the app use its defaults
+else
+    echo "--- No user/pass or CONFIG_YAML provided. App will use its default settings. ---"
+fi
+
+# --- BEGIN: Update SillyTavern Core at Runtime ---
+echo '--- Attempting to update SillyTavern Core from GitHub (staging branch) ---'
+if [ -d ".git" ] && [ "$(git rev-parse --abbrev-ref HEAD)" = "staging" ]; then
+  echo 'Existing staging branch found. Resetting and pulling latest changes...'
+  git reset --hard HEAD && \
+  git pull origin staging || echo 'WARN: git pull failed, continuing with code from build time.'
+  echo '--- SillyTavern Core update check finished. ---'
+else
+  echo 'WARN: .git directory not found or not on staging branch. Skipping runtime update. Code from build time will be used.'
+fi
+# --- END: Update SillyTavern Core at Runtime ---
+
+# --- BEGIN: Configure Git default identity at Runtime ---
+echo '--- Configuring Git default user identity at runtime ---'
+git config --global user.name "SillyTavern Sync" && \
+git config --global user.email "sillytavern-sync@example.com" && \
+git config --global --add safe.directory "${APP_HOME}/data"
+echo '--- Git identity configured for runtime user. ---'
+# --- END: Configure Git default identity at Runtime ---
+
+# --- BEGIN: Dynamically Install Plugins at Runtime ---
+echo '--- Checking for PLUGINS environment variable ---'
+if [ -n "$PLUGINS" ]; then
+  echo "*** Installing Plugins specified in PLUGINS environment variable: $PLUGINS ***"
+  # Ensure plugins directory exists
+  mkdir -p ./plugins && chown node:node ./plugins
+  # Set comma as delimiter
+  IFS=','
+  # Loop through each plugin URL
+  for plugin_url in $PLUGINS; do
+    # Trim leading/trailing whitespace
+    plugin_url=$(echo "$plugin_url" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
+    if [ -z "$plugin_url" ]; then continue; fi
+    # Extract plugin name
+    plugin_name_git=$(basename "$plugin_url")
+    plugin_name=${plugin_name_git%.git}
+    plugin_dir="./plugins/$plugin_name"
+    echo "--- Installing plugin: $plugin_name from $plugin_url into $plugin_dir ---"
+    # Remove existing dir if it exists
+    rm -rf "$plugin_dir"
+    # Clone the plugin (run as root, fix perms later)
+    git clone --depth 1 "$plugin_url" "$plugin_dir"
+    if [ -f "$plugin_dir/package.json" ]; then
+      echo "--- Installing dependencies for $plugin_name ---"
+      (cd "$plugin_dir" && npm install --no-audit --no-fund --loglevel=error --no-progress --omit=dev --force && npm cache clean --force) || echo "WARN: Failed to install dependencies for $plugin_name"
+    else
+       echo "--- No package.json found for $plugin_name, skipping dependency install. ---"
+    fi || echo "WARN: Failed to clone $plugin_name from $plugin_url, skipping..."
+    
+    # Configure cloud-saves plugin if this is the cloud-saves plugin
+    if [ "$plugin_name" = "cloud-saves" ]; then
+      echo "--- Detected cloud-saves plugin, checking for configuration environment variables ---"
+      
+      # Set default values
+      REPO_URL_VALUE=${REPO_URL:-"https://github.com/fuwei99/sillytravern"}
+      GITHUB_TOKEN_VALUE=${GITHUB_TOKEN:-""}
+      AUTOSAVE_INTERVAL_VALUE=${AUTOSAVE_INTERVAL:-30}
+      AUTOSAVE_TARGET_TAG_VALUE=${AUTOSAVE_TARGET_TAG:-""}
+      
+      # Always set autosave to false as required
+      AUTOSAVE_ENABLED="false"
+      
+      echo "--- Creating cloud-saves plugin configuration file ---"
+      CONFIG_JSON_FILE="$plugin_dir/config.json"
+      
+      # Generate config.json file
+      cat <<EOT > ${CONFIG_JSON_FILE}
+{
+  "repo_url": "${REPO_URL_VALUE}",
+  "branch": "main",
+  "username": "cloud-saves",
+  "github_token": "${GITHUB_TOKEN_VALUE}",
+  "display_name": "",
+  "is_authorized": true,
+  "last_save": null,
+  "current_save": null,
+  "has_temp_stash": false,
+  "autoSaveEnabled": ${AUTOSAVE_ENABLED},
+  "autoSaveInterval": ${AUTOSAVE_INTERVAL_VALUE},
+  "autoSaveTargetTag": "${AUTOSAVE_TARGET_TAG_VALUE}"
+}
+EOT
+      
+      # Set correct permissions for config file
+      chown node:node ${CONFIG_JSON_FILE}
+      
+      echo "--- cloud-saves plugin configuration file created at: ${CONFIG_JSON_FILE} ---"
+    fi
+  done
+  # Reset IFS
+  unset IFS
+  # Fix permissions for plugins directory after installation
+  echo "--- Setting permissions for plugins directory ---"
+  chown -R node:node ./plugins
+  echo "*** Plugin installation finished. ***"
+else
+  echo 'PLUGINS environment variable is not set or empty, skipping runtime plugin installation.'
+fi
+# --- END: Dynamically Install Plugins at Runtime ---
+
+echo "*** Starting SillyTavern... ***"
+node ${APP_HOME}/server.js &
+SERVER_PID=$!
+
+echo "SillyTavern server started with PID ${SERVER_PID}. Waiting for it to become responsive..."
+
+# --- Health Check Logic ---
+HEALTH_CHECK_URL="http://localhost:8000/"
+CURL_COMMAND="curl -sf"
+
+# If basic auth is enabled, provide credentials to curl for health checks
+if [ -n "${USERNAME}" ] && [ -n "${PASSWORD}" ]; then
+    echo "--- Health check will use basic auth credentials. ---"
+    # The -u flag provides user:password for basic auth
+    CURL_COMMAND="curl -sf -u \"${USERNAME}:${PASSWORD}\""
+fi
+
+# Health check loop
+RETRY_COUNT=0
+MAX_RETRIES=12 # Wait for 60 seconds max
+# Use eval to correctly execute the command string with quotes
+while ! eval "${CURL_COMMAND} ${HEALTH_CHECK_URL}" > /dev/null; do
+    RETRY_COUNT=$((RETRY_COUNT+1))
+    if [ ${RETRY_COUNT} -ge ${MAX_RETRIES} ]; then
+        echo "SillyTavern failed to start. Exiting."
+        kill ${SERVER_PID}
+        exit 1
+    fi
+    echo "SillyTavern is still starting or not responsive on port 8000, waiting 5 seconds..."
+    sleep 5
+done
+
+echo "SillyTavern started successfully! Beginning periodic keep-alive..."
+
+# --- BEGIN: Install Extensions after SillyTavern startup ---
+install_extensions() {
+    echo "--- Waiting 40 seconds before installing extensions... ---"
+    sleep 40
+    
+    echo "--- Checking for EXTENSIONS environment variable ---"
+    if [ -n "$EXTENSIONS" ]; then
+        echo "*** Installing Extensions specified in EXTENSIONS environment variable: $EXTENSIONS ***"
+        
+        # Determine installation directory based on INSTALL_FOR_ALL_USERS
+        if [ "$INSTALL_FOR_ALL_USERS" = "true" ]; then
+            # System-level installation (for all users)
+            EXTENSIONS_DIR="./public/scripts/extensions/third-party"
+            echo "Installing extensions for all users in: $EXTENSIONS_DIR"
+        else
+            # User-level installation (for default user only)
+            EXTENSIONS_DIR="./data/default-user/extensions"
+            echo "Installing extensions for default user in: $EXTENSIONS_DIR"
+        fi
+        
+        # Ensure extensions directory exists
+        mkdir -p "$EXTENSIONS_DIR" && chown node:node "$EXTENSIONS_DIR"
+        
+        # Set comma as delimiter
+        IFS=','
+        
+        # Loop through each extension URL
+        for extension_url in $EXTENSIONS; do
+            # Trim leading/trailing whitespace
+            extension_url=$(echo "$extension_url" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
+            if [ -z "$extension_url" ]; then continue; fi
+            
+            # Extract extension name
+            extension_name_git=$(basename "$extension_url")
+            extension_name=${extension_name_git%.git}
+            extension_dir="$EXTENSIONS_DIR/$extension_name"
+            
+            echo "--- Installing extension: $extension_name from $extension_url into $extension_dir ---"
+            
+            # Remove existing dir if it exists
+            rm -rf "$extension_dir"
+            
+            # Clone the extension
+            git clone --depth 1 "$extension_url" "$extension_dir"
+            
+            # Check if extension has package.json and install dependencies if needed
+            if [ -f "$extension_dir/package.json" ]; then
+                echo "--- Installing dependencies for $extension_name ---"
+                (cd "$extension_dir" && npm install --no-audit --no-fund --loglevel=error --no-progress --omit=dev --force && npm cache clean --force) || echo "WARN: Failed to install dependencies for $extension_name"
+            else
+                echo "--- No package.json found for $extension_name, skipping dependency install. ---"
+            fi || echo "WARN: Failed to clone $extension_name from $extension_url, skipping..."
+        done
+        
+        # Reset IFS
+        unset IFS
+        
+        # Fix permissions for extensions directory after installation
+        echo "--- Setting permissions for extensions directory ---"
+        chown -R node:node "$EXTENSIONS_DIR"
+        
+        echo "*** Extensions installation finished. ***"
+    else
+        echo 'EXTENSIONS environment variable is not set or empty, skipping extensions installation.'
+    fi
+}
+
+# Run the extension installation in the background
+install_extensions &
+# --- END: Install Extensions after SillyTavern startup ---
+
+# Keep-alive loop
+while kill -0 ${SERVER_PID} 2>/dev/null; do
+    echo "Sending keep-alive request to ${HEALTH_CHECK_URL}"
+    # Use eval here as well for the keep-alive command
+    eval "${CURL_COMMAND} ${HEALTH_CHECK_URL}" > /dev/null || echo "Keep-alive request failed."
+    echo "Keep-alive request sent. Sleeping for 30 minutes."
+    sleep 1800
+done &
+
+wait ${SERVER_PID}