package be.nikiroo.fanfix.library; import java.io.File; import java.io.IOException; import java.net.URL; import java.net.UnknownHostException; import java.util.ArrayList; import java.util.List; import javax.net.ssl.SSLException; import be.nikiroo.fanfix.Instance; import be.nikiroo.fanfix.data.MetaData; import be.nikiroo.fanfix.data.Story; import be.nikiroo.utils.Image; import be.nikiroo.utils.Progress; import be.nikiroo.utils.Version; 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. * * @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; private final String subkey; // informative only (server will make the actual checks) private boolean rw; /** * Create a {@link RemoteLibrary} linked to the given server. *

* Note that the key is structured: * xxx(|yyy|wl)(|rw) *

* Note that anything before the first pipe (|) character is * considered to be the encryption key, anything after that character is * called the subkey (including the other pipe characters and flags!). *

* This is important because the subkey (including the pipe characters and * flags) must be present as-is in the server configuration file to be * allowed. *

*

* Some examples: *

* * @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 key, String host, int port) { int index = -1; if (key != null) { index = key.indexOf('|'); } if (index >= 0) { this.key = key.substring(0, index); this.subkey = key.substring(index + 1); } else { this.key = key; this.subkey = ""; } this.host = host; this.port = port; } @Override public String getLibraryName() { return (rw ? "[READ-ONLY] " : "") + host + ":" + port; } @Override public Status getStatus() { Instance.getInstance().getTraceHandler().trace("Getting remote lib status..."); Status status = getStatusDo(); Instance.getInstance().getTraceHandler().trace("Remote lib status: " + status); return status; } private Status getStatusDo() { final Status[] result = new Status[1]; result[0] = Status.INVALID; try { 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.READ_WRITE; } else if ("r/o".equals(rep)) { rw = false; result[0] = Status.READ_ONLY; } else { result[0] = Status.UNAUTHORIZED; } } @Override protected void onError(Exception e) { if (e instanceof SSLException) { result[0] = Status.UNAUTHORIZED; } else { result[0] = Status.UNAVAILABLE; } } }.connect(); } catch (UnknownHostException e) { result[0] = Status.INVALID; } catch (IllegalArgumentException e) { result[0] = Status.INVALID; } catch (Exception e) { result[0] = Status.UNAVAILABLE; } return result[0]; } @Override public Image getCover(final String luid) throws IOException { final Image[] result = new Image[1]; 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) throws IOException { return getCustomCover(source, "SOURCE"); } @Override 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) throws IOException { final Image[] result = new Image[1]; 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) throws IOException { final Progress pgF = pg; final Story[] result = new Story[1]; connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { Progress pg = pgF; if (pg == null) { pg = new Progress(); } 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()); } } List list = new ArrayList(); for (Object obj = action.send(null); obj != null; obj = action .send(null)) { list.add(obj); pg.add(1); } result[0] = RemoteLibraryServer.rebuildStory(list); pg.done(); } }); return result[0]; } @Override 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(); } pg.setMinMax(0, 10); pg.addProgress(pgSave, 9); pg.addProgress(pgRefresh, 1); final Progress pgF = pgSave; connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { Progress pg = pgF; if (story.getMeta().getWords() <= Integer.MAX_VALUE) { pg.setMinMax(0, (int) story.getMeta().getWords()); } action.send(new Object[] { subkey, "SAVE_STORY", luid }); List list = RemoteLibraryServer.breakStory(story); for (Object obj : list) { action.send(obj); pg.add(1); } luidSaved[0] = (String) action.send(null); pg.done(); } }); // because the meta changed: MetaData meta = getInfo(luidSaved[0]); if (story.getMeta().getClass() != null) { // If already available locally: meta.setCover(story.getMeta().getCover()); } else { // If required: meta.setCover(getCover(meta.getLuid())); } story.setMeta(meta); pg.done(); return story; } @Override public synchronized void delete(final String luid) throws IOException { connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { action.send(new Object[] { subkey, "DELETE_STORY", luid }); } }); } @Override 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) throws IOException { setCover(author, luid, "AUTHOR"); } // type = "SOURCE" | "AUTHOR" private void setCover(final String value, final String luid, 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 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); } // Import it remotely if it is an URL if (pg == null) { pg = new Progress(); } final Progress pgF = pg; final String[] luid = new String[1]; connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { Progress pg = pgF; Object rep = action.send(new Object[] { subkey, "IMPORT", url.toString() }); while (true) { if (!RemoteLibraryServer.updateProgress(pg, rep)) { break; } rep = action.send(null); } pg.done(); luid[0] = (String) rep; } }); if (luid[0] == null) { throw new IOException("Remote failure"); } pg.done(); return getInfo(luid[0]); } @Override // Could work (more slowly) without it protected synchronized void changeSTA(final String luid, final String newSource, final String newTitle, final String newAuthor, Progress pg) throws IOException { final Progress pgF = pg == null ? new Progress() : pg; connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { Progress pg = pgF; Object rep = action.send(new Object[] { subkey, "CHANGE_STA", luid, newSource, newTitle, newAuthor }); while (true) { if (!RemoteLibraryServer.updateProgress(pg, rep)) { break; } rep = action.send(null); } } }); } @Override 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() 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 MetaData getInfo(String luid) throws IOException { List metas = getMetasList(luid, null); if (!metas.isEmpty()) { return metas.get(0); } return null; } @Override protected List getMetas(Progress pg) throws IOException { return getMetasList("*", pg); } @Override protected void updateInfo(MetaData meta) { // Will be taken care of directly server side } @Override protected void invalidateInfo(String luid) { // Will be taken care of directly server side } // The following methods are only used by Save and Delete in BasicLibrary: @Override 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"); } // /** * Return the meta of the given story or a list of all known metas if the * luid is "*". *

* Will not get the covers. * * @param luid * the luid of the story or * * @param pg * the optional progress * * @return the metas * * @throws IOException * in case of I/O error or bad key (SSLException) */ private List getMetasList(final String luid, Progress pg) throws IOException { final Progress pgF = pg; final List metas = new ArrayList(); connectRemoteAction(new RemoteAction() { @Override public void action(ConnectActionClientObject action) throws Exception { Progress pg = pgF; if (pg == null) { pg = new Progress(); } Object rep = action.send(new Object[] { subkey, "GET_METADATA", luid }); while (true) { if (!RemoteLibraryServer.updateProgress(pg, rep)) { break; } 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 IOException)) { Instance.getInstance().getTraceHandler().error(e); return; } err[0] = (IOException) e; } }; array[0] = ra; ra.connect(); } catch (Exception e) { err[0] = (IOException) e; } if (err[0] != null) { throw err[0]; } } }