Update ProductImageController.java

src/main/java/com/how2java/tmall/controller/ProductImageController.java

@@ -1,4 +1,126 @@
   	
 
+  	
+
 package com.how2java.tmall.controller;
 
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.util.List;
+
+import javax.imageio.ImageIO;
+import javax.servlet.http.HttpServletResponse;
+import javax.servlet.http.HttpSession;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Controller;
+import org.springframework.ui.Model;
+import org.springframework.web.bind.annotation.RequestMapping;
+
+import com.how2java.tmall.pojo.Product;
+import com.how2java.tmall.pojo.ProductImage;
+import com.how2java.tmall.service.ProductImageService;
+import com.how2java.tmall.service.ProductService;
+import com.how2java.tmall.util.ImageUtil;
+import com.how2java.tmall.util.Page;
+import com.how2java.tmall.util.UploadedImageFile;
+
+
+@Controller
+@RequestMapping("")
+public class ProductImageController {
+    @Autowired
+    ProductService productService;
+
+    @Autowired
+    ProductImageService productImageService;
+
+
+    @RequestMapping("admin_productImage_add")
+    public String add(ProductImage  pi, HttpSession session, UploadedImageFile uploadedImageFile) {
+        productImageService.add(pi);
+        String fileName = pi.getId()+ ".jpg";
+        String imageFolder;
+        String imageFolder_small=null;
+        String imageFolder_middle=null;
+        if(ProductImageService.type_single.equals(pi.getType())){
+            imageFolder= session.getServletContext().getRealPath("img/productSingle");
+            imageFolder_small= session.getServletContext().getRealPath("img/productSingle_small");
+            imageFolder_middle= session.getServletContext().getRealPath("img/productSingle_middle");
+        }
+        else{
+            imageFolder= session.getServletContext().getRealPath("img/productDetail");
+        }
+
+        File f = new File(imageFolder, fileName);
+        f.getParentFile().mkdirs();
+        try {
+            uploadedImageFile.getImage().transferTo(f);
+            BufferedImage img = ImageUtil.change2jpg(f);
+            ImageIO.write(img, "jpg", f);
+
+            if(ProductImageService.type_single.equals(pi.getType())) {
+                File f_small = new File(imageFolder_small, fileName);
+                File f_middle = new File(imageFolder_middle, fileName);
+
+                ImageUtil.resizeImage(f, 56, 56, f_small);
+                ImageUtil.resizeImage(f, 217, 190, f_middle);
+            }
+
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return "redirect:admin_productImage_list?pid="+pi.getPid();
+    }
+
+    @RequestMapping("admin_productImage_delete")
+    public String delete(int id,HttpSession session) {
+        ProductImage pi = productImageService.get(id);
+
+        String fileName = pi.getId()+ ".jpg";
+        String imageFolder;
+        String imageFolder_small=null;
+        String imageFolder_middle=null;
+
+        if(ProductImageService.type_single.equals(pi.getType())){
+            imageFolder= session.getServletContext().getRealPath("img/productSingle");
+            imageFolder_small= session.getServletContext().getRealPath("img/productSingle_small");
+            imageFolder_middle= session.getServletContext().getRealPath("img/productSingle_middle");
+            File imageFile = new File(imageFolder,fileName);
+            File f_small = new File(imageFolder_small,fileName);
+            File f_middle = new File(imageFolder_middle,fileName);
+            imageFile.delete();
+            f_small.delete();
+            f_middle.delete();
+
+        }
+        else{
+            imageFolder= session.getServletContext().getRealPath("img/productDetail");
+            File imageFile = new File(imageFolder,fileName);
+            imageFile.delete();
+        }
+
+
+        productImageService.delete(id);
+
+
+        return "redirect:admin_productImage_list?pid="+pi.getPid();
+    }
+
+    @RequestMapping("admin_productImage_list")
+    public String list(int pid, Model model) {
+        Product p =productService.get(pid);
+        List<ProductImage> pisSingle = productImageService.list(pid, ProductImageService.type_single);
+        List<ProductImage> pisDetail = productImageService.list(pid, ProductImageService.type_detail);
+
+
+        model.addAttribute("p", p);
+        model.addAttribute("pisSingle", pisSingle);
+        model.addAttribute("pisDetail", pisDetail);
+
+        return "admin/listProductImage";
+    }
+}
+
+  	
+