KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > tigris > scarab > screens > ViewAttachment


1 package org.tigris.scarab.screens;
2
3 /* ================================================================
4  * Copyright (c) 2000-2002 CollabNet. All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions are
8  * met:
9  *
10  * 1. Redistributions of source code must retain the above copyright
11  * notice, this list of conditions and the following disclaimer.
12  *
13  * 2. Redistributions in binary form must reproduce the above copyright
14  * notice, this list of conditions and the following disclaimer in the
15  * documentation and/or other materials provided with the distribution.
16  *
17  * 3. The end-user documentation included with the redistribution, if
18  * any, must include the following acknowlegement: "This product includes
19  * software developed by Collab.Net <http://www.Collab.Net/>."
20  * Alternately, this acknowlegement may appear in the software itself, if
21  * and wherever such third-party acknowlegements normally appear.
22  *
23  * 4. The hosted project names must not be used to endorse or promote
24  * products derived from this software without prior written
25  * permission. For written permission, please contact info@collab.net.
26  *
27  * 5. Products derived from this software may not use the "Tigris" or
28  * "Scarab" names nor may "Tigris" or "Scarab" appear in their names without
29  * prior written permission of Collab.Net.
30  *
31  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
32  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
33  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
34  * IN NO EVENT SHALL COLLAB.NET OR ITS CONTRIBUTORS BE LIABLE FOR ANY
35  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
36  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
37  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
38  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
39  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
40  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
41  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
42  *
43  * ====================================================================
44  *
45  * This software consists of voluntary contributions made by many
46  * individuals on behalf of Collab.Net.
47  */

48
49 // Turbine Stuff
50
import java.io.File JavaDoc;
51 import java.io.BufferedInputStream JavaDoc;
52 import java.io.FileInputStream JavaDoc;
53 import java.io.OutputStream JavaDoc;
54 import javax.servlet.http.HttpServletResponse JavaDoc;
55 import org.apache.turbine.RunData;
56 import org.apache.turbine.TemplateContext;
57 import org.apache.torque.om.NumberKey;
58
59 // Scarab Stuff
60
import org.tigris.scarab.util.Log;
61 import org.tigris.scarab.om.Attachment;
62 import org.tigris.scarab.om.AttachmentManager;
63
64 /**
65  * Sends file contents directly to the output stream.
66  *
67  * @author <a HREF="mailto:jmcnally@collab.net">John McNally</a>
68  * @version $Id: ViewAttachment.java 7365 2003-03-15 21:56:59Z jon $
69  */

70 public class ViewAttachment extends Default
71 {
72     /**
73      * builds up the context for display of variables on the page.
74      */

75     public void doBuildTemplate(RunData data, TemplateContext context)
76         throws Exception JavaDoc
77     {
78         super.doBuildTemplate(data, context);
79         
80         String JavaDoc attachId = data.getParameters().getString("attachId");
81         Attachment attachment = AttachmentManager
82             .getInstance(new NumberKey(attachId), false);
83         String JavaDoc contentType = attachment.getMimeType();
84         HttpServletResponse JavaDoc res = data.getResponse();
85         res.setContentType(contentType);
86         // disposition header does not seem to have that much value, but
87
// adding it in a blind attempt to help win2K, IE client
88
// to handle the downloads better
89
if (!contentType.startsWith("text"))
90         {
91             res.setHeader("Content-Disposition",
92                           "attachment; filename=" + attachment.getFileName());
93         }
94         
95         File JavaDoc f = new File JavaDoc(attachment.getFullPath());
96         res.setContentLength((int)f.length());
97
98         BufferedInputStream JavaDoc bis = null;
99         OutputStream JavaDoc os = data.getResponse().getOutputStream();
100         try
101         {
102             bis = new BufferedInputStream JavaDoc(new FileInputStream JavaDoc(f));
103             byte[] bytes = new byte[2048];
104             int s = 0;
105             while ((s = bis.read(bytes)) != -1)
106             {
107                 try
108                 {
109                     os.write(bytes,0,s);
110                 }
111                 catch (java.io.IOException JavaDoc ioe)
112                 {
113                     Log.get().debug("File download was aborted: " +
114                                     attachment.getFullPath());
115                     break;
116                 }
117             }
118         }
119         finally
120         {
121             if (bis != null)
122             {
123                 bis.close();
124             }
125         }
126
127         // we already sent the response, there is no target to render
128
data.setTarget(null);
129     }
130 }
131
132
Popular Tags