-
Notifications
You must be signed in to change notification settings - Fork 836
/
module.go
65 lines (55 loc) · 2.09 KB
/
module.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright 2022 Evmos Foundation
// This file is part of the Evmos Network packages.
//
// Evmos is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The Evmos packages are distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the Evmos packages. If not, see https://github.com/evmos/evmos/blob/main/LICENSE
package transfer
import (
"fmt"
"github.com/cosmos/cosmos-sdk/types/module"
ibctransfer "github.com/cosmos/ibc-go/v6/modules/apps/transfer"
ibctransferkeeper "github.com/cosmos/ibc-go/v6/modules/apps/transfer/keeper"
"github.com/cosmos/ibc-go/v6/modules/apps/transfer/types"
"github.com/evmos/evmos/v12/x/ibc/transfer/keeper"
)
var (
_ module.AppModule = AppModule{}
_ module.AppModuleBasic = AppModuleBasic{}
)
// AppModuleBasic embeds the IBC Transfer AppModuleBasic
type AppModuleBasic struct {
*ibctransfer.AppModuleBasic
}
// AppModule represents the AppModule for this module
type AppModule struct {
*ibctransfer.AppModule
keeper keeper.Keeper
}
// NewAppModule creates a new 20-transfer module
func NewAppModule(k keeper.Keeper) AppModule {
am := ibctransfer.NewAppModule(*k.Keeper)
return AppModule{
AppModule: &am,
keeper: k,
}
}
// RegisterServices registers module services.
func (am AppModule) RegisterServices(cfg module.Configurator) {
// Override Transfer Msg Server
types.RegisterMsgServer(cfg.MsgServer(), am.keeper)
types.RegisterQueryServer(cfg.QueryServer(), am.keeper)
m := ibctransferkeeper.NewMigrator(*am.keeper.Keeper)
if err := cfg.RegisterMigration(types.ModuleName, 1, m.MigrateTraces); err != nil {
panic(fmt.Sprintf("failed to migrate transfer app from version 1 to 2: %v", err))
}
}