71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package rest
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/mux"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
auth "github.com/cosmos/cosmos-sdk/x/auth/commands"
|
|
)
|
|
|
|
// register REST routes
|
|
func RegisterRoutes(r *mux.Router, cdc *wire.Codec, storeName string) {
|
|
r.HandleFunc(
|
|
"/accounts/{address}",
|
|
QueryAccountRequestHandler(storeName, cdc, auth.GetAccountDecoder(cdc)),
|
|
).Methods("GET")
|
|
}
|
|
|
|
// query accountREST Handler
|
|
func QueryAccountRequestHandler(storeName string, cdc *wire.Codec, decoder sdk.AccountDecoder) func(http.ResponseWriter, *http.Request) {
|
|
ctx := context.NewCoreContextFromViper()
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
addr := vars["address"]
|
|
|
|
bz, err := hex.DecodeString(addr)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
w.Write([]byte(err.Error()))
|
|
return
|
|
}
|
|
key := sdk.Address(bz)
|
|
|
|
res, err := ctx.Query(key, storeName)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(fmt.Sprintf("Could't query account. Error: %s", err.Error())))
|
|
return
|
|
}
|
|
|
|
// the query will return empty if there is no data for this account
|
|
if len(res) == 0 {
|
|
w.WriteHeader(http.StatusNoContent)
|
|
return
|
|
}
|
|
|
|
// decode the value
|
|
account, err := decoder(res)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(fmt.Sprintf("Could't parse query result. Result: %s. Error: %s", res, err.Error())))
|
|
return
|
|
}
|
|
|
|
// print out whole account
|
|
output, err := cdc.MarshalJSON(account)
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte(fmt.Sprintf("Could't marshall query result. Error: %s", err.Error())))
|
|
return
|
|
}
|
|
|
|
w.Write(output)
|
|
}
|
|
}
|