Add --aws-redirect-host cli arg to allow using Iamlive with Localstack

iamlivecore/proxy.go

@@ -165,7 +165,7 @@ func dumpReq(req *http.Request) {
 	fmt.Printf("%v\n", string(dump))
 }
 
-func createProxy(addr string) {
+func createProxy(addr string, awsRedirectHost string) {
 	err := loadCAKeys()
 	if err != nil {
 		log.Fatal(err)
@@ -187,6 +187,11 @@ func createProxy(addr string) {
 			}
 			body, _ = ioutil.ReadAll(req.Body)
 			handleAWSRequest(req, body, 200)
+
+			if awsRedirectHost != "" {
+				req.URL.Host = awsRedirectHost
+				req.Host = awsRedirectHost
+			}
 		} else if isAzureHostname && *providerFlag == "azure" {
 			if *debugFlag {
 				dumpReq(req)

iamlivecore/service.go

@@ -31,6 +31,7 @@ var debugFlag *bool
 var forceWildcardResourceFlag *bool
 var cpuProfileFlag = flag.String("cpu-profile", "", "write a CPU profile to this file (for performance testing purposes)")
 var csmPortFlag *int
+var awsRedirectHostFlag *string
 
 func parseConfig() {
 	provider := "aws"
@@ -50,6 +51,7 @@ func parseConfig() {
 	debug := false
 	forceWildcardResource := false
 	csmPort := 31000
+	awsRedirectHost := ""
 
 	cfgfile, err := homedir.Expand("~/.iamlive/config")
 	if err == nil {
@@ -103,6 +105,10 @@ func parseConfig() {
 			if cfg.Section("").HasKey("force-wildcard-resource") {
 				forceWildcardResource, _ = cfg.Section("").Key("force-wildcard-resource").Bool()
 			}
+			if cfg.Section("").HasKey("aws-redirect-host") {
+				awsRedirectHost = cfg.Section("").Key("aws-redirect-host").String()
+			}
+
 		}
 	}
 
@@ -123,6 +129,7 @@ func parseConfig() {
 	debugFlag = flag.Bool("debug", debug, "dumps associated HTTP requests when set in proxy mode")
 	forceWildcardResourceFlag = flag.Bool("force-wildcard-resource", forceWildcardResource, "when set, the Resource will always be a wildcard")
 	csmPortFlag = flag.Int("csm-port", csmPort, "port to listen on for CSM")
+	awsRedirectHostFlag = flag.String("aws-redirect-host", awsRedirectHost, "redirect all AWS API calls to this endpoint")
 }
 
 func Run() {
@@ -172,13 +179,13 @@ func Run() {
 		handleLoggedCall()
 	} else if *modeFlag == "proxy" {
 		readServiceFiles()
-		createProxy(*bindAddrFlag)
+		createProxy(*bindAddrFlag, *awsRedirectHostFlag)
 	} else {
 		fmt.Println("ERROR: unknown mode")
 	}
 }
 
-func RunWithArgs(provider string, setIni bool, profile string, failsOnly bool, outputFile string, refreshRate int, sortAlphabetical bool, host, mode, bindAddr, caBundle, caKey, accountID string, background, debug, forceWildcardResource bool) {
+func RunWithArgs(provider string, setIni bool, profile string, failsOnly bool, outputFile string, refreshRate int, sortAlphabetical bool, host, mode, bindAddr, caBundle, caKey, accountID string, background, debug, forceWildcardResource bool, awsRedirectHost string) {
 	providerFlag = &provider
 	setiniFlag = &setIni
 	profileFlag = &profile
@@ -195,6 +202,7 @@ func RunWithArgs(provider string, setIni bool, profile string, failsOnly bool, o
 	backgroundFlag = &background
 	debugFlag = &debug
 	forceWildcardResourceFlag = &forceWildcardResource
+	awsRedirectHostFlag = &awsRedirectHost
 
 	if *cpuProfileFlag != "" {
 		f, err := os.Create(*cpuProfileFlag)
@@ -220,7 +228,7 @@ func RunWithArgs(provider string, setIni bool, profile string, failsOnly bool, o
 		handleLoggedCall()
 	} else if *modeFlag == "proxy" {
 		readServiceFiles()
-		createProxy(*bindAddrFlag)
+		createProxy(*bindAddrFlag, *awsRedirectHostFlag)
 	} else {
 		fmt.Println("ERROR: unknown mode")
 	}