mirror of
https://github.com/grassrootseconomics/cic-custodial.git
synced 2024-12-23 02:07:32 +01:00
Mohammed Sohail
4d13a14dc2
Squashed commit of the following: commit05e1396121
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Wed Feb 15 10:03:44 2023 +0300 feat: add status types to dispatcher commit397cd78ca9
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Wed Feb 15 09:39:31 2023 +0300 deps: bump -> cic-celo-sdk commitf2ba079232
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Sun Feb 12 16:53:53 2023 +0300 snapshot: 12-ebening commit4f7909e4ee
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Sun Feb 12 12:50:43 2023 +0300 xnapshot: 12-02 commit773474cad9
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Thu Feb 9 14:23:37 2023 +0300 update: deps initializers commit8a0880fcfc
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Thu Feb 9 10:42:15 2023 +0300 wip: refactor taskers commit8676450122
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Fri Feb 3 12:29:27 2023 +0300 refactor: decouple sql queries, remove transfer * add inline docs * removed transfer taks in prep for re-write commitb4c09cd11a
Author: Mohammed Sohail <sohailsameja@gmail.com> Date: Thu Feb 2 12:29:43 2023 +0000 refactor: cmd/service/* and api
557 lines
12 KiB
Go
557 lines
12 KiB
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/big"
|
|
|
|
"github.com/bsm/redislock"
|
|
"github.com/celo-org/celo-blockchain/common/hexutil"
|
|
celo "github.com/grassrootseconomics/cic-celo-sdk"
|
|
"github.com/grassrootseconomics/cic-custodial/internal/nonce"
|
|
"github.com/grassrootseconomics/cic-custodial/internal/store"
|
|
"github.com/grassrootseconomics/cic-custodial/internal/tasker"
|
|
"github.com/grassrootseconomics/w3-celo-patch"
|
|
"github.com/grassrootseconomics/w3-celo-patch/module/eth"
|
|
"github.com/hibiken/asynq"
|
|
"github.com/nats-io/nats.go"
|
|
)
|
|
|
|
type (
|
|
AccountPayload struct {
|
|
PublicKey string `json:"publicKey"`
|
|
TrackingId string `json:"trackingId"`
|
|
}
|
|
|
|
accountEventPayload struct {
|
|
TrackingId string `json:"trackingId"`
|
|
}
|
|
)
|
|
|
|
func PrepareAccount(
|
|
noncestore nonce.Noncestore,
|
|
taskerClient *tasker.TaskerClient,
|
|
js nats.JetStreamContext,
|
|
) func(context.Context, *asynq.Task) error {
|
|
return func(ctx context.Context, t *asynq.Task) error {
|
|
var (
|
|
p AccountPayload
|
|
)
|
|
|
|
if err := json.Unmarshal(t.Payload(), &p); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := noncestore.SetNewAccountNonce(ctx, p.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err := taskerClient.CreateTask(
|
|
tasker.RegisterAccountOnChain,
|
|
tasker.DefaultPriority,
|
|
&tasker.Task{
|
|
Payload: t.Payload(),
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.GiftTokenTask,
|
|
tasker.DefaultPriority,
|
|
&tasker.Task{
|
|
Payload: t.Payload(),
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
eventPayload := &accountEventPayload{
|
|
TrackingId: p.TrackingId,
|
|
}
|
|
|
|
eventJson, err := json.Marshal(eventPayload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = js.Publish("CUSTODIAL.accountNewNonce", eventJson)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func RegisterAccountOnChainProcessor(
|
|
celoProvider *celo.Provider,
|
|
lockProvider *redislock.Client,
|
|
noncestore nonce.Noncestore,
|
|
pg store.Store,
|
|
system *tasker.SystemContainer,
|
|
taskerClient *tasker.TaskerClient,
|
|
js nats.JetStreamContext,
|
|
) func(context.Context, *asynq.Task) error {
|
|
return func(ctx context.Context, t *asynq.Task) error {
|
|
var (
|
|
p AccountPayload
|
|
)
|
|
|
|
if err := json.Unmarshal(t.Payload(), &p); err != nil {
|
|
return err
|
|
}
|
|
|
|
lock, err := lockProvider.Obtain(ctx, system.LockPrefix+system.PublicKey, system.LockTimeout, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer lock.Release(ctx)
|
|
|
|
nonce, err := noncestore.Acquire(ctx, system.PublicKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
input, err := system.Abis["add"].EncodeArgs(w3.A(p.PublicKey))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// TODO: Review gas params.
|
|
builtTx, err := celoProvider.SignContractExecutionTx(
|
|
system.PrivateKey,
|
|
celo.ContractExecutionTxOpts{
|
|
ContractAddress: system.AccountIndexContract,
|
|
InputData: input,
|
|
GasPrice: big.NewInt(20000000000),
|
|
GasLimit: system.TokenTransferGasLimit,
|
|
Nonce: nonce,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
rawTx, err := builtTx.MarshalBinary()
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
id, err := pg.CreateOTX(ctx, store.OTX{
|
|
TrackingId: p.TrackingId,
|
|
Type: "ACCOUNT_REGISTER",
|
|
RawTx: hexutil.Encode(rawTx),
|
|
TxHash: builtTx.Hash().Hex(),
|
|
From: system.PublicKey,
|
|
Data: hexutil.Encode(builtTx.Data()),
|
|
GasPrice: builtTx.GasPrice().Uint64(),
|
|
Nonce: builtTx.Nonce(),
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
disptachJobPayload, err := json.Marshal(TxPayload{
|
|
OtxId: id,
|
|
Tx: builtTx,
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.TxDispatchTask,
|
|
tasker.HighPriority,
|
|
&tasker.Task{
|
|
Payload: disptachJobPayload,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.GiftGasTask,
|
|
tasker.DefaultPriority,
|
|
&tasker.Task{
|
|
Payload: t.Payload(),
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
eventPayload := &accountEventPayload{
|
|
TrackingId: p.TrackingId,
|
|
}
|
|
|
|
eventJson, err := json.Marshal(eventPayload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = js.Publish("CUSTODIAL.accountRegister", eventJson)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func GiftGasProcessor(
|
|
celoProvider *celo.Provider,
|
|
lockProvider *redislock.Client,
|
|
noncestore nonce.Noncestore,
|
|
pg store.Store,
|
|
system *tasker.SystemContainer,
|
|
taskerClient *tasker.TaskerClient,
|
|
js nats.JetStreamContext,
|
|
) func(context.Context, *asynq.Task) error {
|
|
return func(ctx context.Context, t *asynq.Task) error {
|
|
var (
|
|
p AccountPayload
|
|
)
|
|
|
|
if err := json.Unmarshal(t.Payload(), &p); err != nil {
|
|
return err
|
|
}
|
|
|
|
lock, err := lockProvider.Obtain(ctx, system.LockPrefix+system.PublicKey, system.LockTimeout, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer lock.Release(ctx)
|
|
|
|
nonce, err := noncestore.Acquire(ctx, system.PublicKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// TODO: Review gas params
|
|
builtTx, err := celoProvider.SignGasTransferTx(
|
|
system.PrivateKey,
|
|
celo.GasTransferTxOpts{
|
|
To: w3.A(p.PublicKey),
|
|
Nonce: nonce,
|
|
Value: system.GiftableGasValue,
|
|
GasPrice: celo.FixedMinGas,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
rawTx, err := builtTx.MarshalBinary()
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
id, err := pg.CreateOTX(ctx, store.OTX{
|
|
TrackingId: p.TrackingId,
|
|
Type: "GIFT_GAS",
|
|
RawTx: hexutil.Encode(rawTx),
|
|
TxHash: builtTx.Hash().Hex(),
|
|
From: system.PublicKey,
|
|
Data: hexutil.Encode(builtTx.Data()),
|
|
GasPrice: builtTx.GasPrice().Uint64(),
|
|
Nonce: builtTx.Nonce(),
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
disptachJobPayload, err := json.Marshal(TxPayload{
|
|
OtxId: id,
|
|
Tx: builtTx,
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.TxDispatchTask,
|
|
tasker.HighPriority,
|
|
&tasker.Task{
|
|
Payload: disptachJobPayload,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
eventPayload := &accountEventPayload{
|
|
TrackingId: p.TrackingId,
|
|
}
|
|
|
|
eventJson, err := json.Marshal(eventPayload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = js.Publish("CUSTODIAL.giftNewAccountGas", eventJson)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func GiftTokenProcessor(
|
|
celoProvider *celo.Provider,
|
|
lockProvider *redislock.Client,
|
|
noncestore nonce.Noncestore,
|
|
pg store.Store,
|
|
system *tasker.SystemContainer,
|
|
taskerClient *tasker.TaskerClient,
|
|
js nats.JetStreamContext,
|
|
) func(context.Context, *asynq.Task) error {
|
|
return func(ctx context.Context, t *asynq.Task) error {
|
|
var (
|
|
p AccountPayload
|
|
)
|
|
|
|
if err := json.Unmarshal(t.Payload(), &p); err != nil {
|
|
return err
|
|
}
|
|
|
|
lock, err := lockProvider.Obtain(ctx, system.LockPrefix+system.PublicKey, system.LockTimeout, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer lock.Release(ctx)
|
|
|
|
nonce, err := noncestore.Acquire(ctx, system.PublicKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
input, err := system.Abis["mintTo"].EncodeArgs(w3.A(p.PublicKey), system.GiftableTokenValue)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// TODO: Review gas params.
|
|
builtTx, err := celoProvider.SignContractExecutionTx(
|
|
system.PrivateKey,
|
|
celo.ContractExecutionTxOpts{
|
|
ContractAddress: system.GiftableToken,
|
|
InputData: input,
|
|
GasPrice: big.NewInt(20000000000),
|
|
GasLimit: system.TokenTransferGasLimit,
|
|
Nonce: nonce,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
return err
|
|
}
|
|
|
|
rawTx, err := builtTx.MarshalBinary()
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
id, err := pg.CreateOTX(ctx, store.OTX{
|
|
TrackingId: p.TrackingId,
|
|
Type: "GIFT_VOUCHER",
|
|
RawTx: hexutil.Encode(rawTx),
|
|
TxHash: builtTx.Hash().Hex(),
|
|
From: system.PublicKey,
|
|
Data: hexutil.Encode(builtTx.Data()),
|
|
GasPrice: builtTx.GasPrice().Uint64(),
|
|
Nonce: builtTx.Nonce(),
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
disptachJobPayload, err := json.Marshal(TxPayload{
|
|
OtxId: id,
|
|
Tx: builtTx,
|
|
})
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.TxDispatchTask,
|
|
tasker.HighPriority,
|
|
&tasker.Task{
|
|
Payload: disptachJobPayload,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
eventPayload := &accountEventPayload{
|
|
TrackingId: p.TrackingId,
|
|
}
|
|
|
|
eventJson, err := json.Marshal(eventPayload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = js.Publish("CUSTODIAL.giftNewAccountVoucher", eventJson)
|
|
if err != nil {
|
|
if err := noncestore.Return(ctx, system.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// TODO: https://github.com/grassrootseconomics/cic-custodial/issues/43
|
|
// TODO:
|
|
func RefillGasProcessor(
|
|
celoProvider *celo.Provider,
|
|
nonceProvider nonce.Noncestore,
|
|
lockProvider *redislock.Client,
|
|
system *tasker.SystemContainer,
|
|
taskerClient *tasker.TaskerClient,
|
|
) func(context.Context, *asynq.Task) error {
|
|
return func(ctx context.Context, t *asynq.Task) error {
|
|
var (
|
|
p AccountPayload
|
|
balance big.Int
|
|
)
|
|
if err := json.Unmarshal(t.Payload(), &p); err != nil {
|
|
return fmt.Errorf("json.Unmarshal failed: %v: %w", err, asynq.SkipRetry)
|
|
}
|
|
|
|
if err := celoProvider.Client.CallCtx(
|
|
ctx,
|
|
eth.Balance(w3.A(p.PublicKey), nil).Returns(&balance),
|
|
); err != nil {
|
|
return err
|
|
}
|
|
|
|
if belowThreshold := balance.Cmp(system.GasRefillThreshold); belowThreshold > 0 {
|
|
return nil
|
|
}
|
|
|
|
lock, err := lockProvider.Obtain(ctx, system.LockPrefix+system.PublicKey, system.LockTimeout, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer lock.Release(ctx)
|
|
|
|
nonce, err := nonceProvider.Acquire(ctx, system.PublicKey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
builtTx, err := celoProvider.SignGasTransferTx(
|
|
system.PrivateKey,
|
|
celo.GasTransferTxOpts{
|
|
To: w3.A(p.PublicKey),
|
|
Nonce: nonce,
|
|
Value: system.GasRefillValue,
|
|
GasPrice: celo.FixedMinGas,
|
|
},
|
|
)
|
|
if err != nil {
|
|
if err := nonceProvider.Return(ctx, p.PublicKey); err != nil {
|
|
return err
|
|
}
|
|
return fmt.Errorf("nonce.Return failed: %v: %w", err, asynq.SkipRetry)
|
|
}
|
|
|
|
disptachJobPayload, err := json.Marshal(TxPayload{
|
|
Tx: builtTx,
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("json.Marshal failed: %v: %w", err, asynq.SkipRetry)
|
|
}
|
|
|
|
_, err = taskerClient.CreateTask(
|
|
tasker.TxDispatchTask,
|
|
tasker.HighPriority,
|
|
&tasker.Task{
|
|
Payload: disptachJobPayload,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|