mkgodef: implements rawfunc mode

Signed-off-by: Koichi Shiraishi <[email protected]>

Author: zchee

cmd/mkgodef/main.go

@@ -45,7 +45,7 @@ var log logr.Logger
 
 func main() {
 	flag.StringVar(&flagPackage, fnamePackage, "", "package name of godef")
-	flag.StringSliceVar(&flagMode, fnameMode, []string{"enum, func, type"}, "generate mode")
+	flag.StringSliceVar(&flagMode, fnameMode, []string{"enum, func, type, rawfunc"}, "generate mode")
 	flag.StringSliceVar(&flagHeaders, fnameHeader, nil, "C header to analyze")
 	flag.StringSliceVar(&flagArgs, fnameArg, nil, "arg for analyze")
 	flag.StringSliceVar(&flagSources, fnameSource, nil, "additional C source to analyze")

cmd/mkgodef/mkgodef.go

@@ -44,6 +44,7 @@ const (
 	EnumMode = 1 << iota
 	TypeMode
 	FuncMode
+	RawFuncMode
 )
 
 // Config represents a mkgodef config.
@@ -280,6 +281,8 @@ func run(flags *flag.FlagSet) int {
 			mode |= TypeMode
 		case "func":
 			mode |= FuncMode
+		case "rawfunc":
+			mode |= RawFuncMode
 		}
 	}
 
@@ -493,6 +496,48 @@ func run(flags *flag.FlagSet) int {
 		}
 	}
 
+	if mode&RawFuncMode != 0 {
+		// sort funcMap by DisplayName
+		fns := make([]string, len(funcMap))
+		i := 0
+		for fn := range funcMap {
+			fns[i] = fn
+			i++
+		}
+		sort.Strings(fns)
+
+		var sb strings.Builder
+		seenFn := make(map[string]bool)
+		for _, fn := range fns {
+			cursor := funcMap[fn]
+			if seenFn[fn] {
+				log.V(1).Info("ignore", "s", fn, "kind", cursor.Kind())
+				continue
+			}
+			seenFn[fn] = true
+
+			switch cursor.Kind() {
+			case clang.Cursor_FunctionDecl:
+				p(&sb, "//sys func %s(", cursor.Spelling())
+
+				numArgs := cursor.NumArguments()
+				for i := int32(0); i < numArgs; i++ {
+					argName := cursor.Argument(uint32(i)).DisplayName()
+					argType := cursor.Argument(uint32(i)).Type().CanonicalType().Spelling()
+
+					p(&sb, "%s %s", argName, argType)
+					if i+1 < numArgs {
+						p(&sb, ", ")
+					}
+				}
+				p(&sb, ") %s\n", cursor.ResultType().Spelling())
+
+				bio.WriteString(sb.String())
+				sb.Reset()
+			}
+		}
+	}
+
 	bio.Flush()
 	bio2.Flush()