get case base name from caseName with extension (#55)

* Add function to get case base name from caseName (with extension) Signed-off-by: basseche <[email protected]>
powsybl · Dec 20, 2024 · 0bc7c88 · 0bc7c88
1 parent 4de78a9
commit 0bc7c88
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/src/main/java/com/powsybl/caseserver/CaseController.java b/src/main/java/com/powsybl/caseserver/CaseController.java
@@ -10,6 +10,7 @@
 import com.powsybl.caseserver.elasticsearch.CaseInfosService;
 import com.powsybl.caseserver.service.CaseService;
 import com.powsybl.caseserver.service.MetadataService;
+import com.powsybl.commons.datasource.DataSourceUtil;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.media.Content;
 import io.swagger.v3.oas.annotations.media.Schema;
@@ -224,4 +225,13 @@ public ResponseEntity<List<CaseInfos>> getMetadata(@RequestParam("ids") List<UUI
         LOGGER.debug("get Cases metadata");
         return ResponseEntity.ok().body(caseService.getMetadata(ids));
     }
+
+    @GetMapping("/cases/caseBaseName")
+    @Operation(summary = "Get case base name")
+    @ApiResponse(responseCode = "200", description = "case base name retrieved")
+    public ResponseEntity<String> getCaseBaseName(@RequestParam("caseName") String caseName) {
+        LOGGER.debug("getting base name from case file name");
+        String baseName = DataSourceUtil.getBaseName(caseName);
+        return ResponseEntity.ok(baseName);
+    }
 }
diff --git a/src/test/java/com/powsybl/caseserver/service/AbstractCaseControllerTest.java b/src/test/java/com/powsybl/caseserver/service/AbstractCaseControllerTest.java
@@ -739,4 +739,31 @@ void invalidFileInCaseDirectoryShouldBeIgnored() throws Exception {
                 .andExpect(status().isOk());
         assertNotNull(outputDestination.receive(1000, caseImportDestination));
     }
+
+    @Test
+    void testGetCaseBaseName() throws Exception {
+        MvcResult mvcResult = mvc.perform(get("/v1/cases/caseBaseName?caseName=case.xml"))
+                .andExpect(status().isOk())
+                .andReturn();
+        String response = mvcResult.getResponse().getContentAsString();
+        assertEquals("case", response);
+
+        mvcResult = mvc.perform(get("/v1/cases/caseBaseName?caseName=case.xml.gz"))
+                .andExpect(status().isOk())
+                .andReturn();
+        response = mvcResult.getResponse().getContentAsString();
+        assertEquals("case", response);
+
+        mvcResult = mvc.perform(get("/v1/cases/caseBaseName?caseName=case.v1.xml"))
+                .andExpect(status().isOk())
+                .andReturn();
+        response = mvcResult.getResponse().getContentAsString();
+        assertEquals("case.v1", response);
+
+        mvcResult = mvc.perform(get("/v1/cases/caseBaseName?caseName=case.v1.xml.gz"))
+                .andExpect(status().isOk())
+                .andReturn();
+        response = mvcResult.getResponse().getContentAsString();
+        assertEquals("case.v1", response);
+    }
 }