word

ita/internal/handlers.go

@@ -322,6 +322,15 @@ var Restore = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 
 })
 
+var RewriteAssets = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	err := ret_rewrite_assets(r.Header.Get("old_domain"), r.Header.Get("new_domain"))
+	if err != nil {
+		http.Error(w, "failed", http.StatusInternalServerError)
+		return
+	}
+	fmt.Fprintf(w, "done")
+})
+
 var Z_dummy = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 	fmt.Fprintf(w, "req: %v, rAdr: %v, xff: %v", w.Header().Get("X-Request-Id"), r.RemoteAddr, r.Header["X-Forwarded_for"])
 })

ita/internal/utils.go

@@ -3,6 +3,7 @@ package internal
 import (
 	"archive/tar"
 	"archive/zip"
+	"bytes"
 	"compress/gzip"
 	"context"
 	"crypto/tls"
@@ -12,6 +13,7 @@ import (
 	"hash/fnv"
 	"image"
 	"io"
+	"io/ioutil"
 	"math"
 	"net/http"
 	"os"
@@ -156,6 +158,21 @@ func getRetCcu() (int, error) {
 	return retCcuResp["count"], nil
 }
 
+func ret_rewrite_assets(oldDomain, newDomain string) error {
+	retCcuReq, _ := http.NewRequest("GET", "https://ret."+cfg.PodNS+":4000/api-internal/v1/rewrite_assets",
+		bytes.NewBuffer([]byte(
+			`{"old_domain":"`+oldDomain+`","new_domain":"`+newDomain+`"}`)))
+	retCcuReq.Header.Add("x-ret-dashboard-access-key", cfg.RetApiKey)
+
+	resp, err := _httpClient.Do(retCcuReq)
+
+	if err != nil {
+		body, _ := ioutil.ReadAll(resp.Body)
+		fmt.Println("resp", string(body))
+	}
+	return err
+}
+
 func waitRetCcu() error {
 	timeout := 6 * time.Hour
 	wait := 30 * time.Second