From bc75ab511c32aa020b0424855fddbb39ba5a243b Mon Sep 17 00:00:00 2001 From: Steve Date: Wed, 23 Oct 2024 10:56:03 -0400 Subject: [PATCH] feat: Added hot swaps --- files.go | 143 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ main.go | 42 ++++++++++++++++ types.go | 18 +++++++ 3 files changed, 203 insertions(+) diff --git a/files.go b/files.go index 3d3977f..00967b2 100644 --- a/files.go +++ b/files.go @@ -213,3 +213,146 @@ func ListFiles(amount string, pageToken string, cidPending bool, name string, ci return response, nil } + +func GetSwapHistory(cid string, domain string) (GetSwapHistoryResponse, error) { + jwt, err := findToken() + if err != nil { + return GetSwapHistoryResponse{}, err + } + url := fmt.Sprintf("https://api.pinata.cloud/v3/files/swap/%s?", cid) + + params := []string{} + + if domain != "" { + params = append(params, "domain="+domain) + } else { + internalDomain, err := findGatewayDomain() + if err != nil { + return GetSwapHistoryResponse{}, err + } + params = append(params, "domain="+string(internalDomain)) + } + if len(params) > 0 { + url += strings.Join(params, "&") + } + + req, err := http.NewRequest("GET", url, nil) + + if err != nil { + return GetSwapHistoryResponse{}, errors.Join(err, errors.New("failed to create the request")) + } + req.Header.Set("Authorization", "Bearer "+string(jwt)) + req.Header.Set("content-type", "application/json") + + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return GetSwapHistoryResponse{}, errors.Join(err, errors.New("failed to send the request")) + } + defer resp.Body.Close() + + if resp.StatusCode != 200 { + return GetSwapHistoryResponse{}, fmt.Errorf("server Returned an error %d", resp.StatusCode) + } + + var response GetSwapHistoryResponse + + err = json.NewDecoder(resp.Body).Decode(&response) + if err != nil { + return GetSwapHistoryResponse{}, err + } + formattedJSON, err := json.MarshalIndent(response.Data, "", " ") + if err != nil { + return GetSwapHistoryResponse{}, errors.New("failed to format JSON") + } + + fmt.Println(string(formattedJSON)) + + return response, nil + +} + +func AddSwap(cid string, swapCid string) (AddSwapResponse, error) { + jwt, err := findToken() + if err != nil { + return AddSwapResponse{}, err + } + url := fmt.Sprintf("https://api.pinata.cloud/v3/files/swap/%s", cid) + + payload := AddSwapBody{ + SwapCid: swapCid, + } + + jsonPayload, err := json.Marshal(payload) + + if err != nil { + return AddSwapResponse{}, errors.Join(err, errors.New("Failed to marshal paylod")) + } + + req, err := http.NewRequest("PUT", url, bytes.NewBuffer(jsonPayload)) + + if err != nil { + return AddSwapResponse{}, errors.Join(err, errors.New("failed to create the request")) + } + req.Header.Set("Authorization", "Bearer "+string(jwt)) + req.Header.Set("content-type", "application/json") + + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return AddSwapResponse{}, errors.Join(err, errors.New("failed to send the request")) + } + defer resp.Body.Close() + + if resp.StatusCode != 200 { + return AddSwapResponse{}, fmt.Errorf("server Returned an error %d", resp.StatusCode) + } + + var response AddSwapResponse + + err = json.NewDecoder(resp.Body).Decode(&response) + if err != nil { + return AddSwapResponse{}, err + } + formattedJSON, err := json.MarshalIndent(response.Data, "", " ") + if err != nil { + return AddSwapResponse{}, errors.New("failed to format JSON") + } + + fmt.Println(string(formattedJSON)) + + return response, nil + +} + +func RemoveSwap(cid string) error { + jwt, err := findToken() + if err != nil { + return err + } + url := fmt.Sprintf("https://api.pinata.cloud/v3/files/swap/%s", cid) + + req, err := http.NewRequest("DELETE", url, nil) + + if err != nil { + return errors.Join(err, errors.New("failed to create the request")) + } + req.Header.Set("Authorization", "Bearer "+string(jwt)) + req.Header.Set("content-type", "application/json") + + client := &http.Client{} + resp, err := client.Do(req) + if err != nil { + return errors.Join(err, errors.New("failed to send the request")) + } + defer resp.Body.Close() + + if resp.StatusCode != 200 { + return fmt.Errorf("server Returned an error %d", resp.StatusCode) + } + + fmt.Println("Swap deleted") + + return nil + +} diff --git a/main.go b/main.go index ab6c798..f948e22 100644 --- a/main.go +++ b/main.go @@ -303,6 +303,48 @@ func main() { }, }, }, + { + Name: "swaps", + Aliases: []string{"s"}, + Usage: "Interact and manage hot swaps on Pinata", + Subcommands: []*cli.Command{ + { + Name: "list", + Aliases: []string{"l"}, + Usage: "List swaps for a given gateway domain or for your config gateway domain", + ArgsUsage: "[cid] [optional gateway domain]", + Action: func(ctx *cli.Context) error { + cid := ctx.Args().First() + domain := ctx.Args().Get(1) + _, err := GetSwapHistory(cid, domain) + return err + }, + }, + { + Name: "add", + Aliases: []string{"a"}, + Usage: "Add a swap for a CID", + ArgsUsage: "[cid] [swap cid]", + Action: func(ctx *cli.Context) error { + cid := ctx.Args().First() + swapCid := ctx.Args().Get(1) + _, err := AddSwap(cid, swapCid) + return err + }, + }, + { + Name: "delete", + Aliases: []string{"d"}, + Usage: "Remeove a swap for a CID", + ArgsUsage: "[cid]", + Action: func(ctx *cli.Context) error { + cid := ctx.Args().First() + err := RemoveSwap(cid) + return err + }, + }, + }, + }, { Name: "gateways", Aliases: []string{"gw"}, diff --git a/types.go b/types.go index 1ec8ec9..afa624d 100644 --- a/types.go +++ b/types.go @@ -97,3 +97,21 @@ type GetGatewaysResponse struct { Rows []GetGatewayItem } `json:"data"` } + +type GetSwapHistoryResponse struct { + Data []struct { + MappedCid string `json:"mapped_cid"` + CreatedAt string `json:"created_at"` + } `json:"data"` +} + +type AddSwapBody struct { + SwapCid string `json:"swap_cid"` +} + +type AddSwapResponse struct { + Data struct { + MappedCid string `json:"mapped_cid"` + CreatedAt string `json:"created_at"` + } `json:"data"` +}