Copy source file indexes

This commit is contained in:
Trevor Slocum 2020-02-06 14:53:49 -08:00
parent eb76e8f9b1
commit 152092480f
2 changed files with 15 additions and 2 deletions

View File

@ -1,5 +1,6 @@
0.1.1:
- Link import path
- Copy source file indexes
0.1.0:
- Initial release

16
main.go
View File

@ -257,7 +257,7 @@ func main() {
continue // This is expected for packages without source files
}
sourceFiles := strings.Split(buf.String(), "\n")
sourceFiles := append(strings.Split(buf.String(), "\n"), "index.html")
for _, sourceFile := range sourceFiles {
sourceFile = strings.TrimSpace(sourceFile)
if sourceFile == "" {
@ -280,6 +280,13 @@ func main() {
updatePage(doc, basePath, siteName)
doc.Find(".layout").First().Find("a").Each(func(_ int, selection *goquery.Selection) {
href := selection.AttrOr("href", "")
if !strings.HasSuffix(href, ".") && !strings.HasSuffix(href, "/") && !strings.HasSuffix(href, ".html") {
selection.SetAttr("href", href+".html")
}
})
pkgSrcPath := path.Join(outDir, "src", pkg)
err = os.MkdirAll(pkgSrcPath, 0755)
@ -292,7 +299,12 @@ func main() {
if err != nil {
return
}
err = ioutil.WriteFile(path.Join(pkgSrcPath, sourceFile+".html"), buf.Bytes(), 0755)
outFileName := sourceFile
if !strings.HasSuffix(outFileName, ".html") {
outFileName += ".html"
}
err = ioutil.WriteFile(path.Join(pkgSrcPath, outFileName), buf.Bytes(), 0755)
if err != nil {
log.Fatalf("failed to write docs for %s: %s", pkg, err)
}