1 package org.simantics.workbench.search;
4 import java.io.IOException;
6 import java.net.URLDecoder;
7 import java.net.URLEncoder;
8 import java.util.Collections;
9 import java.util.HashMap;
10 import java.util.HashSet;
14 import org.simantics.db.layer0.genericrelation.IndexQueries;
16 public class SearchQuery {
18 private String originalQuery;
19 private Map<String,String> searchFlags = new HashMap<String, String>();
20 private Set<String> searchParams = new HashSet<String>();
22 public SearchQuery() {
25 public SearchQuery(String originalQuery) {
26 this.originalQuery = originalQuery;
29 public void setOriginalQuery(String originalQuery) {
30 this.originalQuery = originalQuery;
33 public String getOriginalQuery() {
37 public String getQuery(Set<String> supportedParams) {
38 if (originalQuery.trim().isEmpty())
42 for (String key : searchFlags.keySet()) {
43 if (!supportedParams.contains(key))
45 if (!"on".equals(searchFlags.get(key)))
49 // String prefix = originalQuery.startsWith("*") ? "" : "*";
50 // String suffix = originalQuery.endsWith("*") ? "" : "*";
51 // query += key+":"+prefix+originalQuery+suffix;
52 query += key+":"+originalQuery;
58 public String getQuery(String... supportedParams) {
59 Set<String> set = new HashSet<String>();
60 for (String s : supportedParams)
67 public void setSearchFlag(String key, String value) {
68 searchFlags.put(key, value);
71 public void setSearchParam(String key) {
72 searchParams.add(key);
75 public Map<String, String> getSearchFlags() {
76 return Collections.unmodifiableMap(searchFlags);
79 public Set<String> getSearchParams() {
80 return Collections.unmodifiableSet(searchParams);
83 public boolean getBooleanFlag(String key) {
84 if ("on".equals(searchFlags.get(key)))
89 public boolean getBooleanParam(String key) {
90 if (searchParams.contains(key))
96 public String toString() {
100 public SearchQuery withOriginalQuery(String query) {
101 SearchQuery ret = new SearchQuery(query);
102 ret.searchFlags = new HashMap<String, String>(searchFlags);
103 ret.searchParams = new HashSet<String>(searchParams);
107 public SearchQuery escaped(boolean escapeWildcards) {
108 return withOriginalQuery( IndexQueries.escape( originalQuery, escapeWildcards ) );
111 public static URL encode(File file, SearchQuery query) throws IOException {
112 URL url = file.toURI().toURL();
113 String s = url.toString() + "?search=" + URLEncoder.encode(query.getOriginalQuery(), "UTF-8");
114 Map<String, String> flags = query.getSearchFlags();
115 for (String key : flags.keySet()) {
116 s += "&"+key+"="+flags.get(key);
119 for (String param : query.getSearchParams()) {
127 public static SearchQuery decode(URL url) throws IOException{
128 String query = url.getQuery();
129 SearchQuery searchQuery = new SearchQuery();
131 for (String arg : query.split("&")) {
132 String[] keyvalue = arg.split("=");
133 if ("search".equals(keyvalue[0])) {
134 if (keyvalue.length == 2)
135 searchQuery.setOriginalQuery(URLDecoder.decode(keyvalue[1], "UTF-8"));
136 } else if (keyvalue.length == 2){
137 searchQuery.setSearchFlag(URLDecoder.decode(keyvalue[0], "UTF-8"), URLDecoder.decode(keyvalue[1], "UTF-8"));
138 } else if (keyvalue.length == 1) {
139 searchQuery.setSearchParam(URLDecoder.decode(keyvalue[0], "UTF-8"));