generate: Add some basic improve support for protocol types

progrium · Jun 22, 2024 · 9fafbd6 · 9fafbd6
1 parent 0d9358c
commit 9fafbd6
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/generate/codegen/gen_function.go b/generate/codegen/gen_function.go
@@ -194,6 +194,8 @@ func (f *Function) WriteGoCallCode(currentModule *modules.Module, cw *CodeWriter
 			cw.WriteLineF("return *(*%s)(unsafe.Pointer(&%s))", tt.GoName(currentModule, true), resultName)
 		case *typing.CStringType:
 			cw.WriteLineF("return C.GoString(%s)", resultName)
+		case *typing.ProtocolType:
+			cw.WriteLineF("return %s{objc.ObjectFrom(%s)}", returnTypeStr, resultName)
 		default:
 			cw.WriteLineF("return %s(%s)", returnTypeStr, resultName)
 		}

diff --git a/generate/function.go b/generate/function.go
@@ -71,6 +71,9 @@ func (db *Generator) ToFunction(fw string, sym Symbol) *codegen.Function {
 		"CGPDFDictionaryGetName":                   true, // "const char *key, const char * _Nullable *"
 		"CGPDFScannerPopName":                      true, // "const char * _Nullable *"
 	}
+	if sym.Name != "MTLCreateSystemDefaultDevice" {
+		return nil
+	}
 	if knownIssues[sym.Name] {
 		_, err := sym.Parse(db.Platform)
 		log.Printf("skipping function %s %s because of known issue: decl='%s' err='%v'\n", fw, sym.Name, sym.Declaration, err)

diff --git a/generate/typing/pointer_type.go b/generate/typing/pointer_type.go
@@ -1,6 +1,8 @@
 package typing
 
 import (
+	"fmt"
+
 	"github.com/progrium/darwinkit/generate/modules"
 	"github.com/progrium/darwinkit/internal/set"
 )
@@ -40,8 +42,10 @@ func (c *PointerType) GoName(currentModule *modules.Module, receiveFromObjc bool
 		return "*" + c.Type.GoName(currentModule, receiveFromObjc)
 	case *RefType:
 		return "*" + c.Type.GoName(currentModule, receiveFromObjc)
+	case *ProtocolType:
+		return "*" + c.Type.GoName(currentModule, receiveFromObjc)
 	default:
-		panic("not supported pointer to: " + c.Type.ObjcName())
+		panic(fmt.Sprintf("not supported pointer to: %T %v", c.Type, c.Type.ObjcName()))
 	}
 
 }

diff --git a/generate/typing/protocol_type.go b/generate/typing/protocol_type.go
@@ -27,7 +27,11 @@ func (p *ProtocolType) ObjcName() string {
 }
 
 func (p *ProtocolType) CName() string {
-	return "id<" + p.Name + ">"
+	return "void *"
+}
+
+func (p *ProtocolType) CSignature() string {
+	return "void *"
 }
 
 func (p *ProtocolType) GoImports() set.Set[string] {