gwc - handles workspace-based paths (geoserver#380)

src/apps/geoserver/gwc/pom.xml

@@ -49,6 +49,19 @@
       <artifactId>spring-boot-starter-test</artifactId>
       <scope>test</scope>
     </dependency>
+    <dependency>
+      <groupId>org.geoserver.cloud.catalog</groupId>
+      <artifactId>gs-cloud-catalog-plugin</artifactId>
+      <version>${project.version}</version>
+      <type>test-jar</type>
+      <scope>test</scope>
+      <exclusions>
+        <exclusion>
+          <groupId>*</groupId>
+          <artifactId>*</artifactId>
+        </exclusion>
+      </exclusions>
+    </dependency>
   </dependencies>
   <build>
     <plugins>

src/apps/geoserver/gwc/src/test/java/org/geoserver/cloud/gwc/app/GeoWebCacheApplicationTest.java

@@ -5,29 +5,57 @@
 package org.geoserver.cloud.gwc.app;
 
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.springframework.http.MediaType.APPLICATION_JSON;
-import static org.springframework.http.MediaType.APPLICATION_XML;
+import static org.springframework.http.MediaType.*;
 
 import com.google.gson.JsonElement;
 import com.google.gson.JsonParser;
 
+import org.geoserver.catalog.GeoServerCatalogTestData;
+import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.web.servlet.WebMvcRegistrations;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
 import org.springframework.boot.test.web.client.TestRestTemplate;
+import org.springframework.context.ApplicationContext;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.test.context.ActiveProfiles;
+import org.springframework.test.context.DynamicPropertyRegistry;
+import org.springframework.test.context.DynamicPropertySource;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URISyntaxException;
+import java.nio.file.Path;
 
 @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
 @ActiveProfiles("test")
 class GeoWebCacheApplicationTest {
 
     @Autowired private TestRestTemplate restTemplate;
 
+    @Autowired private ApplicationContext context;
+
+    @TempDir private static Path tmpPath;
+
+    @BeforeAll
+    static void initializeDatadir() throws URISyntaxException, IOException {
+        assertThat(tmpPath).isNotNull();
+        GeoServerCatalogTestData.unzipGeoserverCatalogTestData(tmpPath);
+    }
+
+    @DynamicPropertySource
+    static void dynamicProperties(DynamicPropertyRegistry registry)
+            throws URISyntaxException, IOException {
+        File tmpFilePath = tmpPath.toFile();
+        registry.add("geoserver.backend.data-directory.location", tmpFilePath::toString);
+    }
+
     @BeforeEach
     void before() {
         restTemplate = restTemplate.withBasicAuth("admin", "geoserver");
@@ -53,10 +81,86 @@ void testRESTPathExtensionContentNegotiation() {
         testGetRequestContentType("/gwc/rest/layers.xml", APPLICATION_XML);
     }
 
+    @Test
+    void testGwcUrlHandlerMappingArePresentInTheClasspath() {
+        assertThat(context.isTypeMatch("gwcDemoUrlHandlerMapping", WebMvcRegistrations.class))
+                .as("expected a bean gwcDemoUrlHandlerMapping of type GwcUrlHandlerMapping")
+                .isTrue();
+        assertThat(context.isTypeMatch("gwcRestWebUrlHandlerMapping", WebMvcRegistrations.class))
+                .as("expected a bean gwcRestWebUrlHandlerMapping of type GwcUrlHandlerMapping")
+                .isTrue();
+    }
+
+    @Test
+    void testGeneralGwcHome() {
+        ResponseEntity<String> response = testGetRequestContentType("/gwc/home", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGeneralGwcDemo() {
+        ResponseEntity<String> response = testGetRequestContentType("/gwc/demo", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGeneralGwcDemoWithNameSpaceLayer() {
+        ResponseEntity<String> response =
+                testGetRequestContentType("/gwc/demo/ne:boundary_lines", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGeneralGwcServiceWmts() {
+        ResponseEntity<String> response =
+                testGetRequestContentType("/gwc/service/wmts?REQUEST=GetCapabilities", TEXT_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGwcHomePrefixedUrls() {
+        ResponseEntity<String> response = testGetRequestContentType("/ne/gwc/home", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGwcDemoPrefixedUrls() {
+        ResponseEntity<String> response = testGetRequestContentType("/ne/gwc/demo", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGwcDemoWorkspacePrefixedUrlsLayer() {
+        ResponseEntity<String> response =
+                testGetRequestContentType("/ne/gwc/demo/boundary_lines", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGwcDemoWorkspacePrefixedUrlsWorkspaceLayer() {
+        ResponseEntity<String> response =
+                testGetRequestContentType("/ne/gwc/demo/ne:boundary_lines", TEXT_HTML);
+        assertThat(response.getHeaders().getContentType()).isNotEqualTo(APPLICATION_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
+    @Test
+    void testGwcServicePrefixedUrl() {
+        ResponseEntity<String> response =
+                testGetRequestContentType("/ne/gwc/service/wmts?REQUEST=GetCapabilities", TEXT_XML);
+        assertThat(response.getBody()).doesNotContain("ows:Exception");
+    }
+
     protected ResponseEntity<String> testGetRequestContentType(String uri, MediaType expected) {
         ResponseEntity<String> response = restTemplate.getForEntity(uri, String.class);
         assertThat(response.getStatusCode()).isEqualTo(HttpStatus.OK);
-        assertThat(response.getHeaders().getContentType()).isEqualTo(expected);
+        assertThat(response.getHeaders().getContentType().isCompatibleWith(expected)).isTrue();
         return response;
     }
 }

src/catalog/plugin/pom.xml

@@ -116,6 +116,11 @@
       <groupId>org.geotools</groupId>
       <artifactId>gt-process-feature</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.geotools</groupId>
+      <artifactId>gt-sample-data</artifactId>
+      <version>${gt.version}</version>
+    </dependency>
     <dependency>
       <groupId>com.github.f4b6a3</groupId>
       <artifactId>ulid-creator</artifactId>

src/catalog/plugin/src/test/java/org/geoserver/catalog/GeoServerCatalogTestData.java

@@ -0,0 +1,51 @@
+/*
+ * (c) 2024 Open Source Geospatial Foundation - all rights reserved This code is licensed under the
+ * GPL 2.0 license, available at the root application directory.
+ */
+package org.geoserver.catalog;
+
+import lombok.experimental.UtilityClass;
+
+import org.apache.commons.io.FileUtils;
+import org.geotools.test.TestData;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URISyntaxException;
+import java.nio.file.Path;
+
+/**
+ * The purpose of this class is to provide a geoserver datadir which is not fake (it is actually a
+ * copy of the one from GeoServer upstream), which could be used for testing.
+ *
+ * <p>It copies a zip file into a directory - generally a temporary one - an unzip it, so that it is
+ * ready for use.
+ */
+@UtilityClass
+public class GeoServerCatalogTestData {
+
+    /**
+     * This method copies the zipped datadir into the Path object given as argument and unzip it at
+     * the same place.
+     *
+     * <p>It is the caller's responsability to clean up when the datadir is not needed anymore.
+     *
+     * <p>Note: we have to copy the resource into the directory first, because the method from
+     * GeoTools which is being used does not support zip URIs nested into jar files.
+     *
+     * @param tmpPath the temporary path where the datadir has to be unzipped to.
+     * @throws URISyntaxException
+     * @throws IOException
+     */
+    public static void unzipGeoserverCatalogTestData(Path tmpPath)
+            throws URISyntaxException, IOException {
+        InputStream zippedDatadir =
+                GeoServerCatalogTestData.class.getResourceAsStream("/test-data-directory.zip");
+        File tmpDir = tmpPath.toFile();
+        File destFile = new File(tmpDir, "test-data-directory.zip");
+        FileUtils.copyToFile(zippedDatadir, destFile);
+        TestData.unzip(destFile, tmpDir);
+        destFile.delete();
+    }
+}

src/gwc/autoconfigure/src/main/java/org/geoserver/cloud/autoconfigure/web/gwc/GeoWebCacheUIAutoConfiguration.java

@@ -6,15 +6,21 @@
 
 import lombok.extern.slf4j.Slf4j;
 
-import org.geoserver.cloud.autoconfigure.gwc.ConditionalOnGeoWebCacheRestConfigEnabled;
+import org.geoserver.catalog.Catalog;
 import org.geoserver.cloud.autoconfigure.gwc.ConditionalOnWebUIEnabled;
 import org.geoserver.cloud.gwc.config.core.GeoWebCacheConfigurationProperties;
+import org.geoserver.cloud.virtualservice.VirtualServiceVerifier;
+import org.geoserver.gwc.controller.GwcUrlHandlerMapping;
+import org.geoserver.ows.Dispatcher;
 import org.geowebcache.GeoWebCacheDispatcher;
 import org.geowebcache.rest.controller.ByteStreamController;
 import org.gwc.web.rest.GeoWebCacheController;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.boot.autoconfigure.AutoConfiguration;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
+import org.springframework.boot.autoconfigure.web.servlet.WebMvcRegistrations;
 import org.springframework.context.annotation.Bean;
+import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
 
 import javax.annotation.PostConstruct;
 
@@ -28,17 +34,63 @@ public class GeoWebCacheUIAutoConfiguration {
     }
 
     @Bean
-    GeoWebCacheController gwcController(GeoWebCacheDispatcher gwcDispatcher) {
-        return new GeoWebCacheController(gwcDispatcher);
+    GeoWebCacheController gwcController(
+            Dispatcher geoserverDispatcher,
+            GeoWebCacheDispatcher geoWebCacheDispatcher,
+            VirtualServiceVerifier verifier) {
+        return new GeoWebCacheController(geoserverDispatcher, geoWebCacheDispatcher, verifier);
     }
 
-    /**
-     * Provide a handler for static web resources if missing, for example, because {@link
-     * ConditionalOnGeoWebCacheRestConfigEnabled} is disabled
-     */
+    /** ConditionalOnGeoWebCacheRestConfigEnabled} is disabled */
     @Bean
     @ConditionalOnMissingBean(ByteStreamController.class)
     ByteStreamController byteStreamController() {
         return new ByteStreamController();
     }
+
+    @Bean
+    VirtualServiceVerifier virtualServiceVerifier(@Qualifier("rawCatalog") Catalog catalog) {
+        return new VirtualServiceVerifier(catalog);
+    }
+
+    /**
+     * GS's src/web/gwc/src/main/java/applicationContext.xml
+     * <!-- Used for workspace-based demo requests so the requests to GS stay workspace-based -->
+     * <bean id="gwcDemoUrlHandlerMapping"
+     * class="org.geoserver.gwc.controller.GwcUrlHandlerMapping"> <constructor-arg ref="catalog" />
+     * <constructor-arg value="/gwc/demo"/> <property name="alwaysUseFullPath" value="true" />
+     * <property name="order" value="10" /> </bean>
+     * <!-- Used for workspace-based web requests (i.e. for rest/web/openlayer/ol.js) -->
+     * <bean id="gwcRestWebUrlHandlerMapping"
+     * class="org.geoserver.gwc.controller.GwcUrlHandlerMapping"> <constructor-arg ref="catalog" />
+     * <constructor-arg type="java.lang.String" value="/gwc/rest/web"/> <property
+     * name="alwaysUseFullPath" value="true" /> <property name="order" value="10" /> </bean>
+     */
+    @Bean
+    WebMvcRegistrations gwcDemoUrlHandlerMapping(@Qualifier("rawCatalog") Catalog catalog) {
+        GwcUrlHandlerMapping handler = new GwcUrlHandlerMapping(catalog, "/gwc/demo");
+        handler.setAlwaysUseFullPath(true);
+        handler.setOrder(10);
+
+        return new WebMvcRegistrations() {
+            @Override
+            public RequestMappingHandlerMapping getRequestMappingHandlerMapping() {
+                return handler;
+            }
+        };
+    }
+
+    @Bean
+    WebMvcRegistrations gwcRestWebUrlHandlerMapping(@Qualifier("rawCatalog") Catalog catalog) {
+        GwcUrlHandlerMapping handler = new GwcUrlHandlerMapping(catalog, "/gwc/rest/web");
+        handler.setAlwaysUseFullPath(true);
+        handler.setOrder(10);
+
+        return new WebMvcRegistrations() {
+            @Override
+            public RequestMappingHandlerMapping getRequestMappingHandlerMapping() {
+                return handler;
+            }
+        };
+    }
 }

src/gwc/autoconfigure/src/test/java/org/geoserver/cloud/autoconfigure/gwc/web/gwc/GeoWebCacheUIAutoConfigurationTest.java

@@ -0,0 +1,42 @@
+/*
+ * (c) 2024 Open Source Geospatial Foundation - all rights reserved This code is licensed under the
+ * GPL 2.0 license, available at the root application directory.
+ */
+package org.geoserver.cloud.autoconfigure.gwc.web.gwc;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+
+import org.geoserver.cloud.autoconfigure.gwc.GeoWebCacheContextRunner;
+import org.geoserver.cloud.autoconfigure.web.gwc.GeoWebCacheUIAutoConfiguration;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
+import org.springframework.boot.autoconfigure.AutoConfigurations;
+import org.springframework.boot.test.context.runner.WebApplicationContextRunner;
+
+import java.io.File;
+
+public class GeoWebCacheUIAutoConfigurationTest {
+
+    WebApplicationContextRunner runner;
+
+    @TempDir File tmpDir;
+
+    @BeforeEach
+    void setUp() throws Exception {
+        runner =
+                GeoWebCacheContextRunner.newMinimalGeoWebCacheContextRunner(tmpDir)
+                        .withPropertyValues("gwc.web-ui=true")
+                        .withConfiguration(
+                                AutoConfigurations.of(GeoWebCacheUIAutoConfiguration.class));
+    }
+
+    @Test
+    void beansForLocalWorkspacePathsHandlingArePresent() {
+        runner.run(
+                context -> {
+                    assertNotNull(context.getBean("gwcDemoUrlHandlerMapping"));
+                    assertNotNull(context.getBean("gwcRestWebUrlHandlerMapping"));
+                });
+    }
+}

src/gwc/services/pom.xml

@@ -21,5 +21,9 @@
       <groupId>org.geoserver</groupId>
       <artifactId>gs-gwc-rest</artifactId>
     </dependency>
+    <dependency>
+      <groupId>org.geoserver.cloud</groupId>
+      <artifactId>gs-cloud-starter-webmvc</artifactId>
+    </dependency>
   </dependencies>
 </project>