delegationtest: run go fmt, less noisy code

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

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("}")
}
out, err := format.Source(buf.Bytes())
if err != nil {
return err
}
return os.WriteFile("../token_gen.go", out, 0666)
}

View File

@@ -107,21 +107,21 @@ var (
)
var (
TokenCarolDan_InvalidExpiredCID = cid.MustParse("bafyreic7cpi2j65je6rwiyxzq233lpbljvl6ggs5ew53nnye3tc3jdltqm")
TokenCarolDan_InvalidExpiredCID = cid.MustParse("bafyreigvdcarp6426e5aenal3q3c4uyhncwfbmt3fsmb4qzspkpjy57gti")
TokenCarolDan_InvalidExpiredSealed = mustGetBundle(TokenCarolDan_InvalidExpiredCID).Sealed
TokenCarolDan_InvalidExpiredBundle = mustGetBundle(TokenCarolDan_InvalidExpiredCID)
TokenCarolDan_InvalidExpired = mustGetBundle(TokenCarolDan_InvalidExpiredCID).Decoded
)
var (
TokenDanErin_InvalidExpiredCID = cid.MustParse("bafyreiguafxrlpncpjxxfuled2izocjp5s3kcclkdl4edpd22tr2gfcad4")
TokenDanErin_InvalidExpiredCID = cid.MustParse("bafyreifa265mung4nfpjjuiir5nytzgbrhfl66ye2dqmbq4p6dm7bzp4he")
TokenDanErin_InvalidExpiredSealed = mustGetBundle(TokenDanErin_InvalidExpiredCID).Sealed
TokenDanErin_InvalidExpiredBundle = mustGetBundle(TokenDanErin_InvalidExpiredCID)
TokenDanErin_InvalidExpired = mustGetBundle(TokenDanErin_InvalidExpiredCID).Decoded
)
var (
TokenErinFrank_InvalidExpiredCID = cid.MustParse("bafyreid4blvceraqiz32vca4gwk2hcf7dqtu6ydpmq3xlbj5biiq46mo44")
TokenErinFrank_InvalidExpiredCID = cid.MustParse("bafyreicde2per7ynjruvstuae67dxzdg2iepid3nonrmgr6tvodzpbqomu")
TokenErinFrank_InvalidExpiredSealed = mustGetBundle(TokenErinFrank_InvalidExpiredCID).Sealed
TokenErinFrank_InvalidExpiredBundle = mustGetBundle(TokenErinFrank_InvalidExpiredCID)
TokenErinFrank_InvalidExpired = mustGetBundle(TokenErinFrank_InvalidExpiredCID).Decoded