Redirect to unique domain

3 files changed, 179 insertions, 7 deletions

diff --git a/test/acceptance/uniqueDomain_test.go b/test/acceptance/uniqueDomain_test.go
new file mode 100644
index 00000000..1f423170
--- /dev/null
+++ b/test/acceptance/uniqueDomain_test.go
@@ -0,0 +1,97 @@
+package acceptance_test
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"gitlab.com/gitlab-org/gitlab-pages/internal/testhelpers"
+)
+
+func TestRedirectToUniqueDomain(t *testing.T) {
+	RunPagesProcess(t)
+
+	tests := []struct {
+		name          string
+		requestDomain string
+		requestPath   string
+		redirectURL   string
+		httpStatus    int
+	}{
+		{
+			name:          "when project has unique domain",
+			requestDomain: "group.unique-url.gitlab-example.com",
+			requestPath:   "with-unique-url/",
+			redirectURL:   "https://unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com/",
+			httpStatus:    http.StatusPermanentRedirect,
+		},
+		{
+			name:          "when requesting implicit index.html",
+			requestDomain: "group.unique-url.gitlab-example.com",
+			requestPath:   "with-unique-url",
+			redirectURL:   "https://unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			httpStatus:    http.StatusPermanentRedirect,
+		},
+		{
+			name:          "when project is nested",
+			requestDomain: "group.unique-url.gitlab-example.com",
+			requestPath:   "subgroup1/subgroup2/with-unique-url/subdir/index.html",
+			redirectURL:   "https://unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com/subdir/index.html",
+			httpStatus:    http.StatusPermanentRedirect,
+		},
+		{
+			name:          "when serving with a port",
+			requestDomain: "group.unique-url.gitlab-example.com:8080",
+			requestPath:   "with-unique-url/",
+			redirectURL:   "https://unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com:8080/",
+			httpStatus:    http.StatusPermanentRedirect,
+		},
+		{
+			name:          "when serving a path with a port",
+			requestDomain: "group.unique-url.gitlab-example.com:8080",
+			requestPath:   "with-unique-url/subdir/index.html",
+			redirectURL:   "https://unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com:8080/subdir/index.html",
+			httpStatus:    http.StatusPermanentRedirect,
+		},
+		{
+			name:          "when already serving the unique domain",
+			requestDomain: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			httpStatus:    http.StatusOK,
+		},
+		{
+			name:          "when already serving the unique domain with a port",
+			requestDomain: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com:8080",
+			httpStatus:    http.StatusOK,
+		},
+		{
+			name:          "when already serving the unique domain with path",
+			requestDomain: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			requestPath:   "subdir/index.html",
+			httpStatus:    http.StatusOK,
+		},
+		{
+			name:          "when project does not have unique domain with a path",
+			requestDomain: "group.unique-url.gitlab-example.com",
+			requestPath:   "without-unique-url/subdir/index.html",
+			httpStatus:    http.StatusOK,
+		},
+		{
+			name:          "when project does not exist",
+			requestDomain: "group.unique-url.gitlab-example.com",
+			requestPath:   "inexisting-project/",
+			httpStatus:    http.StatusNotFound,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			rsp, err := GetRedirectPage(t, httpsListener, test.requestDomain, test.requestPath)
+			require.NoError(t, err)
+			testhelpers.Close(t, rsp.Body)
+
+			require.Equal(t, test.httpStatus, rsp.StatusCode)
+			require.Equal(t, test.redirectURL, rsp.Header.Get("Location"))
+		})
+	}
+}
diff --git a/test/gitlabstub/api_responses.go b/test/gitlabstub/api_responses.go
index 93bc8995..b9459160 100644
--- a/test/gitlabstub/api_responses.go
+++ b/test/gitlabstub/api_responses.go
@@ -14,6 +14,15 @@ import (
 
 type responseFn func(string) api.VirtualDomain
 
+type projectConfig struct {
+	// refer to makeGitLabPagesAccessStub for custom HTTP responses per projectID
+	projectID     int
+	accessControl bool
+	https         bool
+	pathOnDisk    string
+	uniqueHost    string
+}
+
 // domainResponses holds the predefined API responses for certain domains
 // that can be used with the GitLab API stub in acceptance tests
 // Assume the working dir is inside shared/pages/
@@ -114,6 +123,43 @@ var domainResponses = map[string]responseFn{
 		projectID:  1008,
 		pathOnDisk: "group.acme/with.redirects",
 	}),
+	"group.unique-url.gitlab-example.com": generateVirtualDomain(map[string]projectConfig{
+		"/with-unique-url": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+		"/subgroup1/subgroup2/with-unique-url": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+		"/with-unique-url-with-port": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+		"/with-malformed-unique-url": {
+			uniqueHost: "unique-url@gitlab-example.com:",
+			pathOnDisk: "group/project",
+		},
+		"/with-different-protocol": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+		"/without-unique-url": {
+			pathOnDisk: "group/project",
+		},
+	}),
+	"unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com": generateVirtualDomain(map[string]projectConfig{
+		"/": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+	}),
+	"unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com:8080": generateVirtualDomain(map[string]projectConfig{
+		"/": {
+			uniqueHost: "unique-url-group-unique-url-a1b2c3d4e5f6.gitlab-example.com",
+			pathOnDisk: "group/project",
+		},
+	}),
 	// NOTE: before adding more domains here, generate the zip archive by running (per project)
 	// make zip PROJECT_SUBDIR=group/serving
 	// make zip PROJECT_SUBDIR=group/project2
@@ -187,6 +233,7 @@ func generateVirtualDomainFromDir(dir, rootDomain string, perPrefixConfig map[st
 					Path:   sourcePath,
 					SHA256: sha,
 				},
+				UniqueHost: cfg.uniqueHost,
 			}
 
 			lookupPaths = append(lookupPaths, lookupPath)
@@ -198,12 +245,39 @@ func generateVirtualDomainFromDir(dir, rootDomain string, perPrefixConfig map[st
 	}
 }
 
-type projectConfig struct {
-	// refer to makeGitLabPagesAccessStub for custom HTTP responses per projectID
-	projectID     int
-	accessControl bool
-	https         bool
-	pathOnDisk    string
+func generateVirtualDomain(projectConfigs map[string]projectConfig) responseFn {
+	return func(wd string) api.VirtualDomain {
+		nextID := 1000
+		lookupPaths := make([]api.LookupPath, 0, len(projectConfigs))
+
+		for project, config := range projectConfigs {
+			if config.projectID == 0 {
+				config.projectID = nextID
+				nextID++
+			}
+
+			sourcePath := fmt.Sprintf("file://%s/%s/public.zip", wd, config.pathOnDisk)
+			sum := sha256.Sum256([]byte(sourcePath))
+			sha := hex.EncodeToString(sum[:])
+
+			lookupPaths = append(lookupPaths, api.LookupPath{
+				ProjectID:     config.projectID,
+				AccessControl: config.accessControl,
+				HTTPSOnly:     config.https,
+				Prefix:        ensureEndingSlash(project),
+				UniqueHost:    config.uniqueHost,
+				Source: api.Source{
+					Type:   "zip",
+					Path:   sourcePath,
+					SHA256: sha,
+				},
+			})
+		}
+
+		return api.VirtualDomain{
+			LookupPaths: lookupPaths,
+		}
+	}
 }
 
 // customDomain with per project config
@@ -230,6 +304,7 @@ func customDomain(config projectConfig) responseFn {
 						SHA256: sha,
 						Path:   sourcePath,
 					},
+					UniqueHost: config.uniqueHost,
 				},
 			},
 		}
diff --git a/test/gitlabstub/handlers.go b/test/gitlabstub/handlers.go
index df09d914..8089277b 100644
--- a/test/gitlabstub/handlers.go
+++ b/test/gitlabstub/handlers.go
@@ -29,7 +29,7 @@ func defaultAPIHandler(delay time.Duration, pagesRoot string) http.HandlerFunc {
 		// check if predefined response exists
 		if responseFn, ok := domainResponses[domain]; ok {
 			if err := json.NewEncoder(w).Encode(responseFn(pagesRoot)); err != nil {
-				log.Fatal(err)
+				log.Fatalf("fail to encode response for domain %q: %v", domain, err)
 			}
 			return
 		}