mirror of
https://github.com/h44z/wg-portal
synced 2025-02-26 05:49:14 +00:00
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/swaggo/swag"
|
|
"github.com/swaggo/swag/gen"
|
|
)
|
|
|
|
var apiRootPath = "/internal/app/api"
|
|
var apiDocPath = "core/assets/doc"
|
|
var apiMkDocPath = "/docs/documentation/rest-api"
|
|
|
|
// this replaces the call to: swag init --propertyStrategy pascalcase --parseDependency --parseInternal --generalInfo base.go
|
|
func main() {
|
|
wd, err := os.Getwd() // should be the project root
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
apiBasePath := filepath.Join(wd, apiRootPath)
|
|
apis := []string{"v0", "v1"}
|
|
|
|
hasError := false
|
|
for _, apiVersion := range apis {
|
|
apiPath := filepath.Join(apiBasePath, apiVersion, "handlers")
|
|
|
|
apiVersion = strings.TrimLeft(apiVersion, "api-")
|
|
log.Println("")
|
|
log.Println("Generate swagger docs for API", apiVersion)
|
|
log.Println("Api path:", apiPath)
|
|
|
|
err := generateApi(apiBasePath, apiPath, apiVersion)
|
|
if err != nil {
|
|
hasError = true
|
|
logrus.Errorf("failed to generate API docs for %s: %v", apiVersion, err)
|
|
}
|
|
|
|
// copy the latest version of the API docs for mkdocs
|
|
if apiVersion == apis[len(apis)-1] {
|
|
if err = copyDocForMkdocs(wd, apiBasePath, apiVersion); err != nil {
|
|
logrus.Errorf("failed to copy API docs for mkdocs: %v", err)
|
|
} else {
|
|
log.Println("Copied API docs " + apiVersion + " for mkdocs")
|
|
}
|
|
}
|
|
|
|
log.Println("Generated swagger docs for API", apiVersion)
|
|
}
|
|
|
|
if hasError {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func generateApi(basePath, apiPath, version string) error {
|
|
err := gen.New().Build(&gen.Config{
|
|
SearchDir: apiPath,
|
|
Excludes: "",
|
|
MainAPIFile: "base.go",
|
|
PropNamingStrategy: swag.PascalCase,
|
|
OutputDir: filepath.Join(basePath, apiDocPath),
|
|
OutputTypes: []string{"json", "yaml"},
|
|
ParseVendor: false,
|
|
ParseDependency: 3,
|
|
MarkdownFilesDir: "",
|
|
ParseInternal: true,
|
|
GeneratedTime: false,
|
|
CodeExampleFilesDir: "",
|
|
ParseDepth: 3,
|
|
InstanceName: version,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("swag failed: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func copyDocForMkdocs(workingDir, basePath, version string) error {
|
|
srcPath := filepath.Join(basePath, apiDocPath, fmt.Sprintf("%s_swagger.yaml", version))
|
|
dstPath := filepath.Join(workingDir, apiMkDocPath, "swagger.yaml")
|
|
|
|
// copy the file
|
|
input, err := os.ReadFile(srcPath)
|
|
if err != nil {
|
|
return fmt.Errorf("error while reading swagger doc: %w", err)
|
|
}
|
|
|
|
err = os.WriteFile(dstPath, input, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("error while writing swagger doc: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|