Changeset 635 for trunk/grails-app/services/InventoryItemService.groovy

Timestamp:

Jul 26, 2010, 1:33:50 PM (14 years ago)

Author:

gav

Message:

Add feature to import inventory item pictures from zip file.

File:

• trunk/grails-app/services/InventoryItemService.groovy (modified) (4 diffs)

trunk/grails-app/services/InventoryItemService.groovy

@@ +1 @@
+import org.codehaus.groovy.grails.commons.ConfigurationHolder
+
 /**
 * Provides a service class for the InventoryItem domain class.
@@ -228 +230 @@
     /**
     * Save an inventory item picture.
+    * @param params An object or map containing at least the inventoryItem ID.
     * @param pictureSource A supported source to get the picture image from.
     * Supported sources:
     * HttpServletRequest e.g: 'request' var from controller to run getFile('file') against.
     * ServletContextResource e.g: grailsApplication.mainContext.getResource('images/logo.png')
+    * File e.g: new File('picture.jpg')
     */
     def savePicture(params, pictureSource) {
@@ -294 +298 @@
                 pictureInputStream = pictureSource.inputStream
             }
+            else if(pictureSource instanceof File) {
+                pictureFile = pictureSource
+                pictureFileName = pictureFile.name
+
+                if ( !pictureFile.isFile() || (pictureFile.length() == 0) )
+                    return fail(code:"default.file.not.supplied")
+
+                if (pictureFile.length() > Image.MAX_SIZE)
+                    return fail(code:"default.file.over.max.size", args: [Image.MAX_SIZE/kByteMultiplier, "kB"])
+
+                pictureInputStream = new FileInputStream(pictureSource)
+            }
             else {
                     return fail(code:"inventory.item.picture.source.not.supported")
@@ -329 +345 @@
             return result
 
-        } //end withTransaction
-    }
+        } // end withTransaction
+    } // savePicture
+
+    /**
+    * Import inventory pictures from an uploaded zip file.
+    * @param request The http request to run getFile against.
+    * Get file should return a zip format file containing the inventory item pictures.
+    */
+    def importInventoryItemPictures(request) {
+            def result = [:]
+
+            def kByteMultiplier = 1000
+            def mByteMultiplier = 1000 * kByteMultiplier
+            def fileMaxSize = 500 * mByteMultiplier
+
+            def fail = { Map m ->
+                result.error = [ code: m.code, args: m.args ]
+                return result
+            }
+
+            // Get file from request.
+            def multiPartFile = request.getFile('file')
+            def zipFileName = multiPartFile.originalFilename
+
+            if(!multiPartFile || multiPartFile.isEmpty())
+                return fail(code: "default.file.not.supplied")
+
+            if (multiPartFile.getSize() > fileMaxSize)
+                return fail(code: "default.file.over.max.size", args: [fileMaxSize/mByteMultiplier, "MB"])
+
+            // Check create import dir.
+            def dir = new File(ConfigurationHolder.config.globalDirs.tempInventoryItemPicturesDirectory)
+            def imageFiles = []
+
+            if(!dir.exists())
+                dir.mkdirs()
+
+            if(!dir.isDirectory()) {
+                return fail(code:'inventoryItemPictures.import.failure.no.directory')
+            }
+
+            // Write zip file to disk.
+            def zipOutputFile = new File(dir.absolutePath + File.separator + zipFileName)
+            multiPartFile.transferTo(zipOutputFile)
+
+            // Use ant to unzip.
+            def ant = new AntBuilder()
+            try {
+                ant.unzip(  src: zipOutputFile.absolutePath,
+                                    dest: dir.absolutePath,
+                                    overwrite:"true" )
+            }
+            catch(e) {
+                log.error e
+                return fail(code:'inventoryItemPictures.import.failure.to.unzip')
+            }
+
+            // Recurse through dir building list of imageFiles.
+            def imageFilePattern = ~/[^\s].+(\.(?i)(jpg|png|gif|bmp))$/
+
+            dir.eachFileMatch(imageFilePattern) {
+                imageFiles << it
+            }
+
+            dir.eachDirRecurse { subDir ->
+                subDir.eachFileMatch(imageFilePattern) {
+                    imageFiles << it
+                }
+            }
+
+            // Find inventoryItems by name of picture and call savePicture.
+            def inventoryItemInstance
+            def itemName
+            def savePictureResult
+
+            for(imageFile in imageFiles) {
+                itemName = imageFile.name[0..-5]
+                inventoryItemInstance = InventoryItem.findByName(itemName)
+                if(!inventoryItemInstance) {
+                    log.warn 'InventoryItem not found with name: ' + itemName
+                    continue
+                }
+                if(inventoryItemInstance.picture) {
+                    log.warn 'InventoryItem already has picture: ' + itemName
+                    continue
+                }
+                savePictureResult = savePicture(inventoryItemInstance, imageFile)
+                if(savePictureResult.error)
+                    log.error savePictureResult.error
+                else
+                    log.info 'InventoryItem picture saved: ' + itemName
+            }
+
+            // Success.
+            return result
+
+    } // importInventoryItemPictures
 
 } // end class