ShadowsocksX-NG/ShadowsocksX-NG2/Domain/RuntimeFileStore.swift

import Foundation

/// 运行时文件存取（spec #21 D5，issue #27）：`v2/sslocal-active.json` 是 GUI
/// 写、wrapper 读的跨进程交接契约。写入复用 `AtomicFileWriter` 的物理安全
/// 基线（目录 0700 含自愈、临时文件创建即 0600、写全校验后原子替换）；显式
/// 停止后的清理只做普通 unlink，不承诺安全擦除（D5）。
struct RuntimeFileStore {
  enum PersistenceError: Error, Equatable {
    /// 写入时的文件系统错误。
    case ioFailure(detail: String)
  }

  let fileURL: URL

  init(fileURL: URL = RuntimePaths.runtimeFileURL()) {
    self.fileURL = fileURL
  }

  /// wrapper pid 文件与契约同目录（v2/）：GUI 判活与 SIGUSR1 投递依据
  /// （SMAppService 不暴露运行中 agent 的 pid）。
  var pidFileURL: URL {
    fileURL.deletingLastPathComponent().appendingPathComponent("agent.pid")
  }

  /// 原子写盘；任一步失败保留原文件。
  func write(_ document: SslocalRuntimeDocument) throws {
    let data: Data
    do {
      data = try document.jsonData()
    } catch {
      throw PersistenceError.ioFailure(detail: String(describing: error))
    }
    do {
      try AtomicFileWriter.write(data, to: fileURL)
    } catch {
      throw PersistenceError.ioFailure(detail: String(describing: error))
    }
  }

  /// 读取侧判定与 wrapper 同源：缺失、损坏或结构性无效一律 `nil`。
  func loadDocument() -> SslocalRuntimeDocument? {
    guard let data = try? Data(contentsOf: fileURL) else { return nil }
    return SslocalRuntimeDocument.decodeValidated(data)
  }

  /// 磁盘原始字节，供与派生文档比较以跳过相同内容的写入（幂等）。
  func readData() -> Data? {
    try? Data(contentsOf: fileURL)
  }

  /// 显式停止清理（D2 停止协议末端）：运行时契约文件、wrapper pid 文件与
  /// 原子写残留的临时文件。异常崩溃路径不调用本方法（文件保留供 KeepAlive
  /// 重放，D5）。文件缺失时调用无害。
  func deleteRuntimeFiles() {
    let fileManager = FileManager.default
    try? fileManager.removeItem(at: fileURL)
    try? fileManager.removeItem(at: pidFileURL)
    let directory = fileURL.deletingLastPathComponent()
    guard let contents = try? fileManager.contentsOfDirectory(atPath: directory.path) else {
      return
    }
    let temporaryPrefix = ".\(fileURL.lastPathComponent).tmp-"
    for name in contents where name.hasPrefix(temporaryPrefix) {
      try? fileManager.removeItem(at: directory.appendingPathComponent(name))
    }
  }
}