Add Testing Macros

Don't build. Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
2026-01-30 08:26:17 -05:00
parent d8f6ff3b0b
commit 78a1121a3c
3 changed files with 4 additions and 4 deletions
@@ -0,0 +1,102 @@
+// 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 SwiftCompilerPlugin
+import SwiftSyntax
+import SwiftSyntaxMacros
+
+public struct UseOkResult: ExpressionMacro {
+    public static func expansion(
+        of node: some FreestandingMacroExpansionSyntax,
+        in context: some MacroExpansionContext
+    ) throws -> ExprSyntax {
+
+        guard let argument = node.argumentList.first?.expression else {
+            throw Require.Error.SyntaxError
+        }
+
+        return """
+            {
+                if case Result.Ok(let __runtime) =  \(argument) {
+                    return __runtime
+                } else {
+                    print("Oh no")
+                    throw Require.Error.UnexpectedResult
+                }
+            }()
+            """
+    }
+}
+
+public struct Require {
+    public enum Error: Swift.Error {
+        case UnexpectedResult
+        case SyntaxError
+    }
+}
+
+public struct RequireResult: ExpressionMacro {
+    public static func expansion(
+        of node: some FreestandingMacroExpansionSyntax,
+        in context: some MacroExpansionContext
+    ) throws -> ExprSyntax {
+
+        guard let argument = node.argumentList.first?.expression else {
+            throw Require.Error.SyntaxError
+        }
+
+        return """
+            {
+                if case Result.Ok(_) =  \(argument) {
+                    true
+                } else {
+                    false
+                }
+            }()
+            """
+    }
+}
+
+public struct RequireErrorResult: ExpressionMacro {
+    public static func expansion(
+        of node: some FreestandingMacroExpansionSyntax,
+        in context: some MacroExpansionContext
+    ) throws -> ExprSyntax {
+
+        let arguments = node.argumentList.indices
+        let expected_error = node.argumentList[arguments.startIndex]
+        let error_producer = node.argumentList[arguments.index(after: arguments.startIndex)]
+
+        return """
+            {
+                if case Result.Error(\(expected_error)) = \(error_producer) {
+                    true
+                } else {
+                    false
+                }
+            }()
+            """
+    }
+}
+
+
+@main
+struct P4Macros: CompilerPlugin {
+    var providingMacros: [Macro.Type] = [
+        RequireResult.self, RequireErrorResult.self, UseOkResult.self,
+    ]
+}