101 lines
2.4 KiB
Bash
Executable File
101 lines
2.4 KiB
Bash
Executable File
#!/bin/bash
|
||
# llm-wiki source 页面写入脚本
|
||
# 原子写入 source 页面 + 自动更新缓存，绑定为一项操作
|
||
|
||
set -euo pipefail
|
||
|
||
usage() {
|
||
cat <<'EOF'
|
||
用法：
|
||
bash scripts/create-source-page.sh <raw_file> <output_path> <content_file>
|
||
|
||
参数：
|
||
raw_file : 原始素材文件路径（绝对或相对路径）
|
||
output_path : 目标页面路径（相对于知识库根目录，如 wiki/sources/2026-04-16-rlhf.md）
|
||
content_file : 包含待写入内容的临时文件路径
|
||
EOF
|
||
}
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
|
||
# 参数校验
|
||
if [ "$#" -ne 3 ]; then
|
||
usage
|
||
exit 1
|
||
fi
|
||
|
||
raw_file="$1"
|
||
output_path="$2"
|
||
content_file="$3"
|
||
|
||
# raw_file 和 content_file 必须存在
|
||
if [ ! -f "$raw_file" ]; then
|
||
echo "ERROR: 原始素材文件不存在：$raw_file" >&2
|
||
exit 1
|
||
fi
|
||
|
||
if [ ! -f "$content_file" ]; then
|
||
echo "ERROR: 内容文件不存在：$content_file" >&2
|
||
exit 1
|
||
fi
|
||
|
||
# 通过 cache.sh 的 find_wiki_root 逻辑找到知识库根目录
|
||
# 复用 cache.sh 里的函数
|
||
source_cache_helpers() {
|
||
# 内联 find_wiki_root（与 cache.sh 保持一致）
|
||
find_wiki_root() {
|
||
local file_path="$1"
|
||
local dir parent
|
||
|
||
dir="$(cd "$(dirname "$file_path")" && pwd)"
|
||
|
||
while true; do
|
||
if [ -f "$dir/.wiki-cache.json" ] || [ -f "$dir/.wiki-schema.md" ]; then
|
||
printf '%s\n' "$dir"
|
||
return 0
|
||
fi
|
||
|
||
parent="$(dirname "$dir")"
|
||
[ "$parent" = "$dir" ] && return 1
|
||
dir="$parent"
|
||
done
|
||
}
|
||
}
|
||
|
||
source_cache_helpers
|
||
|
||
wiki_root="$(find_wiki_root "$raw_file")" || {
|
||
echo "ERROR: 未找到知识库根目录：$raw_file" >&2
|
||
exit 1
|
||
}
|
||
|
||
# 拼接完整目标路径
|
||
full_output="$wiki_root/$output_path"
|
||
|
||
# 确保目标目录存在
|
||
mkdir -p "$(dirname "$full_output")"
|
||
|
||
# 第一步：原子写入（临时文件 + rename，防止写一半崩溃）
|
||
tmp_output="${full_output}.tmp.$$"
|
||
if ! cp "$content_file" "$tmp_output"; then
|
||
rm -f "$tmp_output" 2>/dev/null || true
|
||
echo "ERROR: 写入临时文件失败" >&2
|
||
exit 1
|
||
fi
|
||
|
||
if ! mv "$tmp_output" "$full_output"; then
|
||
rm -f "$tmp_output" 2>/dev/null || true
|
||
echo "ERROR: 原子重命名失败" >&2
|
||
exit 1
|
||
fi
|
||
|
||
# 第二步：更新缓存
|
||
if ! bash "$SCRIPT_DIR/cache.sh" update "$raw_file" "$output_path"; then
|
||
# 缓存更新失败 → 回滚：删除已写入的文件
|
||
rm -f "$full_output"
|
||
echo "ERROR: 缓存更新失败，已回滚写入" >&2
|
||
exit 1
|
||
fi
|
||
|
||
echo "SUCCESS"
|