run_update_and_commit.sh 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279
  1. #!/usr/bin/env bash
  2. set -euo pipefail
  3. timestamp() {
  4. date '+%Y-%m-%d %H:%M:%S'
  5. }
  6. log() {
  7. printf '[%s] %s\n' "$(timestamp)" "$*"
  8. }
  9. log_err() {
  10. printf '[%s] %s\n' "$(timestamp)" "$*" >&2
  11. }
  12. SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
  13. DEFAULT_APP_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
  14. APP_DIR="$(git -C "$DEFAULT_APP_DIR" rev-parse --show-toplevel 2>/dev/null || printf '%s' "$DEFAULT_APP_DIR")"
  15. force_commit="${GIT_FORCE_COMMIT:-0}"
  16. if [[ "${1:-}" == "--force-commit" ]]; then
  17. force_commit="1"
  18. shift
  19. fi
  20. if [[ ! "$force_commit" =~ ^[01]$ ]]; then
  21. log "[vmess-domain-rotator] invalid GIT_FORCE_COMMIT=${force_commit}, expected 0 or 1"
  22. exit 1
  23. fi
  24. CONFIG_PATH="${1:-${APP_DIR}/config.server.json}"
  25. export GIT_TERMINAL_PROMPT=0
  26. commit_name="${GIT_COMMIT_NAME:-vmess-domain-rotator}"
  27. commit_email="${GIT_COMMIT_EMAIL:-vmess-domain-rotator@localhost}"
  28. runtime_branch="${GIT_RUNTIME_BRANCH:-runtime-state}"
  29. push_remote="${GIT_PUSH_REMOTE:-origin}"
  30. push_enabled="${GIT_PUSH_ENABLED:-1}"
  31. push_required="${GIT_PUSH_REQUIRED:-${push_enabled}}"
  32. http_user="${GIT_HTTP_USERNAME:-git}"
  33. http_token="${GIT_HTTP_TOKEN:-}"
  34. http_token_file="${GIT_HTTP_TOKEN_FILE:-}"
  35. credential_helper="${GIT_CREDENTIAL_HELPER:-}"
  36. ts="$(date -u +"%Y-%m-%dT%H:%M:%SZ")"
  37. if [[ -z "$http_token" ]] && [[ -n "$http_token_file" ]] && [[ -r "$http_token_file" ]]; then
  38. http_token="$(tr -d '\r\n' < "$http_token_file")"
  39. fi
  40. git_auth_opts=()
  41. if [[ -n "$http_token" ]]; then
  42. auth_b64="$(printf '%s:%s' "$http_user" "$http_token" | base64 | tr -d '\n')"
  43. git_auth_opts=(-c "http.extraHeader=Authorization: Basic ${auth_b64}")
  44. fi
  45. git_auth() {
  46. local dir="$1"
  47. shift
  48. if [[ -n "$credential_helper" ]]; then
  49. git -C "$dir" -c credential.helper="$credential_helper" "${git_auth_opts[@]}" "$@"
  50. else
  51. git -C "$dir" "${git_auth_opts[@]}" "$@"
  52. fi
  53. }
  54. output_settings_json="$(/usr/bin/python3 "${APP_DIR}/scripts/domain_updater.py" --config "$CONFIG_PATH" --print-output-settings)"
  55. mapfile -t output_settings < <(
  56. printf '%s' "$output_settings_json" | /usr/bin/python3 -c '
  57. import json, sys
  58. settings = json.load(sys.stdin)
  59. for key in ["runtime_dir", "selected_text_path", "selected_json_path", "state_path", "vars_path"]:
  60. print(settings[key])
  61. '
  62. )
  63. RUNTIME_DIR="${output_settings[0]}"
  64. SELECTED_TEXT_FILE="${output_settings[1]}"
  65. SELECTED_JSON_FILE="${output_settings[2]}"
  66. STATE_FILE="${output_settings[3]}"
  67. VARS_FILE="${output_settings[4]}"
  68. repo_relpath() {
  69. /usr/bin/python3 - "$APP_DIR" "$1" <<'PY'
  70. import os
  71. import sys
  72. base = os.path.realpath(sys.argv[1])
  73. path = os.path.realpath(sys.argv[2])
  74. try:
  75. common = os.path.commonpath([base, path])
  76. except ValueError:
  77. common = ""
  78. if common != base:
  79. print("")
  80. else:
  81. print(os.path.relpath(path, base))
  82. PY
  83. }
  84. updater_output="$(/usr/bin/python3 "${APP_DIR}/scripts/domain_updater.py" --config "$CONFIG_PATH")"
  85. printf '%s\n' "$updater_output"
  86. updater_status="$(printf '%s' "$updater_output" | /usr/bin/python3 -c 'import json,sys; print(json.load(sys.stdin).get("status",""))' 2>/dev/null || echo "")"
  87. if [[ "$updater_status" == "error_use_last_good" ]]; then
  88. log "[vmess-domain-rotator] warning: updater using fallback value (source error)"
  89. fi
  90. if [[ ! -f "$SELECTED_TEXT_FILE" ]]; then
  91. log "[vmess-domain-rotator] selected value file missing after updater run (${SELECTED_TEXT_FILE}), skip git commit"
  92. exit 0
  93. fi
  94. after="$(tr -d '\r\n' < "$SELECTED_TEXT_FILE")"
  95. if [[ -z "$after" ]]; then
  96. log "[vmess-domain-rotator] empty selected value, skip git commit"
  97. exit 0
  98. fi
  99. selected_rel="$(repo_relpath "$SELECTED_TEXT_FILE")"
  100. if [[ -z "$selected_rel" ]]; then
  101. log "[vmess-domain-rotator] selected value file is outside repo (${SELECTED_TEXT_FILE}), skip git commit"
  102. exit 0
  103. fi
  104. if ! command -v git >/dev/null 2>&1; then
  105. log "[vmess-domain-rotator] git not found, skip git commit"
  106. exit 0
  107. fi
  108. if ! git -C "$APP_DIR" rev-parse --is-inside-work-tree >/dev/null 2>&1; then
  109. log "[vmess-domain-rotator] not a git repo at ${APP_DIR}, skip git commit"
  110. log "[vmess-domain-rotator] hint: reinstall service from a git clone path"
  111. exit 0
  112. fi
  113. if ! git -C "$APP_DIR" rev-parse --verify HEAD >/dev/null 2>&1; then
  114. log "[vmess-domain-rotator] repo has no commits yet, skip git commit"
  115. exit 0
  116. fi
  117. if ! git -C "$APP_DIR" remote get-url "$push_remote" >/dev/null 2>&1; then
  118. push_remote=""
  119. while IFS= read -r r; do
  120. push_remote="$r"
  121. break
  122. done < <(git -C "$APP_DIR" remote)
  123. fi
  124. work_dir=""
  125. cleanup_worktree="0"
  126. current_branch="$(git -C "$APP_DIR" symbolic-ref --quiet --short HEAD 2>/dev/null || true)"
  127. if [[ "$current_branch" == "$runtime_branch" ]]; then
  128. work_dir="$APP_DIR"
  129. else
  130. work_dir="$(mktemp -d "${TMPDIR:-/tmp}/vmess-runtime-state.XXXXXX")"
  131. cleanup_worktree="1"
  132. if git -C "$APP_DIR" show-ref --verify --quiet "refs/heads/${runtime_branch}"; then
  133. git -C "$APP_DIR" worktree add --force "$work_dir" "$runtime_branch"
  134. else
  135. if [[ -n "$push_remote" ]] && git_auth "$APP_DIR" ls-remote --exit-code --heads "$push_remote" "$runtime_branch" >/dev/null 2>&1; then
  136. git_auth "$APP_DIR" fetch "$push_remote" "$runtime_branch:$runtime_branch"
  137. git -C "$APP_DIR" worktree add --force "$work_dir" "$runtime_branch"
  138. else
  139. git -C "$APP_DIR" worktree add --force --detach "$work_dir"
  140. git -C "$work_dir" checkout --orphan "$runtime_branch"
  141. git -C "$work_dir" rm -rf . >/dev/null 2>&1 || true
  142. log "[vmess-domain-rotator] initialized branch ${runtime_branch}"
  143. fi
  144. fi
  145. fi
  146. if [[ "$cleanup_worktree" == "1" ]]; then
  147. cleanup() {
  148. git -C "$APP_DIR" worktree remove --force "$work_dir" >/dev/null 2>&1 || rm -rf "$work_dir"
  149. }
  150. trap cleanup EXIT
  151. fi
  152. work_branch="$(git -C "$work_dir" symbolic-ref --quiet --short HEAD 2>/dev/null || true)"
  153. if [[ "$work_branch" != "$runtime_branch" ]]; then
  154. log "[vmess-domain-rotator] safety check failed: worktree branch is '${work_branch:-detached}', expected '${runtime_branch}'"
  155. exit 1
  156. fi
  157. before=""
  158. if before_raw="$(git -C "$work_dir" show "HEAD:${selected_rel}" 2>/dev/null)"; then
  159. before="$(printf '%s' "$before_raw" | tr -d '\r\n')"
  160. fi
  161. if [[ "$force_commit" != "1" ]] && [[ -n "$before" ]] && [[ "$after" == "$before" ]]; then
  162. log "[vmess-domain-rotator] selected value unchanged (${after}), skip git commit and push"
  163. exit 0
  164. fi
  165. tracked_src_files=("$SELECTED_TEXT_FILE" "$SELECTED_JSON_FILE" "$STATE_FILE" "$VARS_FILE")
  166. tracked_rel_files=()
  167. for src in "${tracked_src_files[@]}"; do
  168. rel="$(repo_relpath "$src")"
  169. if [[ -z "$rel" ]]; then
  170. log "[vmess-domain-rotator] skip non-repo output file: ${src}"
  171. continue
  172. fi
  173. tracked_rel_files+=("$rel")
  174. dst="$work_dir/$rel"
  175. mkdir -p "$(dirname "$dst")"
  176. if [[ -f "$src" ]]; then
  177. cp "$src" "$dst"
  178. else
  179. rm -f "$dst"
  180. fi
  181. done
  182. if [[ "${#tracked_rel_files[@]}" -eq 0 ]]; then
  183. log "[vmess-domain-rotator] no repo-local output files from config (${CONFIG_PATH}), skip git commit"
  184. exit 0
  185. fi
  186. git -C "$work_dir" add -A -- "${tracked_rel_files[@]}" || true
  187. staged_changed="1"
  188. if git -C "$work_dir" diff --cached --quiet; then
  189. staged_changed="0"
  190. fi
  191. if [[ "$staged_changed" == "0" ]] && [[ "$force_commit" != "1" ]]; then
  192. log "[vmess-domain-rotator] no staged changes for ${runtime_branch}, skip git commit"
  193. exit 0
  194. fi
  195. commit_extra_args=()
  196. if [[ "$staged_changed" == "0" ]] && [[ "$force_commit" == "1" ]]; then
  197. commit_extra_args+=(--allow-empty)
  198. log "[vmess-domain-rotator] force commit enabled with unchanged content, creating empty commit"
  199. fi
  200. commit_message="chore: rotate preferred value to ${after} (${ts})"
  201. if [[ "$updater_status" == "error_use_last_good" ]]; then
  202. commit_message="chore(fallback): rotate preferred value to ${after} (${ts})"
  203. fi
  204. if [[ "$force_commit" == "1" ]]; then
  205. commit_message="manual: value ${after}, updated at ${ts}"
  206. fi
  207. git -C "$work_dir" \
  208. -c user.name="$commit_name" \
  209. -c user.email="$commit_email" \
  210. commit "${commit_extra_args[@]}" -m "$commit_message"
  211. if [[ "$push_enabled" != "1" ]]; then
  212. log "[vmess-domain-rotator] git push disabled by GIT_PUSH_ENABLED=${push_enabled}"
  213. elif [[ -z "$push_remote" ]]; then
  214. if [[ "$push_required" == "1" ]]; then
  215. log "[vmess-domain-rotator] no remote found but push is required"
  216. exit 1
  217. fi
  218. log "[vmess-domain-rotator] no remote found, skip git push"
  219. else
  220. push_ok="0"
  221. if git -C "$work_dir" rev-parse --abbrev-ref --symbolic-full-name "@{u}" >/dev/null 2>&1; then
  222. if git_auth "$work_dir" push "$push_remote" "$runtime_branch:$runtime_branch"; then
  223. push_ok="1"
  224. log "[vmess-domain-rotator] pushed to ${push_remote}/${runtime_branch}"
  225. fi
  226. else
  227. if git_auth "$work_dir" push -u "$push_remote" "$runtime_branch:$runtime_branch"; then
  228. push_ok="1"
  229. log "[vmess-domain-rotator] pushed to ${push_remote}/${runtime_branch} (set upstream)"
  230. fi
  231. fi
  232. if [[ "$push_ok" != "1" ]]; then
  233. log "[vmess-domain-rotator] git push failed"
  234. log "[vmess-domain-rotator] hint: configure non-interactive auth (credential.helper store, SSH deploy key, or GIT_HTTP_USERNAME/GIT_HTTP_TOKEN)"
  235. if [[ "$push_required" == "1" ]]; then
  236. exit 1
  237. fi
  238. fi
  239. fi
  240. log "[vmess-domain-rotator] committed output changes on ${runtime_branch}: selected value ${after} from ${RUNTIME_DIR}"