divkit/client/ios/DivKit/StoredValues/DivPersistentValuesStorage.swift

import Foundation
import VGSL

public final class DivPersistentValuesStorage {
  static let storageFileName = "divkit.values_storage"
  private let timestampProvider: Variable<Milliseconds>

  public init(
    timestampProvider: Variable<Milliseconds> = Variable {
      Date().timeIntervalSince1970.milliseconds
    }
  ) {
    self.timestampProvider = timestampProvider
    removeOutdatedStoredValues()
  }

  private let storage = Property<StoredValues>(
    fileName: storageFileName,
    initialValue: StoredValues(items: [:]),
    onError: { error in
      DivKitLogger.error("Failed to create storage: \(error)")
    }
  )

  func set(value: DivStoredValue) {
    var items = storage.value.items
    items[value.name] = StoredValue(
      timestamp: timestampProvider.value,
      value: value.value,
      type: value.type,
      lifetimeInSec: value.lifetimeInSec
    )
    storage.value = StoredValues(items: items)
  }

  func get<T>(name: String) -> T? {
    let items = storage.value.items
    let currentTimestamp = timestampProvider.value
    guard let storedValue = items[name],
          currentTimestamp - storedValue.timestamp < storedValue.lifetimeInSec * 1000 else {
      return nil
    }
    let divStoredValue = DivStoredValue(
      name: name,
      value: storedValue.value,
      type: storedValue.type,
      lifetimeInSec: storedValue.lifetimeInSec
    )
    guard let value: T = divStoredValue.variable?.typedValue() else {
      DivKitLogger.error("The type of the stored value \(name) is not \(T.self)")
      return nil
    }
    return value
  }

  func reset() {
    storage.value = StoredValues(items: [:])
  }

  private func removeOutdatedStoredValues() {
    let currentTimestamp = timestampProvider.value
    let items = storage.value.items
    let newItems = items.filter { _, value in
      currentTimestamp - value.timestamp < value.lifetimeInSec * 1000
    }

    if newItems.count != items.count {
      storage.value = StoredValues(items: newItems)
    }
  }
}

private struct StoredValues: Equatable, Codable {
  let items: [String: StoredValue]
}

private struct StoredValue: Equatable, Codable {
  let timestamp: Milliseconds
  let value: String
  let type: DivStoredValue.ValueType
  let lifetimeInSec: Int
}

extension DivStoredValue {
  var variable: DivVariableValue? {
    switch type {
    case .string:
      .string(value)
    case .number:
      if let value = Double(value) {
        .number(value)
      } else {
        nil
      }
    case .integer:
      if let value = Int(value) {
        .integer(value)
      } else {
        nil
      }
    case .bool, .boolean:
      switch value.lowercased() {
      case "0", "false":
        .bool(false)
      case "1", "true":
        .bool(true)
      default:
        nil
      }
    case .color:
      if let value = Color.color(withHexString: value) {
        .color(value)
      } else {
        nil
      }
    case .url:
      if let value = URL(string: value) {
        .url(value)
      } else {
        nil
      }
    case .array:
      if let rawValue = try? JSONSerialization.jsonObject(jsonString: value) as? [Any],
         let array = DivArray.fromAny(rawValue) {
        .array(array)
      } else {
        nil
      }
    case .dict:
      if let rawValue = try? JSONSerialization.jsonObject(jsonString: value) as? [String: Any],
         let dict = DivDictionary.fromAny(rawValue) {
        .dict(dict)
      } else {
        nil
      }
    }
  }

  var isValueValid: Bool {
    variable != nil
  }
}