Fix remote saving
[fanfix.git] / src / be / nikiroo / fanfix / library / RemoteLibrary.java
index 83766396499fa783712b0558e51c3ad63def314e..f3104135f2977b689a222ea68242c8bc5be75d27 100644 (file)
@@ -3,20 +3,21 @@ package be.nikiroo.fanfix.library;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
+import java.net.UnknownHostException;
 import java.util.ArrayList;
 import java.util.List;
 
 import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.data.MetaData;
 import be.nikiroo.fanfix.data.Story;
-import be.nikiroo.fanfix.output.BasicOutput.OutputType;
 import be.nikiroo.utils.Progress;
+import be.nikiroo.utils.StringUtils;
 import be.nikiroo.utils.Version;
-import be.nikiroo.utils.serial.ConnectActionClient;
+import be.nikiroo.utils.serial.server.ConnectActionClientObject;
 
 /**
  * This {@link BasicLibrary} will access a remote server to list the available
- * stories, and download the one you try to load to the local directory
+ * stories, and download the ones you try to load to the local directory
  * specified in the configuration.
  * 
  * @author niki
@@ -24,139 +25,343 @@ import be.nikiroo.utils.serial.ConnectActionClient;
 public class RemoteLibrary extends BasicLibrary {
        private String host;
        private int port;
-       private File baseDir;
-
-       private LocalLibrary lib;
-       private List<MetaData> metas;
+       private final String md5;
 
        /**
         * Create a {@link RemoteLibrary} linked to the given server.
         * 
+        * @param key
+        *            the key that will allow us to exchange information with the
+        *            server
         * @param host
         *            the host to contact or NULL for localhost
         * @param port
         *            the port to contact it on
         */
-       public RemoteLibrary(String host, int port) {
+       public RemoteLibrary(String key, String host, int port) {
+               this.md5 = StringUtils.getMd5Hash(key);
                this.host = host;
                this.port = port;
+       }
 
-               this.baseDir = Instance.getRemoteDir(host);
-               this.baseDir.mkdirs();
-
-               this.lib = new LocalLibrary(baseDir, OutputType.INFO_TEXT,
-                               OutputType.CBZ);
+       @Override
+       public String getLibraryName() {
+               return host + ":" + port;
        }
 
        @Override
-       protected List<MetaData> getMetas(Progress pg) {
-               // TODO: progress
+       public Status getStatus() {
+               final Status[] result = new Status[1];
 
-               if (metas == null) {
-                       metas = new ArrayList<MetaData>();
+               result[0] = Status.INVALID;
 
-                       try {
-                               new ConnectActionClient(host, port, true, null) {
-                                       @Override
-                                       public void action(Version serverVersion) throws Exception {
-                                               try {
-                                                       Object rep = send("GET_METADATA *");
-                                                       for (MetaData meta : (MetaData[]) rep) {
-                                                               metas.add(meta);
-                                                       }
-                                               } catch (Exception e) {
-                                                       Instance.syserr(e);
-                                               }
+               ConnectActionClientObject action = null;
+               try {
+                       action = new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       Object rep = send(new Object[] { md5, "PING" });
+                                       if ("PONG".equals(rep)) {
+                                               result[0] = Status.READY;
+                                       } else {
+                                               result[0] = Status.UNAUTORIZED;
                                        }
-                               }.connect();
-                       } catch (IOException e) {
-                               Instance.syserr(e);
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       result[0] = Status.UNAVAILABLE;
+                               }
+                       };
+
+               } catch (UnknownHostException e) {
+                       result[0] = Status.INVALID;
+               } catch (IllegalArgumentException e) {
+                       result[0] = Status.INVALID;
+               } catch (Exception e) {
+                       result[0] = Status.UNAVAILABLE;
+               }
+
+               if (action != null) {
+                       try {
+                               action.connect();
+                       } catch (Exception e) {
+                               result[0] = Status.UNAVAILABLE;
                        }
                }
 
-               return metas;
+               return result[0];
        }
 
        @Override
-       public synchronized File getFile(final String luid) {
-               File file = lib.getFile(luid);
-               if (file == null) {
-                       final File[] tmp = new File[1];
-                       try {
-                               new ConnectActionClient(host, port, true, null) {
-                                       @Override
-                                       public void action(Version serverVersion) throws Exception {
-                                               try {
-                                                       Object rep = send("GET_STORY " + luid);
-                                                       Story story = (Story) rep;
-                                                       if (story != null) {
-                                                               lib.save(story, luid, null);
-                                                               tmp[0] = lib.getFile(luid);
-                                                       }
-                                               } catch (Exception e) {
-                                                       Instance.syserr(e);
+       public BufferedImage getCover(final String luid) {
+               final BufferedImage[] result = new BufferedImage[1];
+
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       Object rep = send(new Object[] { md5, "GET_COVER", luid });
+                                       result[0] = (BufferedImage) rep;
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (Exception e) {
+                       Instance.getTraceHandler().error(e);
+               }
+
+               return result[0];
+       }
+
+       @Override
+       public BufferedImage getSourceCover(final String source) {
+               final BufferedImage[] result = new BufferedImage[1];
+
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       Object rep = send(new Object[] { md5, "GET_SOURCE_COVER",
+                                                       source });
+                                       result[0] = (BufferedImage) rep;
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (Exception e) {
+                       Instance.getTraceHandler().error(e);
+               }
+
+               return result[0];
+       }
+
+       @Override
+       public synchronized Story getStory(final String luid, Progress pg) {
+               final Progress pgF = pg;
+               final Story[] result = new Story[1];
+
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       Progress pg = pgF;
+                                       if (pg == null) {
+                                               pg = new Progress();
+                                       }
+
+                                       Object rep = send(new Object[] { md5, "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());
                                                }
                                        }
-                               }.connect();
-                       } catch (IOException e) {
-                               Instance.syserr(e);
-                       }
 
-                       file = tmp[0];
-               }
+                                       List<Object> list = new ArrayList<Object>();
+                                       for (Object obj = send(null); obj != null; obj = send(null)) {
+                                               list.add(obj);
+                                               pg.add(1);
+                                       }
 
-               if (file != null) {
-                       MetaData meta = getInfo(luid);
-                       metas.add(meta);
+                                       result[0] = RemoteLibraryServer.rebuildStory(list);
+                                       pg.done();
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (Exception e) {
+                       Instance.getTraceHandler().error(e);
                }
 
-               return file;
+               return result[0];
        }
 
        @Override
-       public BufferedImage getCover(String luid) {
-               // Retrieve it from the network if needed:
-               if (lib.getInfo(luid) == null) {
-                       getFile(luid);
+       public synchronized Story save(final Story story, final String luid,
+                       Progress pg) throws IOException {
+               final String[] luidSaved = new String[1];
+               Progress pgSave = new Progress();
+               Progress pgRefresh = new Progress();
+               if (pg == null) {
+                       pg = new Progress();
                }
 
-               return lib.getCover(luid);
+               pg.setMinMax(0, 10);
+               pg.addProgress(pgSave, 9);
+               pg.addProgress(pgRefresh, 1);
+
+               final Progress pgF = pgSave;
+
+               new ConnectActionClientObject(host, port, true) {
+                       @Override
+                       public void action(Version serverVersion) throws Exception {
+                               Progress pg = pgF;
+                               if (story.getMeta().getWords() <= Integer.MAX_VALUE) {
+                                       pg.setMinMax(0, (int) story.getMeta().getWords());
+                               }
+
+                               send(new Object[] { md5, "SAVE_STORY", luid });
+
+                               List<Object> list = RemoteLibraryServer.breakStory(story);
+                               for (Object obj : list) {
+                                       send(obj);
+                                       pg.add(1);
+                               }
+
+                               send(null); // done sending the story
+                               luidSaved[0] = (String) send(null); // get LUID
+
+                               pg.done();
+                       }
+
+                       @Override
+                       protected void onError(Exception e) {
+                               Instance.getTraceHandler().error(e);
+                       }
+               }.connect();
+
+               // because the meta changed:
+               clearCache();
+               refresh(pgRefresh);
+               story.setMeta(getInfo(luidSaved[0]));
+
+               pg.done();
+
+               return story;
        }
 
        @Override
-       protected void clearCache() {
-               metas = null;
-               lib.clearCache();
+       public synchronized void delete(final String luid) throws IOException {
+               new ConnectActionClientObject(host, port, true) {
+                       @Override
+                       public void action(Version serverVersion) throws Exception {
+                               send(new Object[] { md5, "DELETE_STORY", luid });
+                       }
+
+                       @Override
+                       protected void onError(Exception e) {
+                               Instance.getTraceHandler().error(e);
+                       }
+               }.connect();
        }
 
        @Override
-       public synchronized Story save(Story story, String luid, Progress pg)
-                       throws IOException {
-               throw new java.lang.InternalError(
-                               "No write support allowed on remote Libraries");
+       public void setSourceCover(final String source, final String luid) {
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       send(new Object[] { md5, "SET_SOURCE_COVER", source, luid });
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (IOException e) {
+                       Instance.getTraceHandler().error(e);
+               }
        }
 
        @Override
-       protected int getNextId() {
+       public synchronized File getFile(final String luid, Progress pg) {
                throw new java.lang.InternalError(
-                               "No write support allowed on remote Libraries");
+                               "Operation not supportorted on remote Libraries");
+       }
+
+       /**
+        * Stop the server.
+        */
+       public void exit() {
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       send(new Object[] { md5, "EXIT" });
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (IOException e) {
+                       Instance.getTraceHandler().error(e);
+               }
        }
 
        @Override
-       protected void doDelete(String luid) throws IOException {
-               throw new java.lang.InternalError(
-                               "No write support allowed on remote Libraries");
+       protected List<MetaData> getMetas(Progress pg) {
+               final Progress pgF = pg;
+               final List<MetaData> metas = new ArrayList<MetaData>();
+
+               try {
+                       new ConnectActionClientObject(host, port, true) {
+                               @Override
+                               public void action(Version serverVersion) throws Exception {
+                                       Progress pg = pgF;
+                                       if (pg == null) {
+                                               pg = new Progress();
+                                       }
+
+                                       Object rep = send(new Object[] { md5, "GET_METADATA", "*" });
+
+                                       while (true) {
+                                               if (!RemoteLibraryServer.updateProgress(pg, rep)) {
+                                                       break;
+                                               }
+
+                                               rep = send(null);
+                                       }
+
+                                       for (MetaData meta : (MetaData[]) rep) {
+                                               metas.add(meta);
+                                       }
+                               }
+
+                               @Override
+                               protected void onError(Exception e) {
+                                       Instance.getTraceHandler().error(e);
+                               }
+                       }.connect();
+               } catch (Exception e) {
+                       Instance.getTraceHandler().error(e);
+               }
+
+               return metas;
        }
 
        @Override
-       protected Story doSave(Story story, Progress pg) throws IOException {
-               throw new java.lang.InternalError(
-                               "No write support allowed on remote Libraries");
+       protected void clearCache() {
        }
 
+       // The following methods are only used by Save and Delete in BasicLibrary:
+
        @Override
-       public void setSourceCover(String source, String luid) {
-               throw new java.lang.InternalError(
-                               "No write support allowed on remote Libraries");
+       protected int getNextId() {
+               throw new java.lang.InternalError("Should not have been called");
+       }
+
+       @Override
+       protected void doDelete(String luid) throws IOException {
+               throw new java.lang.InternalError("Should not have been called");
+       }
+
+       @Override
+       protected Story doSave(Story story, Progress pg) throws IOException {
+               throw new java.lang.InternalError("Should not have been called");
        }
 }