1 package be
.nikiroo
.fanfix
.library
;
3 import java
.io
.IOException
;
5 import java
.util
.ArrayList
;
9 import be
.nikiroo
.fanfix
.Instance
;
10 import be
.nikiroo
.fanfix
.data
.Chapter
;
11 import be
.nikiroo
.fanfix
.data
.MetaData
;
12 import be
.nikiroo
.fanfix
.data
.Paragraph
;
13 import be
.nikiroo
.fanfix
.data
.Story
;
14 import be
.nikiroo
.utils
.Progress
;
15 import be
.nikiroo
.utils
.Progress
.ProgressListener
;
16 import be
.nikiroo
.utils
.StringUtils
;
17 import be
.nikiroo
.utils
.Version
;
18 import be
.nikiroo
.utils
.serial
.server
.ConnectActionServerObject
;
19 import be
.nikiroo
.utils
.serial
.server
.ServerObject
;
22 * Create a new remote server that will listen for order on the given port.
24 * The available commands are given as arrays of objects (first item is the key,
25 * second is the command, the rest are the arguments).
27 * The md5 is always a String (the MD5 hash of the access key), the commands are
28 * also Strings; the parameters vary depending upon the command.
30 * <li>[md5] PING: will return PONG if the key is accepted</li>
31 * <li>[md5] GET_METADATA *: will return the metadata of all the stories in the
32 * library (array)</li>
34 * <li>[md5] GET_METADATA [luid]: will return the metadata of the story of LUID
36 * <li>[md5] GET_STORY [luid]: will return the given story if it exists (or NULL
38 * <li>[md5] SAVE_STORY [luid]: save the story (that must be sent just after the
39 * command) with the given LUID, then return the LUID</li>
40 * <li>[md5] IMPORT [url]: save the story found at the given URL, then return
42 * <li>[md5] DELETE_STORY [luid]: delete the story of LUID luid</li>
43 * <li>[md5] GET_COVER [luid]: return the cover of the story</li>
44 * <li>[md5] GET_SOURCE_COVER [source]: return the cover for this source</li>
45 * <li>[md5] SET_SOURCE_COVER [source], [luid]: set the default cover for the
46 * given source to the cover of the story denoted by luid</li>
47 * <li>[md5] CHANGE_SOURCE [luid] [new source]: change the source of the story
49 * <li>[md5] EXIT: stop the server</li>
54 public class RemoteLibraryServer
extends ServerObject
{
55 private final String md5
;
58 * Create a new remote server (will not be active until
59 * {@link RemoteLibraryServer#start()} is called).
62 * the key that will restrict access to this server
64 * the port to listen on
67 * in case of I/O error
69 public RemoteLibraryServer(String key
, int port
) throws IOException
{
70 super("Fanfix remote library", port
, true);
71 this.md5
= StringUtils
.getMd5Hash(key
);
73 setTraceHandler(Instance
.getTraceHandler());
77 protected Object
onRequest(ConnectActionServerObject action
,
78 Version clientVersion
, Object data
) throws Exception
{
81 Object
[] args
= new Object
[0];
82 if (data
instanceof Object
[]) {
83 Object
[] dataArray
= (Object
[]) data
;
84 if (dataArray
.length
>= 2) {
85 md5
= "" + dataArray
[0];
86 command
= "" + dataArray
[1];
88 args
= new Object
[dataArray
.length
- 2];
89 for (int i
= 2; i
< dataArray
.length
; i
++) {
90 args
[i
- 2] = dataArray
[i
];
95 String trace
= "[" + command
+ "] ";
96 for (Object arg
: args
) {
99 getTraceHandler().trace(trace
);
101 if (!md5
.equals(this.md5
)) {
102 getTraceHandler().trace("Key rejected.");
106 if ("PING".equals(command
)) {
108 } else if ("GET_METADATA".equals(command
)) {
109 if ("*".equals(args
[0])) {
110 Progress pg
= createPgForwarder(action
);
111 List
<MetaData
> metas
= Instance
.getLibrary().getMetas(pg
);
113 return metas
.toArray(new MetaData
[] {});
116 return new MetaData
[] { Instance
.getLibrary().getInfo(
118 } else if ("GET_STORY".equals(command
)) {
119 MetaData meta
= Instance
.getLibrary().getInfo((String
) args
[0]);
126 Story story
= Instance
.getLibrary()
127 .getStory((String
) args
[0], null);
128 for (Object obj
: breakStory(story
)) {
132 } else if ("SAVE_STORY".equals(command
)) {
133 List
<Object
> list
= new ArrayList
<Object
>();
136 Object obj
= action
.rec();
137 while (obj
!= null) {
143 Story story
= rebuildStory(list
);
144 Instance
.getLibrary().save(story
, (String
) args
[0], null);
145 return story
.getMeta().getLuid();
146 } else if ("IMPORT".equals(command
)) {
147 Progress pg
= createPgForwarder(action
);
148 Story story
= Instance
.getLibrary().imprt(
149 new URL((String
) args
[0]), pg
);
151 return story
.getMeta().getLuid();
152 } else if ("DELETE_STORY".equals(command
)) {
153 Instance
.getLibrary().delete((String
) args
[0]);
154 } else if ("GET_COVER".equals(command
)) {
155 return Instance
.getLibrary().getCover((String
) args
[0]);
156 } else if ("GET_SOURCE_COVER".equals(command
)) {
157 return Instance
.getLibrary().getSourceCover((String
) args
[0]);
158 } else if ("SET_SOURCE_COVER".equals(command
)) {
159 Instance
.getLibrary().setSourceCover((String
) args
[0],
161 } else if ("CHANGE_SOURCE".equals(command
)) {
162 Progress pg
= createPgForwarder(action
);
163 Instance
.getLibrary().changeSource((String
) args
[0],
164 (String
) args
[1], pg
);
166 } else if ("EXIT".equals(command
)) {
174 protected void onError(Exception e
) {
175 getTraceHandler().error(e
);
179 * Break a story in multiple {@link Object}s for easier serialisation.
182 * the {@link Story} to break
184 * @return the list of {@link Object}s
186 static List
<Object
> breakStory(Story story
) {
187 List
<Object
> list
= new ArrayList
<Object
>();
189 story
= story
.clone();
192 if (story
.getMeta().isImageDocument()) {
193 for (Chapter chap
: story
) {
195 list
.addAll(chap
.getParagraphs());
196 chap
.setParagraphs(new ArrayList
<Paragraph
>());
198 story
.setChapters(new ArrayList
<Chapter
>());
205 * Rebuild a story from a list of broke up {@link Story} parts.
208 * the list of {@link Story} parts
210 * @return the reconstructed {@link Story}
212 static Story
rebuildStory(List
<Object
> list
) {
216 for (Object obj
: list
) {
217 if (obj
instanceof Story
) {
219 } else if (obj
instanceof Chapter
) {
220 chap
= (Chapter
) obj
;
221 story
.getChapters().add(chap
);
222 } else if (obj
instanceof Paragraph
) {
223 chap
.getParagraphs().add((Paragraph
) obj
);
231 * Update the {@link Progress} with the adequate {@link Object} received
232 * from the network via {@link RemoteLibraryServer}.
235 * the {@link Progress} to update
237 * the object received from the network
239 * @return TRUE if it was a progress event, FALSE if not
241 static boolean updateProgress(Progress pg
, Object rep
) {
242 if (rep
instanceof Integer
[]) {
243 Integer
[] a
= (Integer
[]) rep
;
249 if (min
>= 0 && min
<= max
) {
250 pg
.setMinMax(min
, max
);
251 pg
.setProgress(progress
);
262 * Create a {@link Progress} that will forward its progress over the
266 * the {@link ConnectActionServerObject} to use to forward it
268 * @return the {@link Progress}
270 private static Progress
createPgForwarder(
271 final ConnectActionServerObject action
) {
272 final Boolean
[] isDoneForwarded
= new Boolean
[] { false };
273 final Progress pg
= new Progress() {
275 public boolean isDone() {
276 return isDoneForwarded
[0];
280 final Integer
[] p
= new Integer
[] { -1, -1, -1 };
281 final Long
[] lastTime
= new Long
[] { new Date().getTime() };
282 pg
.addProgressListener(new ProgressListener() {
284 public void progress(Progress progress
, String name
) {
285 int min
= pg
.getMin();
286 int max
= pg
.getMax();
287 int relativeProgress
= min
288 + (int) Math
.round(pg
.getRelativeProgress()
291 // Do not re-send the same value twice over the wire,
292 // unless more than 2 seconds have elapsed (to maintain the
294 if ((p
[0] != min
|| p
[1] != max
|| p
[2] != relativeProgress
)
295 || (new Date().getTime() - lastTime
[0] > 2000)) {
298 p
[2] = relativeProgress
;
301 action
.send(new Integer
[] { min
, max
, relativeProgress
});
303 } catch (Exception e
) {
304 Instance
.getTraceHandler().error(e
);
307 isDoneForwarded
[0] = pg
.isDone();
308 lastTime
[0] = new Date().getTime();
316 // with 30 seconds timeout
317 private static void forcePgDoneSent(Progress pg
) {
318 long start
= new Date().getTime();
320 while (!pg
.isDone() && new Date().getTime() - start
< 30000) {
323 } catch (InterruptedException e
) {
324 Instance
.getTraceHandler().error(e
);