script/genmakefile/genmakefile.go

   1 package main
   2
   3 import (
   4         "fmt"
   5         "go/build"
   6         "os"
   7         "strings"
   8 )
   9
  10 var packages map[string]string = make(map[string]string)
  11
  12 func generate_target(srcdir string, pkgdir string, prefix string, ctx build.Context) string {
  13         pkg, _ := ctx.ImportDir(srcdir+pkgdir, 0)
  14         name := pkg.Name
  15         var deps []string
  16         for _, imp := range pkg.Imports {
  17                 if strings.HasPrefix(imp, prefix) {
  18                         imp = strings.TrimPrefix(imp, prefix)
  19                         if packages[imp] == "" {
  20                                 packages[imp] = generate_target(srcdir, imp, prefix, ctx)
  21                         }
  22                         deps = append(deps, "$(LIBS_"+packages[imp]+")")
  23                 }
  24         }
  25         if pkgdir != "" {
  26                 fmt.Printf("SRCDIR_%s := $(SRCDIR)%s/\n", name, pkgdir)
  27         } else {
  28                 fmt.Printf("SRCDIR_%s := $(SRCDIR)\n", name)
  29         }
  30         fmt.Printf("SRC_%s := $(addprefix $(SRCDIR_%s), %s)\n", name, name, strings.Join(pkg.GoFiles, " "))
  31         fmt.Printf("DEPS_%s := %s\n", name, strings.Join(deps, " "))
  32         if pkgdir != "" {
  33                 fmt.Printf("OBJ_%s := $(LIBDIR)/%s.o\n", name, pkgdir)
  34                 fmt.Printf("LIB_%s := $(LIBDIR)/%s.a\n", name, pkgdir)
  35                 fmt.Printf("LIBS_%s := $(LIB_%s) $(DEPS_%s)\n", name, name, name)
  36                 fmt.Printf("$(OBJ_%s) : $(SRC_%s) $(DEPS_%s)\n", name, name, name)
  37                 fmt.Printf("\t@mkdir -p $(dir $@)\n")
  38                 fmt.Printf("\t$(GOC) $(GOFLAGS) -c -o $@ $(SRC_%s)\n", name)
  39         }
  40         return name
  41 }
  42
  43 func main() {
  44         srcdir := os.Args[1]
  45         prefix := os.Args[2]
  46         ctx := build.Default
  47         ctx.CgoEnabled = false
  48         generate_target(srcdir, "", prefix, ctx)
  49 }