#!/bin/bash

SOURCE_DIR="/home/wuyingwen/ai_projects/ComfyUI/output"
DEST_DIR="/home/wuyingwen/ai_projects/ComfyUI/workflows"
mkdir -p "$DEST_DIR"

# 存放已处理文件的哈希
HASH_TRACK_FILE="/tmp/processed_workflow_hashes.txt"
touch "$HASH_TRACK_FILE"

echo "🔁 每 3 秒轮询 $SOURCE_DIR，避免重复保存..."

while true; do
    for file in "$SOURCE_DIR"/*.json; do
        [ -e "$file" ] || continue

        # 计算 MD5 哈希
        hash=$(md5sum "$file" | awk '{print $1}')
        if grep -q "$hash" "$HASH_TRACK_FILE"; then
            continue
        fi

        # 保存副本并记录 hash
        filename=$(basename "$file")
        timestamped="$DEST_DIR/$(date +%Y%m%d_%H%M%S)_$filename"
        cp "$file" "$timestamped"
        echo "$hash" >> "$HASH_TRACK_FILE"
        echo "✅ 新工作流文件已保存：$timestamped"
    done
    sleep 3
done