cleanup / abuse system fix / prepping for a release
[mir.git] / source / mircoders / media / MediaUploadProcessor.java
1 /*
2  * Copyright (C) 2001, 2002 The Mir-coders group
3  *
4  * This file is part of Mir.
5  *
6  * Mir is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * Mir is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with Mir; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  * In addition, as a special exception, The Mir-coders gives permission to link
21  * the code of this program with  any library licensed under the Apache Software License,
22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
23  * (or with modified versions of the above that use the same license as the above),
24  * and distribute linked combinations including the two.  You must obey the
25  * GNU General Public License in all respects for all of the code used other than
26  * the above mentioned libraries.  If you modify this file, you may extend this
27  * exception to your version of the file, but you are not obligated to do so.
28  * If you do not wish to do so, delete this exception statement from your version.
29  */
30 package mircoders.media;
31
32 import java.util.GregorianCalendar;
33 import java.util.HashMap;
34 import java.util.Map;
35
36 import mir.entity.Entity;
37 import mir.log.LoggerWrapper;
38 import mir.media.MediaFailure;
39 import mir.media.MediaHandler;
40 import mir.misc.StringUtil;
41 import mir.session.UploadedFile;
42 import mir.storage.Database;
43 import mir.util.FileRoutines;
44 import mircoders.module.ModuleMediaType;
45
46 public class MediaUploadProcessor {
47   private static LoggerWrapper logger = new LoggerWrapper("Media.UploadProcessor");
48   private static ModuleMediaType mediaTypeModule = new ModuleMediaType();
49
50   /**
51    * Processes an uploaded media file.
52    * Will create the media entity and so on. 
53    */
54   public static Entity processMediaUpload(UploadedFile aFile, Map aValues) throws UnsupportedMediaTypeExc, MediaFailure {
55     MediaHandler mediaHandler;
56     Entity mediaType = null;
57     Database mediaStorage;
58     Map values = new HashMap();
59     Entity mediaEntity;
60
61     String contentType = aFile.getContentType();
62     logger.debug("processing media upload of " + aFile.getFileName() + " (content type = " + contentType + ")");
63
64     if (contentType!=null) {
65       mediaType = mediaTypeModule.findMediaTypeForMimeType(contentType);
66     }
67
68     if (mediaType==null) {
69       String extension = FileRoutines.getExtension(aFile.getFileName());
70       mediaType = mediaTypeModule.findMediaTypeForExtension(extension);
71     }
72
73     if (mediaType==null) {
74       throw new UnsupportedMediaTypeExc("Media " + aFile.getFileName() + " " +
75           (aFile!=null?"("+aFile.getContentType()+") ":"") + " not supported", aFile.getContentType());
76     }
77
78     values.putAll(aValues);
79     values.put("date", StringUtil.date2webdbDate(new GregorianCalendar()));
80     values.put("is_produced", "0");
81
82     try {
83       mediaHandler = MediaHelper.getHandler(mediaType);
84       mediaStorage = MediaHelper.getStorage(mediaType, mediaType.getFieldValue("tablename"));
85     }
86     catch (Throwable e) {
87       throw new MediaFailure(e);
88     }
89
90     values.put("to_media_type", mediaType.getId());
91
92     try {
93       mediaEntity = mediaStorage.createNewEntity();
94       mediaEntity.setFieldValues(values);
95       mediaEntity.insert();
96       mediaHandler.store(aFile, mediaEntity, mediaType);
97     }
98     catch (Throwable e) {
99       throw new MediaFailure(e);
100     }
101
102     return mediaEntity;
103   }
104 }