wehub-resource-sync
135 lines
4.3 KiB
Bash
Executable File
135 lines
4.3 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
#
|
|
# douyin-session adapter wrapper
|
|
#
|
|
# Called by /cheat-retro when state.data_collection=adapter and platform=douyin.
|
|
#
|
|
# Usage:
|
|
# bash run.sh <aweme_id> <video_folder> [<script_path>]
|
|
#
|
|
# Example:
|
|
# bash run.sh 7234567890123456789 ~/my-channel/videos/2026-05-04_abc123_停止期待
|
|
#
|
|
# Output: writes report.md INTO the video_folder.
|
|
# Exit codes:
|
|
# 0 = success (report.md written)
|
|
# 1 = login expired or required
|
|
# 2 = adapter dependency missing (playwright not installed)
|
|
# 3 = other failure (network, parse error, etc.)
|
|
|
|
set -uo pipefail
|
|
|
|
AWEME_ID="${1:-}"
|
|
VIDEO_FOLDER="${2:-}"
|
|
SCRIPT_PATH="${3:-}"
|
|
|
|
if [[ -z "$AWEME_ID" || -z "$VIDEO_FOLDER" ]]; then
|
|
echo "Usage: bash run.sh <aweme_id> <video_folder> [<script_path>]" >&2
|
|
exit 3
|
|
fi
|
|
|
|
# Resolve adapter source dir (where this script lives)
|
|
ADAPTER_DIR="$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
|
|
|
|
# Find Python — prefer venv in user's project root if exists
|
|
PYTHON=""
|
|
# Walk up from VIDEO_FOLDER to find project root (.cheat-state.json)
|
|
PROJECT_ROOT="$( realpath "$VIDEO_FOLDER" )"
|
|
while [[ "$PROJECT_ROOT" != "/" && ! -f "$PROJECT_ROOT/.cheat-state.json" ]]; do
|
|
PROJECT_ROOT="$( dirname "$PROJECT_ROOT" )"
|
|
done
|
|
if [[ ! -f "$PROJECT_ROOT/.cheat-state.json" ]]; then
|
|
echo "❌ Cannot find project root (.cheat-state.json) from $VIDEO_FOLDER" >&2
|
|
exit 3
|
|
fi
|
|
if [[ -x "$PROJECT_ROOT/.venv/bin/python" ]]; then
|
|
PYTHON="$PROJECT_ROOT/.venv/bin/python"
|
|
elif command -v python3 >/dev/null 2>&1; then
|
|
PYTHON="python3"
|
|
else
|
|
echo "❌ python3 not found — install Python 3.10+ first" >&2
|
|
exit 2
|
|
fi
|
|
|
|
# Verify playwright is installed
|
|
if ! "$PYTHON" -c "import playwright" 2>/dev/null; then
|
|
cat >&2 <<EOF
|
|
❌ playwright not installed.
|
|
|
|
Install in your project venv:
|
|
cd "$PROJECT_ROOT"
|
|
python3 -m venv .venv
|
|
source .venv/bin/activate
|
|
pip install -r "$ADAPTER_DIR/requirements.txt"
|
|
playwright install chromium
|
|
|
|
Then re-run /cheat-retro.
|
|
EOF
|
|
exit 2
|
|
fi
|
|
|
|
# Verify auth/ exists in project root (cookie persistence)
|
|
if [[ ! -d "$PROJECT_ROOT/.auth" ]]; then
|
|
cat >&2 <<EOF
|
|
❌ Not logged in to 抖音 创作者中心.
|
|
|
|
First-time login (one-shot):
|
|
cd "$PROJECT_ROOT"
|
|
source .venv/bin/activate
|
|
$PYTHON "$ADAPTER_DIR/crawler.py" login
|
|
|
|
A Chromium window will pop up — scan QR with your phone to log in.
|
|
Cookie will be saved to .auth/ for future runs.
|
|
EOF
|
|
exit 1
|
|
fi
|
|
|
|
# Make sure video_folder exists
|
|
mkdir -p "$VIDEO_FOLDER"
|
|
|
|
# Resolve script path (optional — if given, copy to video_folder/script.txt for renderer)
|
|
SCRIPT_ARG=""
|
|
if [[ -n "$SCRIPT_PATH" && -f "$SCRIPT_PATH" ]]; then
|
|
SCRIPT_ARG="$SCRIPT_PATH"
|
|
fi
|
|
|
|
# Run from PROJECT_ROOT so .auth/ is found and outputs go to expected paths
|
|
cd "$PROJECT_ROOT"
|
|
export CHEAT_PROJECT_ROOT="$PROJECT_ROOT"
|
|
|
|
# Override VIDEOS_DIR via env var so review.py writes to user's videos/ not its own
|
|
# (review.py uses ROOT/videos by default; we override to use user's project)
|
|
export CHEAT_VIDEOS_DIR="$( dirname "$VIDEO_FOLDER" )" # = user's videos/
|
|
|
|
echo "[douyin-session] fetching aweme_id=$AWEME_ID into $VIDEO_FOLDER"
|
|
if [[ -n "$SCRIPT_ARG" ]]; then
|
|
"$PYTHON" "$ADAPTER_DIR/review.py" video "$AWEME_ID" "$SCRIPT_ARG"
|
|
else
|
|
"$PYTHON" "$ADAPTER_DIR/review.py" video "$AWEME_ID"
|
|
fi
|
|
|
|
# review.py writes to ROOT/videos/<auto-named-folder>/report.md.
|
|
# We need to find the just-written report.md and move it to our video_folder if names differ.
|
|
# (review.py uses video.title for folder name; ours uses <date>_<id>_<short>.)
|
|
LATEST_REPORT=$(find "$( dirname "$VIDEO_FOLDER" )" -name "report.md" -newer "$VIDEO_FOLDER" -type f 2>/dev/null | head -1)
|
|
if [[ -n "$LATEST_REPORT" && "$( dirname "$LATEST_REPORT" )" != "$VIDEO_FOLDER" ]]; then
|
|
# Move the autonamed folder's report into our canonical folder
|
|
cp "$LATEST_REPORT" "$VIDEO_FOLDER/report.md"
|
|
AUTO_DIR=$( dirname "$LATEST_REPORT" )
|
|
# Optionally also copy script.txt if review.py wrote it
|
|
if [[ -f "$AUTO_DIR/script.txt" ]]; then
|
|
cp "$AUTO_DIR/script.txt" "$VIDEO_FOLDER/script.txt"
|
|
fi
|
|
# Clean up the auto-generated directory to avoid duplicates
|
|
rm -rf "$AUTO_DIR"
|
|
echo "[douyin-session] moved auto-named output and cleaned up $AUTO_DIR"
|
|
fi
|
|
|
|
if [[ ! -f "$VIDEO_FOLDER/report.md" ]]; then
|
|
echo "❌ report.md not produced — see review.py output above for details" >&2
|
|
exit 3
|
|
fi
|
|
|
|
echo "✅ report.md written to $VIDEO_FOLDER/report.md"
|
|
exit 0
|