go/ucan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

delegationtest: run go fmt, less noisy code

Browse Source
This commit is contained in:
Michael Muré
2024-12-05 15:30:23 +01:00
parent 7f1adbd945
commit 78825f4f55
5 changed files with 67 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
token/delegation/delegationtest/generator/generator.go
View File

@@ -1,7 +1,9 @@
package main
import (
"bytes"
"fmt"
"go/format"
"os"
"path/filepath"
"slices"
@@ -202,72 +204,71 @@ func (g *generator) createProofChain(name string, prf []cid.Cid) {
}
func (g *generator) writeGoFile() error {
var err error
buf := bytes.NewBuffer(nil)
f, err := os.Create("../token_gen.go")
if err != nil {
return err
}
Println := func(a ...any) { _, _ = fmt.Fprintln(buf, a...) }
Printf := func(format string, a ...any) { _, _ = fmt.Fprintf(buf, format, a...) }
defer func() {
err = f.Close()
}()
_, _ = fmt.Fprintln(f, "// Code generated by delegationtest - DO NOT EDIT.")
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "package delegationtest")
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "import (")
_, _ = fmt.Fprintln(f, "\t\"github.com/ipfs/go-cid\"")
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "\t\"github.com/ucan-wg/go-ucan/token/delegation\"")
_, _ = fmt.Fprintln(f, ")")
Println("// Code generated by delegationtest - DO NOT EDIT.")
Println()
Println("package delegationtest")
Println()
Println("import (")
Println("\t\"github.com/ipfs/go-cid\"")
Println()
Println("\t\"github.com/ucan-wg/go-ucan/token/delegation\"")
Println(")")
refs := make(map[cid.Cid]string, len(g.dlgs))
for _, d := range g.dlgs {
refs[d.id] = d.name + "CID"
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "var (")
_, _ = fmt.Fprintf(f, "\t%sCID = cid.MustParse(\"%s\")\n", d.name, d.id.String())
_, _ = fmt.Fprintf(f, "\t%sSealed = mustGetBundle(%s).Sealed\n", d.name, d.name+"CID")
_, _ = fmt.Fprintf(f, "\t%sBundle = mustGetBundle(%s)\n", d.name, d.name+"CID")
_, _ = fmt.Fprintf(f, "\t%s = mustGetBundle(%s).Decoded\n", d.name, d.name+"CID")
_, _ = fmt.Fprintln(f, ")")
Println()
Println("var (")
Printf("\t%sCID = cid.MustParse(\"%s\")\n", d.name, d.id.String())
Printf("\t%sSealed = mustGetBundle(%s).Sealed\n", d.name, d.name+"CID")
Printf("\t%sBundle = mustGetBundle(%s)\n", d.name, d.name+"CID")
Printf("\t%s = mustGetBundle(%s).Decoded\n", d.name, d.name+"CID")
Println(")")
}
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "var AllTokens = []*delegation.Token{")
Println()
Println("var AllTokens = []*delegation.Token{")
for _, d := range g.dlgs {
_, _ = fmt.Fprintf(f, "\t%s,\n", d.name)
Printf("\t%s,\n", d.name)
}
_, _ = fmt.Fprintln(f, "}")
Println("}")
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "var AllBundles = []*delegation.Bundle{")
Println()
Println("var AllBundles = []*delegation.Bundle{")
for _, d := range g.dlgs {
_, _ = fmt.Fprintf(f, "\t%sBundle,\n", d.name)
Printf("\t%sBundle,\n", d.name)
}
_, _ = fmt.Fprintln(f, "}")
Println("}")
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintln(f, "var cidToName = map[cid.Cid]string{")
Println()
Println("var cidToName = map[cid.Cid]string{")
for _, d := range g.dlgs {
_, _ = fmt.Fprintf(f, "\t%sCID: \"%s\",\n", d.name, d.name)
Printf("\t%sCID: \"%s\",\n", d.name, d.name)
}
_, _ = fmt.Fprintln(f, "}")
Println("}")
for _, c := range g.chains {
_, _ = fmt.Fprintln(f)
_, _ = fmt.Fprintf(f, "var %s = []cid.Cid{\n", c.name)
Println()
Printf("var %s = []cid.Cid{\n", c.name)
for _, d := range slices.Backward(c.prf) {
_, _ = fmt.Fprintf(f, "\t%s,\n", refs[d])
Printf("\t%s,\n", refs[d])
}
_, _ = fmt.Fprintln(f, "}")
Println("}")
}
return err
out, err := format.Source(buf.Bytes())
if err != nil {
return err
}
return os.WriteFile("../token_gen.go", out, 0666)
}