[학성] 지하철 정보 관리 - 인수 테스트 미션 제출합니다.

.gitignore

@@ -30,3 +30,4 @@ out/
 
 ### VS Code ###
 .vscode/
+.DS_Store

build.gradle

@@ -1,28 +1,29 @@
 plugins {
-	id 'org.springframework.boot' version '2.2.5.RELEASE'
-	id 'io.spring.dependency-management' version '1.0.9.RELEASE'
-	id 'java'
+    id 'org.springframework.boot' version '2.2.5.RELEASE'
+    id 'io.spring.dependency-management' version '1.0.9.RELEASE'
+    id 'java'
 }
 
 group = 'com.example'
 version = '0.0.1-SNAPSHOT'
 sourceCompatibility = '1.8'
 
 repositories {
-	mavenCentral()
+    mavenCentral()
 }
 
 dependencies {
-	implementation 'org.springframework.boot:spring-boot-starter-web'
-	implementation 'org.springframework.boot:spring-boot-starter-data-jdbc'
-	implementation 'net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1'
-	testImplementation 'io.rest-assured:rest-assured:3.3.0'
-	testImplementation('org.springframework.boot:spring-boot-starter-test') {
-		exclude group: 'org.junit.vintage', module: 'junit-vintage-engine'
-	}
-	runtimeOnly 'com.h2database:h2'
+    implementation 'org.springframework.boot:spring-boot-starter-web'
+    implementation 'org.springframework.boot:spring-boot-starter-data-jdbc'
+    implementation 'net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1'
+    implementation 'pl.allegro.tech.boot:handlebars-spring-boot-starter:0.3.1'
+    testImplementation 'io.rest-assured:rest-assured:3.3.0'
+    testImplementation('org.springframework.boot:spring-boot-starter-test') {
+        exclude group: 'org.junit.vintage', module: 'junit-vintage-engine'
+    }
+    runtimeOnly 'com.h2database:h2'
 }
 
 test {
-	useJUnitPlatform()
+    useJUnitPlatform()
 }

src/main/java/wooteco/subway/admin/SubwayAdminApplication.java

@@ -6,8 +6,8 @@
 @SpringBootApplication
 public class SubwayAdminApplication {
 
-	public static void main(String[] args) {
-		SpringApplication.run(SubwayAdminApplication.class, args);
-	}
+    public static void main(String[] args) {
+        SpringApplication.run(SubwayAdminApplication.class, args);
+    }
 
 }

src/main/java/wooteco/subway/admin/controller/LineController.java

@@ -0,0 +1,105 @@
+package wooteco.subway.admin.controller;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.PutMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import wooteco.subway.admin.domain.Line;
+import wooteco.subway.admin.domain.Station;
+import wooteco.subway.admin.dto.LineRequest;
+import wooteco.subway.admin.dto.LineResponse;
+import wooteco.subway.admin.dto.LineStationCreateRequest;
+import wooteco.subway.admin.service.LineService;
+import wooteco.subway.admin.service.StationService;
+
+import java.net.URI;
+import java.util.HashMap;
+import java.util.List;
+import java.util.stream.Collectors;
+
+@RestController
+@RequestMapping("/lines")
+public class LineController {
+    private final LineService lineService;
+    private final StationService stationService;
+
+    public LineController(final LineService lineService, final StationService stationService) {
+        this.lineService = lineService;
+        this.stationService = stationService;
+    }
+
+    @PostMapping
+    public ResponseEntity createLine(@RequestBody LineRequest lineRequest) {
+        lineService.validateTitle(lineRequest);
+        Line line = new Line(
+                lineRequest.getTitle(),
+                lineRequest.getStartTime(),
+                lineRequest.getEndTime(),
+                lineRequest.getIntervalTime(),
+                lineRequest.getBgColor()
+        );
+        Line savedLine = lineService.save(line);
+        return ResponseEntity.created(URI.create("/lines/" + savedLine.getId())).body(savedLine);
+    }
+
+    @GetMapping
+    public ResponseEntity findAllLine() {
+        List<LineResponse> lineResponses = lineService.findAll().stream()
+                .map(line -> lineService.findLineWithStationsById(line.getId()))
+                .collect(Collectors.toList());
+        return ResponseEntity.ok(lineResponses);
+    }
+
+    @GetMapping("/{id}")
+    public ResponseEntity findLine(@PathVariable Long id) {
+        return ResponseEntity.ok(lineService.findLineWithStationsById(id));
+    }
+
+    @PutMapping("/{id}")
+    public ResponseEntity updateLine(@PathVariable Long id, @RequestBody LineRequest lineRequest) {
+        lineService.validateTitleWhenUpdateInfo(id, lineRequest);
+        Line line = new Line(
+                lineRequest.getTitle(),
+                lineRequest.getStartTime(),
+                lineRequest.getEndTime(),
+                lineRequest.getIntervalTime(),
+                lineRequest.getBgColor()
+        );
+        lineService.updateLine(id, line);
+        return ResponseEntity.ok().build();
+    }
+
+    @DeleteMapping("/{id:\\d+}")
+    public ResponseEntity deleteLine(@PathVariable Long id) {
+        lineService.deleteLineById(id);
+        return ResponseEntity.ok().build();
+    }
+
+    @PostMapping("/addStation/{id}")
+    public ResponseEntity addStation(@PathVariable Long id,
+                                     @RequestBody HashMap<String, String> map) {
+        Long preStationId = stationService.findStationId(map.get("preStationName"));
+        Station inputStation = stationService.save(map.get("stationName"));
+
+        LineStationCreateRequest lineStationCreateRequest =
+                new LineStationCreateRequest(
+                        preStationId,
+                        inputStation.getId(),
+                        Integer.parseInt(map.get("distance")),
+                        Integer.parseInt(map.get("duration")));
+
+        lineService.addLineStation(id, lineStationCreateRequest);
+        return ResponseEntity.ok().build();
+    }
+
+    @DeleteMapping("/station/{lineId}/{stationId}")
+    public ResponseEntity deleteStation(@PathVariable Long lineId, @PathVariable Long stationId) {
+        lineService.removeLineStation(lineId, stationId);
+        return ResponseEntity.ok().build();
+    }
+}

src/main/java/wooteco/subway/admin/controller/LineStationController.java

@@ -0,0 +1,21 @@
+package wooteco.subway.admin.controller;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RestController;
+import wooteco.subway.admin.service.LineService;
+
+@RestController
+public class LineStationController {
+    private final LineService lineService;
+
+    public LineStationController(final LineService lineService) {
+        this.lineService = lineService;
+    }
+
+    @GetMapping("/lineStations/{lineId}")
+    public ResponseEntity findAllLineStations(@PathVariable Long lineId) {
+        return ResponseEntity.ok(lineService.findLineStationByLineId(lineId));
+    }
+}

src/main/java/wooteco/subway/admin/controller/StationController.java

@@ -1,7 +1,12 @@
 package wooteco.subway.admin.controller;
 
 import org.springframework.http.ResponseEntity;
-import org.springframework.web.bind.annotation.*;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
 import wooteco.subway.admin.domain.Station;
 import wooteco.subway.admin.dto.StationCreateRequest;
 import wooteco.subway.admin.dto.StationResponse;
@@ -20,11 +25,19 @@ public StationController(StationRepository stationRepository) {
     @PostMapping("/stations")
     public ResponseEntity createStation(@RequestBody StationCreateRequest view) {
         Station station = view.toStation();
-        Station persistStation = stationRepository.save(station);
+        validateName(station.getName());
 
+        Station persistStation = stationRepository.save(station);
         return ResponseEntity
-                .created(URI.create("/stations/" + persistStation.getId()))
-                .body(StationResponse.of(persistStation));
+            .created(URI.create("/stations/" + persistStation.getId()))
+            .body(StationResponse.of(persistStation));
+    }
+
+    private void validateName(final String stationName) {
+        stationRepository.findByName(stationName)
+                .ifPresent(station -> {
+                    throw new IllegalArgumentException("존재하는 역 이름입니다");
+                });
     }
 
     @GetMapping("/stations")

src/main/java/wooteco/subway/admin/controller/WebController.java

@@ -0,0 +1,27 @@
+package wooteco.subway.admin.controller;
+
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.GetMapping;
+
+@Controller
+public class WebController {
+    @GetMapping("/")
+    public String index() {
+        return "index";
+    }
+
+    @GetMapping("/admin-line")
+    public String adminLine() {
+        return "admin-line";
+    }
+
+    @GetMapping("/admin-edge")
+    public String adminEdge() {
+        return "admin-edge";
+    }
+
+    @GetMapping("/admin-station")
+    public String adminStation() {
+        return "admin-station";
+    }
+}

src/main/java/wooteco/subway/admin/domain/Line.java

@@ -4,41 +4,48 @@
 
 import java.time.LocalDateTime;
 import java.time.LocalTime;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.stream.Collectors;
 
 public class Line {
     @Id
     private Long id;
-    private String name;
+    private String title;
     private LocalTime startTime;
     private LocalTime endTime;
     private int intervalTime;
     private Set<LineStation> stations;
     private LocalDateTime createdAt;
     private LocalDateTime updatedAt;
+    private String bgColor;
 
     public Line() {
     }
 
-    public Line(Long id, String name, LocalTime startTime, LocalTime endTime, int intervalTime) {
-        this.name = name;
+    public Line(Long id, String title, LocalTime startTime, LocalTime endTime, int intervalTime, String bgColor) {
+        this.title = title;
         this.startTime = startTime;
         this.endTime = endTime;
         this.intervalTime = intervalTime;
         this.createdAt = LocalDateTime.now();
         this.updatedAt = LocalDateTime.now();
+        this.bgColor = bgColor;
+        this.stations = new LinkedHashSet<>();
     }
 
-    public Line(String name, LocalTime startTime, LocalTime endTime, int intervalTime) {
-        this(null, name, startTime, endTime, intervalTime);
+    public Line(String title, LocalTime startTime, LocalTime endTime, int intervalTime, String bgColor) {
+        this(null, title, startTime, endTime, intervalTime, bgColor);
     }
 
     public Long getId() {
         return id;
     }
 
-    public String getName() {
-        return name;
+    public String getTitle() {
+        return title;
     }
 
     public LocalTime getStartTime() {
@@ -65,9 +72,13 @@ public LocalDateTime getUpdatedAt() {
         return updatedAt;
     }
 
+    public String getBgColor() {
+        return bgColor;
+    }
+
     public void update(Line line) {
-        if (line.getName() != null) {
-            this.name = line.getName();
+        if (line.getTitle() != null) {
+            this.title = line.getTitle();
         }
         if (line.getStartTime() != null) {
             this.startTime = line.getStartTime();
@@ -78,20 +89,68 @@ public void update(Line line) {
         if (line.getIntervalTime() != 0) {
             this.intervalTime = line.getIntervalTime();
         }
+        if (line.getBgColor() != null) {
+            this.bgColor = line.getBgColor();
+        }
 
         this.updatedAt = LocalDateTime.now();
     }
 
     public void addLineStation(LineStation lineStation) {
-        // TODO: 구현
+        this.stations.add(lineStation);
     }
 
-    public void removeLineStationById(Long stationId) {
-        // TODO: 구현
+    public void removeLineStationById(Long id) {
+        this.stations.stream()
+                .filter(x -> x.isStationIdEquals(id))
+                .findAny()
+                .ifPresent(target -> {
+                    this.stations.remove(target);
+                });
     }
 
     public List<Long> getLineStationsId() {
-        // TODO: 구현
-        return new ArrayList<>();
+        List<LineStation> result = new ArrayList<>();
+        stations.stream()
+                .filter(x -> x.isPreStationIdEquals(LineStation.NULL_PRE_STATION_VALUE))
+                .findAny()
+                .ifPresent(result::add);
+
+        for (int i = 0; i < stations.size() - 1; i++) {
+            result.add(findNextLineStation(result.get(i)));
+        }
+        return result.stream()
+                .map(LineStation::getStationId)
+                .collect(Collectors.toList());
+    }
+
+    private LineStation findNextLineStation(final LineStation preStation) {
+        return stations.stream()
+                .filter(x -> x.isPreStationIdEquals(preStation.getStationId()))
+                .findAny().orElseThrow(IllegalArgumentException::new);
+    }
+
+    public void updatePreStationWhenAdd(final LineStation toInput) {
+        stations.stream()
+                .filter(x -> x.isPreStationIdEquals(toInput.getPreStationId()))
+                .findAny().ifPresent(lineStation -> {
+            lineStation.updatePreLineStation(toInput.getStationId());
+        });
+    }
+
+    public void updatePreStationWhenRemove(final Long toRemoveId) {
+        LineStation removeStation =
+                stations.stream().filter(x -> x.isStationIdEquals(toRemoveId))
+                        .findAny().orElseThrow(IllegalArgumentException::new);
+
+        stations.stream()
+                .filter(x -> x.isPreStationIdEquals(toRemoveId))
+                .findAny().ifPresent(lineStation -> {
+            lineStation.updatePreLineStation(removeStation.getPreStationId());
+        });
+    }
+
+    public boolean isTitleEquals(final String title) {
+        return this.title.equals(title);
     }
 }