mirror of
https://github.com/onsonr/sonr.git
synced 2025-03-10 21:09:11 +00:00
129 lines
4.5 KiB
Go
129 lines
4.5 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/onsonr/crypto/mpc"
|
|
didtypes "github.com/onsonr/sonr/x/did/types"
|
|
|
|
"github.com/onsonr/sonr/x/vault/types"
|
|
)
|
|
|
|
var _ types.QueryServer = Querier{}
|
|
|
|
type Querier struct{ Keeper }
|
|
|
|
func NewQuerier(keeper Keeper) Querier {
|
|
return Querier{Keeper: keeper}
|
|
}
|
|
|
|
// ╭───────────────────────────────────────────────────────────╮
|
|
// │ Fixed Query Methods │
|
|
// ╰───────────────────────────────────────────────────────────╯
|
|
|
|
// Params implements types.QueryServer.
|
|
func (k Querier) Params(c context.Context, req *types.QueryParamsRequest) (*types.QueryParamsResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
p, err := k.Keeper.Params.Get(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &types.QueryParamsResponse{Params: &p}, nil
|
|
}
|
|
|
|
// Schema implements types.QueryServer.
|
|
func (k Querier) Schema(goCtx context.Context, req *types.QuerySchemaRequest) (*types.QuerySchemaResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
|
|
p, err := k.Keeper.Params.Get(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &types.QuerySchemaResponse{
|
|
Schema: p.Schema,
|
|
}, nil
|
|
}
|
|
|
|
// ╭───────────────────────────────────────────────────────────╮
|
|
// │ Pre-Authenticated Queries │
|
|
// ╰───────────────────────────────────────────────────────────╯
|
|
|
|
// Allocate implements types.QueryServer.
|
|
func (k Querier) Allocate(goCtx context.Context, req *types.QueryAllocateRequest) (*types.QueryAllocateResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
|
|
// 1. Get current schema
|
|
sch, err := k.currentSchema(ctx)
|
|
if err != nil {
|
|
ctx.Logger().Error(fmt.Sprintf("Error getting current schema: %s", err.Error()))
|
|
return nil, types.ErrInvalidSchema.Wrap(err.Error())
|
|
}
|
|
|
|
// 2. Generate MPC Keyshares for new Account
|
|
shares, err := mpc.GenerateKeyshares()
|
|
if err != nil {
|
|
ctx.Logger().Error(fmt.Sprintf("Error generating keyshares: %s", err.Error()))
|
|
return nil, types.ErrInvalidSchema.Wrap(err.Error())
|
|
}
|
|
|
|
// 3. Create Controller from Keyshares
|
|
con, err := didtypes.NewController(shares)
|
|
if err != nil {
|
|
ctx.Logger().Error(fmt.Sprintf("Error creating controller: %s", err.Error()))
|
|
return nil, types.ErrControllerCreation.Wrap(err.Error())
|
|
}
|
|
|
|
// 4. Create a new vault PWA for service-worker
|
|
v, err := types.NewVault("", con.SonrAddress(), con.ChainID(), sch)
|
|
if err != nil {
|
|
ctx.Logger().Error(fmt.Sprintf("Error creating vault: %s", err.Error()))
|
|
return nil, types.ErrInvalidSchema.Wrap(err.Error())
|
|
}
|
|
|
|
// 5. Add to IPFS and Return CID for User Claims in Gateway
|
|
cid, err := k.ipfsClient.Unixfs().Add(context.Background(), v.FS)
|
|
if err != nil {
|
|
ctx.Logger().Error(fmt.Sprintf("Error adding to IPFS: %s", err.Error()))
|
|
return nil, types.ErrVaultAssembly.Wrap(err.Error())
|
|
}
|
|
|
|
// 6. Return final response
|
|
return &types.QueryAllocateResponse{
|
|
Success: true,
|
|
Cid: cid.String(),
|
|
ExpiryBlock: calculateBlockExpiry(ctx, time.Second*30),
|
|
}, nil
|
|
}
|
|
|
|
// ╭───────────────────────────────────────────────────────────╮
|
|
// │ Authenticated Endpoints │
|
|
// ╰───────────────────────────────────────────────────────────╯
|
|
|
|
// Sync implements types.QueryServer.
|
|
func (k Querier) Sync(goCtx context.Context, req *types.QuerySyncRequest) (*types.QuerySyncResponse, error) {
|
|
ctx := sdk.UnwrapSDKContext(goCtx)
|
|
p, err := k.Keeper.Params.Get(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
c, _ := k.DIDKeeper.ResolveController(ctx, req.Did)
|
|
if c == nil {
|
|
return &types.QuerySyncResponse{
|
|
Success: false,
|
|
Schema: p.Schema,
|
|
ChainID: ctx.ChainID(),
|
|
}, nil
|
|
}
|
|
return &types.QuerySyncResponse{
|
|
Success: true,
|
|
Schema: p.Schema,
|
|
ChainID: ctx.ChainID(),
|
|
Address: c.SonrAddress(),
|
|
}, nil
|
|
}
|