node: add initNode and export asNodePtr (#93)

urso · web-flow · commit ac878b8f67b4 · 2024-08-13T21:43:31.000+02:00
diff --git a/src/pgzx/node.zig b/src/pgzx/node.zig
@@ -21,6 +21,12 @@ pub inline fn create(initFrom: anytype) *@TypeOf(initFrom) {
     return node;
 }
 
+pub inline fn initNode(initFrom: anytype) @TypeOf(initFrom) {
+    var node = initFrom;
+    setTag(&node, mustFindTag(@TypeOf(initFrom)));
+    return node;
+}
+
 fn mustFindTag(comptime T: type) Tag {
     return generated.findTag(T) orelse @compileError("No tag found for type");
 }
@@ -57,7 +63,7 @@ pub inline fn copyTypedNode(node: anytype) @TypeOf(node) {
     return @ptrCast(@alignCast(copy(node)));
 }
 
-inline fn asNodePtr(node: anytype) *pg.Node {
+pub inline fn asNodePtr(node: anytype) *pg.Node {
     checkIsPotentialNodePtr(node);
     return @ptrCast(@alignCast(node));
 }

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,12 @@ pub inline fn create(initFrom: anytype) *@TypeOf(initFrom) {`
`21`	`21`	`return node;`
`22`	`22`	`}`
`23`	`23`
	`24`	`+pub inline fn initNode(initFrom: anytype) @TypeOf(initFrom) {`
	`25`	`+ var node = initFrom;`
	`26`	`+ setTag(&node, mustFindTag(@TypeOf(initFrom)));`
	`27`	`+ return node;`
	`28`	`+}`
	`29`	`+`
`24`	`30`	`fn mustFindTag(comptime T: type) Tag {`
`25`	`31`	`return generated.findTag(T) orelse @compileError("No tag found for type");`
`26`	`32`	`}`
`@@ -57,7 +63,7 @@ pub inline fn copyTypedNode(node: anytype) @TypeOf(node) {`
`57`	`63`	`return @ptrCast(@alignCast(copy(node)));`
`58`	`64`	`}`
`59`	`65`
`60`		`-inline fn asNodePtr(node: anytype) *pg.Node {`
	`66`	`+pub inline fn asNodePtr(node: anytype) *pg.Node {`
`61`	`67`	`checkIsPotentialNodePtr(node);`
`62`	`68`	`return @ptrCast(@alignCast(node));`
`63`	`69`	`}`