From d6b2519843ffe12657b8b9c2d7f5fd89cb57223c Mon Sep 17 00:00:00 2001
From: laser <l@s3r.me>
Date: Thu, 11 Jun 2020 11:29:59 -0700
Subject: [PATCH] config-driven IsAcceptingStorageDeals flag

Makes incremental progress towards #1920.
---
 node/builder.go              |  1 +
 node/config/def.go           |  7 ++++++-
 node/modules/dtypes/miner.go |  4 ++++
 node/modules/storageminer.go | 34 ++++++++++++++++++++++++++++++++--
 4 files changed, 43 insertions(+), 3 deletions(-)

diff --git a/node/builder.go b/node/builder.go
index f503bff5c..adbb25efe 100644
--- a/node/builder.go
+++ b/node/builder.go
@@ -313,6 +313,7 @@ func Online() Option {
 			Override(HandleDealsKey, modules.HandleDeals),
 			Override(new(gen.WinningPoStProver), storage.NewWinningPoStProver),
 			Override(new(*miner.Miner), modules.SetupBlockProducer),
+			Override(new(dtypes.IsAcceptingStorageDealsFunc), modules.NewIsAcceptingStorageDealsFunc),
 		),
 	)
 }
diff --git a/node/config/def.go b/node/config/def.go
index 9ca97bb47..a1b03b899 100644
--- a/node/config/def.go
+++ b/node/config/def.go
@@ -27,7 +27,12 @@ type FullNode struct {
 type StorageMiner struct {
 	Common
 
-	Storage sectorstorage.SealerConfig
+	StorageDeals StorageDealConfig
+	Storage      sectorstorage.SealerConfig
+}
+
+type StorageDealConfig struct {
+	IsAcceptingStorageDeals bool
 }
 
 // API contains configs for API endpoint
diff --git a/node/modules/dtypes/miner.go b/node/modules/dtypes/miner.go
index c872fdf69..9a3fcc2cd 100644
--- a/node/modules/dtypes/miner.go
+++ b/node/modules/dtypes/miner.go
@@ -7,3 +7,7 @@ import (
 
 type MinerAddress address.Address
 type MinerID abi.ActorID
+
+// IsAcceptingStorageDealsFunc is a function which reads from miner config to
+// determine if the user has disabled storage deals (or not).
+type IsAcceptingStorageDealsFunc func() (bool, error)
diff --git a/node/modules/storageminer.go b/node/modules/storageminer.go
index 1d358628b..4fd1443d1 100644
--- a/node/modules/storageminer.go
+++ b/node/modules/storageminer.go
@@ -35,6 +35,7 @@ import (
 	paramfetch "github.com/filecoin-project/go-paramfetch"
 	"github.com/filecoin-project/go-statestore"
 	"github.com/filecoin-project/go-storedcounter"
+	"github.com/filecoin-project/lotus/node/config"
 	sectorstorage "github.com/filecoin-project/sector-storage"
 	"github.com/filecoin-project/sector-storage/ffiwrapper"
 	"github.com/filecoin-project/sector-storage/stores"
@@ -304,14 +305,27 @@ func NewStorageAsk(ctx helpers.MetricsCtx, fapi lapi.FullNode, ds dtypes.Metadat
 	return storedAsk, nil
 }
 
-func StorageProvider(minerAddress dtypes.MinerAddress, ffiConfig *ffiwrapper.Config, storedAsk *storedask.StoredAsk, h host.Host, ds dtypes.MetadataDS, ibs dtypes.StagingBlockstore, r repo.LockedRepo, pieceStore dtypes.ProviderPieceStore, dataTransfer dtypes.ProviderDataTransfer, spn storagemarket.StorageProviderNode) (storagemarket.StorageProvider, error) {
+func StorageProvider(minerAddress dtypes.MinerAddress, ffiConfig *ffiwrapper.Config, storedAsk *storedask.StoredAsk, h host.Host, ds dtypes.MetadataDS, ibs dtypes.StagingBlockstore, r repo.LockedRepo, pieceStore dtypes.ProviderPieceStore, dataTransfer dtypes.ProviderDataTransfer, spn storagemarket.StorageProviderNode, isAcceptingStorageDealsFunc dtypes.IsAcceptingStorageDealsFunc) (storagemarket.StorageProvider, error) {
 	net := smnet.NewFromLibp2pHost(h)
 	store, err := piecefilestore.NewLocalFileStore(piecefilestore.OsPath(r.Path()))
 	if err != nil {
 		return nil, err
 	}
 
-	p, err := storageimpl.NewProvider(net, namespace.Wrap(ds, datastore.NewKey("/deals/provider")), ibs, store, pieceStore, dataTransfer, spn, address.Address(minerAddress), ffiConfig.SealProofType, storedAsk)
+	opt := storageimpl.CustomDealDecisionLogic(func(ctx context.Context, deal storagemarket.MinerDeal) (bool, string, error) {
+		willEntertainProposals, err := isAcceptingStorageDealsFunc()
+		if err != nil {
+			return false, "IsAcceptingStorageDealsFunc error", err
+		}
+
+		if !willEntertainProposals {
+			return false, "user has disabled storage deals", nil
+		}
+
+		return true, "", nil
+	})
+
+	p, err := storageimpl.NewProvider(net, namespace.Wrap(ds, datastore.NewKey("/deals/provider")), ibs, store, pieceStore, dataTransfer, spn, address.Address(minerAddress), ffiConfig.SealProofType, storedAsk, opt)
 	if err != nil {
 		return p, err
 	}
@@ -361,3 +375,19 @@ func StorageAuth(ctx helpers.MetricsCtx, ca lapi.Common) (sectorstorage.StorageA
 	headers.Add("Authorization", "Bearer "+string(token))
 	return sectorstorage.StorageAuth(headers), nil
 }
+
+func NewIsAcceptingStorageDealsFunc(r repo.LockedRepo) (dtypes.IsAcceptingStorageDealsFunc, error) {
+	return func() (bool, error) {
+		raw, err := r.Config()
+		if err != nil {
+			return false, err
+		}
+
+		cfg, ok := raw.(*config.StorageMiner)
+		if !ok {
+			return false, xerrors.New("expected address of config.StorageMiner")
+		}
+
+		return cfg.StorageDeals.IsAcceptingStorageDeals, nil
+	}, nil
+}