b6d5dadbb2913793ca355f3cf2d9e41eed297b2b
[mir.git] / source / mircoders / module / ModuleMediaType.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
31 package mircoders.module;
32
33 import java.util.List;
34
35 import mir.entity.Entity;
36 import mir.entity.EntityList;
37 import mir.log.LoggerWrapper;
38 import mir.module.AbstractModule;
39 import mir.module.ModuleExc;
40 import mir.module.ModuleFailure;
41 import mir.storage.StorageObject;
42 import mir.util.JDBCStringRoutines;
43 import mir.util.StringRoutines;
44 import mircoders.storage.DatabaseMediaType;
45
46 public class ModuleMediaType extends AbstractModule {
47   static LoggerWrapper logger = new LoggerWrapper("Module.Content");
48
49   public ModuleMediaType() {
50     this (DatabaseMediaType.getInstance());
51   }
52
53   public ModuleMediaType(StorageObject aStorage) {
54     this.theStorage = aStorage;
55   }
56
57   public Entity findMediaTypeForMimeType(String aMimeType) throws ModuleExc, ModuleFailure {
58     List contentTypeParts = StringRoutines.splitString(aMimeType, "/");
59
60     if (contentTypeParts.size()!=2) {
61       throw new InvalidMimeTypeExc("Invalid mimetype: " + aMimeType, aMimeType);
62     }
63     String mimeTypeMajor = (String) contentTypeParts.get(0);
64
65     EntityList mediaTypes;
66
67     mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(aMimeType)+"'");
68     if (mediaTypes.size() == 0) {
69       mediaTypes = DatabaseMediaType.getInstance().selectByWhereClause("mime_type = '"+JDBCStringRoutines.escapeStringLiteral(mimeTypeMajor+"/*")+"'");
70     }
71     if (mediaTypes.size() == 0) {
72       throw new UnsupportedMimeTypeExc("Unsupported mimetype: " + aMimeType, aMimeType);
73     }
74
75     return (Entity) mediaTypes.elementAt(0);
76   }
77
78   public static class MimeTypeExc extends ModuleExc {
79     private String mimeType;
80
81     public MimeTypeExc(String aMessage, String aMimeType) {
82       super (aMessage);
83       mimeType = aMimeType;
84     }
85
86     public String getMimeType() {
87       return mimeType;
88     }
89   }
90
91   public static class UnsupportedMimeTypeExc extends MimeTypeExc {
92     public UnsupportedMimeTypeExc(String aMessage, String aMimeType) {
93       super(aMessage, aMimeType);
94     }
95   }
96
97   public static class InvalidMimeTypeExc extends MimeTypeExc {
98     public InvalidMimeTypeExc(String aMessage, String aMimeType) {
99       super(aMessage, aMimeType);
100     }
101   }
102 }