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