forked from cerc-io/ipld-eth-server
35 lines
630 B
Go
35 lines
630 B
Go
|
package printer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/mmcloughlin/avo/internal/prnt"
|
||
|
|
"github.com/mmcloughlin/avo/ir"
|
||
|
|
)
|
||
|
|
|
||
|
|
type stubs struct {
|
||
|
|
cfg Config
|
||
|
|
prnt.Generator
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewStubs constructs a printer for writing stub function declarations.
|
||
|
|
func NewStubs(cfg Config) Printer {
|
||
|
|
return &stubs{cfg: cfg}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stubs) Print(f *ir.File) ([]byte, error) {
|
||
|
|
s.Comment(s.cfg.GeneratedWarning())
|
||
|
|
|
||
|
|
if len(f.Constraints) > 0 {
|
||
|
|
s.NL()
|
||
|
|
s.Printf(f.Constraints.GoString())
|
||
|
|
}
|
||
|
|
|
||
|
|
s.NL()
|
||
|
|
s.Printf("package %s\n", s.cfg.Pkg)
|
||
|
|
for _, fn := range f.Functions() {
|
||
|
|
s.NL()
|
||
|
|
s.Comment(fn.Doc...)
|
||
|
|
s.Printf("%s\n", fn.Stub())
|
||
|
|
}
|
||
|
|
return s.Result()
|
||
|
|
}
|