81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
|
package genesis
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
"github.com/filecoin-project/go-amt-ipld/v2"
|
||
|
"github.com/filecoin-project/specs-actors/actors/builtin/market"
|
||
|
"github.com/ipfs/go-cid"
|
||
|
bstore "github.com/ipfs/go-ipfs-blockstore"
|
||
|
cbor "github.com/ipfs/go-ipld-cbor"
|
||
|
cbg "github.com/whyrusleeping/cbor-gen"
|
||
|
"golang.org/x/xerrors"
|
||
|
|
||
|
"github.com/filecoin-project/lotus/chain/actors"
|
||
|
"github.com/filecoin-project/lotus/chain/state"
|
||
|
"github.com/filecoin-project/lotus/chain/store"
|
||
|
"github.com/filecoin-project/lotus/chain/types"
|
||
|
)
|
||
|
|
||
|
func SetupStorageMarketActor(bs bstore.Blockstore, sroot cid.Cid, deals []market.DealProposal) (cid.Cid, error) {
|
||
|
ctx := context.TODO()
|
||
|
cst := cbor.NewCborStore(bs)
|
||
|
ast := store.ActorStore(context.TODO(), bs)
|
||
|
|
||
|
cdeals := make([]cbg.CBORMarshaler, len(deals))
|
||
|
sdeals := make([]cbg.CBORMarshaler, len(deals))
|
||
|
for i, deal := range deals {
|
||
|
d := deal // copy
|
||
|
|
||
|
cdeals[i] = &d
|
||
|
sdeals[i] = &market.DealState{
|
||
|
SectorStartEpoch: 1,
|
||
|
LastUpdatedEpoch: -1,
|
||
|
SlashEpoch: -1,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
dealAmt, err := amt.FromArray(ctx, cst, cdeals)
|
||
|
if err != nil {
|
||
|
return cid.Undef, xerrors.Errorf("amt build failed: %w", err)
|
||
|
}
|
||
|
|
||
|
stateAmt, err := amt.FromArray(ctx, cst, sdeals)
|
||
|
if err != nil {
|
||
|
return cid.Undef, xerrors.Errorf("amt build failed: %w", err)
|
||
|
}
|
||
|
|
||
|
sms, err := market.ConstructState(ast)
|
||
|
if err != nil {
|
||
|
return cid.Cid{}, err
|
||
|
}
|
||
|
|
||
|
sms.Proposals = dealAmt
|
||
|
sms.States = stateAmt
|
||
|
|
||
|
stcid, err := cst.Put(context.TODO(), sms)
|
||
|
if err != nil {
|
||
|
return cid.Undef, err
|
||
|
}
|
||
|
|
||
|
// TODO: MARKET BALANCES!!!!!!111
|
||
|
|
||
|
act := &types.Actor{
|
||
|
Code: actors.StorageMarketCodeCid,
|
||
|
Head: stcid,
|
||
|
Nonce: 0,
|
||
|
Balance: types.NewInt(0),
|
||
|
}
|
||
|
|
||
|
state, err := state.LoadStateTree(cst, sroot)
|
||
|
if err != nil {
|
||
|
return cid.Undef, xerrors.Errorf("making new state tree: %w", err)
|
||
|
}
|
||
|
|
||
|
if err := state.SetActor(actors.StorageMarketAddress, act); err != nil {
|
||
|
return cid.Undef, xerrors.Errorf("set storage market actor: %w", err)
|
||
|
}
|
||
|
|
||
|
return state.Flush(context.TODO())
|
||
|
}
|