@@ -19,7 +19,7 @@ import (
1919func main () {
2020ctx := context .Background ()
2121log := slog .Make (sloghuman .Sink (os .Stderr ))
22- code ,err := GenerateFromDirectory (ctx ,os .Args [1 ],os .Args [2 ] )
22+ code ,err := GenerateFromDirectory (ctx ,os .Args [1 ],os .Args [2 :] ... )
2323if err != nil {
2424log .Fatal (ctx ,"generate" ,slog .Error (err ))
2525}
@@ -28,14 +28,14 @@ func main() {
2828}
2929
3030// GenerateFromDirectory will return all the typescript code blocks for a directory
31- func GenerateFromDirectory (ctx context.Context ,directory , typName string ) (string ,error ) {
31+ func GenerateFromDirectory (ctx context.Context ,directory string , typeNames ... string ) (string ,error ) {
3232g := Generator {}
3333err := g .parsePackage (ctx ,directory )
3434if err != nil {
3535return "" ,xerrors .Errorf ("parse package %q: %w" ,directory ,err )
3636}
3737
38- str ,err := g .generate (typName )
38+ str ,err := g .generate (typeNames ... )
3939if err != nil {
4040return "" ,xerrors .Errorf ("parse package %q: %w" ,directory ,err )
4141}
@@ -74,38 +74,40 @@ func (g *Generator) parsePackage(ctx context.Context, patterns ...string) error
7474return nil
7575}
7676
77- func (g * Generator )generate (typName string ) (string ,error ) {
77+ func (g * Generator )generate (typeNames ... string ) (string ,error ) {
7878sb := strings.Builder {}
7979
8080_ ,_ = fmt .Fprint (& sb ,"Copy the following code into the audit.AuditableResources table\n \n " )
8181
82- obj := g .pkg .Types .Scope ().Lookup (typName )
83- if obj == nil || obj .Type ()== nil {
84- return "" ,xerrors .Errorf ("type doesn't exist %q" ,typName )
85- }
86-
87- switch obj := obj .(type ) {
88- case * types.TypeName :
89- named ,ok := obj .Type ().(* types.Named )
90- if ! ok {
91- panic ("all typenames should be named types" )
82+ for _ ,typName := range typeNames {
83+ obj := g .pkg .Types .Scope ().Lookup (typName )
84+ if obj == nil || obj .Type ()== nil {
85+ return "" ,xerrors .Errorf ("type doesn't exist %q" ,typName )
9286}
9387
94- switch typ := named .Underlying ().(type ) {
95- case * types.Struct :
96- g .writeStruct (& sb ,typ ,typName )
88+ switch obj := obj .(type ) {
89+ case * types.TypeName :
90+ named ,ok := obj .Type ().(* types.Named )
91+ if ! ok {
92+ panic ("all typenames should be named types" )
93+ }
94+
95+ switch typ := named .Underlying ().(type ) {
96+ case * types.Struct :
97+ g .writeStruct (& sb ,typ ,typName )
9798
99+ default :
100+ return "" ,xerrors .Errorf ("invalid type %T" ,obj )
101+ }
98102default :
99103return "" ,xerrors .Errorf ("invalid type %T" ,obj )
100104}
101- default :
102- return "" ,xerrors .Errorf ("invalid type %T" ,obj )
103105}
104106
105107return sb .String (),nil
106108}
107109
108- func (g * Generator )writeStruct (w io.Writer ,st * types.Struct ,name string ) {
110+ func (* Generator )writeStruct (w io.Writer ,st * types.Struct ,name string ) {
109111_ ,_ = fmt .Fprintf (w ,"\t &database.%s{}: {\n " ,name )
110112
111113for i := 0 ;i < st .NumFields ();i ++ {