Stage 11: Store and change products in MongoDB

package com.edupoly.products;

import java.math.BigDecimal;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

@Document("products")
public class Product {
    @Id
    private int id;
    private String name;
    private String category;
    private BigDecimal price;
    private String brand;
    private int stock;
    private String description;
    private int warrantyMonths;

    public int getId() {
        return id;
    }

    public void setId(int id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getCategory() {
        return category;
    }

    public void setCategory(String category) {
        this.category = category;
    }

    public BigDecimal getPrice() {
        return price;
    }

    public void setPrice(BigDecimal price) {
        this.price = price;
    }

    public String getBrand() {
        return brand;
    }

    public void setBrand(String brand) {
        this.brand = brand;
    }

    public int getStock() {
        return stock;
    }

    public void setStock(int stock) {
        this.stock = stock;
    }

    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }

    public int getWarrantyMonths() {
        return warrantyMonths;
    }

    public void setWarrantyMonths(int warrantyMonths) {
        this.warrantyMonths = warrantyMonths;
    }
}

package com.edupoly.products;

import org.springframework.data.mongodb.repository.MongoRepository;

public interface ProductRepository extends MongoRepository<Product, Integer> {
}

package com.edupoly.products;

import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import org.springframework.stereotype.Service;

@Service
public class ProductService {
    private final ProductRepository repository;

    public ProductService(ProductRepository repository) {
        this.repository = repository;
    }

    public List<Product> findAll(boolean inStockOnly) {
        List<Product> selected = new ArrayList<>();
        for (Product product : repository.findAll()) {
            if (!inStockOnly || product.getStock() > 0) {
                selected.add(product);
            }
        }
        return selected;
    }

    public Optional<Product> findById(int id) {
        return repository.findById(id);
    }
    public Product createSample() {
        Optional<Product> existing = repository.findById(31);
        if (existing.isPresent()) {
            return existing.get();
        }
        Product product = new Product();
        product.setId(31);
        product.setName("Practice Product");
        product.setCategory("Study Essentials");
        product.setPrice(new java.math.BigDecimal("250.00"));
        product.setBrand("StudyWorks");
        product.setStock(8);
        product.setDescription("Created through the application's Create sample button.");
        product.setWarrantyMonths(6);
        return repository.insert(product);
    }

    public Product updateStock(Product product, int stock) {
        product.setStock(stock);
        return repository.save(product);
    }

    public void delete(int id) {
        repository.deleteById(id);
    }
}

package com.edupoly.products;

import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.server.ResponseStatusException;

@Controller
public class ProductController {
    private final ProductService service;

    public ProductController(ProductService service) {
        this.service = service;
    }

    @GetMapping({"/", "/products"})
    public String listProducts(
            @RequestParam(name = "inStockOnly", defaultValue = "false") boolean inStockOnly,
            Model model) {
        model.addAttribute("products", service.findAll(inStockOnly));
        return "products";
    }

    @GetMapping("/products/{id}")
    public String productDetails(@PathVariable("id") int id, Model model) {
        Product product = service.findById(id).orElseThrow(() ->
                new ResponseStatusException(HttpStatus.NOT_FOUND, "Product not found"));
        model.addAttribute("product", product);
        return "product-details";
    }
    @PostMapping("/products/sample")
    public String createSample() {
        Product product = service.createSample();
        return "redirect:/products/" + product.getId();
    }

    @PostMapping("/products/{id}/stock")
    public String updateStock(@PathVariable("id") int id,
                              @RequestParam("stock") int stock) {
        if (stock < 0) {
            throw new ResponseStatusException(HttpStatus.BAD_REQUEST, "Stock cannot be negative");
        }
        Product product = service.findById(id).orElseThrow(() ->
                new ResponseStatusException(HttpStatus.NOT_FOUND, "Product not found"));
        service.updateStock(product, stock);
        return "redirect:/products/" + id;
    }

    @PostMapping("/products/{id}/delete")
    public String delete(@PathVariable("id") int id) {
        if (service.findById(id).isEmpty()) {
            throw new ResponseStatusException(HttpStatus.NOT_FOUND, "Product not found");
        }
        service.delete(id);
        return "redirect:/products";
    }
}

package com.edupoly.products;

import java.util.Arrays;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

@Component
public class ProductDataSeeder implements CommandLineRunner {
    private final ProductRepository repository;
    private final ProductFileReader reader;

    public ProductDataSeeder(ProductRepository repository, ProductFileReader reader) {
        this.repository = repository;
        this.reader = reader;
    }

    @Override
    public void run(String... args) {
        if (repository.count() == 0) {
            repository.saveAll(Arrays.asList(reader.getProducts()));
            System.out.println("Seeded 30 products into the empty demo database");
        }
    }
}