package be.nikiroo.fanfix.library; import java.io.IOException; import java.security.InvalidParameterException; import java.util.ArrayList; import java.util.List; import be.nikiroo.fanfix.Instance; import be.nikiroo.fanfix.data.Chapter; import be.nikiroo.fanfix.data.MetaData; import be.nikiroo.fanfix.data.Paragraph; import be.nikiroo.fanfix.data.Story; import be.nikiroo.utils.Progress; import be.nikiroo.utils.Progress.ProgressListener; import be.nikiroo.utils.Version; import be.nikiroo.utils.serial.server.ConnectActionServerObject; import be.nikiroo.utils.serial.server.ServerObject; /** * Create a new remote server that will listen for order on the given port. *

* The available commands are given as arrays of objects (first item is the key, * second is the command, the rest are the arguments). *

* The key is always a String, the commands are also Strings; the parameters * vary depending upon the command. *

* * @author niki */ public class RemoteLibraryServer extends ServerObject { private final String key; /** * Create a new remote server (will not be active until * {@link RemoteLibraryServer#start()} is called). * * @param key * the key that will restrict access to this server * @param port * the port to listen on * * @throws IOException * in case of I/O error */ public RemoteLibraryServer(String key, int port) throws IOException { super("Fanfix remote library", port, true); this.key = key; setTraceHandler(Instance.getTraceHandler()); } @Override protected Object onRequest(ConnectActionServerObject action, Version clientVersion, Object data) throws Exception { String key = ""; String command = ""; Object[] args = new Object[0]; if (data instanceof Object[]) { Object[] dataArray = (Object[]) data; if (dataArray.length >= 2) { args = new Object[dataArray.length - 2]; for (int i = 2; i < dataArray.length; i++) { args[i - 2] = dataArray[i]; } key = "" + dataArray[0]; command = "" + dataArray[1]; } } String trace = "[" + command + "] "; for (Object arg : args) { trace += arg + " "; } getTraceHandler().trace(trace); if (!key.equals(this.key)) { getTraceHandler().trace("Key rejected."); return null; } if ("PING".equals(command)) { return "PONG"; } else if ("GET_METADATA".equals(command)) { if (args[0].equals("*")) { List metas = Instance.getLibrary().getMetas( createPgForwarder(action)); return metas.toArray(new MetaData[] {}); } throw new InvalidParameterException( "only * is valid here, but you passed: " + args[0]); } else if ("GET_STORY".equals(command)) { MetaData meta = Instance.getLibrary().getInfo("" + args[0]); meta = meta.clone(); meta.setCover(null); action.send(meta); action.rec(); Story story = Instance.getLibrary().getStory("" + args[0], null); for (Object obj : breakStory(story)) { action.send(obj); action.rec(); } } else if ("SAVE_STORY".equals(command)) { List list = new ArrayList(); action.send(null); Object obj = action.rec(); while (obj != null) { list.add(obj); action.send(null); obj = action.rec(); } Story story = rebuildStory(list); Instance.getLibrary().save(story, "" + args[1], null); } else if ("DELETE_STORY".equals(command)) { Instance.getLibrary().delete("" + args[0]); } else if ("GET_COVER".equals(command)) { return Instance.getLibrary().getCover("" + args[0]); } else if ("GET_SOURCE_COVER".equals(command)) { return Instance.getLibrary().getSourceCover("" + args[0]); } else if ("SET_SOURCE_COVER".equals(command)) { Instance.getLibrary().setSourceCover("" + args[0], "" + args[1]); } else if ("EXIT".equals(command)) { stop(0, false); } return null; } @Override protected void onError(Exception e) { getTraceHandler().error(e); } /** * Break a story in multiple {@link Object}s for easier serialisation. * * @param story * the {@link Story} to break * * @return the list of {@link Object}s */ static List breakStory(Story story) { List list = new ArrayList(); story = story.clone(); list.add(story); if (story.getMeta().isImageDocument()) { for (Chapter chap : story) { list.add(chap); list.addAll(chap.getParagraphs()); chap.setParagraphs(new ArrayList()); } story.setChapters(new ArrayList()); } return list; } /** * Rebuild a story from a list of broke up {@link Story} parts. * * @param list * the list of {@link Story} parts * * @return the reconstructed {@link Story} */ static Story rebuildStory(List list) { Story story = null; Chapter chap = null; for (Object obj : list) { if (obj instanceof Story) { story = (Story) obj; } else if (obj instanceof Chapter) { chap = (Chapter) obj; story.getChapters().add(chap); } else if (obj instanceof Paragraph) { chap.getParagraphs().add((Paragraph) obj); } } return story; } /** * Update the {@link Progress} with the adequate {@link Object} received * from the network via {@link RemoteLibraryServer}. * * @param pg * the {@link Progress} to update * @param rep * the object received from the network * * @return TRUE if it was a progress event, FALSE if not */ static boolean updateProgress(Progress pg, Object rep) { if (rep instanceof Integer[]) { Integer[] a = (Integer[]) rep; if (a.length == 3) { int min = a[0]; int max = a[1]; int progress = a[2]; if (min >= 0 && min <= max) { pg.setMinMax(min, max); pg.setProgress(progress); return true; } } } return false; } /** * Create a {@link Progress} that will forward its progress over the * network. * * @param action * the {@link ConnectActionServerObject} to use to forward it * * @return the {@link Progress} */ private static Progress createPgForwarder( final ConnectActionServerObject action) { final Progress pg = new Progress(); final Integer[] p = new Integer[] { -1, -1, -1 }; pg.addProgressListener(new ProgressListener() { @Override public void progress(Progress progress, String name) { int min = pg.getMin(); int max = pg.getMax(); int relativeProgress = min + (int) Math.round(pg.getRelativeProgress() * (max - min)); // Do not re-send the same value twice over the wire if (p[0] != min || p[1] != max || p[2] != relativeProgress) { p[0] = min; p[1] = max; p[2] = relativeProgress; try { action.send(new Integer[] { min, max, relativeProgress }); action.rec(); } catch (Exception e) { Instance.getTraceHandler().error(e); } } } }); return pg; } }