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, j=0; i<len; i++, j++) {\r
 332                         if (search.charAt(i)==nextChar && !quote) {\r
 333                                 b.setCharAt(j,'\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                                                 j++;\r
 342                                                 b.insert(j, "\0");\r
 343                                         }\r
 344                                 }\r
 345                         }\r
 346                         if (((i+2)<len) && search.charAt(i) == '\\') {\r
 347                                 i=i+2;\r
 348                         }\r
 349                 }\r
 350                 \r
 351                 search = b.toString();\r
 352                 int pos = 0;\r
 353                 for (int i=0; i<search.length(); i++) {\r
 354                         if (search.charAt(i) == '\0') {\r
 355                                 search = search.substring(1);\r
 356                                 i=0;\r
 357                         } else {\r
 358                                 pos = search.indexOf('\0');\r
 359                                 if (pos > 0) {\r
 360                                         words.add(search.substring(0,pos).toLowerCase());\r
 361                                         search = search.substring(pos);\r
 362                                         i=0;\r
 363                                 }\r
 364                         }\r
 365                 }\r
 366                 if (search.charAt(0)=='\0')     \r
 367                         words.add(search.substring(1).toLowerCase());\r
 368                 else\r
 369                         words.add(search.toLowerCase());\r
 370                 parseTerms(words);\r
 371         }\r
 372 \r
 373         \r
 374         // Parse out individual words into separate lists\r
 375         // Supported options\r
 376         // Tags\r
 377         // Notebooks\r
 378         // Intitle\r
 379         // author\r
 380         // source\r
 381         // source application\r
 382         // created\r
 383         // updated\r
 384         // subject date\r
 385 \r
 386         private void parseTerms(List<String> words) {\r
 387                 for (int i=0; i<words.size(); i++) {\r
 388                         String word = words.get(i);\r
 389                         int pos = word.indexOf(":");\r
 390                         if (word.startsWith("any:")) {\r
 391                                 any = true;\r
 392                                 word = word.substring(4).trim();\r
 393                                 pos = word.indexOf(":");\r
 394                         }\r
 395                         boolean searchPhrase = false;\r
 396                         if (pos < 0 && word.indexOf(" ") > 0) {\r
 397                                 searchPhrase=true;\r
 398                                 searchPhrases.add(word.toLowerCase());\r
 399                         }\r
 400                         if (!searchPhrase && pos < 0) \r
 401                                 if (word != null && word.length() > 0)\r
 402                                         getWords().add(word); \r
 403 //                              getWords().add("*"+word+"*");           //// WILDCARD\r
 404                         if (word.startsWith("intitle:")) \r
 405                                 intitle.add("*"+word+"*");\r
 406                         if (word.startsWith("-intitle:")) \r
 407                                 intitle.add("*"+word+"*");\r
 408                         if (word.startsWith("notebook:")) \r
 409                                 notebooks.add(word);\r
 410                         if (word.startsWith("-notebook:")) \r
 411                                 notebooks.add(word);\r
 412                         if (word.startsWith("tag:")) \r
 413                                 tags.add(word);\r
 414                         if (word.startsWith("-tag:")) \r
 415                                 tags.add(word);\r
 416                         if (word.startsWith("resource:")) \r
 417                                 resource.add(word);\r
 418                         if (word.startsWith("-resource:")) \r
 419                                 resource.add(word);\r
 420                         if (word.startsWith("author:")) \r
 421                                 author.add(word);\r
 422                         if (word.startsWith("-author:")) \r
 423                                 author.add(word);\r
 424                         if (word.startsWith("source:")) \r
 425                                 source.add(word);\r
 426                         if (word.startsWith("-source:")) \r
 427                                 source.add(word);\r
 428                         if (word.startsWith("sourceapplication:")) \r
 429                                 sourceApplication.add(word);\r
 430                         if (word.startsWith("-sourceapplication:")) \r
 431                                 sourceApplication.add(word);\r
 432                         if (word.startsWith("recotype:")) \r
 433                                 recoType.add(word);\r
 434                         if (word.startsWith("-recotype:")) \r
 435                                 recoType.add(word);\r
 436                         if (word.startsWith("todo:")) \r
 437                                 todo.add(word);\r
 438                         if (word.startsWith("-todo:")) \r
 439                                 todo.add(word);\r
 440                         if (word.startsWith("stack:"))\r
 441                                 stack.add(word);\r
 442                         if (word.startsWith("-stack:"))\r
 443                                 stack.add(word);\r
 444 \r
 445                         if (word.startsWith("latitude:")) \r
 446                                 latitude.add(word);\r
 447                         if (word.startsWith("-latitude:")) \r
 448                                 latitude.add(word);\r
 449                         if (word.startsWith("longitude:")) \r
 450                                 longitude.add(word);\r
 451                         if (word.startsWith("-longitude:")) \r
 452                                 longitude.add(word);\r
 453                         if (word.startsWith("altitude:")) \r
 454                                 altitude.add(word);\r
 455                         if (word.startsWith("-altitude:")) \r
 456                                 altitude.add(word);\r
 457 \r
 458                         if (word.startsWith("created:")) \r
 459                                 created.add(word);\r
 460                         if (word.startsWith("-created:")) \r
 461                                 created.add(word);\r
 462                         if (word.startsWith("updated:")) \r
 463                                 updated.add(word);\r
 464                         if (word.startsWith("-updated:")) \r
 465                                 updated.add(word);\r
 466                         if (word.startsWith("subjectdate:")) \r
 467                                 created.add(word);\r
 468                         if (word.startsWith("-subjectdate:")) \r
 469                                 created.add(word);\r
 470 \r
 471                 }\r
 472         }\r
 473         // Match notebooks in search terms against notes\r
 474         private boolean matchListAll(List<String> list, String title) {\r
 475                 if (list.size() == 0)\r
 476                         return true;\r
 477                 boolean negative = false;\r
 478                 for (int i=0; i<list.size(); i++) {\r
 479                         int pos = list.get(i).indexOf(":");\r
 480                         negative = false;\r
 481                         if (list.get(i).startsWith("-"))\r
 482                                 negative = true;\r
 483                         String filterName = cleanupWord(list.get(i).substring(pos+1));\r
 484                         filterName = filterName.replace("*", ".*");   // setup for regular expression pattern match\r
 485                         boolean matches = Pattern.matches(filterName.toLowerCase(),title.toLowerCase());\r
 486                         if (matches && negative)\r
 487                                 return false;\r
 488                         if (matches && !negative)\r
 489                                 return true;\r
 490                 }\r
 491                 if (negative)\r
 492                         return true;\r
 493                 else\r
 494                         return false;\r
 495         }\r
 496         // Match notebooks in search terms against notes\r
 497         private boolean matchContentAll(Note n) {\r
 498                 if (todo.size() == 0 && resource.size() == 0 && searchPhrases.size() == 0)\r
 499                         return true;\r
 500                 \r
 501                 boolean returnTodo = false;\r
 502                 boolean returnResource = false;\r
 503                 boolean returnPhrase = false;\r
 504                 \r
 505                 if (todo.size() == 0)\r
 506                         returnTodo = true;\r
 507                 if (resource.size() == 0)\r
 508                         returnResource = true;\r
 509                 if (searchPhrases.size() == 0)\r
 510                         returnPhrase = true;\r
 511                 \r
 512                 \r
 513                 n = conn.getNoteTable().getNote(n.getGuid(), true, true, false, false, false);\r
 514                 \r
 515                 // Check for search phrases\r
 516                 String text = StringEscapeUtils.unescapeHtml(n.getContent().replaceAll("\\<.*?\\>", "")).toLowerCase();\r
 517                 boolean negative = false;\r
 518                 for (int i=0; i<searchPhrases.size(); i++) {\r
 519                         String phrase = searchPhrases.get(i);\r
 520                         if (phrase.startsWith("-")) {\r
 521                                 negative = true;\r
 522                                 phrase = phrase.substring(1);\r
 523                         } else\r
 524                                 negative = false;\r
 525                         phrase = phrase.substring(1);\r
 526                         phrase = phrase.substring(0,phrase.length()-1);\r
 527                         if (text.indexOf(phrase)>=0) {\r
 528                                 if (!negative)\r
 529                                         returnPhrase = true;\r
 530                         }\r
 531                         if (text.indexOf(phrase)<0 && negative)\r
 532                                 returnPhrase = true;\r
 533                 }\r
 534 \r
 535                 \r
 536                 for (int i=0; i<todo.size(); i++) {\r
 537                         String value = todo.get(i);\r
 538                         value = value.replace("\"", "");\r
 539                         boolean desiredState;\r
 540                         if (!value.endsWith(":false") && !value.endsWith(":true") && !value.endsWith(":*") && !value.endsWith("*"))\r
 541                                 return false;\r
 542                         if (value.endsWith(":false"))\r
 543                                 desiredState = false;\r
 544                         else\r
 545                                 desiredState = true;\r
 546                         if (value.startsWith("-"))\r
 547                                 desiredState = !desiredState;\r
 548                         int pos = n.getContent().indexOf("<en-todo");\r
 549                         if (pos == -1 && value.startsWith("-") && (value.endsWith("*") || value.endsWith(":")))\r
 550                                 return true;\r
 551                         if (pos > -1 && value.startsWith("-") && (value.endsWith("*") || value.endsWith(":")))\r
 552                                 return false;\r
 553                         if (pos == -1) \r
 554                                 return false;\r
 555                         if (value.endsWith("*"))\r
 556                                 returnTodo = true;\r
 557                         while (pos > -1) {\r
 558                                 int endPos = n.getContent().indexOf("/>", pos);\r
 559                                 String segment = n.getContent().substring(pos, endPos);\r
 560                                 boolean currentState;\r
 561                                 if (segment.toLowerCase().indexOf("checked=\"true\"") == -1)\r
 562                                         currentState = false;\r
 563                                 else\r
 564                                         currentState = true;\r
 565                                 if (desiredState == currentState)\r
 566                                         returnTodo = true;\r
 567                                 \r
 568                                 pos = n.getContent().indexOf("<en-todo", pos+1);\r
 569                         }\r
 570                 }\r
 571                 \r
 572                 // Check resources\r
 573                 for (int i=0; i<resource.size(); i++) {\r
 574                         String resourceString = resource.get(i);\r
 575                         resourceString = resourceString.replace("\"", "");\r
 576                         negative = false;\r
 577                         if (resourceString.startsWith("-"))\r
 578                                 negative = true;\r
 579                         resourceString = resourceString.substring(resourceString.indexOf(":")+1);\r
 580                         if (resourceString.equals(""))\r
 581                                 return false;\r
 582                         for (int j=0; j<n.getResourcesSize(); j++) {\r
 583                                 boolean match = stringMatch(n.getResources().get(j).getMime(), resourceString, negative);\r
 584                                 if (!match)\r
 585                                         return false;\r
 586                                 returnResource = true;\r
 587                         }\r
 588                 }\r
 589                 \r
 590                 return returnResource && returnTodo && returnPhrase;\r
 591         }\r
 592         \r
 593         private boolean stringMatch(String content, String text, boolean negative) {\r
 594                 String regex;\r
 595                 if (content == null && !negative)\r
 596                         return false;\r
 597                 if (content == null && negative)\r
 598                         return true;\r
 599                 \r
 600                 if (text.endsWith("*")) {\r
 601                         text = text.substring(0,text.length()-1);\r
 602                         regex = text;\r
 603                 } else {\r
 604                         regex = text;\r
 605                 }\r
 606                 content = content.toLowerCase();\r
 607                 regex = regex.toLowerCase();\r
 608                 boolean matches = content.startsWith(regex);\r
 609                 if (negative)\r
 610                         return !matches;\r
 611                 return matches;\r
 612         }\r
 613         \r
 614         // Remove odd strings from search terms\r
 615         private String cleanupWord(String word) {\r
 616                 if (word.startsWith("\""))\r
 617                         word = word.substring(1);\r
 618                 if (word.endsWith("\""))\r
 619             word = word.substring(0,word.length()-1);\r
 620                 word = word.replace("\\\"","\"");\r
 621                 word = word.replace("\\\\","\\");\r
 622                 \r
 623                 return word;\r
 624         }\r
 625 \r
 626         \r
 627         // Match dates\r
 628         private boolean matchDatesAll(List<String> dates, long noteDate) {\r
 629                 if (dates.size()== 0) \r
 630                         return true;\r
 631                 \r
 632                 boolean negative = false;\r
 633                 for (int i=0; i<dates.size(); i++) {\r
 634                         String requiredDate = dates.get(i);\r
 635                         if (requiredDate.startsWith("-"))\r
 636                                 negative = true;\r
 637                         \r
 638                         int response = 0;\r
 639                         requiredDate = requiredDate.substring(requiredDate.indexOf(":")+1);\r
 640                         try {\r
 641                                 response = dateCheck(requiredDate, noteDate);\r
 642                         } catch (java.lang.NumberFormatException e) {return false;}  {\r
 643                                 if (negative && response < 0)\r
 644                                         return false;\r
 645                                 if (!negative && response > 0)\r
 646                                         return false;\r
 647                         }\r
 648                 }\r
 649                 return true;\r
 650         }\r
 651         private boolean matchDatesAny(List<String> dates, long noteDate) {\r
 652                 if (dates.size()== 0) \r
 653                         return true;\r
 654                 \r
 655                 boolean negative = false;\r
 656                 for (int i=0; i<dates.size(); i++) {\r
 657                         String requiredDate = dates.get(i);\r
 658                         if (requiredDate.startsWith("-"))\r
 659                                 negative = true;\r
 660                         \r
 661                         int response = 0;\r
 662                         requiredDate = requiredDate.substring(requiredDate.indexOf(":")+1);\r
 663                         try {\r
 664                                 response = dateCheck(requiredDate, noteDate);\r
 665                         } catch (java.lang.NumberFormatException e) {return false;}  {\r
 666                                 if (negative && response > 0)\r
 667                                         return true;\r
 668                                 if (!negative && response < 0)\r
 669                                         return true;\r
 670                         }\r
 671                 }\r
 672                 return false;\r
 673         }\r
 674         \r
 675         @SuppressWarnings("unused")\r
 676         private void printCalendar(Calendar calendar) {\r
 677                 // define output format and print\r
 678                 SimpleDateFormat sdf = new SimpleDateFormat("d MMM yyyy hh:mm:ss aaa");\r
 679                 String date = sdf.format(calendar.getTime());\r
 680                 System.err.print(date);\r
 681                 calendar = new GregorianCalendar();\r
 682         }\r
 683         \r
 684         \r
 685         //****************************************\r
 686         //****************************************\r
 687         // Match search terms against notes\r
 688         //****************************************\r
 689         //****************************************\r
 690         public List<Note> matchWords() {\r
 691                 logger.log(logger.EXTREME, "Inside EnSearch.matchWords()");\r
 692                 boolean subSelect = false;\r
 693                 \r
 694                 NoteTable noteTable = new NoteTable(logger, conn);  \r
 695                 List<String> validGuids = new ArrayList<String>();\r
 696                 \r
 697                 if (searchWords.size() > 0) \r
 698                         subSelect = true;\r
 699 \r
 700                 NSqlQuery query = new NSqlQuery(conn.getConnection());\r
 701                 // Build a temp table for GUID results\r
 702                 if (!conn.dbTableExists("SEARCH_RESULTS")) {\r
 703                         query.exec("create temporary table SEARCH_RESULTS (guid varchar)");\r
 704                         query.exec("create temporary table SEARCH_RESULTS_MERGE (guid varchar)");\r
 705                 } else {\r
 706                         query. exec("Delete from SEARCH_RESULTS");\r
 707                         query. exec("Delete from SEARCH_RESULTS_MERGE");\r
 708                 }\r
 709 \r
 710                 NSqlQuery insertQuery = new NSqlQuery(conn.getConnection());\r
 711                 NSqlQuery indexQuery = new NSqlQuery(conn.getIndexConnection());\r
 712                 NSqlQuery mergeQuery = new NSqlQuery(conn.getConnection());\r
 713                 NSqlQuery deleteQuery = new NSqlQuery(conn.getConnection());\r
 714                 \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                                 if (getWords().get(i).indexOf("*") == 0) {\r
 721                                         indexQuery.prepare("Select distinct guid from words where weight >= " +minimumRecognitionWeight +\r
 722                                                         " and word=:word");\r
 723                                         indexQuery.bindValue(":word", getWords().get(i));\r
 724                                 } else {\r
 725                                         indexQuery.prepare("Select distinct guid from words where weight >= " +minimumRecognitionWeight +\r
 726                                                 " and word like :word");\r
 727                                         indexQuery.bindValue(":word", getWords().get(i).replace("*", "%"));\r
 728                                 }\r
 729                                 indexQuery.exec();\r
 730                                 String guid = null;\r
 731                                 while(indexQuery.next()) {\r
 732                                         guid = indexQuery.valueString(0);\r
 733                                         if (i==0 || any) {\r
 734                                                 insertQuery.bindValue(":guid", guid);\r
 735                                                 insertQuery.exec();\r
 736                                         } else {\r
 737                                                 mergeQuery.bindValue(":guid", guid);\r
 738                                                 mergeQuery.exec();\r
 739                                         }\r
 740                                 }\r
 741                                 if (i>0 && !any) {\r
 742                                         deleteQuery.exec("Delete from SEARCH_RESULTS where guid not in (select guid from SEARCH_RESULTS_MERGE)");\r
 743                                         deleteQuery.exec("Delete from SEARCH_RESULTS_MERGE");\r
 744                                 }\r
 745                         }\r
 746 \r
 747                         query.prepare("Select distinct guid from Note where guid in (Select guid from SEARCH_RESULTS)");\r
 748                         if (!query.exec()) \r
 749                                 logger.log(logger.LOW, "Error merging search results:" + query.lastError());\r
 750                 \r
 751                         while (query.next()) {\r
 752                                 validGuids.add(query.valueString(0));\r
 753                         }\r
 754                 }\r
 755                 \r
 756                 List<Note> noteIndex = noteTable.getAllNotes();\r
 757                 List<Note> guids = new ArrayList<Note>();\r
 758                 for (int i=0; i<noteIndex.size(); i++) {\r
 759                         Note n = noteIndex.get(i);\r
 760                         boolean good = true;\r
 761                         \r
 762                         if (!validGuids.contains(n.getGuid()) && subSelect)\r
 763                                 good = false;\r
 764                                                 \r
 765                         // Start matching special stuff, like tags & notebooks\r
 766                         if (any) {\r
 767                                 if (good && !matchTagsAny(n.getTagNames()))\r
 768                                         good = false;\r
 769                                 if (good && !matchNotebook(n.getNotebookGuid()))\r
 770                                         good = false;\r
 771                                 if (good && !matchNotebookStack(n.getNotebookGuid()))\r
 772                                         good = false;\r
 773                                 if (good && !matchListAny(getIntitle(), n.getTitle()))\r
 774                                         good = false;\r
 775                                 if (good && !matchListAny(getAuthor(), n.getAttributes().getAuthor()))\r
 776                                         good = false;\r
 777                                 if (good && !matchListAny(getSource(), n.getAttributes().getSource()))\r
 778                                         good = false;\r
 779                                 if (good && !matchListAny(getSourceApplication(), n.getAttributes().getSourceApplication()))\r
 780                                         good = false;\r
 781                                 if (good && !matchContentAny(n))\r
 782                                         good = false;\r
 783                                 if (good && !matchDatesAny(getCreated(), n.getCreated()))\r
 784                                         good = false;\r
 785                                 if (good && !matchDatesAny(getUpdated(), n.getUpdated()))\r
 786                                         good = false;\r
 787                                 if (good && n.getAttributes() != null && !matchDatesAny(getSubjectDate(), n.getAttributes().getSubjectDate()))\r
 788                                         good = false;\r
 789                         } else {\r
 790                                 if (good && !matchTagsAll(n.getTagNames()))\r
 791                                         good = false;\r
 792                                 if (good && !matchNotebook(n.getNotebookGuid()))\r
 793                                         good = false;\r
 794                                 if (good && !matchNotebookStack(n.getNotebookGuid()))\r
 795                                         good = false;\r
 796                                 if (good && !matchListAll(getIntitle(), n.getTitle()))\r
 797                                         good = false;\r
 798                                 if (good && !matchListAll(getAuthor(), n.getAttributes().getAuthor()))\r
 799                                         good = false;\r
 800                                 if (good && !matchListAll(getSource(), n.getAttributes().getSource()))\r
 801                                         good = false;\r
 802                                 if (good && !matchListAll(getSourceApplication(), n.getAttributes().getSourceApplication()))\r
 803                                         good = false;\r
 804                                 if (good && !matchContentAll(n))\r
 805                                         good = false;\r
 806                                 if (good && !matchDatesAll(getCreated(), n.getCreated()))\r
 807                                         good = false;\r
 808                                 if (good && !matchDatesAll(getUpdated(), n.getUpdated()))\r
 809                                         good = false;\r
 810                                 if (good && n.getAttributes() != null && !matchDatesAll(getSubjectDate(), n.getAttributes().getSubjectDate()))\r
 811                                         good = false;\r
 812                         }\r
 813                         if (good) {\r
 814                                 guids.add(n);\r
 815                         }\r
 816                 }\r
 817                 \r
 818                 // For performance reasons, we didn't get the tags for every note individually.  We now need to \r
 819                 // get them\r
 820                 List<NoteTagsRecord> noteTags = noteTable.noteTagsTable.getAllNoteTags();\r
 821                 for (int i=0; i<guids.size(); i++) {\r
 822                         List<String> tags = new ArrayList<String>();\r
 823                         List<String> names = new ArrayList<String>();\r
 824                         for (int j=0; j<noteTags.size(); j++) {\r
 825                                 if (guids.get(i).getGuid().equals(noteTags.get(j).noteGuid)) {\r
 826                                         tags.add(noteTags.get(j).tagGuid);\r
 827                                         names.add(getTagNameByGuid(noteTags.get(j).tagGuid));\r
 828                                 }\r
 829                         }\r
 830                         \r
 831                         guids.get(i).setTagGuids(tags);\r
 832                         guids.get(i).setTagNames(names);\r
 833                 };\r
 834                 logger.log(logger.EXTREME, "Leaving EnSearch.matchWords()");\r
 835                 return guids;\r
 836         }\r
 837         \r
 838         \r
 839         \r
 840         private String getTagNameByGuid(String guid) {\r
 841                 for (int i=0; i<tagIndex.size(); i++) {\r
 842                         if (tagIndex.get(i).getGuid().equals(guid)) \r
 843                                         return tagIndex.get(i).getName();\r
 844                 }               \r
 845                 return "";\r
 846         }\r
 847 \r
 848         // Compare dates\r
 849         public int dateCheck(String date, long noteDate)  throws java.lang.NumberFormatException  {\r
 850                 int offset = 0;\r
 851                 boolean found = false;\r
 852                 GregorianCalendar calendar = new GregorianCalendar();\r
 853                 \r
 854                 if (date.contains("-")) {\r
 855                         String modifier = date.substring(date.indexOf("-")+1);\r
 856                         offset = new Integer(modifier);\r
 857                         offset = 0-offset;\r
 858                         date = date.substring(0,date.indexOf("-"));\r
 859                 }\r
 860                 \r
 861                 if (date.contains("+")) {\r
 862                         String modifier = date.substring(date.indexOf("+")+1);\r
 863                         offset = new Integer(modifier);\r
 864                         date = date.substring(0,date.indexOf("+"));\r
 865                 }\r
 866                 \r
 867                 if (date.equalsIgnoreCase("today")) {\r
 868                         calendar.add(Calendar.DATE, offset);\r
 869                         calendar.set(Calendar.HOUR, 0);\r
 870                         calendar.set(Calendar.MINUTE, 0);\r
 871                         calendar.set(Calendar.SECOND, 1);\r
 872                         found = true;\r
 873                 }\r
 874                 \r
 875                 if (date.equalsIgnoreCase("month")) {\r
 876                         calendar.add(Calendar.MONTH, offset);\r
 877                         calendar.set(Calendar.DAY_OF_MONTH, 1);\r
 878                         calendar.set(Calendar.HOUR, 0);\r
 879                         calendar.set(Calendar.MINUTE, 0);\r
 880                         calendar.set(Calendar.SECOND, 1);\r
 881                         found = true;\r
 882                 }\r
 883 \r
 884                 if (date.equalsIgnoreCase("year")) {\r
 885                         calendar.add(Calendar.YEAR, offset);\r
 886                         calendar.set(Calendar.MONTH, Calendar.JANUARY);\r
 887                         calendar.set(Calendar.DAY_OF_MONTH, 1);\r
 888                         calendar.set(Calendar.HOUR, 0);\r
 889                         calendar.set(Calendar.MINUTE, 0);\r
 890                         calendar.set(Calendar.SECOND, 1);\r
 891                         found = true;\r
 892                 }\r
 893 \r
 894                 if (date.equalsIgnoreCase("week")) {\r
 895                         calendar.add(Calendar.DATE, 0-calendar.get(Calendar.DAY_OF_WEEK)+1);\r
 896                         calendar.add(Calendar.DATE,(offset*7));\r
 897                         calendar.set(Calendar.HOUR, 0);\r
 898                         calendar.set(Calendar.MINUTE, 0);\r
 899                         calendar.set(Calendar.SECOND, 1);\r
 900 \r
 901                         found = true;\r
 902                 }\r
 903                 \r
 904                 // If nothing was found, then we have a date number\r
 905                 if (!found) {\r
 906                         calendar = stringToGregorianCalendar(date);\r
 907                 }\r
 908                 \r
 909                 \r
 910                 String dateTimeFormat = new String("yyyyMMdd-HHmmss");\r
 911                 SimpleDateFormat simple = new SimpleDateFormat(dateTimeFormat);\r
 912                 StringBuilder creationDate = new StringBuilder(simple.format(noteDate));\r
 913                 GregorianCalendar nCalendar = stringToGregorianCalendar(creationDate.toString().replace("-", "T"));\r
 914                 if (calendar == null || nCalendar == null)  // If we have something invalid, it automatically fails\r
 915                         return 1;\r
 916                 return calendar.compareTo(nCalendar);\r
 917         }\r
 918         private GregorianCalendar stringToGregorianCalendar(String date) {\r
 919                 String datePart = date;\r
 920                 GregorianCalendar calendar = new GregorianCalendar();\r
 921                 boolean GMT = false;\r
 922                 String timePart = "";\r
 923                 if (date.contains("T")) {\r
 924                         datePart = date.substring(0,date.indexOf("T"));\r
 925                         timePart = date.substring(date.indexOf("T")+1);\r
 926                 } else {\r
 927                         timePart = "000001";\r
 928                 }\r
 929                 if (datePart.length() != 8)\r
 930                         return null;\r
 931                 calendar.set(Calendar.YEAR, new Integer(datePart.substring(0,4)));\r
 932                 calendar.set(Calendar.MONTH, new Integer(datePart.substring(4,6))-1);\r
 933                 calendar.set(Calendar.DAY_OF_MONTH, new Integer(datePart.substring(6)));\r
 934                 if (timePart.endsWith("Z")) {\r
 935                         GMT = true;\r
 936                         timePart = timePart.substring(0,timePart.length()-1);\r
 937                 }\r
 938                 timePart = timePart.concat("000000");\r
 939                 timePart = timePart.substring(0,6);\r
 940                 calendar.set(Calendar.HOUR, new Integer(timePart.substring(0,2)));\r
 941                 calendar.set(Calendar.MINUTE, new Integer(timePart.substring(2,4)));\r
 942                 calendar.set(Calendar.SECOND, new Integer(timePart.substring(4)));\r
 943                 if (GMT)\r
 944                         calendar.set(Calendar.ZONE_OFFSET, -1*(calendar.get(Calendar.ZONE_OFFSET)/(1000*60*60)));\r
 945                 return calendar;\r
 946 \r
 947         }\r
 948                 \r
 949 }\r