wip actor wrapper codegen

chain/actors/agen/main.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"golang.org/x/xerrors"
+	"io/ioutil"
+	"path/filepath"
+	"text/template"
+)
+
+func main() {
+	if err := run(); err != nil {
+		fmt.Println(err)
+		return
+	}
+}
+
+func run() error {
+	versions := []int{0, 2, 3, 4}
+
+	versionImports := map[int]string{
+		0: "/",
+		2: "/v2/",
+		3: "/v3/",
+		4: "/v4/",
+	}
+
+	actors := map[string][]int{
+		"account": versions,
+	}
+
+	for act, versions := range actors {
+		actDir := filepath.Join("chain/actors/builtin", act)
+
+		{
+			af, err := ioutil.ReadFile(filepath.Join(actDir, "state.go.template"))
+			if err != nil {
+				return xerrors.Errorf("loading state adapter template: %w", err)
+			}
+
+			for _, version := range versions {
+				tpl := template.Must(template.New("").Funcs(template.FuncMap{}).Parse(string(af)))
+
+				var b bytes.Buffer
+
+				err := tpl.Execute(&b, map[string]interface{}{
+					"v": version,
+					"import": versionImports[version],
+				})
+				if err != nil {
+					return err
+				}
+
+				if err := ioutil.WriteFile(filepath.Join(actDir, fmt.Sprintf("v%d.go", version)), b.Bytes(), 0666); err != nil {
+					return err
+				}
+			}
+		}
+
+		{
+			af, err := ioutil.ReadFile(filepath.Join(actDir, "actor.go.template"))
+			if err != nil {
+				return xerrors.Errorf("loading actor template: %w", err)
+			}
+
+			tpl := template.Must(template.New("").Funcs(template.FuncMap{
+				"import": func(v int) string { return versionImports[v] },
+			}).Parse(string(af)))
+
+			var b bytes.Buffer
+
+			err = tpl.Execute(&b, map[string]interface{}{
+				"versions": versions,
+			})
+			if err != nil {
+				return err
+			}
+
+			if err := ioutil.WriteFile(filepath.Join(actDir, fmt.Sprintf("%s.go", act)), b.Bytes(), 0666); err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}

chain/actors/builtin/account/actor.go.template

@@ -0,0 +1,37 @@
+package account
+
+import (
+	"golang.org/x/xerrors"
+
+	"github.com/filecoin-project/go-address"
+	"github.com/filecoin-project/go-state-types/cbor"
+	"github.com/ipfs/go-cid"
+
+	"github.com/filecoin-project/lotus/chain/actors/adt"
+	"github.com/filecoin-project/lotus/chain/actors/builtin"
+	"github.com/filecoin-project/lotus/chain/types"
+{{range .versions}}
+	builtin{{.}} "github.com/filecoin-project/specs-actors{{import .}}actors/builtin"{{end}}
+)
+
+func init() {
+{{range .versions}}	builtin.RegisterActorState(builtin{{.}}.AccountActorCodeID, func(store adt.Store, root cid.Cid) (cbor.Marshaler, error) {
+		return load{{.}}(store, root)
+	})
+{{end}}}
+
+var Methods = builtin4.MethodsAccount
+
+func Load(store adt.Store, act *types.Actor) (State, error) {
+	switch act.Code {
+{{range .versions}}	case builtin{{.}}.AccountActorCodeID:
+		return load{{.}}(store, act.Head)
+{{end}}	}
+	return nil, xerrors.Errorf("unknown actor code %s", act.Code)
+}
+
+type State interface {
+	cbor.Marshaler
+
+	PubkeyAddress() (address.Address, error)
+}

chain/actors/builtin/account/state.go.template

@@ -0,0 +1,30 @@
+package account
+
+import (
+	"github.com/filecoin-project/go-address"
+	"github.com/ipfs/go-cid"
+
+	"github.com/filecoin-project/lotus/chain/actors/adt"
+
+	account{{.v}} "github.com/filecoin-project/specs-actors{{.import}}actors/builtin/account"
+)
+
+var _ State = (*state{{.v}})(nil)
+
+func load{{.v}}(store adt.Store, root cid.Cid) (State, error) {
+	out := state{{.v}}{store: store}
+	err := store.Get(store.Context(), root, &out)
+	if err != nil {
+		return nil, err
+	}
+	return &out, nil
+}
+
+type state{{.v}} struct {
+	account{{.v}}.State
+	store adt.Store
+}
+
+func (s *state{{.v}}) PubkeyAddress() (address.Address, error) {
+	return s.Address, nil
+}