package client
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
kzgceremony "github.com/arnaucube/eth-kzg-ceremony-alt"
|
|
)
|
|
|
|
type Client struct {
|
|
url string
|
|
c *http.Client
|
|
}
|
|
|
|
func (c *Client) postWithAuth(url, contentType string, body io.Reader, bearer string) (resp *http.Response, err error) {
|
|
req, err := http.NewRequest("POST", url, body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Content-Type", contentType)
|
|
req.Header.Add("Authorization", bearer)
|
|
return c.c.Do(req)
|
|
}
|
|
|
|
func NewClient(sequencerURL string) *Client {
|
|
httpClient := &http.Client{}
|
|
return &Client{
|
|
url: sequencerURL,
|
|
c: httpClient,
|
|
}
|
|
}
|
|
|
|
func (c *Client) GetCurrentStatus() (*MsgStatus, error) {
|
|
resp, err := c.c.Get(
|
|
c.url + "/info/status")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var msg MsgStatus
|
|
err = json.Unmarshal(body, &msg)
|
|
return &msg, err
|
|
}
|
|
|
|
func (c *Client) GetCurrentState() (*kzgceremony.State, error) {
|
|
resp, err := c.c.Get(
|
|
c.url + "/info/current_state")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var state *kzgceremony.State
|
|
err = json.Unmarshal(body, &state)
|
|
return state, err
|
|
}
|
|
|
|
func (c *Client) GetRequestLink() (*MsgRequestLink, error) {
|
|
resp, err := c.c.Get(
|
|
c.url + "/auth/request_link")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
switch resp.StatusCode {
|
|
case http.StatusBadRequest:
|
|
return nil, fmt.Errorf("Invalid request. Missing parameters.")
|
|
case http.StatusUnauthorized:
|
|
return nil, fmt.Errorf("Eth address doesn't match message signer, or account nonce is too low")
|
|
case http.StatusForbidden:
|
|
return nil, fmt.Errorf("Invalid HTTP method")
|
|
default:
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var msg MsgRequestLink
|
|
err = json.Unmarshal(body, &msg)
|
|
return &msg, err
|
|
}
|
|
|
|
func (c *Client) PostAuthCallback() (*MsgRequestLink, error) {
|
|
resp, err := c.c.Get(
|
|
c.url + "/auth/request_link")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
switch resp.StatusCode {
|
|
case http.StatusBadRequest:
|
|
return nil, fmt.Errorf("Invalid request. Missing parameters.")
|
|
case http.StatusUnauthorized:
|
|
return nil, fmt.Errorf("Eth address doesn't match message signer, or account nonce is too low")
|
|
case http.StatusForbidden:
|
|
return nil, fmt.Errorf("Invalid HTTP method")
|
|
default:
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var msg MsgRequestLink
|
|
err = json.Unmarshal(body, &msg)
|
|
return &msg, err
|
|
}
|
|
|
|
func (c *Client) PostTryContribute(sessionID string) (*kzgceremony.BatchContribution, bool, error) {
|
|
bearer := "Bearer " + sessionID
|
|
resp, err := c.postWithAuth(
|
|
c.url+"/lobby/try_contribute", "application/json", nil, bearer)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
fmt.Println(string(body))
|
|
switch resp.StatusCode {
|
|
case http.StatusBadRequest:
|
|
return nil, true, fmt.Errorf("call came to early. rate limited")
|
|
case http.StatusUnauthorized:
|
|
return nil, false, fmt.Errorf("unkown session id. unauthorized access")
|
|
default:
|
|
return nil, false, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
err = ioutil.WriteFile("prevBatchContribution.json", body, 0600)
|
|
if err != nil {
|
|
return nil, false, err
|
|
}
|
|
bc := &kzgceremony.BatchContribution{}
|
|
err = json.Unmarshal(body, bc)
|
|
return bc, false, err
|
|
}
|
|
|
|
func (c *Client) PostAbortContribution(sessionID string) ([]byte, error) {
|
|
bearer := "Bearer " + sessionID
|
|
resp, err := c.postWithAuth(
|
|
c.url+"/contribution/abort", "application/json", nil, bearer)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
fmt.Println("body", string(body))
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
switch resp.StatusCode {
|
|
case http.StatusBadRequest:
|
|
return nil, fmt.Errorf("invalid request")
|
|
case http.StatusUnauthorized:
|
|
return nil, fmt.Errorf("unkown session id. unauthorized access")
|
|
default:
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
return body, nil
|
|
}
|
|
|
|
func (c *Client) PostContribute(sessionID string, bc *kzgceremony.BatchContribution) (*MsgContributeReceipt, error) {
|
|
bearer := "Bearer " + sessionID
|
|
|
|
jsonBC, err := json.Marshal(bc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
resp, err := c.postWithAuth(
|
|
c.url+"/contribute", "application/json", bytes.NewBuffer(jsonBC), bearer)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
switch resp.StatusCode {
|
|
case http.StatusBadRequest:
|
|
return nil, fmt.Errorf("invalid request")
|
|
default:
|
|
return nil, fmt.Errorf("unexpected http code: %d", resp.StatusCode)
|
|
}
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var msg MsgContributeReceipt
|
|
err = json.Unmarshal(body, &msg)
|
|
return &msg, err
|
|
}
|