Will Hawkins
796 lines
27 KiB
Swift
796 lines
27 KiB
Swift
// 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
|
|
import P4Lang
|
|
import P4Runtime
|
|
import SwiftTreeSitter
|
|
import TreeSitterExtensions
|
|
import TreeSitterP4
|
|
|
|
extension Declaration: CompilableDeclaration {
|
|
public static func Compile(
|
|
node: Node, withContext context: CompilerContext
|
|
) -> Result<(P4Type, CompilerContext)?> {
|
|
|
|
let declaration_compilers: [String: CompilableDeclaration.Type] = [
|
|
"function_declaration": FunctionDeclaration.self,
|
|
"control_declaration": Control.self,
|
|
"type_declaration": StructDeclaration.self, // Assume that type declarations are struct declarations.
|
|
]
|
|
|
|
guard let declaration_compiler = declaration_compilers[node.nodeType!] else {
|
|
return .Ok(.none)
|
|
}
|
|
|
|
return declaration_compiler.Compile(node: node, withContext: context)
|
|
}
|
|
}
|
|
|
|
extension FunctionDeclaration: CompilableDeclaration {
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(any Common.P4Type, CompilerContext)?> {
|
|
let function_declaration_node = node
|
|
#RequireNodeType<Node, (ParameterList, CompilerContext)>(
|
|
node: function_declaration_node, type: "function_declaration",
|
|
nice_type_name: "Function Declaration")
|
|
|
|
var context = context
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
var currentChild: Node? = .none
|
|
if function_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: function_declaration_node, withError: "Missing function declaration component"))
|
|
}
|
|
currentChild = function_declaration_node.child(at: currentChildIdx)
|
|
|
|
let maybe_function_type = Types.CompileType(type: currentChild!, withContext: context)
|
|
guard case .Ok(let function_type) = maybe_function_type else {
|
|
return .Error(maybe_function_type.error()!)
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if function_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: function_declaration_node, withError: "Missing function declaration component"))
|
|
}
|
|
currentChild = function_declaration_node.child(at: currentChildIdx)
|
|
|
|
let maybe_function_name = Identifier.Compile(node: currentChild!, withContext: context)
|
|
guard case .Ok(let function_name) = maybe_function_name else {
|
|
return .Error(maybe_function_name.error()!)
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if function_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: function_declaration_node, withError: "Missing function declaration component"))
|
|
}
|
|
currentChild = function_declaration_node.child(at: currentChildIdx)
|
|
|
|
let maybe_function_parameters = ParameterList.Compile(node: currentChild!, withContext: context)
|
|
guard case .Ok((let function_parameters, let updated_context)) = maybe_function_parameters
|
|
else {
|
|
return .Error(maybe_function_parameters.error()!)
|
|
}
|
|
context = updated_context
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if function_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: function_declaration_node, withError: "Missing function declaration component"))
|
|
}
|
|
currentChild = function_declaration_node.child(at: currentChildIdx)
|
|
|
|
// Add the parameters into scope.
|
|
var function_scope = context.instances.enter()
|
|
for parameter in function_parameters.parameters {
|
|
function_scope = function_scope.declare(identifier: parameter.name, withValue: parameter.type)
|
|
}
|
|
|
|
let maybe_function_body = Parser.Statement.Compile(
|
|
node: currentChild!, withContext: context.update(newInstances: function_scope))
|
|
guard case .Ok((let function_body, _)) = maybe_function_body else {
|
|
return .Error(maybe_function_body.error()!)
|
|
}
|
|
|
|
let function_declaration = FunctionDeclaration(
|
|
named: function_name, ofType: function_type, withParameters: function_parameters,
|
|
withBody: function_body)
|
|
|
|
// Do not use the updated context returned by parsing the body
|
|
// and do not use the function_scope, either.
|
|
return .Ok(
|
|
(
|
|
function_declaration,
|
|
context.update(
|
|
newTypes: context.types.declare(
|
|
identifier: function_name, withValue: function_declaration))
|
|
))
|
|
}
|
|
}
|
|
|
|
struct StructDeclaration {}
|
|
|
|
extension StructDeclaration: CompilableDeclaration {
|
|
static func Compile(
|
|
node: Node, withContext context: CompilerContext
|
|
) -> Result<(P4Type, CompilerContext)?> {
|
|
|
|
let struct_declaration_node = node.child(at: 0)!
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
|
|
var currentChild: Node? = .none
|
|
|
|
guard let node_type = struct_declaration_node.nodeType,
|
|
node_type == "struct_declaration"
|
|
else {
|
|
return Result.Error(
|
|
ErrorOnNode(node: struct_declaration_node, withError: "Did not find a struct declaration"))
|
|
}
|
|
|
|
if struct_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: struct_declaration_node, withError: "Missing elements in struct declaration"))
|
|
}
|
|
|
|
// Skip the keyword struct
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if struct_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: struct_declaration_node, withError: "Missing elements in struct declaration"))
|
|
}
|
|
|
|
// The name of the struct type.
|
|
currentChild = struct_declaration_node.child(at: currentChildIdx)
|
|
let maybe_struct_identifier = Identifier.Compile(
|
|
node: currentChild!, withContext: context)
|
|
guard case Result.Ok(let struct_identifier) = maybe_struct_identifier else {
|
|
return Result.Error(maybe_struct_identifier.error()!)
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
|
|
// Skip the '{'
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if struct_declaration_node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: struct_declaration_node, withError: "Missing element of struct declaration"))
|
|
}
|
|
|
|
currentChild = struct_declaration_node.child(at: currentChildIdx)
|
|
|
|
// If there are no fields, it will be a "}"
|
|
if currentChild!.nodeType == "}" {
|
|
let struc = P4Struct(withName: struct_identifier, andFields: P4StructFields([]))
|
|
return Result.Ok(
|
|
(
|
|
struc,
|
|
context.update(
|
|
newTypes: context.types.declare(identifier: struct_identifier, withValue: struc))
|
|
))
|
|
}
|
|
|
|
var parse_errs: [Error] = Array()
|
|
var current_context = context
|
|
var parsed_fields: [P4StructFieldIdentifier] = Array()
|
|
|
|
if currentChild!.nodeType == "struct_declaration_fields" {
|
|
currentChild!.enumerateNamedChildren { declaration_field in
|
|
switch VariableDeclarationStatement.Compile(
|
|
node: declaration_field, withContext: current_context)
|
|
{
|
|
case .Ok((let declaration, let updated_context)):
|
|
let variable_declaration = declaration as! VariableDeclarationStatement
|
|
parsed_fields.append(
|
|
P4StructFieldIdentifier(
|
|
id: variable_declaration.identifier, withType: variable_declaration.initializer.type()
|
|
))
|
|
current_context = updated_context
|
|
case .Error(let e): parse_errs.append(e)
|
|
}
|
|
}
|
|
}
|
|
|
|
if !parse_errs.isEmpty {
|
|
return .Error(
|
|
Error(
|
|
withMessage: "Error(s) parsing select cases: "
|
|
+ (parse_errs.map { error in
|
|
return "\(error.msg)"
|
|
}.joined(separator: ";"))))
|
|
}
|
|
|
|
let declared_struct = P4Struct(
|
|
withName: struct_identifier, andFields: P4StructFields(parsed_fields))
|
|
return .Ok(
|
|
(
|
|
declared_struct,
|
|
current_context.update(
|
|
newTypes: current_context.types.declare(
|
|
identifier: struct_identifier, withValue: declared_struct))
|
|
))
|
|
}
|
|
}
|
|
|
|
extension P4Lang.Parser: CompilableDeclaration {
|
|
public static func Compile(
|
|
node: Node, withContext context: CompilerContext
|
|
) -> Result<(P4Type, CompilerContext)?> {
|
|
let parser_node = node
|
|
#SkipUnlessNodeType<Node, (P4Type, CompilerContext)?>(
|
|
node: parser_node, type: "parserDeclaration")
|
|
|
|
var current_context = context
|
|
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
var currentChild: Node? = .none
|
|
|
|
if parser_node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: parser_node, withError: "Missing elements of parser declaration"))
|
|
}
|
|
|
|
currentChild = parser_node.child(at: currentChildIdx)
|
|
if currentChild!.nodeType != "parserType" {
|
|
return .Error(
|
|
ErrorOnNode(node: currentChild!, withError: "Missing type for parser declaration"))
|
|
}
|
|
|
|
let type_node = currentChild
|
|
var parser_name: Common.Identifier? = .none
|
|
|
|
// Assume that the parameter list is empty!
|
|
var parameter_list = ParameterList()
|
|
|
|
do {
|
|
// Parse the parser type (type_node)
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
|
|
if type_node!.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(
|
|
node: parser_node, withError: "Missing elements of parser type in parser declaration"))
|
|
}
|
|
|
|
var currentChild = type_node!.child(at: currentChildIdx)
|
|
if currentChild!.nodeType == "annotations" {
|
|
return .Error(
|
|
ErrorOnNode(
|
|
node: currentChild!, withError: "Annotations in parser type are not yet handled."))
|
|
// Will increment indexes here.
|
|
}
|
|
|
|
// Skip the parser keyword
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if type_node!.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: type_node!, withError: "Missing name in parser type declaration"))
|
|
}
|
|
currentChild = type_node?.child(at: currentChildIdx)
|
|
|
|
switch Identifier.Compile(node: currentChild!, withContext: current_context) {
|
|
case .Ok(let id): parser_name = id
|
|
case .Error(let e):
|
|
return .Error(e)
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if type_node!.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: type_node!, withError: "Missing parser parameters"))
|
|
}
|
|
|
|
currentChild = type_node?.child(at: currentChildIdx)
|
|
switch ParameterList.Compile(node: currentChild!, withContext: current_context) {
|
|
case .Ok(let (parsed_parameter_list, updated_context)):
|
|
parameter_list = parsed_parameter_list
|
|
current_context = updated_context
|
|
case .Error(let e):
|
|
return .Error(e)
|
|
}
|
|
}
|
|
|
|
// Now, let's put the parameters into scope.
|
|
for parameter in parameter_list.parameters {
|
|
current_context = current_context.update(
|
|
newInstances: current_context.instances.declare(
|
|
identifier: parameter.name, withValue: parameter.type))
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if parser_node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: parser_node, withError: "Missing parser declaration component"))
|
|
}
|
|
|
|
// Skip the '{'
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if parser_node.childCount < currentChildIdxSafe {
|
|
return .Error((Error(withMessage: "Missing body of parser declaration")))
|
|
}
|
|
currentChild = parser_node.child(at: currentChildIdx)
|
|
|
|
if currentChild!.nodeType == "parserLocalElements" {
|
|
return .Error(
|
|
ErrorOnNode(node: currentChild!, withError: "Parser Local Elements are not yet handled."))
|
|
// Will increment indexes here.
|
|
}
|
|
|
|
if parser_node.childCount < currentChildIdxSafe {
|
|
return .Error((Error(withMessage: "Missing body of parser declaration")))
|
|
}
|
|
|
|
if currentChild!.nodeType != "parserStates" {
|
|
return .Error(Error(withMessage: "Missing parser states in parser declaration"))
|
|
}
|
|
|
|
switch Parser.Compile(
|
|
withName: parser_name!, withParameters: parameter_list, node: currentChild!,
|
|
withContext: current_context)
|
|
{
|
|
case Result.Ok((let parser, let updated_context)):
|
|
// Create a new context with the name of the parser that was just compiled in scope.
|
|
return .Ok(
|
|
(
|
|
parser,
|
|
context.update(
|
|
newInstances: updated_context.instances.declare(
|
|
identifier: parser.name, withValue: parser))
|
|
))
|
|
case Result.Error(let error): return .Error(error)
|
|
}
|
|
}
|
|
}
|
|
|
|
extension Control: CompilableDeclaration {
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(any Common.P4Type, CompilerContext)?> {
|
|
|
|
#SkipUnlessNodeType<Node, (P4Type, CompilerContext)?>(
|
|
node: node, type: "control_declaration")
|
|
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
var currentChild: Node? = .none
|
|
|
|
var local_context = context
|
|
|
|
// Skip control keyword
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing control declaration component"))
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing control declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
guard
|
|
case .Ok(let control_name) = Identifier.Compile(
|
|
node: currentChild!, withContext: local_context)
|
|
else {
|
|
return Result.Error(
|
|
Error(withMessage: "Could not parse a parameter name from \(currentChild!.text!)"))
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing control declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
let maybe_control_parameters = ParameterList.Compile(
|
|
node: currentChild!, withContext: local_context)
|
|
guard case .Ok((let control_parameters, let updated_context)) = maybe_control_parameters
|
|
else {
|
|
return .Error(maybe_control_parameters.error()!)
|
|
}
|
|
local_context = updated_context
|
|
|
|
// Before continuing, make sure to put the parameters into context.
|
|
var control_scope = local_context.instances.enter()
|
|
for parameter in control_parameters.parameters {
|
|
control_scope = control_scope.declare(identifier: parameter.name, withValue: parameter.type)
|
|
}
|
|
local_context = local_context.update(newInstances: control_scope)
|
|
|
|
// Skip the '{'
|
|
currentChildIdx += 2
|
|
currentChildIdxSafe += 2
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing control declaration component"))
|
|
}
|
|
|
|
var actions: [Action] = Array()
|
|
var tables: [Table] = Array()
|
|
|
|
// Because the final child
|
|
// is the '}'.
|
|
// \/\/
|
|
for currentChildIdx in currentChildIdx..<(node.childCount - 1) {
|
|
let currentChild = node.child(at: currentChildIdx)!
|
|
if currentChild.nodeType == "action_declaration" {
|
|
let maybe_action_declaration = Action.Compile(
|
|
node: currentChild, withContext: local_context)
|
|
guard
|
|
case .Ok((let action_declaration, let updated_context)) = maybe_action_declaration
|
|
else {
|
|
return .Error(maybe_action_declaration.error()!)
|
|
}
|
|
actions.append(action_declaration)
|
|
local_context = updated_context
|
|
} else if currentChild.nodeType == "table_declaration" {
|
|
let maybe_table_declaration = Table.Compile(
|
|
node: currentChild, withContext: local_context)
|
|
guard
|
|
case .Ok((let table_declaration, let updated_context)) = maybe_table_declaration
|
|
else {
|
|
return .Error(maybe_table_declaration.error()!)
|
|
}
|
|
tables.append(table_declaration)
|
|
local_context = updated_context
|
|
} else {
|
|
return .Error(
|
|
ErrorOnNode(node: currentChild, withError: "Uknown node type in control declaration"))
|
|
}
|
|
}
|
|
|
|
// There should only be a single table!
|
|
// TODO: Check the semantics here.
|
|
if tables.count > 1 {
|
|
// TODO: Make this error message better.
|
|
// IDEA: Add a "compilation context" for the error message into the `CompilationContext`
|
|
// that can be retrieved to make the error messages nicer.
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "More than one table in control declaration"))
|
|
}
|
|
|
|
let declared_control =
|
|
(Control(
|
|
named: control_name, withParameters: control_parameters, withTable: tables[0],
|
|
withActions: Actions(withActions: actions))
|
|
as P4Type)
|
|
|
|
// Don't forget to add the newly declared Control to the instance that we were given
|
|
// (and not the one that we entered to do the parsing of this Control).
|
|
return .Ok(
|
|
(
|
|
declared_control,
|
|
context.update(
|
|
newInstances: context.instances.declare(
|
|
identifier: control_name, withValue: declared_control))
|
|
))
|
|
}
|
|
}
|
|
|
|
extension Action: Compilable {
|
|
public typealias T = Action
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.Action, CompilerContext)> {
|
|
#RequireNodeType<Node, (P4Type, CompilerContext)>(
|
|
node: node, type: "action_declaration", nice_type_name: "Action Declaration")
|
|
|
|
var currentChildIdx = 1
|
|
var currentChildIdxSafe = 2
|
|
var currentChild: Node? = .none
|
|
var current_context = context
|
|
|
|
// Skip action keyword
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing action declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
guard
|
|
case .Ok(let action_name) = Identifier.Compile(
|
|
node: currentChild!, withContext: current_context)
|
|
else {
|
|
return Result.Error(
|
|
Error(withMessage: "Could not parse an action name from \(currentChild!.text!)"))
|
|
}
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing action declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
let maybe_action_parameters = ParameterList.Compile(
|
|
node: currentChild!, withContext: current_context)
|
|
guard case .Ok((let action_parameters, let updated_context)) = maybe_action_parameters
|
|
else {
|
|
return .Error(maybe_action_parameters.error()!)
|
|
}
|
|
current_context = updated_context
|
|
|
|
currentChildIdx += 1
|
|
currentChildIdxSafe += 1
|
|
if node.childCount < currentChildIdxSafe {
|
|
return Result.Error(
|
|
ErrorOnNode(
|
|
node: node, withError: "Missing action declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
// Add the parameters into scope.
|
|
var function_scope = context.instances.enter()
|
|
for parameter in action_parameters.parameters {
|
|
function_scope = function_scope.declare(identifier: parameter.name, withValue: parameter.type)
|
|
}
|
|
|
|
let maybe_action_body = Parser.Statement.Compile(
|
|
node: currentChild!, withContext: context.update(newInstances: function_scope))
|
|
guard case .Ok((let action_body, _)) = maybe_action_body else {
|
|
return .Error(maybe_action_body.error()!)
|
|
}
|
|
|
|
return .Ok(
|
|
(
|
|
Action(named: action_name, withParameters: action_parameters, withBody: action_body),
|
|
current_context
|
|
))
|
|
}
|
|
}
|
|
|
|
extension TableKeyEntry: Compilable {
|
|
public typealias T = TableKeyEntry
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.TableKeyEntry, CompilerContext)> {
|
|
|
|
#RequireNodeType<Node, (P4Type, CompilerContext)>(
|
|
node: node, type: "table_key_entry", nice_type_name: "Table Key Entry")
|
|
|
|
var currentChildIdx = 0
|
|
var currentChildIdxSafe = 1
|
|
var currentChild: Node? = .none
|
|
|
|
let current_context = context
|
|
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing table key entry declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
let maybe_keyset_expression = KeysetExpression.compile(
|
|
node: currentChild!, withContext: current_context)
|
|
guard case .Ok(let keyset_expression) = maybe_keyset_expression else {
|
|
return Result.Error(maybe_keyset_expression.error()!)
|
|
}
|
|
|
|
// Skip the ':'
|
|
currentChildIdx += 2
|
|
currentChildIdxSafe += 2
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "Missing table key entry declaration component"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
let maybe_match_type = TableKeyMatchType.Compile(
|
|
node: currentChild!, withContext: current_context)
|
|
guard case .Ok((let match_type, _)) = maybe_match_type else {
|
|
return .Error(maybe_match_type.error()!)
|
|
}
|
|
|
|
return .Ok((TableKeyEntry(keyset_expression as! KeysetExpression, match_type), current_context))
|
|
}
|
|
}
|
|
|
|
extension TableKeyMatchType: Compilable {
|
|
public typealias T = TableKeyMatchType
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.TableKeyMatchType, CompilerContext)> {
|
|
#RequireNodeType<Node, (TableKeyMatchType, CompilerContext)>(
|
|
node: node, type: "table_key_match_type", nice_type_name: "Table Key Match Type")
|
|
|
|
if node.text! == "exact" {
|
|
return .Ok((TableKeyMatchType.Exact, context))
|
|
}
|
|
return .Error(ErrorOnNode(node: node, withError: "\(node.text!) is not a valid match type)"))
|
|
}
|
|
}
|
|
|
|
extension TableKeys: Compilable {
|
|
public typealias T = TableKeys
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.TableKeys, CompilerContext)> {
|
|
#RequireNodeType<Node, (TableKeyMatchType, CompilerContext)>(
|
|
node: node, type: "table_keys", nice_type_name: "Table Keys")
|
|
|
|
// Skip the
|
|
// keys = {
|
|
// 0 1 2
|
|
let currentChildIdx = 3
|
|
let currentChildIdxSafe = 4
|
|
var currentChild: Node? = .none
|
|
var current_context = context
|
|
|
|
if node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(
|
|
node: node, withError: "Missing table keys declaration component in control declaration"))
|
|
}
|
|
currentChild = node.child(at: currentChildIdx)
|
|
|
|
var entries: [TableKeyEntry] = Array()
|
|
var errors: [Error] = Array()
|
|
|
|
currentChild!.enumerateNamedChildren { entry in
|
|
switch TableKeyEntry.Compile(node: currentChild!, withContext: current_context) {
|
|
case .Ok((let keyset_expression, let updated_context)):
|
|
entries.append(keyset_expression)
|
|
current_context = updated_context
|
|
case .Error(let e): errors.append(e)
|
|
}
|
|
}
|
|
|
|
if !errors.isEmpty {
|
|
return .Error(
|
|
Error(
|
|
withMessage: "Error(s) parsing table key: "
|
|
+ (errors.map { error in
|
|
return "\(error.msg)"
|
|
}.joined(separator: ";"))))
|
|
}
|
|
|
|
return .Ok((TableKeys(withEntries: entries), current_context))
|
|
}
|
|
}
|
|
|
|
extension TablePropertyList: Compilable {
|
|
public typealias T = TablePropertyList
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.TablePropertyList, CompilerContext)> {
|
|
|
|
#RequireNodeType<Node, (P4Type, CompilerContext)>(
|
|
node: node, type: "table_property_list", nice_type_name: "Table Property List")
|
|
|
|
var current_context = context
|
|
|
|
var keys: [TableKeys] = Array()
|
|
var _: [Action] = Array() // Actions are not yet supported
|
|
var errors: [Error] = Array()
|
|
|
|
node.enumerateNamedChildren { child in
|
|
if child.nodeType == "table_keys" {
|
|
switch TableKeys.Compile(node: child, withContext: current_context) {
|
|
case .Ok((let table_key, let updated_context)):
|
|
current_context = updated_context
|
|
keys.append(table_key)
|
|
case .Error(let e): errors.append(e)
|
|
}
|
|
} else if child.nodeType == "table_actions" {
|
|
errors.append(
|
|
ErrorOnNode(
|
|
node: child, withError: "Actions in table property lists are not yet supported"))
|
|
} else {
|
|
errors.append(
|
|
ErrorOnNode(node: child, withError: "Uknown node type in control declaration"))
|
|
}
|
|
}
|
|
|
|
if !errors.isEmpty {
|
|
return .Error(
|
|
Error(
|
|
withMessage: "Error(s) parsing property list: "
|
|
+ (errors.map { error in
|
|
return "\(error.msg)"
|
|
}.joined(separator: ";"))))
|
|
}
|
|
|
|
// There should be only one table keys!
|
|
if keys.count > 1 {
|
|
// Todo: Make this error message better.
|
|
return .Error(
|
|
ErrorOnNode(node: node, withError: "More than one key set in table property list"))
|
|
}
|
|
|
|
return .Ok((TablePropertyList(withActions: TableActions(), withKeys: keys[0]), current_context))
|
|
|
|
}
|
|
}
|
|
|
|
extension Table: Compilable {
|
|
public typealias T = Table
|
|
public static func Compile(
|
|
node: SwiftTreeSitter.Node, withContext context: CompilerContext
|
|
) -> Common.Result<(P4Lang.Table, CompilerContext)> {
|
|
|
|
let table_declaration_node = node
|
|
#RequireNodeType<Node, (P4Type, CompilerContext)>(
|
|
node: table_declaration_node, type: "table_declaration", nice_type_name: "Table Declaration")
|
|
|
|
var currentChildIdx = 1
|
|
var currentChildIdxSafe = 2
|
|
var currentChild: Node? = .none
|
|
|
|
let current_context = context
|
|
|
|
if table_declaration_node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: table_declaration_node, withError: "Missing table declaration component"))
|
|
}
|
|
currentChild = table_declaration_node.child(at: currentChildIdx)
|
|
|
|
guard
|
|
case .Ok(let table_name) = Identifier.Compile(
|
|
node: currentChild!, withContext: current_context)
|
|
else {
|
|
return Result.Error(
|
|
Error(withMessage: "Could not parse a table name from \(currentChild!.text!)"))
|
|
}
|
|
|
|
// Skip the '{'
|
|
currentChildIdx += 2
|
|
currentChildIdxSafe += 2
|
|
if table_declaration_node.childCount < currentChildIdxSafe {
|
|
return .Error(
|
|
ErrorOnNode(node: table_declaration_node, withError: "Missing table declaration component"))
|
|
}
|
|
currentChild = table_declaration_node.child(at: currentChildIdx)
|
|
|
|
let maybe_table_property_list = TablePropertyList.Compile(
|
|
node: currentChild!, withContext: current_context)
|
|
guard case .Ok((let table_property_list, _)) = maybe_table_property_list else {
|
|
return Result.Error(maybe_table_property_list.error()!)
|
|
}
|
|
|
|
return .Ok(
|
|
(Table(withName: table_name, withPropertyList: table_property_list), current_context))
|
|
}
|
|
}
|