Adds option to download any command respond
[mkjsf-moved-to-github.git] / src / main / java / com / indexdata / mkjsf / pazpar2 / data / Responses.java
1 package com.indexdata.mkjsf.pazpar2.data;\r
2 \r
3 import java.io.IOException;\r
4 import java.io.Serializable;\r
5 import java.io.UnsupportedEncodingException;\r
6 import java.util.List;\r
7 import java.util.Map;\r
8 import java.util.concurrent.ConcurrentHashMap;\r
9 \r
10 import javax.enterprise.context.SessionScoped;\r
11 import javax.faces.context.ExternalContext;\r
12 import javax.faces.context.FacesContext;\r
13 import javax.inject.Named;\r
14 \r
15 import org.apache.commons.io.FilenameUtils;\r
16 import org.apache.log4j.Logger;\r
17 \r
18 import com.indexdata.mkjsf.errors.ErrorHelper;\r
19 import com.indexdata.mkjsf.errors.ErrorInterface;\r
20 import com.indexdata.mkjsf.pazpar2.data.sp.SpResponses;\r
21 import com.indexdata.mkjsf.utils.Utils;\r
22 \r
23 @Named("pzresp") @SessionScoped\r
24 public class Responses implements Serializable {\r
25     \r
26   private static final long serialVersionUID = -7543231258346154642L;\r
27   protected Map<String,ResponseDataObject> dataObjects = new ConcurrentHashMap<String,ResponseDataObject>();\r
28   private static Logger logger = Logger.getLogger(Responses.class);\r
29   private ErrorHelper errorHelper = null;\r
30 \r
31   public Responses() {    \r
32   }\r
33   \r
34   public void put(String name, ResponseDataObject responseData) {\r
35     dataObjects.put(name, responseData);\r
36   }\r
37   \r
38   public void setErrorHelper(ErrorHelper helper) {    \r
39     this.errorHelper = helper;\r
40   }\r
41   \r
42   public boolean hasApplicationError () {\r
43     if (getSearch().hasApplicationError()) {\r
44       logger.info("Error detected in search");\r
45       return true;\r
46     }\r
47     for (String name : dataObjects.keySet()) {\r
48       if (dataObjects.get(name).hasApplicationError()) {\r
49         logger.info("Error detected in " + name);\r
50         return true;\r
51       }\r
52     }    \r
53     return false;\r
54   }\r
55   \r
56   /**\r
57    * Returns a search command error, if any, otherwise the first\r
58    * error found for an arbitrary command, if any, otherwise\r
59    * an empty dummy error. \r
60    */    \r
61   public ErrorInterface getCommandError() {\r
62     CommandError error = new CommandError();\r
63     if (dataObjects.get("search").hasApplicationError()) {\r
64       error = dataObjects.get("search").getApplicationError();\r
65       error.setErrorHelper(errorHelper);\r
66     } else {\r
67       for (String name : dataObjects.keySet()) {     \r
68         if (dataObjects.get(name).hasApplicationError()) {     \r
69           error = dataObjects.get(name).getApplicationError();\r
70           error.setErrorHelper(errorHelper);\r
71           break;\r
72         } \r
73       }\r
74     }\r
75     return error;         \r
76   }\r
77   \r
78   public void resetSearchResponses() {\r
79     logger.debug("Resetting show,stat,termlist,bytarget,record,search response objects.");\r
80     dataObjects.put("show", new ShowResponse());\r
81     dataObjects.put("stat", new StatResponse());\r
82     dataObjects.put("termlist", new TermListsResponse());\r
83     dataObjects.put("bytarget", new ByTarget());\r
84     dataObjects.put("record", new RecordResponse());\r
85     dataObjects.put("search", new SearchResponse());    \r
86   }\r
87   \r
88   public void resetAllSessionData () {\r
89     logger.debug("Resetting all response objects");\r
90     dataObjects = new ConcurrentHashMap<String,ResponseDataObject>();    \r
91     resetSearchResponses();\r
92     dataObjects.put("init", new InitResponse());\r
93   }\r
94   \r
95   public InitResponse getInit () {\r
96     logger.info("Request to show init response from " + Utils.objectId(this));\r
97     return ((InitResponse) dataObjects.get("init"));\r
98   }\r
99 \r
100   public ShowResponse getShow () {\r
101     return ((ShowResponse) dataObjects.get("show"));\r
102   }\r
103   \r
104   public StatResponse getStat () {\r
105     return ((StatResponse) dataObjects.get("stat"));\r
106   }\r
107   \r
108   public RecordResponse getRecord() {\r
109     return ((RecordResponse) dataObjects.get("record"));\r
110   }\r
111   \r
112   public SearchResponse getSearch() {\r
113     return ((SearchResponse) dataObjects.get("search"));\r
114   }\r
115   \r
116   public TermListsResponse getTermLists () {\r
117     return ((TermListsResponse) dataObjects.get("termlist"));\r
118   }\r
119   \r
120   public List<TermResponse> getFacetTerms (String facet, int count) {\r
121     return (getTermLists().getTermList(facet).getTerms(count));\r
122   }\r
123     \r
124   public List<TermResponse> getFacetTerms (String facet) {\r
125     return (getTermLists().getTermList(facet).getTerms());\r
126   }\r
127   \r
128   public ByTarget getByTarget() {\r
129     return ((ByTarget) dataObjects.get("bytarget"));\r
130   }\r
131 \r
132   public ResponseDataObject getResponseObject (String name) {\r
133     return dataObjects.get(name);\r
134   }\r
135   \r
136   public boolean hasRecords () {\r
137     return getStat().getRecords() > 0            \r
138            && getShow().getHits() != null \r
139            && getShow().getHits().size()>0;\r
140   }\r
141   \r
142   public String getActiveClients() {    \r
143     if (getShow()!=null) {\r
144       logger.debug("Active clients: "+getShow().getActiveClients());\r
145       return getShow().getActiveClients();\r
146     } else {\r
147       return "";\r
148     }\r
149   }\r
150 \r
151   public SpResponses getSp() {\r
152     // TODO:\r
153     return null;\r
154   }\r
155 \r
156   public void download(String commandName) throws UnsupportedEncodingException, IOException {\r
157     logger.info(Utils.objectId(this) + " got a download request for "\r
158         + commandName);\r
159     FacesContext facesContext = FacesContext.getCurrentInstance();\r
160     ExternalContext externalContext = facesContext.getExternalContext();\r
161     externalContext.setResponseHeader("Content-Type","application/xml; charset=\"utf-8\"");\r
162     externalContext.setResponseHeader("Content-Length",String.valueOf(dataObjects.get(commandName).getXml().getBytes("UTF-8").length));\r
163     externalContext.setResponseHeader("Content-Disposition","attachment;filename=\"" + commandName + ".xml\"");\r
164     externalContext.getResponseOutputStream().write(dataObjects.get(commandName).getXml().getBytes("UTF-8"));\r
165     facesContext.responseComplete();\r
166   }\r
167 }