status-im/status-go

package statusgo

import (
    "encoding/json"

    "github.com/status-im/status-go/account"
    "github.com/status-im/status-go/eth-node/keystore"
    "github.com/status-im/status-go/transactions"
)

const (
    codeUnknown int = iota
    // special codes
    codeFailedParseResponse
    codeFailedParseParams
    // account related codes
    codeErrNoAccountSelected
    codeErrInvalidTxSender
    codeErrDecrypt
)

var errToCodeMap = map[error]int{
    account.ErrNoAccountSelected:    codeErrNoAccountSelected,
    transactions.ErrInvalidTxSender: codeErrInvalidTxSender,
    keystore.ErrDecrypt:             codeErrDecrypt,
}

type jsonrpcSuccessfulResponse struct {
    Result interface{} `json:"result"`
}

type jsonrpcErrorResponse struct {
    Error jsonError `json:"error"`
}

type jsonError struct {
    Code    int    `json:"code,omitempty"`
    Message string `json:"message"`
}

func prepareJSONResponse(result interface{}, err error) string {
    code := codeUnknown
    if c, ok := errToCodeMap[err]; ok {
        code = c
    }

    return prepareJSONResponseWithCode(result, err, code)
}

func prepareJSONResponseWithCode(result interface{}, err error, code int) string {
    if err != nil {
        errResponse := jsonrpcErrorResponse{
            Error: jsonError{Code: code, Message: err.Error()},
        }
        response, _ := json.Marshal(&errResponse)
        return string(response)
    }

    data, err := json.Marshal(jsonrpcSuccessfulResponse{result})
    if err != nil {
        return prepareJSONResponseWithCode(nil, err, codeFailedParseResponse)
    }
    return string(data)
}