forked from mongodb/mongo-go-driver
/
main.go
92 lines (80 loc) · 2.48 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package main
import (
"flag"
"fmt"
"html/template"
"log"
"os"
"path/filepath"
"strings"
"golang.org/x/tools/go/packages"
)
var tmpl = `<!DOCTYPE html>
<html lang="en">
<head>
<meta http-equiv="refresh" content="5; url=https://godoc.org/github.com/wimspaargaren/mongo-go-driver/{{.}}">
<meta name=go-import content="go.mongodb.org/mongo-driver git https://github.com/mongodb/mongo-go-driver.git">
<meta name="go-source" content="go.mongodb.org/mongo-driver https://github.com/mongodb/mongo-go-driver https://github.com/mongodb/mongo-go-driver/tree/master{/dir} https://github.com/mongodb/mongo-go-driver/blob/master{/dir}/{file}#L{line}">
</head>
<body>
Redirecting to docs...
</body>
</html>
`
func main() {
var directory = "."
var destination = "s3-website"
fs := flag.NewFlagSet("", flag.ExitOnError)
fs.Usage = func() {
fmt.Fprintln(fs.Output(), "docbuilder is used to create the static site for the import paths for go.mongodb.org.")
fmt.Fprintln(fs.Output(), "usage: docbuilder [flags] [destination]")
fs.PrintDefaults()
}
fs.StringVar(&directory, "directory", ".", "directory to run docbuilder on")
err := fs.Parse(os.Args[1:])
if err == flag.ErrHelp {
fs.Usage()
os.Exit(0)
}
if err != nil {
log.Fatalf("Could not parse flags: %v", err)
}
args := fs.Args()
if len(args) > 0 {
destination = args[0]
}
pkgs, err := packages.Load(&packages.Config{Dir: directory}, "./...")
if err != nil {
log.Fatalf("Could not load packages: %v", err)
}
dirs := make([]string, 1, len(pkgs)+1)
for _, pkg := range pkgs {
if !strings.HasPrefix(pkg.PkgPath, "go.mongodb.org") {
continue
}
dirs = append(dirs, strings.TrimPrefix(pkg.PkgPath, "go.mongodb.org/mongo-driver"))
}
err = os.MkdirAll(filepath.Join(destination, "mongo-driver"), os.ModeDir|os.FileMode(0755))
if err != nil {
log.Fatalf("Could not make path: %v", err)
}
t, err := template.New("index").Parse(tmpl)
if err != nil {
log.Fatalf("Could not parse template: %v", err)
}
for _, dir := range dirs {
directory := filepath.Join(destination, "mongo-driver", dir)
err = os.MkdirAll(directory, os.ModeDir|os.FileMode(0755))
if err != nil {
log.Fatalf("Could not create directory (%s): %v", directory, err)
}
file, err := os.Create(filepath.Join(directory, "index.html"))
if err != nil {
log.Fatalf("Could not create index.html: %v", err)
}
err = t.Execute(file, dir)
if err != nil {
log.Fatalf("Could not execute template: %v", err)
}
}
}