mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 18:15:19 +00:00
86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
|
package lcd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/gorilla/mux"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
"github.com/tendermint/tmlibs/log"
|
||
|
|
|
||
|
|
tmserver "github.com/tendermint/tendermint/rpc/lib/server"
|
||
|
|
cmn "github.com/tendermint/tmlibs/common"
|
||
|
|
|
||
|
|
client "github.com/cosmos/cosmos-sdk/client"
|
||
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
||
|
|
keys "github.com/cosmos/cosmos-sdk/client/keys"
|
||
|
|
rpc "github.com/cosmos/cosmos-sdk/client/rpc"
|
||
|
|
tx "github.com/cosmos/cosmos-sdk/client/tx"
|
||
|
|
version "github.com/cosmos/cosmos-sdk/version"
|
||
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
||
|
|
auth "github.com/cosmos/cosmos-sdk/x/auth/client/rest"
|
||
|
|
bank "github.com/cosmos/cosmos-sdk/x/bank/client/rest"
|
||
|
|
//ibc "github.com/cosmos/cosmos-sdk/x/ibc/client/rest"
|
||
|
|
//stake "github.com/cosmos/cosmos-sdk/x/stake/client/rest"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ServeCommand will generate a long-running rest server
|
||
|
|
// (aka Light Client Daemon) that exposes functionality similar
|
||
|
|
// to the cli, but over rest
|
||
|
|
func ServeCommand(cdc *wire.Codec) *cobra.Command {
|
||
|
|
flagListenAddr := "laddr"
|
||
|
|
flagCORS := "cors"
|
||
|
|
|
||
|
|
cmd := &cobra.Command{
|
||
|
|
Use: "rest-server",
|
||
|
|
Short: "Start LCD (light-client daemon), a local REST server",
|
||
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||
|
|
listenAddr := viper.GetString(flagListenAddr)
|
||
|
|
handler := createHandler(cdc)
|
||
|
|
logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)).
|
||
|
|
With("module", "rest-server")
|
||
|
|
listener, err := tmserver.StartHTTPServer(listenAddr, handler, logger)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
logger.Info("REST server started")
|
||
|
|
|
||
|
|
// Wait forever and cleanup
|
||
|
|
cmn.TrapSignal(func() {
|
||
|
|
err := listener.Close()
|
||
|
|
logger.Error("Error closing listener", "err", err)
|
||
|
|
})
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
}
|
||
|
|
cmd.Flags().StringP(flagListenAddr, "a", "tcp://localhost:1317", "Address for server to listen on")
|
||
|
|
cmd.Flags().String(flagCORS, "", "Set to domains that can make CORS requests (* for all)")
|
||
|
|
cmd.Flags().StringP(client.FlagChainID, "c", "", "ID of chain we connect to")
|
||
|
|
cmd.Flags().StringP(client.FlagNode, "n", "tcp://localhost:46657", "Node to connect to")
|
||
|
|
return cmd
|
||
|
|
}
|
||
|
|
|
||
|
|
func createHandler(cdc *wire.Codec) http.Handler {
|
||
|
|
r := mux.NewRouter()
|
||
|
|
r.HandleFunc("/version", version.RequestHandler).Methods("GET")
|
||
|
|
|
||
|
|
kb, err := keys.GetKeyBase() //XXX
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := context.NewCoreContextFromViper()
|
||
|
|
|
||
|
|
// TODO make more functional? aka r = keys.RegisterRoutes(r)
|
||
|
|
keys.RegisterRoutes(r)
|
||
|
|
rpc.RegisterRoutes(ctx, r)
|
||
|
|
tx.RegisterRoutes(ctx, r, cdc)
|
||
|
|
auth.RegisterRoutes(ctx, r, cdc, "acc")
|
||
|
|
bank.RegisterRoutes(ctx, r, cdc, kb)
|
||
|
|
//ibc.RegisterRoutes(ctx, r, cdc, kb)
|
||
|
|
//stake.RegisterRoutes(ctx, r, cdc, kb)
|
||
|
|
r.HandleFunc("/decode_tx", DecodeTxRequestHandlerFn(ctx, cdc)).Methods("POST")
|
||
|
|
return r
|
||
|
|
}
|