f423ddf8324c70b2a5f823a95553686ad270d1cf
[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.media.MediaExc;
38 import mir.media.MediaFailure;
39 import mircoders.media.MediaHelper;
40 import mir.media.MediaHandler;
41 import mir.misc.StringUtil;
42 import mir.session.UploadedFile;
43 import mir.storage.Database;
44 import mircoders.module.ModuleMediaType;
45
46 public class MediaUploadProcessor {
47   public static Entity processMediaUpload(UploadedFile aFile, Map aValues) throws MediaExc, MediaFailure {
48     String mediaId;
49     MediaHandler mediaHandler;
50     Entity mediaType;
51     ModuleMediaType mediaTypeModule;
52     Database mediaStorage;
53     Map values = new HashMap();
54     String MediaId;
55     Entity mediaEntity;
56
57
58     try {
59       String contentType = aFile.getContentType();
60
61       if (contentType.equals("text/plain") ||
62           contentType.equals("application/octet-stream") ||
63           contentType == null) {
64         throw new MediaExc("Invalid content-type: " + contentType);
65       }
66
67       values.putAll(aValues);
68       values.put("date", StringUtil.date2webdbDate(new GregorianCalendar()));
69       values.put("is_produced", "0");
70
71       mediaTypeModule = new ModuleMediaType();
72       mediaType = mediaTypeModule.findMediaTypeForMimeType(contentType);
73
74       try {
75         mediaHandler = MediaHelper.getHandler(mediaType);
76         mediaStorage = MediaHelper.getStorage(mediaType, mediaType.getFieldValue("tablename"));
77       }
78       catch (Throwable e) {
79         throw new MediaFailure(e);
80       }
81
82       values.put("to_media_type", mediaType.getId());
83
84       try {
85         mediaEntity = (Entity) mediaStorage.getEntityClass().newInstance();
86         mediaEntity.setStorage(mediaStorage);
87       }
88       catch (Throwable e) {
89         throw new MediaFailure(e);
90       }
91
92       mediaEntity.setFieldValues(values);
93       mediaId = mediaEntity.insert();
94
95       try {
96         mediaHandler.store(aFile.getInputStream(), mediaEntity, mediaType);
97       }
98       catch (Throwable e) {
99         throw new MediaFailure(e);
100       }
101
102       return mediaEntity;
103     }
104     catch (Throwable e) {
105       throw new MediaFailure(e);
106     }
107
108   }
109 }