mirror of
https://github.com/Tyrrrz/DiscordChatExporter.git
synced 2026-06-10 00:02:37 -06:00
57d472f8e8
Discover Discord tokens from env, token files, GUI Settings.dat, and desktop leveldb; bind-mount the host scrape script so container preflight uses partition/--after cursors; skip inaccessible channels without aborting targets; fix set -e and busybox mktemp for incremental exports under ~/Documents. Co-authored-by: Cursor <cursoragent@cursor.com>
108 lines
2.9 KiB
Bash
Executable file
108 lines
2.9 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
set -Eeuo pipefail
|
|
|
|
SCRIPT_DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd -P)
|
|
REPO_ROOT="${DCE_REPO_ROOT:-$(cd "$SCRIPT_DIR/.." && pwd -P)}"
|
|
CONFIG_PATH="${DCE_PRIMARY_CONFIG:-$REPO_ROOT/config/scrape-targets.json}"
|
|
CONTAINER_CONFIG="${DCE_CONTAINER_CONFIG:-/config/scrape-targets.json}"
|
|
HOST_RUNNER="$REPO_ROOT/scripts/run-discord-scrape-host.sh"
|
|
DISCOVER_TOKEN="$REPO_ROOT/scripts/discover-discord-token.sh"
|
|
VERIFY_SCRIPT="$REPO_ROOT/scripts/verify-documents-archives.sh"
|
|
SETUP_AUTH="$REPO_ROOT/scripts/setup-scrape-auth.sh"
|
|
|
|
usage() {
|
|
cat <<EOF
|
|
Usage:
|
|
$(basename "$0") [--dry-run] [--target NAME] [--config PATH]
|
|
|
|
End-to-end Documents scrape workflow:
|
|
1. Verify enabled targets have seeded archives under ~/Documents/<server>/
|
|
2. Bootstrap scrape.env when DISCORD_TOKEN is exported
|
|
3. Preflight against Discord (skipped with --dry-run)
|
|
4. Incremental scrape (append-only merges into existing JSON files)
|
|
|
|
Options:
|
|
--dry-run Verify archives only; do not call Discord
|
|
--target NAME Limit preflight/scrape to one configured target
|
|
--config PATH Scrape target config (default: config/scrape-targets.json)
|
|
EOF
|
|
}
|
|
|
|
die() {
|
|
printf 'ERROR: %s\n' "$*" >&2
|
|
exit 1
|
|
}
|
|
|
|
main() {
|
|
local dry_run=0
|
|
local target=""
|
|
local -a passthrough=()
|
|
|
|
while (($#)); do
|
|
case "$1" in
|
|
--dry-run)
|
|
dry_run=1
|
|
shift
|
|
;;
|
|
--target)
|
|
[[ $# -ge 2 ]] || die "Missing value for --target."
|
|
target=$2
|
|
passthrough+=(--target "$2")
|
|
shift 2
|
|
;;
|
|
--config)
|
|
[[ $# -ge 2 ]] || die "Missing value for --config."
|
|
CONFIG_PATH=$2
|
|
passthrough+=(--config "$2")
|
|
shift 2
|
|
;;
|
|
--help|-h)
|
|
usage
|
|
exit 0
|
|
;;
|
|
*)
|
|
die "Unknown option: $1"
|
|
;;
|
|
esac
|
|
done
|
|
|
|
"$VERIFY_SCRIPT" --config "$CONFIG_PATH"
|
|
|
|
if (( dry_run == 1 )); then
|
|
printf 'Dry run complete: archive paths verified. Export DISCORD_TOKEN or create a token file, then rerun without --dry-run.\n'
|
|
exit 0
|
|
fi
|
|
|
|
if [[ -n "${DISCORD_TOKEN:-}" || -n "${DISCORD_TOKEN_FILE:-}" ]]; then
|
|
"$SETUP_AUTH" 2>/dev/null || true
|
|
elif [[ -x "$DISCOVER_TOKEN" ]] && "$DISCOVER_TOKEN" >/dev/null 2>&1; then
|
|
"$SETUP_AUTH" 2>/dev/null || true
|
|
fi
|
|
|
|
local -a container_args=("${passthrough[@]}")
|
|
local has_config=0 idx=0
|
|
|
|
while (( idx < ${#container_args[@]} )); do
|
|
if [[ "${container_args[idx]}" == "--config" ]]; then
|
|
has_config=1
|
|
case "${container_args[idx + 1]:-}" in
|
|
"$CONFIG_PATH"|config/scrape-targets.json|./config/scrape-targets.json)
|
|
container_args[idx + 1]="$CONTAINER_CONFIG"
|
|
;;
|
|
esac
|
|
break
|
|
fi
|
|
idx=$((idx + 1))
|
|
done
|
|
|
|
if (( has_config == 0 )); then
|
|
container_args=(--config "$CONTAINER_CONFIG" "${container_args[@]}")
|
|
fi
|
|
|
|
"$HOST_RUNNER" preflight "${container_args[@]}"
|
|
"$HOST_RUNNER" scrape "${container_args[@]}"
|
|
}
|
|
|
|
main "$@"
|