Todd Knight
df6ca5289f
* Create user handler stuff. * Fixing Not Found errors from DB. * Fixing param names. * Update the proto's go package generation. * Generate the user resource. * Adding SDK methods for users. * Fix the base type of Users and add tests. * Connect user service to the controller handler. * Attach the default org to the Test Controller's client. * Take into account the provided request paths when the client builds the org/project path prefix. * Adding comments to the API. * rename repo function to repoFn. * Fix typo in update mask errors. * Clean up tests. * Remove special checks for request url content in the api client and fix CORS test.
123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
type readInfo struct {
|
|
baseType string
|
|
targetType string
|
|
path string
|
|
}
|
|
|
|
var readFuncs = map[string][]*readInfo{
|
|
"scopes": {
|
|
{
|
|
baseType: "Organization",
|
|
targetType: "Project",
|
|
path: "projects",
|
|
},
|
|
{
|
|
baseType: "Organization",
|
|
targetType: "users.User",
|
|
path: "users",
|
|
},
|
|
{
|
|
baseType: "Project",
|
|
targetType: "hosts.HostCatalog",
|
|
path: "host-catalogs",
|
|
},
|
|
},
|
|
}
|
|
|
|
func writeReadFuncs() {
|
|
for outPkg, funcs := range readFuncs {
|
|
outFile := os.Getenv("GEN_BASEPATH") + fmt.Sprintf("/api/%s/read.gen.go", outPkg)
|
|
outBuf := bytes.NewBuffer([]byte(fmt.Sprintf(
|
|
`// Code generated by "make api"; DO NOT EDIT.
|
|
package %s
|
|
`, outPkg)))
|
|
for _, readInfo := range funcs {
|
|
readFuncTemplate.Execute(outBuf, struct {
|
|
BaseType string
|
|
TargetType string
|
|
TargetName string
|
|
Path string
|
|
}{
|
|
BaseType: readInfo.baseType,
|
|
TargetType: readInfo.targetType,
|
|
TargetName: strings.Split(readInfo.targetType, ".")[strings.Count(readInfo.targetType, ".")],
|
|
Path: readInfo.path,
|
|
})
|
|
}
|
|
if err := ioutil.WriteFile(outFile, outBuf.Bytes(), 0644); err != nil {
|
|
fmt.Printf("error writing file %q: %v\n", outFile, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|
|
|
|
var readFuncTemplate = template.Must(template.New("").Parse(
|
|
`
|
|
func (s {{ .BaseType }}) Read{{ .TargetName }}(ctx context.Context, r *{{ .TargetType }}) (*{{ .TargetType }}, *api.Error, error) {
|
|
if s.Client == nil {
|
|
return nil, nil, fmt.Errorf("nil client in Read{{ .TargetName }} request")
|
|
}
|
|
if s.Id == "" {
|
|
{{ if (eq .BaseType "Organization") }}
|
|
// Assume the client has been configured with organization already and
|
|
// move on
|
|
{{ else if (eq .BaseType "Project") }}
|
|
// Assume the client has been configured with project already and move
|
|
// on
|
|
{{ else }}
|
|
return nil, nil, fmt.Errorf("missing {{ .BaseType }} ID in Read{{ .TargetType }} request")
|
|
{{ end }}
|
|
} else {
|
|
// If it's explicitly set here, override anything that might be in the
|
|
// client
|
|
{{ if (eq .BaseType "Organization") }}
|
|
ctx = context.WithValue(ctx, "org", s.Id)
|
|
{{ else if (eq .BaseType "Project") }}
|
|
ctx = context.WithValue(ctx, "project", s.Id)
|
|
{{ end }}
|
|
}
|
|
if r.Id == "" {
|
|
return nil, nil, fmt.Errorf("empty {{ .TargetType }} ID field in Read{{ .TargetName }} request")
|
|
}
|
|
|
|
req, err := s.Client.NewRequest(ctx, "GET", fmt.Sprintf("%s/%s", "{{ .Path }}", r.Id), r)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("error creating Read{{ .TargetName }} request: %w", err)
|
|
}
|
|
|
|
resp, err := s.Client.Do(req)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("error performing client request during Read{{ .TargetName }} call: %w", err)
|
|
}
|
|
|
|
target := new({{ .TargetType }})
|
|
apiErr, err := resp.Decode(target)
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("error decoding Read{{ .TargetName }} repsonse: %w", err)
|
|
}
|
|
|
|
{{ if (eq .TargetType "Organization") }}
|
|
target.Client = s.Client.Clone()
|
|
target.Client.SetOrgnization(target.Id)
|
|
{{ else if (eq .TargetType "Project") }}
|
|
target.Client = s.Client.Clone()
|
|
target.Client.SetProject(target.Id)
|
|
{{ else }}
|
|
target.Client = s.Client
|
|
{{ end }}
|
|
|
|
return target, apiErr, nil
|
|
}
|
|
`))
|