New web library (http/https)
[nikiroo-utils.git] / src / be / nikiroo / fanfix / library / RemoteLibrary.java
index 653c25743e6730f59b1bd233ace4d11525ccafce..9fa8c66190174bd1a92536b995d181a16bc8744a 100644 (file)
@@ -21,10 +21,35 @@ import be.nikiroo.utils.serial.server.ConnectActionClientObject;
  * This {@link BasicLibrary} will access a remote server to list the available
  * stories, and download the ones you try to load to the local directory
  * specified in the configuration.
+ * <p>
+ * This remote library uses a custom fanfix:// protocol.
  * 
  * @author niki
  */
 public class RemoteLibrary extends BasicLibrary {
+       interface RemoteAction {
+               public void action(ConnectActionClientObject action) throws Exception;
+       }
+
+       class RemoteConnectAction extends ConnectActionClientObject {
+               public RemoteConnectAction() throws IOException {
+                       super(host, port, key);
+               }
+
+               @Override
+               public Object send(Object data)
+                               throws IOException, NoSuchFieldException, NoSuchMethodException,
+                               ClassNotFoundException {
+                       Object rep = super.send(data);
+                       if (rep instanceof RemoteLibraryException) {
+                               RemoteLibraryException remoteEx = (RemoteLibraryException) rep;
+                               throw remoteEx.unwrapException();
+                       }
+
+                       return rep;
+               }
+       }
+
        private String host;
        private int port;
        private final String key;
@@ -33,8 +58,6 @@ public class RemoteLibrary extends BasicLibrary {
        // informative only (server will make the actual checks)
        private boolean rw;
 
-       // TODO: error handling is not up to par!
-
        /**
         * Create a {@link RemoteLibrary} linked to the given server.
         * <p>
@@ -57,7 +80,7 @@ public class RemoteLibrary extends BasicLibrary {
         * <li><b>wl</b>: flag to allow access to all the stories (bypassing the
         * whitelist if it exists)</li>
         * </ul>
-        * 
+        * <p>
         * Some examples:
         * <ul>
         * <li><b>my_key</b>: normal connection, will take the default server
@@ -92,20 +115,26 @@ public class RemoteLibrary extends BasicLibrary {
                        this.subkey = "";
                }
 
+               if (host.startsWith("fanfix://")) {
+                       host = host.substring("fanfix://".length());
+               }
+
                this.host = host;
                this.port = port;
        }
 
        @Override
        public String getLibraryName() {
-               return host + ":" + port;
+               return (rw ? "[READ-ONLY] " : "") + "fanfix://" + host + ":" + port;
        }
 
        @Override
        public Status getStatus() {
-               Instance.getTraceHandler().trace("Getting remote lib status...");
+               Instance.getInstance().getTraceHandler()
+                               .trace("Getting remote lib status...");
                Status status = getStatusDo();
-               Instance.getTraceHandler().trace("Remote lib status: " + status);
+               Instance.getInstance().getTraceHandler()
+                               .trace("Remote lib status: " + status);
                return status;
        }
 
@@ -115,17 +144,17 @@ public class RemoteLibrary extends BasicLibrary {
                result[0] = Status.INVALID;
 
                try {
-                       new ConnectActionClientObject(host, port, key) {
+                       new RemoteConnectAction() {
                                @Override
                                public void action(Version serverVersion) throws Exception {
                                        Object rep = send(new Object[] { subkey, "PING" });
 
                                        if ("r/w".equals(rep)) {
                                                rw = true;
-                                               result[0] = Status.READY;
+                                               result[0] = Status.READ_WRITE;
                                        } else if ("r/o".equals(rep)) {
                                                rw = false;
-                                               result[0] = Status.READY;
+                                               result[0] = Status.READ_ONLY;
                                        } else {
                                                result[0] = Status.UNAUTHORIZED;
                                        }
@@ -152,121 +181,87 @@ public class RemoteLibrary extends BasicLibrary {
        }
 
        @Override
-       public Image getCover(final String luid) {
+       public Image getCover(final String luid) throws IOException {
                final Image[] result = new Image[1];
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Object rep = send(new Object[] { subkey, "GET_COVER", luid });
-                                       result[0] = (Image) rep;
-                               }
-
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
-                               }
-                       }.connect();
-               } catch (Exception e) {
-                       Instance.getTraceHandler().error(e);
-               }
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Object rep = action
+                                               .send(new Object[] { subkey, "GET_COVER", luid });
+                               result[0] = (Image) rep;
+                       }
+               });
 
                return result[0];
        }
 
        @Override
-       public Image getCustomSourceCover(final String source) {
+       public Image getCustomSourceCover(final String source) throws IOException {
                return getCustomCover(source, "SOURCE");
        }
 
        @Override
-       public Image getCustomAuthorCover(final String author) {
+       public Image getCustomAuthorCover(final String author) throws IOException {
                return getCustomCover(author, "AUTHOR");
        }
 
        // type: "SOURCE" or "AUTHOR"
-       private Image getCustomCover(final String source, final String type) {
+       private Image getCustomCover(final String source, final String type)
+                       throws IOException {
                final Image[] result = new Image[1];
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Object rep = send(new Object[] { subkey,
-                                                       "GET_CUSTOM_COVER", type, source });
-                                       result[0] = (Image) rep;
-                               }
-
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
-                               }
-                       }.connect();
-               } catch (Exception e) {
-                       Instance.getTraceHandler().error(e);
-               }
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Object rep = action.send(new Object[] { subkey,
+                                               "GET_CUSTOM_COVER", type, source });
+                               result[0] = (Image) rep;
+                       }
+               });
 
                return result[0];
        }
 
        @Override
-       public synchronized Story getStory(final String luid, Progress pg) {
+       public synchronized Story getStory(final String luid, Progress pg)
+                       throws IOException {
                final Progress pgF = pg;
                final Story[] result = new Story[1];
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Progress pg = pgF;
-                                       if (pg == null) {
-                                               pg = new Progress();
-                                       }
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Progress pg = pgF;
+                               if (pg == null) {
+                                       pg = new Progress();
+                               }
 
-                                       Object rep = send(new Object[] { subkey, "GET_STORY", luid });
+                               Object rep = action
+                                               .send(new Object[] { subkey, "GET_STORY", luid });
 
-                                       MetaData meta = null;
-                                       if (rep instanceof MetaData) {
-                                               meta = (MetaData) rep;
-                                               if (meta.getWords() <= Integer.MAX_VALUE) {
-                                                       pg.setMinMax(0, (int) meta.getWords());
-                                               }
+                               MetaData meta = null;
+                               if (rep instanceof MetaData) {
+                                       meta = (MetaData) rep;
+                                       if (meta.getWords() <= Integer.MAX_VALUE) {
+                                               pg.setMinMax(0, (int) meta.getWords());
                                        }
-
-                                       List<Object> list = new ArrayList<Object>();
-                                       for (Object obj = send(null); obj != null; obj = send(null)) {
-                                               list.add(obj);
-                                               pg.add(1);
-                                       }
-
-                                       result[0] = RemoteLibraryServer.rebuildStory(list);
-                                       pg.done();
                                }
 
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
+                               List<Object> list = new ArrayList<Object>();
+                               for (Object obj = action.send(null); obj != null; obj = action
+                                               .send(null)) {
+                                       list.add(obj);
+                                       pg.add(1);
                                }
-                       }.connect();
-               } catch (Exception e) {
-                       Instance.getTraceHandler().error(e);
-               }
+
+                               result[0] = RemoteLibraryServer.rebuildStory(list);
+                               pg.done();
+                       }
+               });
 
                return result[0];
        }
@@ -288,37 +283,28 @@ public class RemoteLibrary extends BasicLibrary {
 
                final Progress pgF = pgSave;
 
-               new ConnectActionClientObject(host, port, key) {
+               connectRemoteAction(new RemoteAction() {
                        @Override
-                       public void action(Version serverVersion) throws Exception {
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
                                Progress pg = pgF;
                                if (story.getMeta().getWords() <= Integer.MAX_VALUE) {
                                        pg.setMinMax(0, (int) story.getMeta().getWords());
                                }
 
-                               send(new Object[] { subkey, "SAVE_STORY", luid });
+                               action.send(new Object[] { subkey, "SAVE_STORY", luid });
 
                                List<Object> list = RemoteLibraryServer.breakStory(story);
                                for (Object obj : list) {
-                                       send(obj);
+                                       action.send(obj);
                                        pg.add(1);
                                }
 
-                               luidSaved[0] = (String) send(null);
+                               luidSaved[0] = (String) action.send(null);
 
                                pg.done();
                        }
-
-                       @Override
-                       protected void onError(Exception e) {
-                               if (e instanceof SSLException) {
-                                       Instance.getTraceHandler().error(
-                                                       "Connection refused (bad key)");
-                               } else {
-                                       Instance.getTraceHandler().error(e);
-                               }
-                       }
-               }.connect();
+               });
 
                // because the meta changed:
                MetaData meta = getInfo(luidSaved[0]);
@@ -338,63 +324,45 @@ public class RemoteLibrary extends BasicLibrary {
 
        @Override
        public synchronized void delete(final String luid) throws IOException {
-               new ConnectActionClientObject(host, port, key) {
+               connectRemoteAction(new RemoteAction() {
                        @Override
-                       public void action(Version serverVersion) throws Exception {
-                               send(new Object[] { subkey, "DELETE_STORY", luid });
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               action.send(new Object[] { subkey, "DELETE_STORY", luid });
                        }
-
-                       @Override
-                       protected void onError(Exception e) {
-                               if (e instanceof SSLException) {
-                                       Instance.getTraceHandler().error(
-                                                       "Connection refused (bad key)");
-                               } else {
-                                       Instance.getTraceHandler().error(e);
-                               }
-                       }
-               }.connect();
+               });
        }
 
        @Override
-       public void setSourceCover(final String source, final String luid) {
+       public void setSourceCover(final String source, final String luid)
+                       throws IOException {
                setCover(source, luid, "SOURCE");
        }
 
        @Override
-       public void setAuthorCover(final String author, final String luid) {
+       public void setAuthorCover(final String author, final String luid)
+                       throws IOException {
                setCover(author, luid, "AUTHOR");
        }
 
        // type = "SOURCE" | "AUTHOR"
        private void setCover(final String value, final String luid,
-                       final String type) {
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       send(new Object[] { subkey, "SET_COVER", type, value, luid });
-                               }
-
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
-                               }
-                       }.connect();
-               } catch (IOException e) {
-                       Instance.getTraceHandler().error(e);
-               }
+                       final String type) throws IOException {
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               action.send(new Object[] { subkey, "SET_COVER", type, value,
+                                               luid });
+                       }
+               });
        }
 
        @Override
        // Could work (more slowly) without it
-       public Story imprt(final URL url, Progress pg) throws IOException {
+       public MetaData imprt(final URL url, Progress pg) throws IOException {
                // Import the file locally if it is actually a file
+
                if (url == null || url.getProtocol().equalsIgnoreCase("file")) {
                        return super.imprt(url, pg);
                }
@@ -405,59 +373,37 @@ public class RemoteLibrary extends BasicLibrary {
                        pg = new Progress();
                }
 
-               pg.setMinMax(0, 2);
-               Progress pgImprt = new Progress();
-               Progress pgGet = new Progress();
-               pg.addProgress(pgImprt, 1);
-               pg.addProgress(pgGet, 1);
-
-               final Progress pgF = pgImprt;
+               final Progress pgF = pg;
                final String[] luid = new String[1];
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Progress pg = pgF;
-
-                                       Object rep = send(new Object[] { subkey, "IMPORT",
-                                                       url.toString() });
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Progress pg = pgF;
 
-                                       while (true) {
-                                               if (!RemoteLibraryServer.updateProgress(pg, rep)) {
-                                                       break;
-                                               }
+                               Object rep = action.send(
+                                               new Object[] { subkey, "IMPORT", url.toString() });
 
-                                               rep = send(null);
+                               while (true) {
+                                       if (!RemoteLibraryServer.updateProgress(pg, rep)) {
+                                               break;
                                        }
 
-                                       pg.done();
-                                       luid[0] = (String) rep;
+                                       rep = action.send(null);
                                }
 
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
-                               }
-                       }.connect();
-               } catch (IOException e) {
-                       Instance.getTraceHandler().error(e);
-               }
+                               pg.done();
+                               luid[0] = (String) rep;
+                       }
+               });
 
                if (luid[0] == null) {
                        throw new IOException("Remote failure");
                }
 
-               Story story = getStory(luid[0], pgGet);
-               pgGet.done();
-
                pg.done();
-               return story;
+               return getInfo(luid[0]);
        }
 
        @Override
@@ -468,72 +414,52 @@ public class RemoteLibrary extends BasicLibrary {
 
                final Progress pgF = pg == null ? new Progress() : pg;
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Progress pg = pgF;
-
-                                       Object rep = send(new Object[] { subkey, "CHANGE_STA",
-                                                       luid, newSource, newTitle, newAuthor });
-                                       while (true) {
-                                               if (!RemoteLibraryServer.updateProgress(pg, rep)) {
-                                                       break;
-                                               }
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Progress pg = pgF;
 
-                                               rep = send(null);
+                               Object rep = action.send(new Object[] { subkey, "CHANGE_STA",
+                                               luid, newSource, newTitle, newAuthor });
+                               while (true) {
+                                       if (!RemoteLibraryServer.updateProgress(pg, rep)) {
+                                               break;
                                        }
-                               }
 
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
+                                       rep = action.send(null);
                                }
-                       }.connect();
-               } catch (IOException e) {
-                       Instance.getTraceHandler().error(e);
-               }
+                       }
+               });
        }
 
        @Override
-       public synchronized File getFile(final String luid, Progress pg) {
+       public File getFile(final String luid, Progress pg) {
                throw new java.lang.InternalError(
                                "Operation not supportorted on remote Libraries");
        }
 
        /**
         * Stop the server.
+        * 
+        * @throws IOException
+        *             in case of I/O errors
+        * @throws SSLException
+        *             when the key was not accepted
         */
-       public void exit() {
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       send(new Object[] { subkey, "EXIT" });
-                               }
-
-                               @Override
-                               protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
-                                       }
-                               }
-                       }.connect();
-               } catch (IOException e) {
-                       Instance.getTraceHandler().error(e);
-               }
+       public void exit() throws IOException, SSLException {
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               action.send(new Object[] { subkey, "EXIT" });
+                               Thread.sleep(100);
+                       }
+               });
        }
 
        @Override
-       public synchronized MetaData getInfo(String luid) {
+       public MetaData getInfo(String luid) throws IOException {
                List<MetaData> metas = getMetasList(luid, null);
                if (!metas.isEmpty()) {
                        return metas.get(0);
@@ -543,7 +469,7 @@ public class RemoteLibrary extends BasicLibrary {
        }
 
        @Override
-       protected List<MetaData> getMetas(Progress pg) {
+       protected List<MetaData> getMetas(Progress pg) throws IOException {
                return getMetasList("*", pg);
        }
 
@@ -587,56 +513,78 @@ public class RemoteLibrary extends BasicLibrary {
         * @param pg
         *            the optional progress
         * 
-        * 
         * @return the metas
+        * 
+        * @throws IOException
+        *             in case of I/O error or bad key (SSLException)
         */
-       private List<MetaData> getMetasList(final String luid, Progress pg) {
+       private List<MetaData> getMetasList(final String luid, Progress pg)
+                       throws IOException {
                final Progress pgF = pg;
                final List<MetaData> metas = new ArrayList<MetaData>();
 
-               try {
-                       new ConnectActionClientObject(host, port, key) {
-                               @Override
-                               public void action(Version serverVersion) throws Exception {
-                                       Progress pg = pgF;
-                                       if (pg == null) {
-                                               pg = new Progress();
-                                       }
-
-                                       Object rep = send(new Object[] { subkey, "GET_METADATA",
-                                                       luid });
+               connectRemoteAction(new RemoteAction() {
+                       @Override
+                       public void action(ConnectActionClientObject action)
+                                       throws Exception {
+                               Progress pg = pgF;
+                               if (pg == null) {
+                                       pg = new Progress();
+                               }
 
-                                       while (true) {
-                                               if (!RemoteLibraryServer.updateProgress(pg, rep)) {
-                                                       break;
-                                               }
+                               Object rep = action
+                                               .send(new Object[] { subkey, "GET_METADATA", luid });
 
-                                               rep = send(null);
+                               while (true) {
+                                       if (!RemoteLibraryServer.updateProgress(pg, rep)) {
+                                               break;
                                        }
 
-                                       if (rep instanceof MetaData[]) {
-                                               for (MetaData meta : (MetaData[]) rep) {
-                                                       metas.add(meta);
-                                               }
-                                       } else if (rep != null) {
-                                               metas.add((MetaData) rep);
+                                       rep = action.send(null);
+                               }
+
+                               if (rep instanceof MetaData[]) {
+                                       for (MetaData meta : (MetaData[]) rep) {
+                                               metas.add(meta);
                                        }
+                               } else if (rep != null) {
+                                       metas.add((MetaData) rep);
+                               }
+                       }
+               });
+
+               return metas;
+       }
+
+       private void connectRemoteAction(final RemoteAction runAction)
+                       throws IOException {
+               final IOException[] err = new IOException[1];
+               try {
+                       final RemoteConnectAction[] array = new RemoteConnectAction[1];
+                       RemoteConnectAction ra = new RemoteConnectAction() {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       runAction.action(array[0]);
                                }
 
                                @Override
                                protected void onError(Exception e) {
-                                       if (e instanceof SSLException) {
-                                               Instance.getTraceHandler().error(
-                                                               "Connection refused (bad key)");
-                                       } else {
-                                               Instance.getTraceHandler().error(e);
+                                       if (!(e instanceof IOException)) {
+                                               Instance.getInstance().getTraceHandler().error(e);
+                                               return;
                                        }
+
+                                       err[0] = (IOException) e;
                                }
-                       }.connect();
+                       };
+                       array[0] = ra;
+                       ra.connect();
                } catch (Exception e) {
-                       Instance.getTraceHandler().error(e);
+                       err[0] = (IOException) e;
                }
 
-               return metas;
+               if (err[0] != null) {
+                       throw err[0];
+               }
        }
 }