// p4rse, Copyright 2026, Will Hawkins
//
// This file is part of p4rse.
//
// This file is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

public typealias StatementInterloper = (EvaluatableStatement, ControlFlow, ProgramExecution) -> Void
public typealias ExpressionInterloper = (EvaluatableExpression, Result<P4Value>, ProgramExecution) -> Void

open class ProgramExecution: CustomStringConvertible {
  public var scopes: VarValueScopes = VarValueScopes()
  var globalValues: VarValueScopes?
  var error: Error?
  var debug: DebugLevel = DebugLevel.Error
  var statement_interloper: StatementInterloper?
  var expression_interloper: ExpressionInterloper?

  init(copy: ProgramExecution) {
    self.scopes = copy.scopes
    self.globalValues = copy.globalValues
    self.error = copy.error
    self.debug = copy.debug
    self.statement_interloper = copy.statement_interloper
    self.expression_interloper = copy.expression_interloper
  }

  public init() {
    globalValues = .none
  }

  public init(withGlobalValues values: VarValueScopes) {
    globalValues = values
  }

  open var description: String {
    return "Runtime:\nScopes: \(scopes)"
  }

  public func hasError() -> Bool {
    return self.error != nil
  }

  public func getError() -> Error? {
    return self.error
  }

  public func setError(error: Error) -> ProgramExecution {
    let npe = ProgramExecution(copy: self)
    npe.error = error
    return npe
  }

  public func getDebugLevel() -> DebugLevel {
    return self.debug
  }

  public func setDebugLevel(_ dl: DebugLevel) -> ProgramExecution {
    let pe = ProgramExecution(copy: self)
    pe.debug = dl
    return pe
  }

  public func getStatementInterloper() -> StatementInterloper? {
    return self.statement_interloper
  }

  public func setStatementInterloper(_ interloper: @escaping StatementInterloper) -> ProgramExecution {
    let pe = ProgramExecution(copy: self)
    pe.statement_interloper = interloper
    return pe
  }

  public func getExpressionInterloper() -> ExpressionInterloper? {
    return self.expression_interloper
  }

  public func setExpressionInterloper(_ interloper: @escaping ExpressionInterloper) -> ProgramExecution {
    let pe = ProgramExecution(copy: self)
    pe.expression_interloper = interloper
    return pe
  }

  open func isDone() -> Bool {
    return false
  }

  open func setDone() -> ProgramExecution {
    // For a bare ProgramExecution, setDone is a noop.
    return self
  }

  public func enter_scope() -> ProgramExecution {
    let new_pe = ProgramExecution(copy: self)
    new_pe.scopes = new_pe.scopes.enter()

    return new_pe
  }

  public func exit_scope() -> ProgramExecution {
    let new_pe = ProgramExecution(copy: self)
    new_pe.scopes = new_pe.scopes.exit()

    return new_pe
  }

  public func replaceScopes(_ new_scopes: VarValueScopes) -> ProgramExecution {
    let new_pe = ProgramExecution(copy: self)
    new_pe.scopes = new_scopes
    return new_pe
  }

  public func declare(identifier: Identifier, withValue value: P4Value) -> ProgramExecution {
    let new_pe = ProgramExecution(copy: self)
    let new_scopes = new_pe.scopes.declare(identifier: identifier, withValue: value)

    new_pe.scopes = new_scopes
    return new_pe
  }

  public func getGlobalValues() -> VarValueScopes {
    return self.globalValues ?? VarValueScopes()
  }

  public func setGlobalValues(_ global_values: VarValueScopes) -> ProgramExecution {
    let new_pe = ProgramExecution(copy: self)
    new_pe.globalValues = global_values
    return new_pe
  }


}

/// A scope that resolves variable identifiers to their values.
public typealias VarValueScope = Scope<P4Value>

/// Scopes that resolves variable identifiers to their values.
public typealias VarValueScopes = Scopes<P4Value>

/// Indicate the control flow result of a particular statement.
public enum ControlFlow {
  case Next
  case Continue
  case Break
  case Return(P4Value?)
  case Error
}