mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
53 lines
1020 B
Go
53 lines
1020 B
Go
package servergen
|
|
|
|
import (
|
|
"errors"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/99designs/gqlgen/codegen"
|
|
"github.com/99designs/gqlgen/codegen/templates"
|
|
"github.com/99designs/gqlgen/plugin"
|
|
)
|
|
|
|
func New(filename string) plugin.Plugin {
|
|
return &Plugin{filename}
|
|
}
|
|
|
|
type Plugin struct {
|
|
filename string
|
|
}
|
|
|
|
var _ plugin.CodeGenerator = &Plugin{}
|
|
|
|
func (m *Plugin) Name() string {
|
|
return "servergen"
|
|
}
|
|
|
|
func (m *Plugin) GenerateCode(data *codegen.Data) error {
|
|
serverBuild := &ServerBuild{
|
|
ExecPackageName: data.Config.Exec.ImportPath(),
|
|
ResolverPackageName: data.Config.Resolver.ImportPath(),
|
|
}
|
|
|
|
if _, err := os.Stat(m.filename); errors.Is(err, fs.ErrNotExist) {
|
|
return templates.Render(templates.Options{
|
|
PackageName: "main",
|
|
Filename: m.filename,
|
|
Data: serverBuild,
|
|
Packages: data.Config.Packages,
|
|
})
|
|
}
|
|
|
|
log.Printf("Skipped server: %s already exists\n", m.filename)
|
|
return nil
|
|
}
|
|
|
|
type ServerBuild struct {
|
|
codegen.Data
|
|
|
|
ExecPackageName string
|
|
ResolverPackageName string
|
|
}
|