Will Hawkins
80 lines
2.5 KiB
Swift
80 lines
2.5 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 ArgumentParser
|
|
import Common
|
|
import P4Compiler
|
|
import SystemPackage
|
|
|
|
@main
|
|
struct Cli: ParsableCommand {
|
|
@Flag(help: "Disable ANSI-stylized output.") var plain: Int
|
|
|
|
static let configuration = CommandConfiguration(
|
|
abstract: "P4CE compiler, interpreter and debugger.",
|
|
subcommands: [Compile.self])
|
|
}
|
|
|
|
struct CliOptions: ParsableArguments {
|
|
@ArgumentParser.Argument(help: "File to compile.") // Have to be explicit because Common has an Argument, too!
|
|
var path: String
|
|
|
|
@Option(name: [.customShort("I")], help: "Search paths.")
|
|
var search: [String] = []
|
|
}
|
|
|
|
extension Cli {
|
|
struct Compile: ParsableCommand {
|
|
static let configuration = CommandConfiguration(abstract: "Compile P4CE code.")
|
|
|
|
@ParentCommand var parent: Cli
|
|
|
|
@OptionGroup var options: CliOptions
|
|
|
|
mutating func run() {
|
|
let sm = SourceManager(options.search.map { FilePath($0) })
|
|
let prep = SourceCodePreprocessor(sm)
|
|
let file = FilePath(options.path)
|
|
|
|
let maybe_source = prep.preprocess(file)
|
|
guard case .Ok(let source) = maybe_source else {
|
|
let formatter = FormatterAnsi()
|
|
print(ErrorWithLabel("Preprocessor Error", maybe_source.error()!).format(formatter))
|
|
return
|
|
}
|
|
|
|
let maybe_program = Program.Compile(source.getSource())
|
|
guard case .Ok(_) = maybe_program else {
|
|
let formatter = FormatterAnsi()
|
|
print(ErrorWithLabel("Compiler Error", maybe_source.error()!).format(formatter))
|
|
return
|
|
}
|
|
|
|
let success_formatter: any Formattable =
|
|
if parent.plain != 0 {
|
|
FormatterPlain()
|
|
} else {
|
|
FormatterAnsi()
|
|
}
|
|
|
|
print(
|
|
success_formatter.formatWithStyle(
|
|
"Success", Style(StyleColor.Green, [StyleFormat.Underline])))
|
|
}
|
|
}
|
|
}
|