mkstdlib.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. // +build ignore
  2. // mkstdlib generates the zstdlib.go file, containing the Go standard
  3. // library API symbols. It's baked into the binary to avoid scanning
  4. // GOPATH in the common case.
  5. package main
  6. import (
  7. "bufio"
  8. "bytes"
  9. "fmt"
  10. "go/format"
  11. "io"
  12. "io/ioutil"
  13. "log"
  14. "os"
  15. "path/filepath"
  16. "regexp"
  17. "runtime"
  18. "sort"
  19. "strings"
  20. )
  21. func mustOpen(name string) io.Reader {
  22. f, err := os.Open(name)
  23. if err != nil {
  24. log.Fatal(err)
  25. }
  26. return f
  27. }
  28. func api(base string) string {
  29. return filepath.Join(runtime.GOROOT(), "api", base)
  30. }
  31. var sym = regexp.MustCompile(`^pkg (\S+).*?, (?:var|func|type|const) ([A-Z]\w*)`)
  32. var unsafeSyms = map[string]bool{"Alignof": true, "ArbitraryType": true, "Offsetof": true, "Pointer": true, "Sizeof": true}
  33. func main() {
  34. var buf bytes.Buffer
  35. outf := func(format string, args ...interface{}) {
  36. fmt.Fprintf(&buf, format, args...)
  37. }
  38. outf("// Code generated by mkstdlib.go. DO NOT EDIT.\n\n")
  39. outf("package imports\n")
  40. outf("var stdlib = map[string]map[string]bool{\n")
  41. f := io.MultiReader(
  42. mustOpen(api("go1.txt")),
  43. mustOpen(api("go1.1.txt")),
  44. mustOpen(api("go1.2.txt")),
  45. mustOpen(api("go1.3.txt")),
  46. mustOpen(api("go1.4.txt")),
  47. mustOpen(api("go1.5.txt")),
  48. mustOpen(api("go1.6.txt")),
  49. mustOpen(api("go1.7.txt")),
  50. mustOpen(api("go1.8.txt")),
  51. mustOpen(api("go1.9.txt")),
  52. mustOpen(api("go1.10.txt")),
  53. mustOpen(api("go1.11.txt")),
  54. mustOpen(api("go1.12.txt")),
  55. )
  56. sc := bufio.NewScanner(f)
  57. pkgs := map[string]map[string]bool{
  58. "unsafe": unsafeSyms,
  59. }
  60. paths := []string{"unsafe"}
  61. for sc.Scan() {
  62. l := sc.Text()
  63. has := func(v string) bool { return strings.Contains(l, v) }
  64. if has("struct, ") || has("interface, ") || has(", method (") {
  65. continue
  66. }
  67. if m := sym.FindStringSubmatch(l); m != nil {
  68. path, sym := m[1], m[2]
  69. if _, ok := pkgs[path]; !ok {
  70. pkgs[path] = map[string]bool{}
  71. paths = append(paths, path)
  72. }
  73. pkgs[path][sym] = true
  74. }
  75. }
  76. if err := sc.Err(); err != nil {
  77. log.Fatal(err)
  78. }
  79. sort.Strings(paths)
  80. for _, path := range paths {
  81. outf("\t%q: map[string]bool{\n", path)
  82. pkg := pkgs[path]
  83. var syms []string
  84. for sym := range pkg {
  85. syms = append(syms, sym)
  86. }
  87. sort.Strings(syms)
  88. for _, sym := range syms {
  89. outf("\t\t%q: true,\n", sym)
  90. }
  91. outf("},\n")
  92. }
  93. outf("}\n")
  94. fmtbuf, err := format.Source(buf.Bytes())
  95. if err != nil {
  96. log.Fatal(err)
  97. }
  98. err = ioutil.WriteFile("zstdlib.go", fmtbuf, 0666)
  99. if err != nil {
  100. log.Fatal(err)
  101. }
  102. }