WebsiteProjectController.java 3.2 KB
// website/controller/WebsiteProjectController.java
package com.aigeo.website.controller;

import com.aigeo.website.entity.WebsiteProject;
import com.aigeo.website.entity.WebsiteProjectStatus;
import com.aigeo.website.service.WebsiteProjectService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.Optional;

@RestController
@RequestMapping("/api/website-projects")
public class WebsiteProjectController {
    
    @Autowired
    private WebsiteProjectService websiteProjectService;
    
    @GetMapping
    public List<WebsiteProject> getAllProjects() {
        return websiteProjectService.getAllProjects();
    }
    
    @GetMapping("/{id}")
    public ResponseEntity<WebsiteProject> getProjectById(@PathVariable Integer id) {
        Optional<WebsiteProject> project = websiteProjectService.getProjectById(id);
        return project.map(ResponseEntity::ok)
                .orElse(ResponseEntity.notFound().build());
    }
    
    @GetMapping("/company/{companyId}")
    public List<WebsiteProject> getProjectsByCompanyId(@PathVariable Integer companyId) {
        return websiteProjectService.getProjectsByCompanyId(companyId);
    }
    
    @GetMapping("/user/{userId}")
    public List<WebsiteProject> getProjectsByUserId(@PathVariable Integer userId) {
        return websiteProjectService.getProjectsByUserId(userId);
    }
    
    @GetMapping("/status/{status}")
    public List<WebsiteProject> getProjectsByStatus(@PathVariable WebsiteProjectStatus status) {
        return websiteProjectService.getProjectsByStatus(status);
    }
    
    @GetMapping("/company/{companyId}/status/{status}")
    public List<WebsiteProject> getProjectsByCompanyIdAndStatus(
            @PathVariable Integer companyId, @PathVariable WebsiteProjectStatus status) {
        return websiteProjectService.getProjectsByCompanyIdAndStatus(companyId, status);
    }
    
    @PostMapping
    public WebsiteProject createProject(@RequestBody WebsiteProject project) {
        return websiteProjectService.saveProject(project);
    }
    
    @PutMapping("/{id}")
    public ResponseEntity<WebsiteProject> updateProject(@PathVariable Integer id, 
                                                     @RequestBody WebsiteProject projectDetails) {
        Optional<WebsiteProject> project = websiteProjectService.getProjectById(id);
        if (project.isPresent()) {
            WebsiteProject updatedProject = project.get();
            updatedProject.setProjectName(projectDetails.getProjectName());
            updatedProject.setSiteName(projectDetails.getSiteName());
            updatedProject.setStatus(projectDetails.getStatus());
            
            WebsiteProject savedProject = websiteProjectService.saveProject(updatedProject);
            return ResponseEntity.ok(savedProject);
        } else {
            return ResponseEntity.notFound().build();
        }
    }
    
    @DeleteMapping("/{id}")
    public ResponseEntity<Void> deleteProject(@PathVariable Integer id) {
        websiteProjectService.deleteProject(id);
        return ResponseEntity.noContent().build();
    }
}