OSDN Git Service

Correct a problem where searching on an attribute without specifying a word search...
[neighbornote/NeighborNote.git] / src / cx / fbn / nevernote / sql / REnSearch.java
1 /*\r
2  * This file is part of NeverNote \r
3  * Copyright 2009 Randy Baumgarte\r
4  * \r
5  * This file may be licensed under the terms of of the\r
6  * GNU General Public License Version 2 (the ``GPL'').\r
7  *\r
8  * Software distributed under the License is distributed\r
9  * on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either\r
10  * express or implied. See the GPL for the specific language\r
11  * governing rights and limitations.\r
12  *\r
13  * You should have received a copy of the GPL along with this\r
14  * program. If not, go to http://www.gnu.org/licenses/gpl.html\r
15  * or write to the Free Software Foundation, Inc.,\r
16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.\r
17  *\r
18 */\r
19 \r
20 \r
21 package cx.fbn.nevernote.sql;\r
22 \r
23 import java.text.SimpleDateFormat;\r
24 import java.util.ArrayList;\r
25 import java.util.Calendar;\r
26 import java.util.GregorianCalendar;\r
27 import java.util.List;\r
28 import java.util.regex.Pattern;\r
29 \r
30 import org.apache.commons.lang.StringEscapeUtils;\r
31 \r
32 import com.evernote.edam.type.Note;\r
33 import com.evernote.edam.type.Notebook;\r
34 import com.evernote.edam.type.Tag;\r
35 \r
36 import cx.fbn.nevernote.sql.driver.NSqlQuery;\r
37 import cx.fbn.nevernote.utilities.ApplicationLogger;\r
38 \r
39 public class REnSearch {\r
40         \r
41         private final List<String>      searchWords;\r
42         private final List<String>  searchPhrases;\r
43         private final List<String>      notebooks;\r
44         private final List<String>      tags;\r
45         private final List<String>      intitle;\r
46         private final List<String>      created;\r
47         private final List<String>      updated;\r
48         private final List<String>      resource;\r
49         private final List<String>      subjectDate;\r
50         private final List<String>      longitude;\r
51         private final List<String>      latitude;\r
52         private final List<String>      altitude;\r
53         private final List<String>      author;\r
54         private final List<String>      source;\r
55         private final List<String>      sourceApplication;\r
56         private final List<String>      recoType;\r
57         private final List<String>      todo;\r
58         private final List<String>  stack;\r
59         private final List<Tag>         tagIndex;\r
60         private final ApplicationLogger logger;\r
61 //      private final DatabaseConnection db;\r
62         private boolean any;\r
63         private int minimumRecognitionWeight = 80;\r
64         private final DatabaseConnection conn;\r
65         \r
66         public REnSearch(DatabaseConnection c, ApplicationLogger l, String s, List<Tag> t, int r) {\r
67                 logger = l;\r
68                 conn = c;\r
69                 tagIndex = t;\r
70                 minimumRecognitionWeight = r;\r
71                 searchWords = new ArrayList<String>();\r
72                 searchPhrases = new ArrayList<String>();\r
73                 notebooks = new ArrayList<String>();\r
74                 tags = new ArrayList<String>();\r
75                 intitle = new ArrayList<String>();\r
76                 created = new  ArrayList<String>();\r
77                 updated = new ArrayList<String>();\r
78                 resource = new ArrayList<String>();\r
79                 subjectDate = new ArrayList<String>();\r
80                 longitude = new ArrayList<String>();\r
81                 latitude = new ArrayList<String>();\r
82                 altitude = new ArrayList<String>();\r
83                 author = new ArrayList<String>();\r
84                 source = new ArrayList<String>();\r
85                 sourceApplication = new ArrayList<String>();\r
86                 recoType = new ArrayList<String>();\r
87                 todo = new ArrayList<String>();\r
88                 any = false;\r
89                 stack = new ArrayList<String>();\r
90                 \r
91                 if (s == null) \r
92                         return;\r
93                 if (s.trim().equals(""))\r
94                         return;\r
95                 \r
96                 resolveSearch(s);\r
97         }\r
98                 \r
99         public List<String> getWords() { return searchWords; }\r
100         public List<String> getNotebooks() { return notebooks; }\r
101         public List<String> getIntitle() {      return intitle; }\r
102         public List<String> getTags() { return tags; }\r
103         public List<String> getResource() {     return resource; }\r
104         public List<String> getAuthor() { return author; }      \r
105         public List<String> getSource() { return source; }      \r
106         public List<String> getSourceApplication() { return sourceApplication; }        \r
107         public List<String> getRecoType() {     return recoType; }      \r
108         public List<String> getToDo() { return todo; }\r
109         public List<String> getLongitude() { return longitude; }\r
110         public List<String> getLatitude() { return latitude; }\r
111         public List<String> getAltitude() { return altitude; }\r
112         public List<String> getCreated() { return created; }\r
113         public List<String> getUpdated() { return updated; }\r
114         public List<String> getSubjectDate() { return subjectDate; }\r
115         public List<String> getStack() { return stack; }\r
116 \r
117         // match tag names\r
118         private boolean matchTagsAll(List<String> tagNames) {\r
119                 List<String> list = getTags();\r
120                                 \r
121                 for (int j=0; j<list.size(); j++) {\r
122                         boolean negative = false;\r
123                         negative = false;\r
124                         if (list.get(j).startsWith("-"))\r
125                                 negative = true;\r
126                         int pos = list.get(j).indexOf(":");\r
127                         String filterName = cleanupWord(list.get(j).substring(pos+1));\r
128                         filterName = filterName.replace("*", ".*");   // setup for regular expression pattern match\r
129                         \r
130                         if (tagNames.size() == 0 && !negative)\r
131                                 return false;\r
132                         if (tagNames.size() == 0 && negative)\r
133                                 return true;\r
134                         \r
135                         boolean good = false;\r
136                         for (int i=0; i<tagNames.size() && !good; i++) {                \r
137                                 boolean matches = Pattern.matches(filterName.toLowerCase(),tagNames.get(i).toLowerCase());\r
138                                 if (matches && !negative)\r
139                                         good = true;\r
140                                 if (!matches && negative)\r
141                                         good = true;\r
142                         }\r
143                         if (!good)\r
144                                 return false;\r
145                 }\r
146                 return true;\r
147         }\r
148         \r
149         // match tag names\r
150         private boolean matchTagsAny(List<String> tagNames) {\r
151                 List<String> list = getTags();\r
152                 if (list.size() == 0)\r
153                         return true;\r
154                 \r
155                 boolean negative = false;               \r
156                 boolean found = false;\r
157                 \r
158                 for (int j=0; j<list.size(); j++) {\r
159                         negative = false;\r
160                         if (list.get(j).startsWith("-"))\r
161                                 negative = true;\r
162                         int pos = list.get(j).indexOf(":");\r
163                         String filterName = cleanupWord(list.get(j).substring(pos+1));\r
164                         filterName = filterName.replace("*", ".*");   // setup for regular expression pattern match\r
165                         \r
166                         if (tagNames.size() == 0)\r
167                                 found = false;\r
168 \r
169                         for (int i=0; i<tagNames.size(); i++) {         \r
170                                 boolean matches = Pattern.matches(filterName.toLowerCase(),tagNames.get(i).toLowerCase());\r
171                                 if (matches)\r
172                                         found = true;\r
173                         }\r
174                 }\r
175                 if (negative)\r
176                         return !found;\r
177                 else\r
178                         return found;\r
179         }\r
180         \r
181         // Match notebooks in search terms against notes\r
182         private boolean matchNotebook(String guid) {\r
183                 if (getNotebooks().size() == 0)\r
184                         return true;\r
185                 NotebookTable bookTable = new NotebookTable(logger, conn);\r
186                 List<Notebook> books = bookTable.getAll();\r
187 \r
188                 String name = new String("");\r
189                 for (int i=0; i<books.size(); i++) {\r
190                         if (guid.equalsIgnoreCase(books.get(i).getGuid())) {\r
191                                 name = books.get(i).getName();\r
192                                 i=books.size();\r
193                         }\r
194                 }\r
195                 if (any)\r
196                         return matchListAny(getNotebooks(), name);\r
197                 else\r
198                         return matchListAll(getNotebooks(), name);\r
199         }\r
200         // Match notebooks in search terms against notes\r
201         private boolean matchNotebookStack(String guid) {\r
202                 if (getStack().size() == 0)\r
203                         return true;\r
204                 NotebookTable bookTable = new NotebookTable(logger, conn);\r
205                 List<Notebook> books = bookTable.getAll();\r
206 \r
207                 String name = new String("");\r
208                 for (int i=0; i<books.size(); i++) {\r
209                         if (guid.equalsIgnoreCase(books.get(i).getGuid())) {\r
210                                 name = books.get(i).getStack();\r
211                                 i=books.size();\r
212                         }\r
213                 }\r
214                 if (name == null)\r
215                         name = "";\r
216                 if (any)\r
217                         return matchListAny(getStack(), name);\r
218                 else\r
219                         return matchListAll(getStack(), name);\r
220         }\r
221 \r
222         // Match notebooks in search terms against notes\r
223         private boolean matchListAny(List<String> list, String title) {\r
224                 if (list.size() == 0)\r
225                         return true;\r
226                 boolean negative = false;\r
227                 boolean found = false;\r
228                 for (int i=0; i<list.size(); i++) {\r
229                         int pos = list.get(i).indexOf(":");\r
230                         negative = false;\r
231                         if (list.get(i).startsWith("-"))\r
232                                 negative = true;\r
233                         String filterName = cleanupWord(list.get(i).substring(pos+1));\r
234                         filterName = filterName.replace("*", ".*");   // setup for regular expression pattern match\r
235                         boolean matches = Pattern.matches(filterName.toLowerCase(),title.toLowerCase());\r
236                         if (matches)\r
237                                 found = true;\r
238                 }\r
239                 if (negative)\r
240                         return !found;\r
241                 else\r
242                         return found;\r
243         }\r
244         // Match notebooks in search terms against notes\r
245         private boolean matchContentAny(Note n) {\r
246                 if (todo.size() == 0 && resource.size() == 0 && searchPhrases.size() == 0)\r
247                         return true;\r
248 \r
249                 // pull back the record\r
250                 n = conn.getNoteTable().getNote(n.getGuid(), true, true, false, false, false);\r
251 \r
252                 // Check for search phrases\r
253                 String text = StringEscapeUtils.unescapeHtml(n.getContent().replaceAll("\\<.*?\\>", "")).toLowerCase();\r
254                 boolean negative = false;\r
255                 for (int i=0; i<searchPhrases.size(); i++) {\r
256                         String phrase = searchPhrases.get(i);\r
257                         if (phrase.startsWith("-")) {\r
258                                 negative = true;\r
259                                 phrase = phrase.substring(1);\r
260                         } else\r
261                                 negative = false;\r
262                         phrase = phrase.substring(1);\r
263                         phrase = phrase.substring(0,phrase.length()-1);\r
264                         if (text.indexOf(phrase)>=0) {\r
265                                 if (negative)\r
266                                         return false;\r
267                                 else\r
268                                         return true;\r
269                         }\r
270                         if (text.indexOf(phrase)<0 && negative)\r
271                                 return true;\r
272                 }\r
273                 \r
274                 for (int i=0; i<todo.size(); i++) {\r
275                         String value = todo.get(i);\r
276                         value = value.replace("\"", "");\r
277                         boolean desiredState;\r
278                         if (!value.endsWith(":false") && !value.endsWith(":true") && !value.endsWith(":*") && !value.endsWith("*"))\r
279                                 return false;\r
280                         if (value.endsWith(":false"))\r
281                                 desiredState = false;\r
282                         else\r
283                                 desiredState = true;\r
284                         if (value.startsWith("-"))\r
285                                 desiredState = !desiredState;\r
286                         int pos = n.getContent().indexOf("<en-todo");\r
287                         if (pos == -1 && value.startsWith("-") && (value.endsWith("*") || value.endsWith(":")))\r
288                                 return true;\r
289                         if (value.endsWith("*"))\r
290                                 return true;\r
291                         while (pos > -1) {\r
292                                 int endPos = n.getContent().indexOf("/>", pos);\r
293                                 String segment = n.getContent().substring(pos, endPos);\r
294                                 boolean currentState;\r
295                                 if (segment.toLowerCase().indexOf("checked=\"true\"") == -1)\r
296                                         currentState = false;\r
297                                 else\r
298                                         currentState = true;\r
299                                 if (desiredState == currentState)\r
300                                         return true;\r
301                                 \r
302                                 pos = n.getContent().indexOf("<en-todo", pos+1);\r
303                         }\r
304                 }\r
305                 \r
306                 // Check resources\r
307                 for (int i=0; i<resource.size(); i++) {\r
308                         String resourceString = resource.get(i);\r
309                         resourceString = resourceString.replace("\"", "");\r
310                         if (resourceString.startsWith("-"))\r
311                                 negative = true;\r
312                         resourceString = resourceString.substring(resourceString.indexOf(":")+1);\r
313                         for (int j=0; j<n.getResourcesSize(); j++) {\r
314                                 boolean match = stringMatch(n.getResources().get(j).getMime(), resourceString, negative);\r
315                                 if (match)\r
316                                         return true;\r
317                         }\r
318                 }\r
319                 return false;\r
320         }\r
321         \r
322         \r
323         // Take the initial search & split it apart\r
324         private void resolveSearch(String search) {\r
325                 List<String> words = new ArrayList<String>();\r
326                 StringBuffer b = new StringBuffer(search);\r
327                 \r
328                 int len = search.length();\r
329                 char nextChar = ' ';\r
330                 boolean quote = false;\r
331                 for (int i=0; i<len; i++) {\r
332                         if (search.charAt(i)==nextChar && !quote) {\r
333                                 b.setCharAt(i,'\0');\r
334                                 nextChar = ' ';\r
335                         } else {\r
336                                 if (search.charAt(i)=='\"') {\r
337                                         if (!quote) {\r
338                                                 quote=true;\r
339                                         } else {\r
340                                                 quote=false;\r
341                                         }\r
342                                 }\r
343                         }\r
344                         if (((i+2)<len) && search.charAt(i) == '\\') {\r
345                                 i=i+2;\r
346                         }\r
347                 }\r
348                 \r
349                 search = b.toString();\r
350                 int pos = 0;\r
351                 for (int i=0; i<search.length(); i++) {\r
352                         if (search.charAt(i) == '\0') {\r
353                                 search = search.substring(1);\r
354                                 i=0;\r
355                         } else {\r
356                                 pos = search.indexOf('\0');\r
357                                 if (pos > 0) {\r
358                                         words.add(search.substring(0,pos).toLowerCase());\r
359                                         search = search.substring(pos);\r
360                                         i=0;\r
361                                 }\r
362                         }\r
363                 }\r
364                 if (search.charAt(0)=='\0')     \r
365                         words.add(search.substring(1).toLowerCase());\r
366                 else\r
367                         words.add(search.toLowerCase());\r
368                 parseTerms(words);\r
369         }\r
370 \r
371         \r
372         // Parse out individual words into separate lists\r
373         // Supported options\r
374         // Tags\r
375         // Notebooks\r
376         // Intitle\r
377         // author\r
378         // source\r
379         // source application\r
380         // created\r
381         // updated\r
382         // subject date\r
383 \r
384         private void parseTerms(List<String> words) {\r
385                 \r
386                 for (int i=0; i<words.size(); i++) {\r
387                         String word = words.get(i);\r
388                         int pos = word.indexOf(":");\r
389                         if (word.startsWith("any:")) {\r
390                                 any = true;\r
391                                 word = word.substring(4).trim();\r
392                                 pos = word.indexOf(":");\r
393                         }\r
394                         boolean searchPhrase = false;\r
395                         if (pos < 0 && word.indexOf(" ") > 0) {\r
396                                 searchPhrase=true;\r
397                                 searchPhrases.add(word.toLowerCase());\r
398                         }\r
399                         if (!searchPhrase && pos < 0) \r
400                                 getWords().add(word); \r
401 //                              getWords().add("*"+word+"*");           //// WILDCARD\r
402                         if (word.startsWith("intitle:")) \r
403                                 intitle.add("*"+word+"*");\r
404                         if (word.startsWith("-intitle:")) \r
405                                 intitle.add("*"+word+"*");\r
406                         if (word.startsWith("notebook:")) \r
407                                 notebooks.add(word);\r
408                         if (word.startsWith("-notebook:")) \r
409                                 notebooks.add(word);\r
410                         if (word.startsWith("tag:")) \r
411                                 tags.add(word);\r
412                         if (word.startsWith("-tag:")) \r
413                                 tags.add(word);\r
414                         if (word.startsWith("resource:")) \r
415                                 resource.add(word);\r
416                         if (word.startsWith("-resource:")) \r
417                                 resource.add(word);\r
418                         if (word.startsWith("author:")) \r
419                                 author.add(word);\r
420                         if (word.startsWith("-author:")) \r
421                                 author.add(word);\r
422                         if (word.startsWith("source:")) \r
423                                 source.add(word);\r
424                         if (word.startsWith("-source:")) \r
425                                 source.add(word);\r
426                         if (word.startsWith("sourceapplication:")) \r
427                                 sourceApplication.add(word);\r
428                         if (word.startsWith("-sourceapplication:")) \r
429                                 sourceApplication.add(word);\r
430                         if (word.startsWith("recotype:")) \r
431                                 recoType.add(word);\r
432                         if (word.startsWith("-recotype:")) \r
433                                 recoType.add(word);\r
434                         if (word.startsWith("todo:")) \r
435                                 todo.add(word);\r
436                         if (word.startsWith("-todo:")) \r
437                                 todo.add(word);\r
438                         if (word.startsWith("stack:"))\r
439                                 stack.add(word);\r
440                         if (word.startsWith("-stack:"))\r
441                                 stack.add(word);\r
442 \r
443                         if (word.startsWith("latitude:")) \r
444                                 latitude.add(word);\r
445                         if (word.startsWith("-latitude:")) \r
446                                 latitude.add(word);\r
447                         if (word.startsWith("longitude:")) \r
448                                 longitude.add(word);\r
449                         if (word.startsWith("-longitude:")) \r
450                                 longitude.add(word);\r
451                         if (word.startsWith("altitude:")) \r
452                                 altitude.add(word);\r
453                         if (word.startsWith("-altitude:")) \r
454                                 altitude.add(word);\r
455 \r
456                         if (word.startsWith("created:")) \r
457                                 created.add(word);\r
458                         if (word.startsWith("-created:")) \r
459                                 created.add(word);\r
460                         if (word.startsWith("updated:")) \r
461                                 updated.add(word);\r
462                         if (word.startsWith("-updated:")) \r
463                                 updated.add(word);\r
464                         if (word.startsWith("subjectdate:")) \r
465                                 created.add(word);\r
466                         if (word.startsWith("-subjectdate:")) \r
467                                 created.add(word);\r
468 \r
469                 }\r
470         }\r
471         // Match notebooks in search terms against notes\r
472         private boolean matchListAll(List<String> list, String title) {\r
473                 if (list.size() == 0)\r
474                         return true;\r
475                 boolean negative = false;\r
476                 for (int i=0; i<list.size(); i++) {\r
477                         int pos = list.get(i).indexOf(":");\r
478                         negative = false;\r
479                         if (list.get(i).startsWith("-"))\r
480                                 negative = true;\r
481                         String filterName = cleanupWord(list.get(i).substring(pos+1));\r
482                         filterName = filterName.replace("*", ".*");   // setup for regular expression pattern match\r
483                         boolean matches = Pattern.matches(filterName.toLowerCase(),title.toLowerCase());\r
484                         if (matches && negative)\r
485                                 return false;\r
486                         if (matches && !negative)\r
487                                 return true;\r
488                 }\r
489                 if (negative)\r
490                         return true;\r
491                 else\r
492                         return false;\r
493         }\r
494         // Match notebooks in search terms against notes\r
495         private boolean matchContentAll(Note n) {\r
496                 if (todo.size() == 0 && resource.size() == 0 && searchPhrases.size() == 0)\r
497                         return true;\r
498                 \r
499                 boolean returnTodo = false;\r
500                 boolean returnResource = false;\r
501                 boolean returnPhrase = false;\r
502                 \r
503                 if (todo.size() == 0)\r
504                         returnTodo = true;\r
505                 if (resource.size() == 0)\r
506                         returnResource = true;\r
507                 if (searchPhrases.size() == 0)\r
508                         returnPhrase = true;\r
509                 \r
510                 \r
511                 n = conn.getNoteTable().getNote(n.getGuid(), true, true, false, false, false);\r
512                 \r
513                 // Check for search phrases\r
514                 String text = StringEscapeUtils.unescapeHtml(n.getContent().replaceAll("\\<.*?\\>", "")).toLowerCase();\r
515                 boolean negative = false;\r
516                 for (int i=0; i<searchPhrases.size(); i++) {\r
517                         String phrase = searchPhrases.get(i);\r
518                         if (phrase.startsWith("-")) {\r
519                                 negative = true;\r
520                                 phrase = phrase.substring(1);\r
521                         } else\r
522                                 negative = false;\r
523                         phrase = phrase.substring(1);\r
524                         phrase = phrase.substring(0,phrase.length()-1);\r
525                         if (text.indexOf(phrase)>=0) {\r
526                                 if (!negative)\r
527                                         returnPhrase = true;\r
528                         }\r
529                         if (text.indexOf(phrase)<0 && negative)\r
530                                 returnPhrase = true;\r
531                 }\r
532 \r
533                 \r
534                 for (int i=0; i<todo.size(); i++) {\r
535                         String value = todo.get(i);\r
536                         value = value.replace("\"", "");\r
537                         boolean desiredState;\r
538                         if (!value.endsWith(":false") && !value.endsWith(":true") && !value.endsWith(":*") && !value.endsWith("*"))\r
539                                 return false;\r
540                         if (value.endsWith(":false"))\r
541                                 desiredState = false;\r
542                         else\r
543                                 desiredState = true;\r
544                         if (value.startsWith("-"))\r
545                                 desiredState = !desiredState;\r
546                         int pos = n.getContent().indexOf("<en-todo");\r
547                         if (pos == -1 && value.startsWith("-") && (value.endsWith("*") || value.endsWith(":")))\r
548                                 return true;\r
549                         if (pos > -1 && value.startsWith("-") && (value.endsWith("*") || value.endsWith(":")))\r
550                                 return false;\r
551                         if (pos == -1) \r
552                                 return false;\r
553                         if (value.endsWith("*"))\r
554                                 returnTodo = true;\r
555                         while (pos > -1) {\r
556                                 int endPos = n.getContent().indexOf("/>", pos);\r
557                                 String segment = n.getContent().substring(pos, endPos);\r
558                                 boolean currentState;\r
559                                 if (segment.toLowerCase().indexOf("checked=\"true\"") == -1)\r
560                                         currentState = false;\r
561                                 else\r
562                                         currentState = true;\r
563                                 if (desiredState == currentState)\r
564                                         returnTodo = true;\r
565                                 \r
566                                 pos = n.getContent().indexOf("<en-todo", pos+1);\r
567                         }\r
568                 }\r
569                 \r
570                 // Check resources\r
571                 for (int i=0; i<resource.size(); i++) {\r
572                         String resourceString = resource.get(i);\r
573                         resourceString = resourceString.replace("\"", "");\r
574                         negative = false;\r
575                         if (resourceString.startsWith("-"))\r
576                                 negative = true;\r
577                         resourceString = resourceString.substring(resourceString.indexOf(":")+1);\r
578                         if (resourceString.equals(""))\r
579                                 return false;\r
580                         for (int j=0; j<n.getResourcesSize(); j++) {\r
581                                 boolean match = stringMatch(n.getResources().get(j).getMime(), resourceString, negative);\r
582                                 if (!match)\r
583                                         return false;\r
584                                 returnResource = true;\r
585                         }\r
586                 }\r
587                 \r
588                 return returnResource && returnTodo && returnPhrase;\r
589         }\r
590         \r
591         private boolean stringMatch(String content, String text, boolean negative) {\r
592                 String regex;\r
593                 if (content == null && !negative)\r
594                         return false;\r
595                 if (content == null && negative)\r
596                         return true;\r
597                 \r
598                 if (text.endsWith("*")) {\r
599                         text = text.substring(0,text.length()-1);\r
600                         regex = text;\r
601                 } else {\r
602                         regex = text;\r
603                 }\r
604                 content = content.toLowerCase();\r
605                 regex = regex.toLowerCase();\r
606                 boolean matches = content.startsWith(regex);\r
607                 if (negative)\r
608                         return !matches;\r
609                 return matches;\r
610         }\r
611         \r
612         // Remove odd strings from search terms\r
613         private String cleanupWord(String word) {\r
614                 if (word.startsWith("\""))\r
615                         word = word.substring(1);\r
616                 if (word.endsWith("\""))\r
617             word = word.substring(0,word.length()-1);\r
618                 word = word.replace("\\\"","\"");\r
619                 word = word.replace("\\\\","\\");\r
620                 \r
621                 return word;\r
622         }\r
623 \r
624         \r
625         // Match dates\r
626         private boolean matchDatesAll(List<String> dates, long noteDate) {\r
627                 if (dates.size()== 0) \r
628                         return true;\r
629                 \r
630                 boolean negative = false;\r
631                 for (int i=0; i<dates.size(); i++) {\r
632                         String requiredDate = dates.get(i);\r
633                         if (requiredDate.startsWith("-"))\r
634                                 negative = true;\r
635                         \r
636                         int response = 0;\r
637                         requiredDate = requiredDate.substring(requiredDate.indexOf(":")+1);\r
638                         try {\r
639                                 response = dateCheck(requiredDate, noteDate);\r
640                         } catch (java.lang.NumberFormatException e) {return false;}  {\r
641                                 if (negative && response < 0)\r
642                                         return false;\r
643                                 if (!negative && response > 0)\r
644                                         return false;\r
645                         }\r
646                 }\r
647                 return true;\r
648         }\r
649         private boolean matchDatesAny(List<String> dates, long noteDate) {\r
650                 if (dates.size()== 0) \r
651                         return true;\r
652                 \r
653                 boolean negative = false;\r
654                 for (int i=0; i<dates.size(); i++) {\r
655                         String requiredDate = dates.get(i);\r
656                         if (requiredDate.startsWith("-"))\r
657                                 negative = true;\r
658                         \r
659                         int response = 0;\r
660                         requiredDate = requiredDate.substring(requiredDate.indexOf(":")+1);\r
661                         try {\r
662                                 response = dateCheck(requiredDate, noteDate);\r
663                         } catch (java.lang.NumberFormatException e) {return false;}  {\r
664                                 if (negative && response > 0)\r
665                                         return true;\r
666                                 if (!negative && response < 0)\r
667                                         return true;\r
668                         }\r
669                 }\r
670                 return false;\r
671         }\r
672         \r
673         @SuppressWarnings("unused")\r
674         private void printCalendar(Calendar calendar) {\r
675                 // define output format and print\r
676                 SimpleDateFormat sdf = new SimpleDateFormat("d MMM yyyy hh:mm:ss aaa");\r
677                 String date = sdf.format(calendar.getTime());\r
678                 System.err.print(date);\r
679                 calendar = new GregorianCalendar();\r
680         }\r
681         \r
682         \r
683         //****************************************\r
684         //****************************************\r
685         // Match search terms against notes\r
686         //****************************************\r
687         //****************************************\r
688         public List<Note> matchWords() {\r
689                 logger.log(logger.EXTREME, "Inside EnSearch.matchWords()");\r
690                 boolean subSelect = false;\r
691                 \r
692                 NoteTable noteTable = new NoteTable(logger, conn);  \r
693                 List<String> validGuids = new ArrayList<String>();\r
694                 \r
695                 if (searchWords.size() > 0) \r
696                         subSelect = true;\r
697 \r
698                 NSqlQuery query = new NSqlQuery(conn.getConnection());\r
699                 // Build a temp table for GUID results\r
700                 if (!conn.dbTableExists("SEARCH_RESULTS")) {\r
701                         query.exec("create temporary table SEARCH_RESULTS (guid varchar)");\r
702                         query.exec("create temporary table SEARCH_RESULTS_MERGE (guid varchar)");\r
703                 } else {\r
704                         query. exec("Delete from SEARCH_RESULTS");\r
705                         query. exec("Delete from SEARCH_RESULTS_MERGE");\r
706                 }\r
707 \r
708                 NSqlQuery insertQuery = new NSqlQuery(conn.getConnection());\r
709                 NSqlQuery indexQuery = new NSqlQuery(conn.getIndexConnection());\r
710                 NSqlQuery mergeQuery = new NSqlQuery(conn.getConnection());\r
711                 NSqlQuery deleteQuery = new NSqlQuery(conn.getConnection());\r
712                 \r
713                 indexQuery.prepare("Select distinct guid from words where weight >= " +minimumRecognitionWeight +\r
714                 " and word=:word");\r
715                 insertQuery.prepare("Insert into SEARCH_RESULTS (guid) values (:guid)");\r
716                 mergeQuery.prepare("Insert into SEARCH_RESULTS_MERGE (guid) values (:guid)");\r
717                 \r
718                 if (subSelect) {\r
719                         for (int i=0; i<getWords().size(); i++) {\r
720                                 indexQuery.bindValue(":word", getWords().get(i));\r
721                                 indexQuery.exec();\r
722                                 String guid = null;\r
723                                 while(indexQuery.next()) {\r
724                                         guid = indexQuery.valueString(0);\r
725                                         if (i==0 || any) {\r
726                                                 insertQuery.bindValue(":guid", guid);\r
727                                                 insertQuery.exec();\r
728                                         } else {\r
729                                                 mergeQuery.bindValue(":guid", guid);\r
730                                                 mergeQuery.exec();\r
731                                         }\r
732                                 }\r
733                                 if (i>0 && !any) {\r
734                                         deleteQuery.exec("Delete from SEARCH_RESULTS where guid not in (select guid from SEARCH_RESULTS_MERGE)");\r
735                                         deleteQuery.exec("Delete from SEARCH_RESULTS_MERGE");\r
736                                 }\r
737                         }\r
738 \r
739                         query.prepare("Select distinct guid from Note where guid in (Select guid from SEARCH_RESULTS)");\r
740                         if (!query.exec()) \r
741                                 logger.log(logger.LOW, "Error merging search results:" + query.lastError());\r
742                 \r
743                         while (query.next()) {\r
744                                 validGuids.add(query.valueString(0));\r
745                         }\r
746                 }\r
747                 \r
748                 List<Note> noteIndex = noteTable.getAllNotes();\r
749                 List<Note> guids = new ArrayList<Note>();\r
750                 for (int i=0; i<noteIndex.size(); i++) {\r
751                         Note n = noteIndex.get(i);\r
752                         boolean good = true;\r
753                         \r
754                         if (!validGuids.contains(n.getGuid()) && subSelect)\r
755                                 good = false;\r
756                                                 \r
757                         // Start matching special stuff, like tags & notebooks\r
758                         if (any) {\r
759                                 if (good && !matchTagsAny(n.getTagNames()))\r
760                                         good = false;\r
761                                 if (good && !matchNotebook(n.getNotebookGuid()))\r
762                                         good = false;\r
763                                 if (good && !matchNotebookStack(n.getNotebookGuid()))\r
764                                         good = false;\r
765                                 if (good && !matchListAny(getIntitle(), n.getTitle()))\r
766                                         good = false;\r
767                                 if (good && !matchListAny(getAuthor(), n.getAttributes().getAuthor()))\r
768                                         good = false;\r
769                                 if (good && !matchListAny(getSource(), n.getAttributes().getSource()))\r
770                                         good = false;\r
771                                 if (good && !matchListAny(getSourceApplication(), n.getAttributes().getSourceApplication()))\r
772                                         good = false;\r
773                                 if (good && !matchContentAny(n))\r
774                                         good = false;\r
775                                 if (good && !matchDatesAny(getCreated(), n.getCreated()))\r
776                                         good = false;\r
777                                 if (good && !matchDatesAny(getUpdated(), n.getUpdated()))\r
778                                         good = false;\r
779                                 if (good && n.getAttributes() != null && !matchDatesAny(getSubjectDate(), n.getAttributes().getSubjectDate()))\r
780                                         good = false;\r
781                         } else {\r
782                                 if (good && !matchTagsAll(n.getTagNames()))\r
783                                         good = false;\r
784                                 if (good && !matchNotebook(n.getNotebookGuid()))\r
785                                         good = false;\r
786                                 if (good && !matchNotebookStack(n.getNotebookGuid()))\r
787                                         good = false;\r
788                                 if (good && !matchListAll(getIntitle(), n.getTitle()))\r
789                                         good = false;\r
790                                 if (good && !matchListAll(getAuthor(), n.getAttributes().getAuthor()))\r
791                                         good = false;\r
792                                 if (good && !matchListAll(getSource(), n.getAttributes().getSource()))\r
793                                         good = false;\r
794                                 if (good && !matchListAll(getSourceApplication(), n.getAttributes().getSourceApplication()))\r
795                                         good = false;\r
796                                 if (good && !matchContentAll(n))\r
797                                         good = false;\r
798                                 if (good && !matchDatesAll(getCreated(), n.getCreated()))\r
799                                         good = false;\r
800                                 if (good && !matchDatesAll(getUpdated(), n.getUpdated()))\r
801                                         good = false;\r
802                                 if (good && n.getAttributes() != null && !matchDatesAll(getSubjectDate(), n.getAttributes().getSubjectDate()))\r
803                                         good = false;\r
804                         }\r
805                         if (good) {\r
806                                 guids.add(n);\r
807                         }\r
808                 }\r
809                 \r
810                 // For performance reasons, we didn't get the tags for every note individually.  We now need to \r
811                 // get them\r
812                 List<NoteTagsRecord> noteTags = noteTable.noteTagsTable.getAllNoteTags();\r
813                 for (int i=0; i<guids.size(); i++) {\r
814                         List<String> tags = new ArrayList<String>();\r
815                         List<String> names = new ArrayList<String>();\r
816                         for (int j=0; j<noteTags.size(); j++) {\r
817                                 if (guids.get(i).getGuid().equals(noteTags.get(j).noteGuid)) {\r
818                                         tags.add(noteTags.get(j).tagGuid);\r
819                                         names.add(getTagNameByGuid(noteTags.get(j).tagGuid));\r
820                                 }\r
821                         }\r
822                         \r
823                         guids.get(i).setTagGuids(tags);\r
824                         guids.get(i).setTagNames(names);\r
825                 };\r
826                 logger.log(logger.EXTREME, "Leaving EnSearch.matchWords()");\r
827                 return guids;\r
828         }\r
829         \r
830         \r
831         \r
832         private String getTagNameByGuid(String guid) {\r
833                 for (int i=0; i<tagIndex.size(); i++) {\r
834                         if (tagIndex.get(i).getGuid().equals(guid)) \r
835                                         return tagIndex.get(i).getName();\r
836                 }               \r
837                 return "";\r
838         }\r
839 \r
840         // Compare dates\r
841         public int dateCheck(String date, long noteDate)  throws java.lang.NumberFormatException  {\r
842                 int offset = 0;\r
843                 boolean found = false;\r
844                 GregorianCalendar calendar = new GregorianCalendar();\r
845                 \r
846                 if (date.contains("-")) {\r
847                         String modifier = date.substring(date.indexOf("-")+1);\r
848                         offset = new Integer(modifier);\r
849                         offset = 0-offset;\r
850                         date = date.substring(0,date.indexOf("-"));\r
851                 }\r
852                 \r
853                 if (date.contains("+")) {\r
854                         String modifier = date.substring(date.indexOf("+")+1);\r
855                         offset = new Integer(modifier);\r
856                         date = date.substring(0,date.indexOf("+"));\r
857                 }\r
858                 \r
859                 if (date.equalsIgnoreCase("today")) {\r
860                         calendar.add(Calendar.DATE, offset);\r
861                         calendar.set(Calendar.HOUR, 0);\r
862                         calendar.set(Calendar.MINUTE, 0);\r
863                         calendar.set(Calendar.SECOND, 1);\r
864                         found = true;\r
865                 }\r
866                 \r
867                 if (date.equalsIgnoreCase("month")) {\r
868                         calendar.add(Calendar.MONTH, offset);\r
869                         calendar.set(Calendar.DAY_OF_MONTH, 1);\r
870                         calendar.set(Calendar.HOUR, 0);\r
871                         calendar.set(Calendar.MINUTE, 0);\r
872                         calendar.set(Calendar.SECOND, 1);\r
873                         found = true;\r
874                 }\r
875 \r
876                 if (date.equalsIgnoreCase("year")) {\r
877                         calendar.add(Calendar.YEAR, offset);\r
878                         calendar.set(Calendar.MONTH, Calendar.JANUARY);\r
879                         calendar.set(Calendar.DAY_OF_MONTH, 1);\r
880                         calendar.set(Calendar.HOUR, 0);\r
881                         calendar.set(Calendar.MINUTE, 0);\r
882                         calendar.set(Calendar.SECOND, 1);\r
883                         found = true;\r
884                 }\r
885 \r
886                 if (date.equalsIgnoreCase("week")) {\r
887                         calendar.add(Calendar.DATE, 0-calendar.get(Calendar.DAY_OF_WEEK)+1);\r
888                         calendar.add(Calendar.DATE,(offset*7));\r
889                         calendar.set(Calendar.HOUR, 0);\r
890                         calendar.set(Calendar.MINUTE, 0);\r
891                         calendar.set(Calendar.SECOND, 1);\r
892 \r
893                         found = true;\r
894                 }\r
895                 \r
896                 // If nothing was found, then we have a date number\r
897                 if (!found) {\r
898                         calendar = stringToGregorianCalendar(date);\r
899                 }\r
900                 \r
901                 \r
902                 String dateTimeFormat = new String("yyyyMMdd-HHmmss");\r
903                 SimpleDateFormat simple = new SimpleDateFormat(dateTimeFormat);\r
904                 StringBuilder creationDate = new StringBuilder(simple.format(noteDate));\r
905                 GregorianCalendar nCalendar = stringToGregorianCalendar(creationDate.toString().replace("-", "T"));\r
906                 if (calendar == null || nCalendar == null)  // If we have something invalid, it automatically fails\r
907                         return 1;\r
908                 return calendar.compareTo(nCalendar);\r
909         }\r
910         private GregorianCalendar stringToGregorianCalendar(String date) {\r
911                 String datePart = date;\r
912                 GregorianCalendar calendar = new GregorianCalendar();\r
913                 boolean GMT = false;\r
914                 String timePart = "";\r
915                 if (date.contains("T")) {\r
916                         datePart = date.substring(0,date.indexOf("T"));\r
917                         timePart = date.substring(date.indexOf("T")+1);\r
918                 } else {\r
919                         timePart = "000001";\r
920                 }\r
921                 if (datePart.length() != 8)\r
922                         return null;\r
923                 calendar.set(Calendar.YEAR, new Integer(datePart.substring(0,4)));\r
924                 calendar.set(Calendar.MONTH, new Integer(datePart.substring(4,6))-1);\r
925                 calendar.set(Calendar.DAY_OF_MONTH, new Integer(datePart.substring(6)));\r
926                 if (timePart.endsWith("Z")) {\r
927                         GMT = true;\r
928                         timePart = timePart.substring(0,timePart.length()-1);\r
929                 }\r
930                 timePart = timePart.concat("000000");\r
931                 timePart = timePart.substring(0,6);\r
932                 calendar.set(Calendar.HOUR, new Integer(timePart.substring(0,2)));\r
933                 calendar.set(Calendar.MINUTE, new Integer(timePart.substring(2,4)));\r
934                 calendar.set(Calendar.SECOND, new Integer(timePart.substring(4)));\r
935                 if (GMT)\r
936                         calendar.set(Calendar.ZONE_OFFSET, -1*(calendar.get(Calendar.ZONE_OFFSET)/(1000*60*60)));\r
937                 return calendar;\r
938 \r
939         }\r
940                 \r
941 }\r