CLI search, step 1
[fanfix.git] / src / be / nikiroo / fanfix / Main.java
index 53c4a8629f952075f4519131f5541363f5701f45..bbb335e97afb9395b8cabb7196e5f1845ca0fcf7 100644 (file)
@@ -4,6 +4,7 @@ import java.io.File;
 import java.io.IOException;
 import java.net.MalformedURLException;
 import java.net.URL;
+import java.util.ArrayList;
 import java.util.List;
 
 import be.nikiroo.fanfix.bundles.StringId;
@@ -20,6 +21,7 @@ import be.nikiroo.fanfix.output.BasicOutput.OutputType;
 import be.nikiroo.fanfix.reader.BasicReader;
 import be.nikiroo.fanfix.reader.Reader;
 import be.nikiroo.fanfix.reader.Reader.ReaderType;
+import be.nikiroo.fanfix.searchable.BasicSearchable;
 import be.nikiroo.fanfix.supported.BasicSupport;
 import be.nikiroo.fanfix.supported.SupportType;
 import be.nikiroo.utils.Progress;
@@ -33,7 +35,7 @@ import be.nikiroo.utils.serial.server.ServerObject;
  */
 public class Main {
        private enum MainAction {
-               IMPORT, EXPORT, CONVERT, READ, READ_URL, LIST, HELP, SET_READER, START, VERSION, SERVER, STOP_SERVER, REMOTE,
+               IMPORT, EXPORT, CONVERT, READ, READ_URL, LIST, HELP, SET_READER, START, VERSION, SERVER, STOP_SERVER, REMOTE, SET_SOURCE, SET_TITLE, SET_AUTHOR, SEARCH, SEARCH_TAG
        }
 
        /**
@@ -59,7 +61,15 @@ public class Main {
         * </li>
         * <li>--read-url [URL] ([chapter number]): convert on the fly and read the
         * story, without saving it</li>
+        * <li>--search WEBSITE [free text] ([page] ([item])): search for the given terms, 
+        * show the given page (page 0 means "how many page do we have", starts at page 1)</li>
+        * <li>--search-tag WEBSITE ([tag 1] [tag2...] ([page] ([item]))): list the known 
+        * tags or search the stories for the given tag(s), show the given page of results
+        * (page 0 means "how many page do we have", starts at page 1)</li>
         * <li>--list ([type]): list the stories present in the library</li>
+        * <li>--set-source [id] [new source]: change the source of the given story</li>
+        * <li>--set-title [id] [new title]: change the title of the given story</li>
+        * <li>--set-author [id] [new author]: change the author of the given story</li>
         * <li>--set-reader [reader type]: set the reader type to CLI, TUI or LOCAL
         * for this command</li>
         * <li>--version: get the version of the program</li>
@@ -76,6 +86,8 @@ public class Main {
                String urlString = null;
                String luid = null;
                String sourceString = null;
+               String titleString = null;
+               String authorString = null;
                String chapString = null;
                String target = null;
                String key = null;
@@ -83,6 +95,11 @@ public class Main {
                Boolean plusInfo = null;
                String host = null;
                Integer port = null;
+               SupportType searchOn = null;
+               String search = null;
+               List<String> tags = new ArrayList<String>();
+               Integer page = null;
+               Integer item = null;
 
                boolean noMoreActions = false;
 
@@ -150,6 +167,33 @@ public class Main {
                                        exitCode = 255;
                                }
                                break;
+                       case SET_SOURCE:
+                               if (luid == null) {
+                                       luid = args[i];
+                               } else if (sourceString == null) {
+                                       sourceString = args[i];
+                               } else {
+                                       exitCode = 255;
+                               }
+                               break;
+                       case SET_TITLE:
+                               if (luid == null) {
+                                       luid = args[i];
+                               } else if (sourceString == null) {
+                                       titleString = args[i];
+                               } else {
+                                       exitCode = 255;
+                               }
+                               break;
+                       case SET_AUTHOR:
+                               if (luid == null) {
+                                       luid = args[i];
+                               } else if (sourceString == null) {
+                                       authorString = args[i];
+                               } else {
+                                       exitCode = 255;
+                               }
+                               break;
                        case READ:
                                if (luid == null) {
                                        luid = args[i];
@@ -168,6 +212,64 @@ public class Main {
                                        exitCode = 255;
                                }
                                break;
+                       case SEARCH:
+                               if (searchOn == null) {
+                                       searchOn = SupportType.valueOfAllOkUC(args[i]);
+                                       
+                                       if (searchOn == null) {
+                                               Instance.getTraceHandler().error(
+                                                               "Website not known: <" + args[i] + ">");
+                                               exitCode = 255;
+                                       }
+                                       
+                                       if (BasicSearchable.getSearchable(searchOn) == null) {
+                                               Instance.getTraceHandler().error(
+                                                               "Website not supported: " + searchOn);
+                                               exitCode = 255;
+                                       }
+                               } else if (search == null) {
+                                       search = args[i];
+                               } else if (page == null) {
+                                       try {
+                                               page = Integer.parseInt(args[i]);
+                                       } catch (NumberFormatException e) {
+                                               Instance.getTraceHandler().error(
+                                                               new Exception("Incorrect page number: <"
+                                                                               + args[i] + ">", e));
+                                               exitCode = 255;
+                                       }
+                               } else if (item == null) {
+                                       try {
+                                               item = Integer.parseInt(args[i]);
+                                       } catch (NumberFormatException e) {
+                                               Instance.getTraceHandler().error(
+                                                               new Exception("Incorrect item number: <"
+                                                                               + args[i] + ">", e));
+                                               exitCode = 255;
+                                       }
+                               } else {
+                                       exitCode = 255;
+                               }
+                               break;
+                       case SEARCH_TAG:
+                               if (searchOn == null) {
+                                       searchOn = SupportType.valueOfAllOkUC(args[i]);
+                                       
+                                       if (searchOn == null) {
+                                               Instance.getTraceHandler().error(
+                                                               "Website not known: <" + args[i] + ">");
+                                               exitCode = 255;
+                                       }
+                                       
+                                       if (BasicSearchable.getSearchable(searchOn) == null) {
+                                               Instance.getTraceHandler().error(
+                                                               "Website not supported: " + searchOn);
+                                               exitCode = 255;
+                                       }
+                               } else {
+                                       tags.add(args[i]);
+                               }
+                               break;
                        case HELP:
                                exitCode = 255;
                                break;
@@ -250,8 +352,8 @@ public class Main {
                                System.err.println("\tVersion " + v);
                                System.err.println("\t-------------");
                                System.err.println("");
-                               for (String item : updates.getChanges().get(v)) {
-                                       System.err.println("\t- " + item);
+                               for (String it : updates.getChanges().get(v)) {
+                                       System.err.println("\t- " + it);
                                }
                                System.err.println("");
                        }
@@ -282,6 +384,30 @@ public class Main {
                                }
                                exitCode = list(sourceString);
                                break;
+                       case SET_SOURCE:
+                               try {
+                                       Instance.getLibrary().changeSource(luid, sourceString, pg);
+                               } catch (IOException e1) {
+                                       Instance.getTraceHandler().error(e1);
+                                       exitCode = 21;
+                               }
+                               break;
+                       case SET_TITLE:
+                               try {
+                                       Instance.getLibrary().changeTitle(luid, titleString, pg);
+                               } catch (IOException e1) {
+                                       Instance.getTraceHandler().error(e1);
+                                       exitCode = 22;
+                               }
+                               break;
+                       case SET_AUTHOR:
+                               try {
+                                       Instance.getLibrary().changeAuthor(luid, authorString, pg);
+                               } catch (IOException e1) {
+                                       Instance.getTraceHandler().error(e1);
+                                       exitCode = 23;
+                               }
+                               break;
                        case READ:
                                if (BasicReader.getReader() == null) {
                                        Instance.getTraceHandler()
@@ -301,6 +427,73 @@ public class Main {
                                        break;
                                }
                                exitCode = read(urlString, chapString, false);
+                               break;
+                       case SEARCH:
+                               if (searchOn == null || search == null) {
+                                       exitCode = 255;
+                                       break;
+                               }
+                               
+                               if (page == null) {
+                                       page = 1;
+                               }
+                               if (item == null) {
+                                       item = 0;
+                               }
+                               
+                               if (BasicReader.getReader() == null) {
+                                       Instance.getTraceHandler()
+                                                       .error(new Exception(
+                                                                       "No reader type has been configured"));
+                                       exitCode = 10;
+                                       break;
+                               }
+                               
+                               try {
+                                       BasicReader.getReader().search(searchOn, search, page, item);
+                               } catch (IOException e1) {
+                                       Instance.getTraceHandler().error(e1);
+                               }
+                               
+                               break;
+                       case SEARCH_TAG:
+                               if (searchOn == null) {
+                                       exitCode = 255;
+                                       break;
+                               }
+                               
+                               item = 0;
+                               page = 1;
+
+                               try {
+                                       item = Integer.parseInt(tags.get(tags.size()-1));
+                                       tags.remove(tags.size() - 1);
+                                       
+                                       try {
+                                               int tmp = Integer.parseInt(tags.get(tags.size()-1));
+                                               tags.remove(tags.size() - 1);
+                                               
+                                               page = item;
+                                               item = tmp;
+                                       } catch (Exception e) {
+                                       }
+                               } catch (Exception e) {
+                               }
+                               
+                               if (BasicReader.getReader() == null) {
+                                       Instance.getTraceHandler()
+                                                       .error(new Exception(
+                                                                       "No reader type has been configured"));
+                                       exitCode = 10;
+                                       break;
+                               }
+                               
+                               try {
+                                       BasicReader.getReader().searchTag(searchOn, page, item, tags.toArray(new String[]{}));
+                               } catch (IOException e1) {
+                                       Instance.getTraceHandler().error(e1);
+                               }
+                               
                                break;
                        case HELP:
                                syntax(true);
@@ -354,27 +547,19 @@ public class Main {
                        }
                }
 
-               // We cannot do it when in GUI mode, because it is async...
-               // So if we close the temp files before it is actually used,
-               // we have a problem...
-               // TODO: close it at the correct time (for now, finalize try to do it)
-               if (false) {
-                       try {
-                               Instance.getTempFiles().close();
-                       } catch (IOException e) {
-                               Instance.getTraceHandler().error(
-                                               new IOException(
-                                                               "Cannot dispose of the temporary files", e));
-                       }
+               try {
+                       Instance.getTempFiles().close();
+               } catch (IOException e) {
+                       Instance.getTraceHandler()
+                                       .error(new IOException(
+                                                       "Cannot dispose of the temporary files", e));
                }
 
                if (exitCode == 255) {
                        syntax(false);
                }
 
-               if (exitCode != 0) {
-                       System.exit(exitCode);
-               }
+               System.exit(exitCode);
        }
 
        /**
@@ -462,7 +647,7 @@ public class Main {
        }
 
        /**
-        * Start the CLI reader for this {@link Story}.
+        * Start the current reader for this {@link Story}.
         * 
         * @param story
         *            the LUID of the {@link Story} in the {@link LocalLibrary}
@@ -488,7 +673,7 @@ public class Main {
                        if (chapString != null) {
                                try {
                                        reader.setChapter(Integer.parseInt(chapString));
-                                       reader.read();
+                                       reader.read(true);
                                } catch (NumberFormatException e) {
                                        Instance.getTraceHandler().error(
                                                        new IOException("Chapter number cannot be parsed: "
@@ -496,7 +681,7 @@ public class Main {
                                        return 2;
                                }
                        } else {
-                               reader.read();
+                               reader.read(true);
                        }
                } catch (IOException e) {
                        Instance.getTraceHandler().error(e);
@@ -548,7 +733,8 @@ public class Main {
                                        BasicSupport support = BasicSupport.getSupport(source);
 
                                        if (support != null) {
-                                               Instance.getTraceHandler().trace("Support found: " + support.getClass());
+                                               Instance.getTraceHandler().trace(
+                                                               "Support found: " + support.getClass());
                                                Progress pgIn = new Progress();
                                                Progress pgOut = new Progress();
                                                if (pg != null) {