app-check/AppCheckCore/Sources/AppAttestProvider/AppCheckCoreAppAttestProviderState.swift

// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import Foundation

enum AppCheckCoreAppAttestAttestationState: Int {
  case unsupported
  case supportedInitial
  case keyGenerated
  case keyRegistered
}

class AppCheckCoreAppAttestProviderState: NSObject {
  let state: AppCheckCoreAppAttestAttestationState
  let appAttestUnsupportedError: Error?
  let appAttestKeyID: String?
  let attestationArtifact: Data?

  init(unsupportedWithError error: Error) {
    state = .unsupported
    appAttestUnsupportedError = error
    appAttestKeyID = nil
    attestationArtifact = nil
    super.init()
  }

  init(supportedInitialState: Void = ()) {
    state = .supportedInitial
    appAttestUnsupportedError = nil
    appAttestKeyID = nil
    attestationArtifact = nil
    super.init()
  }

  init(generatedKeyID keyID: String) {
    state = .keyGenerated
    appAttestKeyID = keyID
    appAttestUnsupportedError = nil
    attestationArtifact = nil
    super.init()
  }

  init(registeredKeyID keyID: String, artifact: Data) {
    state = .keyRegistered
    appAttestKeyID = keyID
    attestationArtifact = artifact
    appAttestUnsupportedError = nil
    super.init()
  }
}