NFT Module

go.mod

@@ -9,7 +9,6 @@ require (
 	github.com/fortytw2/leaktest v1.3.0 // indirect
 	github.com/gogo/protobuf v1.2.1
 	github.com/golang/mock v1.3.1-0.20190508161146-9fa652df1129
-	github.com/golang/protobuf v1.3.0
 	github.com/gorilla/mux v1.7.0
 	github.com/gorilla/websocket v1.4.0 // indirect
 	github.com/mattn/go-isatty v0.0.6

go.sum

@@ -215,6 +215,7 @@ golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/tools v0.0.0-20180221164845-07fd8470d635/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
 golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20190425150028-36563e24a262 h1:qsl9y/CJx34tuA7QCPNp86JNJe4spst6Ff8MjvPUdPg=
 golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
 google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
 google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=

simapp/app.go

@@ -23,6 +23,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/x/genutil"
 	"github.com/cosmos/cosmos-sdk/x/gov"
 	"github.com/cosmos/cosmos-sdk/x/mint"
+	"github.com/cosmos/cosmos-sdk/x/nft"
 	"github.com/cosmos/cosmos-sdk/x/params"
 	paramsclient "github.com/cosmos/cosmos-sdk/x/params/client"
 	"github.com/cosmos/cosmos-sdk/x/slashing"
@@ -54,6 +55,7 @@ var (
 		params.AppModuleBasic{},
 		crisis.AppModuleBasic{},
 		slashing.AppModuleBasic{},
+		nft.AppModuleBasic{},
 		supply.AppModuleBasic{},
 	)
 )
@@ -87,6 +89,7 @@ type SimApp struct {
 	keyGov      *sdk.KVStoreKey
 	keyParams   *sdk.KVStoreKey
 	tkeyParams  *sdk.TransientStoreKey
+	keyNFT      *sdk.KVStoreKey
 
 	// keepers
 	accountKeeper  auth.AccountKeeper
@@ -99,6 +102,7 @@ type SimApp struct {
 	govKeeper      gov.Keeper
 	crisisKeeper   crisis.Keeper
 	paramsKeeper   params.Keeper
+	nftKeeper      nft.Keeper
 
 	// the module manager
 	mm *module.Manager
@@ -130,6 +134,7 @@ func NewSimApp(logger log.Logger, db dbm.DB, traceStore io.Writer, loadLatest bo
 		keyGov:         sdk.NewKVStoreKey(gov.StoreKey),
 		keyParams:      sdk.NewKVStoreKey(params.StoreKey),
 		tkeyParams:     sdk.NewTransientStoreKey(params.TStoreKey),
+		keyNFT:         sdk.NewKVStoreKey(nft.StoreKey),
 	}
 
 	// init params keeper and subspaces
@@ -165,6 +170,7 @@ func NewSimApp(logger log.Logger, db dbm.DB, traceStore io.Writer, loadLatest bo
 	app.slashingKeeper = slashing.NewKeeper(app.cdc, app.keySlashing, &stakingKeeper,
 		slashingSubspace, slashing.DefaultCodespace)
 	app.crisisKeeper = crisis.NewKeeper(crisisSubspace, invCheckPeriod, app.supplyKeeper, auth.FeeCollectorName)
+	app.nftKeeper = nft.NewKeeper(app.cdc, app.keyNFT)
 
 	// register the proposal types
 	govRouter := gov.NewRouter()
@@ -191,6 +197,7 @@ func NewSimApp(logger log.Logger, db dbm.DB, traceStore io.Writer, loadLatest bo
 		mint.NewAppModule(app.mintKeeper),
 		slashing.NewAppModule(app.slashingKeeper, app.stakingKeeper),
 		staking.NewAppModule(app.stakingKeeper, app.distrKeeper, app.accountKeeper, app.supplyKeeper),
+		nft.NewAppModule(app.nftKeeper),
 	)
 
 	// During begin block slashing happens after distr.BeginBlocker so that
@@ -212,7 +219,7 @@ func NewSimApp(logger log.Logger, db dbm.DB, traceStore io.Writer, loadLatest bo
 	// initialize stores
 	app.MountStores(app.keyMain, app.keyAccount, app.keySupply, app.keyStaking,
 		app.keyMint, app.keyDistr, app.keySlashing, app.keyGov, app.keyParams,
-		app.tkeyParams, app.tkeyStaking, app.tkeyDistr)
+		app.tkeyParams, app.tkeyStaking, app.tkeyDistr, app.keyNFT)
 
 	// initialize BaseApp
 	app.SetInitChainer(app.InitChainer)

simapp/params.go

@@ -20,4 +20,6 @@ const (
 	OpWeightMsgUndelegate                              = "op_weight_msg_undelegate"
 	OpWeightMsgBeginRedelegate                         = "op_weight_msg_begin_redelegate"
 	OpWeightMsgUnjail                                  = "op_weight_msg_unjail"
+	OpWeightMsgTransferNFT                             = "op_weight_msg_transfer_nft"
+	OpWeightMsgEditNFTMetadata                         = "op_weight_msg_edit_nft_metadata"
 )

simapp/sim_test.go

@@ -27,6 +27,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/x/simulation"
 	slashingsim "github.com/cosmos/cosmos-sdk/x/slashing/simulation"
 	"github.com/cosmos/cosmos-sdk/x/staking"
+	"github.com/cosmos/cosmos-sdk/x/nft"
 	stakingsim "github.com/cosmos/cosmos-sdk/x/staking/simulation"
 )
 
@@ -126,6 +127,7 @@ func appStateRandomizedFn(
 	GenDistrGenesisState(cdc, r, appParams, genesisState)
 	stakingGen := GenStakingGenesisState(cdc, r, accs, amount, numAccs, numInitiallyBonded, appParams, genesisState)
 	GenSlashingGenesisState(cdc, r, stakingGen, appParams, genesisState)
+	GenNFTGenesisState(cdc, r, accs, appParams, genesisState)
 
 	appState, err := MakeCodec().MarshalJSON(genesisState)
 	if err != nil {
@@ -325,6 +327,28 @@ func testAndRunTxs(app *SimApp) []simulation.WeightedOperation {
 			}(nil),
 			slashingsim.SimulateMsgUnjail(app.slashingKeeper),
 		},
+		{
+			func(_ *rand.Rand) int {
+				var v int
+				ap.GetOrGenerate(cdc, OpWeightMsgTransferNFT, &v, nil,
+					func(_ *rand.Rand) {
+						v = 100
+					})
+				return v
+			}(nil),
+			nft.SimulateMsgEditNFTMetadata(app.nftKeeper),
+		},
+		{
+			func(_ *rand.Rand) int {
+				var v int
+				ap.GetOrGenerate(cdc, OpWeightMsgTransferNFT, &v, nil,
+					func(_ *rand.Rand) {
+						v = 100
+					})
+				return v
+			}(nil),
+			nft.SimulateMsgTransferNFT(app.nftKeeper),
+		},
 	}
 }
 

simapp/utils.go

@@ -30,6 +30,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/x/genaccounts"
 	"github.com/cosmos/cosmos-sdk/x/gov"
 	"github.com/cosmos/cosmos-sdk/x/mint"
+	"github.com/cosmos/cosmos-sdk/x/nft"
 	"github.com/cosmos/cosmos-sdk/x/simulation"
 	"github.com/cosmos/cosmos-sdk/x/slashing"
 	"github.com/cosmos/cosmos-sdk/x/staking"
@@ -494,6 +495,45 @@ func GenStakingGenesisState(
 	return stakingGenesis
 }
 
+func GenNFTGenesisState(cdc *codec.Codec, r *rand.Rand, accs []simulation.Account, ap simulation.AppParams, genesisState map[string]json.RawMessage) {
+	const (
+		Kitties = "crypto-kitties"
+		Doggos  = "crypto-doggos"
+	)
+
+	collections := nft.NewCollections(nft.NewCollection(Kitties, nft.NFTs{}), nft.NewCollection(Doggos, nft.NFTs{}))
+	var ownerships []nft.Owner
+
+	for _, acc := range accs {
+		if r.Intn(100) < 50 {
+			baseNFT := nft.NewBaseNFT(
+				simulation.RandStringOfLength(r, 10), // id
+				acc.Address,
+				simulation.RandStringOfLength(r, 15), // name
+				simulation.RandStringOfLength(r, 50), // description
+				simulation.RandStringOfLength(r, 30), // image
+				simulation.RandStringOfLength(r, 45), // tokenURI
+			)
+
+			var idCollection nft.IDCollection
+			if r.Intn(100) < 50 {
+				collections[0].AddNFT(&baseNFT)
+				idCollection = nft.NewIDCollection(Kitties, []string{baseNFT.ID})
+			} else {
+				collections[1].AddNFT(&baseNFT)
+				idCollection = nft.NewIDCollection(Doggos, []string{baseNFT.ID})
+			}
+			ownership := nft.NewOwner(acc.Address, idCollection)
+			ownerships = append(ownerships, ownership)
+		}
+	}
+
+	nftGenesis := nft.NewGenesisState(ownerships, collections)
+
+	fmt.Printf("Selected randomly generated NFT parameters:\n%s\n", codec.MustMarshalJSONIndent(cdc, nftGenesis))
+	genesisState[nft.ModuleName] = cdc.MustMarshalJSON(nftGenesis)
+}
+
 // GetSimulationLog unmarshals the KVPair's Value to the corresponding type based on the
 // each's module store key and the prefix bytes of the KVPair's key.
 func GetSimulationLog(storeName string, cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) (log string) {

types/address.go

@@ -8,7 +8,7 @@ import (
 	"fmt"
 	"strings"
 
-	"gopkg.in/yaml.v2"
+	yaml "gopkg.in/yaml.v2"
 
 	"github.com/tendermint/tendermint/crypto"
 	cryptoAmino "github.com/tendermint/tendermint/crypto/encoding/amino"

x/gov/invariants.go

@@ -24,7 +24,7 @@ func AllInvariants(keeper Keeper) sdk.Invariant {
 func ModuleAccountInvariant(keeper Keeper) sdk.Invariant {
 	return func(ctx sdk.Context) (string, bool) {
 		var expectedDeposits sdk.Coins
-
+		
 		keeper.IterateAllDeposits(ctx, func(deposit types.Deposit) bool {
 			expectedDeposits = expectedDeposits.Add(deposit.Amount)
 			return false

x/nft/alias.go

@@ -0,0 +1,54 @@
+// nolint
+// autogenerated code using github.com/rigelrozanski/multitool
+// aliases generated for the following subdirectories:
+// ALIASGEN: github.com/cosmos/cosmos-sdk/x/nft/internal/keeper
+// ALIASGEN: github.com/cosmos/cosmos-sdk/x/nft/internal/types
+package nft
+
+import (
+	"github.com/cosmos/cosmos-sdk/x/nft/internal/keeper"
+	"github.com/cosmos/cosmos-sdk/x/nft/internal/types"
+)
+
+type (
+	Keeper             = keeper.Keeper
+	NFT                = types.NFT
+	BaseNFT            = types.BaseNFT
+	NFTs               = types.NFTs
+	Owner              = types.Owner
+	IDCollection       = types.IDCollection
+	Collection         = types.Collection
+	Collections        = types.Collections
+	GenesisState       = types.GenesisState
+	MsgTransferNFT     = types.MsgTransferNFT
+	MsgEditNFTMetadata = types.MsgEditNFTMetadata
+	MsgMintNFT         = types.MsgMintNFT
+	MsgBurnNFT         = types.MsgBurnNFT
+)
+
+var (
+	NewKeeper             = keeper.NewKeeper
+	RegisterInvariants    = keeper.RegisterInvariants
+	NewQuerier            = keeper.NewQuerier
+	NewBaseNFT            = types.NewBaseNFT
+	NewNFTs               = types.NewNFTs
+	NewIDCollection       = types.NewIDCollection
+	NewOwner              = types.NewOwner
+	NewCollection         = types.NewCollection
+	NewCollections        = types.NewCollections
+	EmptyCollection       = types.EmptyCollection
+	NewGenesisState       = types.NewGenesisState
+	NewMsgTransferNFT     = types.NewMsgTransferNFT
+	NewMsgEditNFTMetadata = types.NewMsgEditNFTMetadata
+	DefaultGenesisState   = types.DefaultGenesisState
+	ValidateGenesis       = types.ValidateGenesis
+	RegisterCodec         = types.RegisterCodec
+	ModuleCdc             = types.ModuleCdc
+)
+
+const (
+	StoreKey     = types.StoreKey
+	RouterKey    = types.RouterKey
+	QuerierRoute = types.QuerierRoute
+	ModuleName   = types.ModuleName
+)