chore: add support for vcr compression and trim k8s cassettes

cmd/vcr-compressor/main.go

@@ -0,0 +1,105 @@
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"os"
+
+	"github.com/scaleway/scaleway-sdk-go/api/k8s/v1"
+	"gopkg.in/dnaeon/go-vcr.v3/cassette"
+)
+
+var transientStates = map[string]bool{
+	k8s.ClusterStatusCreating.String(): true,
+	k8s.ClusterStatusDeleting.String(): true,
+	k8s.ClusterStatusUpdating.String(): true,
+	k8s.PoolStatusDeleting.String():    true,
+	k8s.PoolStatusScaling.String():     true,
+	k8s.PoolStatusUpgrading.String():   true,
+}
+
+func main() {
+	if len(os.Args) < 2 {
+		log.Fatalf("Usage: %s <cassette_file_name_without_yaml>\n", os.Args[0])
+	}
+
+	chemin := os.Args[1]
+
+	inputCassette, err := cassette.Load(chemin)
+	if err != nil {
+		log.Fatalf("Error while reading file : %v\n", err)
+	}
+
+	outputCassette := cassette.New(chemin)
+	transitioning := false
+
+	for i := range len(inputCassette.Interactions) {
+		interaction := inputCassette.Interactions[i]
+		responseBody := interaction.Response.Body
+		requestMethod := interaction.Request.Method
+
+		if requestMethod != "GET" {
+			transitioning = false
+
+			log.Printf("Interaction %d is not a GET request. Recording it\n", i)
+			outputCassette.AddInteraction(interaction)
+
+			continue
+		}
+
+		if responseBody == "" {
+			log.Printf("Interaction %d got an empty response body. Recording it\n", i)
+			outputCassette.AddInteraction(interaction)
+
+			continue
+		}
+
+		var m map[string]interface{}
+
+		err := json.Unmarshal([]byte(responseBody), &m)
+		if err != nil {
+			log.Printf("Interaction %d have an error with unmarshalling response body: %v. Recording it\n", i, err)
+			outputCassette.AddInteraction(interaction)
+
+			continue
+		}
+
+		if m["status"] == nil {
+			log.Printf("Interaction %d does not contain a status field. Recording it\n", i)
+			outputCassette.AddInteraction(interaction)
+
+			continue
+		}
+
+		if m["status"] != nil {
+			// We test if the state is transient
+			if _, ok := transientStates[m["status"].(string)]; ok {
+				if transitioning {
+					log.Printf("Interaction %d is in a transient state while we are already in transitient state. No need to record it: %s\n", i, m["status"])
+
+					continue
+				} else {
+					transitioning = true
+
+					log.Printf("Interaction %d is in a transient state: %s, Recording it\n", i, m["status"])
+					outputCassette.AddInteraction(interaction)
+				}
+			} else {
+				if transitioning {
+					log.Printf("Interaction %d is not in a transient state anymore: %s, Recording it\n", i, m["status"])
+					outputCassette.AddInteraction(interaction)
+
+					transitioning = false
+				} else {
+					log.Printf("Interaction %d is not in a transient state: %s, Recording it\n", i, m["status"])
+					outputCassette.AddInteraction(interaction)
+				}
+			}
+		}
+	}
+
+	err = outputCassette.Save()
+	if err != nil {
+		log.Fatalf("error while saving file: %v", err)
+	}
+}

cmd/vcr-viewer/main.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"encoding/json"
+	"log"
+	"os"
+
+	"gopkg.in/dnaeon/go-vcr.v3/cassette"
+)
+
+func main() {
+	if len(os.Args) < 2 {
+		log.Fatalf("Usage: %s <cassette_file_name_without_yaml>\n", os.Args[0])
+	}
+
+	chemin := os.Args[1]
+
+	data, err := cassette.Load(chemin)
+	if err != nil {
+		log.Fatalf("Error while reading file: %v\n", err)
+	}
+
+	for i := range len(data.Interactions) {
+		interaction := data.Interactions[i]
+
+		log.Println("--------------")
+		log.Printf("Interaction %d:\n", i+1)
+		log.Printf("  Request:\n")
+		log.Printf("    Method: %s\n", interaction.Request.Method)
+		log.Printf("    URL: %s\n", interaction.Request.URL)
+
+		if interaction.Request.Body != "" {
+			log.Printf("    Body: %s\n", interaction.Request.Body)
+		}
+
+		log.Printf("  Response:\n")
+		log.Printf("    Status: %s\n", interaction.Response.Status)
+		log.Printf("    Body: %s\n", interaction.Response.Body)
+
+		var m map[string]interface{}
+
+		err := json.Unmarshal([]byte(interaction.Response.Body), &m)
+		if err != nil {
+			continue
+		}
+
+		if m["status"] != nil {
+			log.Println("++++++++++++++++")
+			log.Printf("status: %s\n", m["status"]) // Modifie le champ "status" pour qu'il soit "ok"
+			log.Println("++++++++++++++++")
+		}
+
+		log.Println("--------------")
+	}
+}