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.
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
type deleteInfo struct {
|
|
baseType string
|
|
targetType string
|
|
path string
|
|
}
|
|
|
|
var deleteFuncs = map[string][]*deleteInfo{
|
|
"scopes": {
|
|
{
|
|
"Organization",
|
|
"Project",
|
|
"projects",
|
|
},
|
|
{
|
|
"Organization",
|
|
"users.User",
|
|
"users",
|
|
},
|
|
{
|
|
"Project",
|
|
"hosts.HostCatalog",
|
|
"host-catalogs",
|
|
},
|
|
},
|
|
}
|
|
|
|
func writeDeleteFuncs() {
|
|
for outPkg, funcs := range deleteFuncs {
|
|
outFile := os.Getenv("GEN_BASEPATH") + fmt.Sprintf("/api/%s/delete.gen.go", outPkg)
|
|
outBuf := bytes.NewBuffer([]byte(fmt.Sprintf(
|
|
`// Code generated by "make api"; DO NOT EDIT.
|
|
package %s
|
|
`, outPkg)))
|
|
for _, deleteInfo := range funcs {
|
|
deleteFuncTemplate.Execute(outBuf, struct {
|
|
BaseType string
|
|
TargetType string
|
|
TargetName string
|
|
Path string
|
|
}{
|
|
BaseType: deleteInfo.baseType,
|
|
TargetType: deleteInfo.targetType,
|
|
TargetName: strings.Split(deleteInfo.targetType, ".")[strings.Count(deleteInfo.targetType, ".")],
|
|
Path: deleteInfo.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 deleteFuncTemplate = template.Must(template.New("").Parse(
|
|
`
|
|
// Delete{{ .TargetName }} returns true iff the {{ .TargetType }} existed when the delete attempt was made.
|
|
func (s {{ .BaseType }}) Delete{{ .TargetName }}(ctx context.Context, r *{{ .TargetType }}) (bool, *api.Error, error) {
|
|
if s.Client == nil {
|
|
return false, nil, fmt.Errorf("nil client in Delete{{ .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 Delete{{ .TargetName }} 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 false, nil, fmt.Errorf("empty {{ .TargetType }} ID field in Delete{{ .TargetName }} request")
|
|
}
|
|
|
|
req, err := s.Client.NewRequest(ctx, "DELETE", fmt.Sprintf("%s/%s", "{{ .Path }}", r.Id), nil)
|
|
if err != nil {
|
|
return false, nil, fmt.Errorf("error creating Delete{{ .TargetName }} request: %w", err)
|
|
}
|
|
|
|
resp, err := s.Client.Do(req)
|
|
if err != nil {
|
|
return false, nil, fmt.Errorf("error performing client request during Delete{{ .TargetName }} call: %w", err)
|
|
}
|
|
|
|
type deleteResponse struct {
|
|
Existed bool
|
|
}
|
|
target := &deleteResponse{}
|
|
|
|
apiErr, err := resp.Decode(target)
|
|
if err != nil {
|
|
return false, nil, fmt.Errorf("error decoding Delete{{ .TargetName }} repsonse: %w", err)
|
|
}
|
|
|
|
return target.Existed, apiErr, nil
|
|
}
|
|
`))
|