// 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/>.

import Common

public struct LocalElements {

}

public struct LocalElement {

}

public struct ParserAssignmentStatement {
  public let lvalue: EvaluatableLValueExpression
  public let value: EvaluatableExpression

  public init(
    withLValue lvalue: EvaluatableLValueExpression, withValue value: EvaluatableExpression
  ) {
    self.lvalue = lvalue
    self.value = value
  }
}

/// A P4 Parser State
///
/// Note: A P4 Parser State is both a type and a value.
/// This "bare" parser state represents the state more as a type than a value.
public class ParserState: P4DataType, P4DataValue, Equatable, CustomStringConvertible {
  public static func == (lhs: ParserState, rhs: ParserState) -> Bool {
    // Two "bare" parser states are always equal.
    return true
  }

  public func eq(rhs: any Common.P4DataType) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public func type() -> any Common.P4DataType {
    return self
  }

  // Any operation between two "bare" parser states is always true.
  public func eq(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public func lt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public func lte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public func gt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public func gte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public var description: String {
    return "Bare Parser State"
  }

  /// Construct a ParserState
  public init() {}

  public func def() -> any P4DataValue {
    return ParserState()
  }
}

/// Instantiated Parser State
///
/// A parser state is both a type and a value. The Instantiated
/// Parser State is the base class for parser states that act more
/// as a value than a type.
public class InstantiatedParserState: ParserState {
  public static func == (lhs: InstantiatedParserState, rhs: InstantiatedParserState) -> Bool {
    return lhs.state == rhs.state
  }

  public override func eq(rhs: any Common.P4DataType) -> Bool {
    return switch rhs {
    case is ParserState: true
    default: false
    }
  }

  public override func type() -> any Common.P4DataType {
    return self
  }

  public override func eq(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let other as InstantiatedParserState: self.state == other.state
    default: false
    }
  }

  public override func lt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let other as InstantiatedParserState: self.state < other.state
    default: false
    }
  }

  public override func lte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let other as InstantiatedParserState: self.state <= other.state
    default: false
    }
  }

  public override func gt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let other as InstantiatedParserState: self.state > other.state
    default: false
    }
  }

  public override func gte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let other as InstantiatedParserState: self.state >= other.state
    default: false
    }
  }

  public private(set) var state: Identifier
  public private(set) var statements: [EvaluatableStatement]

  public override var description: String {
    return "Name: \(state)"
  }

  /// Construct a ParserState
  public init(
    name: Identifier, withStatements stmts: [EvaluatableStatement],
  ) {
    state = name
    statements = stmts
  }

  /// (private) constructor (no transition)
  ///
  /// accept and reject are the only final states and they are constructed internally.
  private init(name: Identifier) {
    state = name
    statements = Array()
  }

  public override func def() -> any P4DataValue {
    return InstantiatedParserState(name: Identifier(name: ""))
  }
}

public class ParserStateDirectTransition: InstantiatedParserState {

  private let next_state: Identifier

  public init(
    name: Identifier, withStatements stmts: [EvaluatableStatement],
    withNextState next_state: Identifier
  ) {
    self.next_state = next_state
    super.init(name: name, withStatements: stmts)
  }

  public override var description: String {
    return "State (Name: \(super.state) (direct transition))"
  }

  public func get_next_state() -> Identifier {
    return self.next_state
  }
}

public class ParserStateNoTransition: InstantiatedParserState {
  public override init(name: Identifier, withStatements stmts: [any EvaluatableStatement]) {
    super.init(name: name, withStatements: stmts)
  }
  public override var description: String {
    return "State (Name: \(super.state) (no transition))"
  }
}

public class ParserStateSelectTransition: InstantiatedParserState {

  public let selectExpression: SelectExpression

  public override var description: String {
    return "State (Name: \(super.state) (select transition))"
  }

  public init(
    name: Identifier, withStatements stmts: [any EvaluatableStatement],
    withTransitioniExpression te: SelectExpression
  ) {
    self.selectExpression = te
    super.init(name: name, withStatements: stmts)
  }
}

nonisolated(unsafe) public let accept = ParserStateNoTransition(
  name: Identifier(name: "accept"), withStatements: [])
nonisolated(unsafe) public let reject = ParserStateNoTransition(
  name: Identifier(name: "reject"), withStatements: [])

public struct ParserStates {
  public var states: [InstantiatedParserState] = Array()

  public func count() -> Int {
    return states.count
  }

  public func find(withIdentifier id: Identifier) -> ParserState? {
    for state in states {
      if state.state == id {
        return .some(state)
      }
    }
    return .none
  }

  public init() {
    self.states = Array()
  }

  private init(withStates states: [InstantiatedParserState]) {
    self.states = states
  }

  public func append(state: InstantiatedParserState) -> ParserStates {
    var new_states = self.states
    new_states.append(state)
    return ParserStates(withStates: new_states)
  }
}

/// A P4 Parser
///
/// Note: A Parser is both a type _and_ a value.
public struct Parser: P4DataType, P4DataValue {
  public func type() -> any Common.P4DataType {
    return self
  }

  public func eq(rhs: any Common.P4DataType) -> Bool {
    return switch rhs {
    case let parser_rhs as Parser: self.name == parser_rhs.name
    default: false
    }
  }

  public func lt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let parser_rhs as Parser: self.name < parser_rhs.name
    default: false
    }
  }

  public func lte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let parser_rhs as Parser: self.name <= parser_rhs.name
    default: false
    }
  }

  public func gt(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let parser_rhs as Parser: self.name > parser_rhs.name
    default: false
    }
  }

  public func gte(rhs: any Common.P4DataValue) -> Bool {
    return switch rhs {
    case let parser_rhs as Parser: self.name >= parser_rhs.name
    default: false
    }
  }

  public var states: ParserStates

  public var name: Identifier
  public var parameters: ParameterList

  public init(withName name: Identifier) {
    self.states = ParserStates()
    self.parameters = ParameterList()
    self.name = name
  }

  public init(withName name: Identifier, withParameters parameters: ParameterList) {
    self.states = ParserStates()
    self.parameters = parameters
    self.name = name
  }

  public func findStartState() -> ParserState? {
    for state in states.states {
      if state.state == Identifier(name: "start") {
        return state
      }
    }
    return .none
  }

  public func eq(rhs: any P4DataValue) -> Bool {
    return switch rhs {
    case let other as Parser: self.name == other.name
    default: false
    }
  }

  public var description: String {
    return "Parser \(self.name) with parameters: \(parameters) and states: \(self.states)"
  }

  public func def() -> any P4DataValue {
    return Parser(withName: Identifier(name: ""))
  }
}

/// Launder a parser state into an instantiated parser state.
public func AsInstantiatedParserState(_ state: ParserState) -> InstantiatedParserState {
  return state as! InstantiatedParserState
}