aoqi@0: /*
aoqi@0:  * Copyright (c) 1997, 2005, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: package javax.activation;
aoqi@0: 
aoqi@0: import java.net.URL;
aoqi@0: import java.net.URLConnection;
aoqi@0: import java.io.InputStream;
aoqi@0: import java.io.OutputStream;
aoqi@0: import java.io.IOException;
aoqi@0: 
aoqi@0: /**
aoqi@0:  * The URLDataSource class provides an object that wraps a <code>URL</code>
aoqi@0:  * object in a DataSource interface. URLDataSource simplifies the handling
aoqi@0:  * of data described by URLs within the JavaBeans Activation Framework
aoqi@0:  * because this class can be used to create new DataHandlers. <i>NOTE: The
aoqi@0:  * DataHandler object creates a URLDataSource internally,
aoqi@0:  * when it is constructed with a URL.</i>
aoqi@0:  *
aoqi@0:  * @see javax.activation.DataSource
aoqi@0:  * @see javax.activation.DataHandler
aoqi@0:  *
aoqi@0:  * @since 1.6
aoqi@0:  */
aoqi@0: public class URLDataSource implements DataSource {
aoqi@0:     private URL url = null;
aoqi@0:     private URLConnection url_conn = null;
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * URLDataSource constructor. The URLDataSource class will
aoqi@0:      * not open a connection to the URL until a method requiring it
aoqi@0:      * to do so is called.
aoqi@0:      *
aoqi@0:      * @param url The URL to be encapsulated in this object.
aoqi@0:      */
aoqi@0:     public URLDataSource(URL url) {
aoqi@0:         this.url = url;
aoqi@0:     }
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * Returns the value of the URL content-type header field.
aoqi@0:      * It calls the URL's <code>URLConnection.getContentType</code> method
aoqi@0:      * after retrieving a URLConnection object.
aoqi@0:      * <i>Note: this method attempts to call the <code>openConnection</code>
aoqi@0:      * method on the URL. If this method fails, or if a content type is not
aoqi@0:      * returned from the URLConnection, getContentType returns
aoqi@0:      * "application/octet-stream" as the content type.</i>
aoqi@0:      *
aoqi@0:      * @return the content type.
aoqi@0:      */
aoqi@0:     public String getContentType() {
aoqi@0:         String type = null;
aoqi@0: 
aoqi@0:         try {
aoqi@0:             if (url_conn == null)
aoqi@0:                 url_conn = url.openConnection();
aoqi@0:         } catch (IOException e) { }
aoqi@0: 
aoqi@0:         if (url_conn != null)
aoqi@0:             type = url_conn.getContentType();
aoqi@0: 
aoqi@0:         if (type == null)
aoqi@0:             type = "application/octet-stream";
aoqi@0: 
aoqi@0:         return type;
aoqi@0:     }
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * Calls the <code>getFile</code> method on the URL used to
aoqi@0:      * instantiate the object.
aoqi@0:      *
aoqi@0:      * @return the result of calling the URL's getFile method.
aoqi@0:      */
aoqi@0:     public String getName() {
aoqi@0:         return url.getFile();
aoqi@0:     }
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * The getInputStream method from the URL. Calls the
aoqi@0:      * <code>openStream</code> method on the URL.
aoqi@0:      *
aoqi@0:      * @return the InputStream.
aoqi@0:      */
aoqi@0:     public InputStream getInputStream() throws IOException {
aoqi@0:         return url.openStream();
aoqi@0:     }
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * The getOutputStream method from the URL. First an attempt is
aoqi@0:      * made to get the URLConnection object for the URL. If that
aoqi@0:      * succeeds, the getOutputStream method on the URLConnection
aoqi@0:      * is returned.
aoqi@0:      *
aoqi@0:      * @return the OutputStream.
aoqi@0:      */
aoqi@0:     public OutputStream getOutputStream() throws IOException {
aoqi@0:         // get the url connection if it is available
aoqi@0:         url_conn = url.openConnection();
aoqi@0: 
aoqi@0:         if (url_conn != null) {
aoqi@0:             url_conn.setDoOutput(true);
aoqi@0:             return url_conn.getOutputStream();
aoqi@0:         } else
aoqi@0:             return null;
aoqi@0:     }
aoqi@0: 
aoqi@0:     /**
aoqi@0:      * Return the URL used to create this DataSource.
aoqi@0:      *
aoqi@0:      * @return The URL.
aoqi@0:      */
aoqi@0:     public URL getURL() {
aoqi@0:         return url;
aoqi@0:     }
aoqi@0: }