Performance improvement:
[jvcard.git] / src / be / nikiroo / jvcard / launcher / Main.java
CommitLineData
7da41ecd
NR
1package be.nikiroo.jvcard.launcher;
2
3import java.io.File;
4import java.io.IOException;
5import java.lang.reflect.Field;
7da41ecd
NR
6import java.net.Socket;
7import java.nio.charset.Charset;
8import java.util.LinkedList;
9import java.util.List;
10
26d254a3
NR
11import javax.imageio.ImageIO;
12
7da41ecd 13import be.nikiroo.jvcard.Card;
26d254a3
NR
14import be.nikiroo.jvcard.Contact;
15import be.nikiroo.jvcard.Data;
16import be.nikiroo.jvcard.TypeInfo;
5ad0e17e 17import be.nikiroo.jvcard.launcher.CardResult.MergeCallback;
7da41ecd 18import be.nikiroo.jvcard.parsers.Format;
845fb1d7 19import be.nikiroo.jvcard.remote.Command;
7da41ecd
NR
20import be.nikiroo.jvcard.remote.SimpleSocket;
21import be.nikiroo.jvcard.resources.Bundles;
22import be.nikiroo.jvcard.resources.StringUtils;
e119a1c1
NR
23import be.nikiroo.jvcard.resources.bundles.ColorBundle;
24import be.nikiroo.jvcard.resources.bundles.DisplayBundle;
25import be.nikiroo.jvcard.resources.bundles.RemoteBundle;
26import be.nikiroo.jvcard.resources.bundles.TransBundle;
27import be.nikiroo.jvcard.resources.enums.DisplayOption;
28import be.nikiroo.jvcard.resources.enums.StringId;
7da41ecd
NR
29
30/**
31 * This class contains the runnable Main method. It will parse the user supplied
32 * parameters and take action based upon those. Most of the time, it will start
33 * a MainWindow.
34 *
35 * @author niki
36 *
37 */
38public class Main {
39 static public final String APPLICATION_TITLE = "jVcard";
6435013a 40 static public final String APPLICATION_VERSION = "1.0-dev";
7da41ecd
NR
41
42 static private final int ERR_NO_FILE = 1;
43 static private final int ERR_SYNTAX = 2;
44 static private final int ERR_INTERNAL = 3;
e119a1c1 45 static private TransBundle transService;
7da41ecd 46
30a4aa17
NR
47 static private String defaultFn;
48 static private boolean forceComputedFn;
49
26d254a3 50 enum Mode {
88eb8122 51 CONTACT_MANAGER, I18N, SERVER, LOAD_PHOTO, SAVE_PHOTO, SAVE_CONFIG, HELP
26d254a3
NR
52 }
53
7da41ecd 54 /**
9b8cb729 55 * Translate the given {@link StringId} into user text.
7da41ecd 56 *
9b8cb729 57 * @param stringId
7da41ecd 58 * the ID to translate
9b8cb729
NR
59 * @param values
60 * the values to insert instead of the place holders in the
61 * translation
7da41ecd 62 *
9b8cb729 63 * @return the translated text with the given value where required
7da41ecd 64 */
e119a1c1 65 static public String trans(StringId id, Object... values) {
59597d59 66 return transService.getString(id, values);
7da41ecd
NR
67 }
68
69 /**
70 * Check if unicode characters should be used.
71 *
72 * @return TRUE to allow unicode
73 */
74 static public boolean isUnicode() {
75 return transService.isUnicode();
76 }
77
78 /**
79 * Start the application.
80 *
81 * <p>
82 * The returned exit codes are:
83 * <ul>
84 * <li>1: no files to open</li>
85 * <li>2: invalid syntax</li>
86 * <li>3: internal error</li>
87 * </ul>
88 * </p>
89 *
90 * @param args
91 * the parameters (see <tt>--help</tt> to know which are
92 * supported)
93 */
94 public static void main(String[] args) {
95 Boolean textMode = null;
96 boolean noMoreParams = false;
97 boolean filesTried = false;
98
99 // get the "system default" language to help translate the --help
100 // message if needed
101 String language = null;
e119a1c1 102 transService = new TransBundle(language);
7da41ecd
NR
103
104 boolean unicode = transService.isUnicode();
26d254a3 105 String dir = null;
7da41ecd 106 List<String> files = new LinkedList<String>();
26d254a3
NR
107 int port = -1;
108 Mode mode = Mode.CONTACT_MANAGER;
109 String format = null;
7da41ecd
NR
110 for (int index = 0; index < args.length; index++) {
111 String arg = args[index];
112 if (!noMoreParams && arg.equals("--")) {
113 noMoreParams = true;
114 } else if (!noMoreParams && arg.equals("--help")) {
88eb8122
NR
115 if (mode != Mode.CONTACT_MANAGER) {
116 SERR(StringId.CLI_SERR_MODES);
117 return;
118 }
119 mode = Mode.HELP;
7da41ecd
NR
120 } else if (!noMoreParams && arg.equals("--tui")) {
121 textMode = true;
122 } else if (!noMoreParams && arg.equals("--gui")) {
123 textMode = false;
124 } else if (!noMoreParams && arg.equals("--noutf")) {
125 unicode = false;
126 transService.setUnicode(unicode);
127 } else if (!noMoreParams && arg.equals("--lang")) {
128 index++;
129 if (index >= args.length) {
88eb8122 130 SERR(StringId.CLI_SERR_NOLANG);
7da41ecd
NR
131 return;
132 }
133
134 language = args[index];
e119a1c1 135 transService = new TransBundle(language);
7da41ecd
NR
136 transService.setUnicode(unicode);
137 } else if (!noMoreParams && arg.equals("--config")) {
138 index++;
139 if (index >= args.length) {
88eb8122 140 SERR(StringId.CLI_SERR_NODIR);
7da41ecd
NR
141 return;
142 }
143
144 Bundles.setDirectory(args[index]);
e119a1c1 145 transService = new TransBundle(language);
7da41ecd 146 transService.setUnicode(unicode);
e119a1c1
NR
147 } else if (!noMoreParams && arg.equals("--save-config")) {
148 index++;
149 if (index >= args.length) {
88eb8122 150 SERR(StringId.CLI_SERR_NODIR);
e119a1c1
NR
151 return;
152 }
153 dir = args[index];
154
155 if (mode != Mode.CONTACT_MANAGER) {
88eb8122 156 SERR(StringId.CLI_SERR_MODES);
e119a1c1
NR
157 return;
158 }
159 mode = Mode.SAVE_CONFIG;
7da41ecd 160 } else if (!noMoreParams && arg.equals("--server")) {
26d254a3 161 if (mode != Mode.CONTACT_MANAGER) {
88eb8122 162 SERR(StringId.CLI_SERR_MODES);
26d254a3
NR
163 return;
164 }
165 mode = Mode.SERVER;
166
7da41ecd
NR
167 index++;
168 if (index >= args.length) {
88eb8122 169 SERR(StringId.CLI_SERR_NOPORT);
7da41ecd
NR
170 return;
171 }
172
173 try {
174 port = Integer.parseInt(args[index]);
175 } catch (NumberFormatException e) {
88eb8122 176 SERR(StringId.CLI_SERR_BADPORT, "" + args[index]);
7da41ecd
NR
177 return;
178 }
179 } else if (!noMoreParams && arg.equals("--i18n")) {
26d254a3 180 if (mode != Mode.CONTACT_MANAGER) {
88eb8122 181 SERR(StringId.CLI_SERR_MODES);
26d254a3
NR
182 return;
183 }
184 mode = Mode.I18N;
185
7da41ecd
NR
186 index++;
187 if (index >= args.length) {
88eb8122 188 SERR(StringId.CLI_SERR_NODIR);
7da41ecd
NR
189 return;
190 }
9b8cb729 191
26d254a3
NR
192 dir = args[index];
193 } else if (!noMoreParams
194 && (arg.equals("--load-photo")
195 || arg.equals("--save-photo") || arg
196 .equals("--only-photo"))) {
197 if (mode != Mode.CONTACT_MANAGER) {
88eb8122 198 SERR(StringId.CLI_SERR_MODES);
26d254a3
NR
199 return;
200 }
201
202 if (arg.equals("--load-photo")) {
203 mode = Mode.LOAD_PHOTO;
204 } else if (arg.equals("--save-photo")) {
205 mode = Mode.SAVE_PHOTO;
26d254a3
NR
206 }
207
208 index++;
209 if (index >= args.length) {
88eb8122 210 SERR(StringId.CLI_SERR_NODIR);
26d254a3
NR
211 return;
212 }
213
214 dir = args[index];
215
216 index++;
217 if (index >= args.length) {
88eb8122 218 SERR(StringId.CLI_SERR_NOFORMAT);
26d254a3
NR
219 return;
220 }
221
222 format = args[index];
7da41ecd
NR
223 } else {
224 filesTried = true;
225 files.addAll(open(arg));
226 }
227 }
9b8cb729 228
f578f3af 229 // Force headless mode if we run in forced-text mode
26d254a3 230 if (mode != Mode.CONTACT_MANAGER || (textMode != null && textMode)) {
f578f3af
NR
231 // same as -Djava.awt.headless=true
232 System.setProperty("java.awt.headless", "true");
233 }
7da41ecd
NR
234
235 if (unicode) {
236 utf8();
237 }
238
30a4aa17
NR
239 // N/FN fix information:
240 readNFN();
241
7da41ecd 242 // Error management:
26d254a3 243 if (mode == Mode.SERVER && files.size() > 0) {
88eb8122
NR
244 SERR(StringId.CLI_SERR_NOLANG, "--server");
245 return;
26d254a3 246 } else if (mode == Mode.I18N && files.size() > 0) {
88eb8122
NR
247 SERR(StringId.CLI_SERR_NOLANG, "--i18n");
248 return;
26d254a3 249 } else if (mode == Mode.I18N && language == null) {
88eb8122 250 SERR(StringId.CLI_SERR_NOLANG);
26d254a3
NR
251 } else if ((mode == Mode.CONTACT_MANAGER || mode == Mode.SAVE_PHOTO || mode == Mode.LOAD_PHOTO)
252 && files.size() == 0) {
7da41ecd
NR
253 if (files.size() == 0 && !filesTried) {
254 files.addAll(open("."));
255 }
256
257 if (files.size() == 0) {
88eb8122 258 ERR(StringId.CLI_ERR, StringId.CLI_ERR_NOFILES, ERR_NO_FILE);
7da41ecd
NR
259 return;
260 }
261 }
262 //
263
26d254a3 264 switch (mode) {
e119a1c1
NR
265 case SAVE_CONFIG: {
266 try {
267 if (!new File(dir).isDirectory()) {
268 if (!new File(dir).mkdir()) {
88eb8122
NR
269 System.err.println(trans(
270 StringId.CLI_ERR_CANNOT_CREATE_CONFDIR, dir));
e119a1c1
NR
271 }
272 }
273
274 transService.updateFile(dir); // current lang TransBundle
275 new TransBundle().updateFile(dir);
276 new ColorBundle().updateFile(dir);
277 new DisplayBundle().updateFile(dir);
278 new RemoteBundle().updateFile(dir);
279 } catch (IOException e) {
280 e.printStackTrace();
59597d59 281 System.err.flush();
e119a1c1
NR
282 System.exit(ERR_INTERNAL);
283 }
284 break;
285 }
26d254a3 286 case SERVER: {
7da41ecd 287 try {
02b341aa 288 Optional.runServer(port);
7da41ecd
NR
289 } catch (Exception e) {
290 if (e instanceof IOException) {
291 System.err
292 .println("I/O Exception: Cannot start the server");
293 } else {
88eb8122
NR
294 ERR(StringId.CLI_ERR, StringId.CLI_ERR_NO_REMOTING,
295 ERR_INTERNAL);
296 return;
7da41ecd
NR
297 }
298 }
26d254a3
NR
299 break;
300 }
301 case I18N: {
7da41ecd 302 try {
e119a1c1 303 transService.updateFile(dir);
7da41ecd 304 } catch (IOException e) {
88eb8122
NR
305 System.err.println(trans(StringId.CLI_ERR_CANNOT_CREATE_LANG,
306 dir));
e119a1c1 307 e.printStackTrace();
7da41ecd 308 }
26d254a3
NR
309 break;
310 }
26d254a3
NR
311 case LOAD_PHOTO: {
312 for (String file : files) {
313 try {
314 Card card = getCard(file, null).getCard();
315 for (Contact contact : card) {
316 String filename = contact.toString(format, "");
f29274a7 317 File f = new File(dir, filename);
26d254a3
NR
318
319 if (f.exists()) {
e3fe9834 320 System.out.println("Loading " + f);
26d254a3
NR
321 try {
322 String b64 = StringUtils.fromImage(ImageIO
323 .read(f));
324
88eb8122
NR
325 // remove previous photos:
326 for (Data photo = contact
327 .getPreferredData("PHOTO"); photo != null; photo = contact
328 .getPreferredData("PHOTO")) {
329 photo.delete();
26d254a3 330 }
88eb8122 331 //
26d254a3
NR
332
333 List<TypeInfo> types = new LinkedList<TypeInfo>();
334 types.add(new TypeInfo("ENCODING", "b"));
59597d59 335 types.add(new TypeInfo("TYPE", "jpeg"));
26d254a3
NR
336 Data photo = new Data(types, "PHOTO", b64, null);
337 contact.add(photo);
338 } catch (IOException e) {
339 System.err.println("Cannot read photo: "
340 + filename);
341 }
342 }
343 }
344 card.save();
345 } catch (IOException e) {
88eb8122
NR
346 System.err
347 .println(trans(StringId.CLI_ERR_CANNOT_OPEN, file));
26d254a3
NR
348 }
349 }
350 break;
351 }
352 case SAVE_PHOTO: {
353 for (String file : files) {
354 try {
355 Card card = getCard(file, null).getCard();
356 for (Contact contact : card) {
357 Data photo = contact.getPreferredData("PHOTO");
358 if (photo != null) {
359 String filename = contact.toString(format, "");
360 File f = new File(dir, filename + ".png");
e3fe9834 361 System.out.println("Saving " + f);
26d254a3
NR
362 try {
363 ImageIO.write(
364 StringUtils.toImage(photo.getValue()),
365 "png", f);
366 } catch (IOException e) {
88eb8122
NR
367 System.err.println(trans(
368 StringId.CLI_ERR_CANNOT_SAVE_PHOTO,
369 contact.getPreferredDataValue("FN")));
26d254a3
NR
370 }
371 }
372 }
373 } catch (IOException e) {
88eb8122
NR
374 System.err
375 .println(trans(StringId.CLI_ERR_CANNOT_OPEN, file));
26d254a3
NR
376 }
377 }
378 break;
379 }
380 case CONTACT_MANAGER: {
7da41ecd 381 try {
02b341aa 382 Optional.startTui(textMode, files);
7da41ecd
NR
383 } catch (Exception e) {
384 if (e instanceof IOException) {
88eb8122
NR
385 ERR(StringId.CLI_ERR, StringId.CLI_ERR_CANNOT_START,
386 ERR_NO_FILE);
387 return;
7da41ecd 388 } else {
88eb8122
NR
389 ERR(StringId.CLI_ERR, StringId.CLI_ERR_NO_TUI, ERR_INTERNAL);
390 return;
7da41ecd
NR
391 }
392 }
26d254a3
NR
393 break;
394 }
88eb8122
NR
395 case HELP: {
396 System.out.println(APPLICATION_TITLE + " " + APPLICATION_VERSION);
397 System.out.println();
398
399 System.out.println(trans(StringId.CLI_HELP));
400 System.out.println();
401
402 System.out.println(trans(StringId.CLI_HELP_MODES));
403 System.out.println("\t--help : "
404 + trans(StringId.CLI_HELP_MODE_HELP));
405 System.out.println("\t(--tui|--gui) (--noutf) ... : "
406 + trans(StringId.CLI_HELP_MODE_CONTACT_MANAGER));
407 System.out.println("\t--server PORT ... : "
408 + trans(StringId.CLI_HELP_MODE_SERVER));
409 System.out.println("\t--save-config DIR : "
410 + trans(StringId.CLI_HELP_MODE_SAVE_CONFIG));
411 System.out.println("\t--i18n DIR ---lang LANG : "
412 + trans(StringId.CLI_HELP_MODE_I18N));
413 System.out.println("\t--load-photo DIR FORMAT ... : "
414 + trans(StringId.CLI_HELP_MODE_LOAD_PHOTO));
415 System.out.println("\t--save-photo DIR FORMAT ... : "
416 + trans(StringId.CLI_HELP_MODE_SAVE_PHOTO));
417 System.out.println();
418
419 System.out.println(trans(StringId.CLI_HELP_OPTIONS));
420 System.out.println("\t-- : " + trans(StringId.CLI_HELP_DD));
421 System.out.println("\t--lang LANG : "
422 + trans(StringId.CLI_HELP_LANG));
423 System.out.println("\t--tui : " + trans(StringId.CLI_HELP_TUI));
424 System.out.println("\t--gui : " + trans(StringId.CLI_HELP_GUI));
425 System.out.println("\t--noutf : " + trans(StringId.CLI_HELP_NOUTF));
426 System.out.println("\t--config : "
427 + trans(StringId.CLI_HELP_CONFIG));
428 System.out.println();
429
430 System.out.println(trans(StringId.CLI_HELP_FOOTER));
431 System.out.println();
432
433 }
7da41ecd
NR
434 }
435 }
436
437 /**
438 * Return the {@link Card} corresponding to the given resource name -- a
30a4aa17
NR
439 * file or a remote jvcard URL.
440 *
441 * <p>
442 * Will also fix the FN if required (see display.properties).
443 * </p>
7da41ecd
NR
444 *
445 * @param input
446 * a filename or a remote jvcard url with named resource (e.g.:
447 * <tt>jvcard://localhost:4444/coworkers.vcf</tt>)
5ad0e17e
NR
448 * @param callback
449 * the {@link MergeCallback} to call in case of conflict, or NULL
450 * to disallow conflict management (the {@link Card} will not be
451 * allowed to synchronise in case of conflicts)
7da41ecd
NR
452 *
453 * @return the {@link Card}
454 *
455 * @throws IOException
456 * in case of IO error or remoting not available
457 */
5ad0e17e
NR
458 static public CardResult getCard(String input, MergeCallback callback)
459 throws IOException {
7da41ecd
NR
460 boolean remote = false;
461 Format format = Format.Abook;
462 String ext = input;
463 if (ext.contains(".")) {
464 String tab[] = ext.split("\\.");
465 if (tab.length > 1 && tab[tab.length - 1].equalsIgnoreCase("vcf")) {
466 format = Format.VCard21;
467 }
468 }
469
470 if (input.contains("://")) {
471 format = Format.VCard21;
472 remote = true;
473 }
474
5ad0e17e 475 CardResult card = null;
7da41ecd
NR
476 try {
477 if (remote) {
5ad0e17e 478 card = Optional.syncCard(input, callback);
7da41ecd 479 } else {
5ad0e17e
NR
480 card = new CardResult(new Card(new File(input), format), false,
481 false, false);
7da41ecd
NR
482 }
483 } catch (IOException ioe) {
484 throw ioe;
485 } catch (Exception e) {
4298276a 486 throw new IOException("Remoting support not available", e);
7da41ecd
NR
487 }
488
30a4aa17
NR
489 // Fix the FN value
490 if (defaultFn != null) {
491 try {
492 for (Contact contact : card.getCard()) {
493 Data name = contact.getPreferredData("FN");
494 if (name == null || name.getValue().length() == 0
495 || forceComputedFn) {
496 name.setValue(contact.toString(defaultFn, ""));
497 }
498 }
499 } catch (Exception e) {
500 // sync failed -> getCard() throws.
501 // do not update.
502 }
503 }
504
7da41ecd
NR
505 return card;
506 }
507
7da41ecd
NR
508 /**
509 * Open the given path and add all its files if it is a directory or just
510 * this one if not to the returned list.
511 *
512 * @param path
513 * the path to open
514 *
515 * @return the list of opened files
516 */
517 static private List<String> open(String path) {
518 List<String> files = new LinkedList<String>();
519
520 if (path != null && path.startsWith("jvcard://")) {
521 if (path.endsWith("/")) {
522 files.addAll(list(path));
523 } else {
524 files.add(path);
525 }
526 } else {
527 File file = new File(path);
528 if (file.exists()) {
529 if (file.isDirectory()) {
530 for (File subfile : file.listFiles()) {
531 if (!subfile.isDirectory())
532 files.add(subfile.getAbsolutePath());
533 }
534 } else {
535 files.add(file.getAbsolutePath());
536 }
537 } else {
538 System.err.println("File or directory not found: \"" + path
539 + "\"");
540 }
541 }
542
543 return files;
544 }
545
546 /**
547 * List all the available {@link Card}s on the given network location (which
548 * is expected to be a jVCard remote server, obviously).
549 *
550 * @param path
551 * the jVCard remote server path (e.g.:
552 * <tt>jvcard://localhost:4444/</tt>)
553 *
554 * @return the list of {@link Card}s
555 */
556 static private List<String> list(String path) {
557 List<String> files = new LinkedList<String>();
558
559 try {
560 String host = path.split("\\:")[1].substring(2);
561 int port = Integer.parseInt(path.split("\\:")[2].replaceAll("/$",
562 ""));
563 SimpleSocket s = new SimpleSocket(new Socket(host, port),
564 "sync client");
565 s.open(true);
566
845fb1d7 567 s.sendCommand(Command.LIST_CARD);
7da41ecd
NR
568 for (String p : s.receiveBlock()) {
569 files.add(path
570 + p.substring(StringUtils.fromTime(0).length() + 1));
571 }
572 s.close();
573 } catch (Exception e) {
574 e.printStackTrace();
575 }
576
577 return files;
578 }
579
580 /**
581 * Really, really ask for UTF-8 encoding.
582 */
583 static private void utf8() {
584 try {
585 System.setProperty("file.encoding", "UTF-8");
586 Field charset = Charset.class.getDeclaredField("defaultCharset");
587 charset.setAccessible(true);
588 charset.set(null, null);
589 } catch (SecurityException e) {
590 } catch (NoSuchFieldException e) {
591 } catch (IllegalArgumentException e) {
592 } catch (IllegalAccessException e) {
593 }
594 }
30a4aa17
NR
595
596 /**
597 * Read display.properties to know if we should fix the FN field when empty,
598 * or always, or never.
599 */
600 static private void readNFN() {
e119a1c1 601 DisplayBundle map = new DisplayBundle();
30a4aa17 602
e119a1c1
NR
603 defaultFn = map.getString(DisplayOption.CONTACT_DETAILS_DEFAULT_FN);
604
605 forceComputedFn = map.getBoolean(
606 DisplayOption.CONTACT_DETAILS_SHOW_COMPUTED_FN, false);
30a4aa17 607 }
88eb8122
NR
608
609 /**
610 * Syntax error detected, closing the application with an error message.
611 *
612 * @param err
613 * the syntax error case
614 */
615 static private void SERR(StringId err, Object... values) {
616 ERR(StringId.CLI_SERR, err, ERR_SYNTAX, values);
617 }
618
619 /**
620 * Error detected, closing the application with an error message.
621 *
622 * @param err
623 * the error case
624 * @param suberr
625 * the suberror or NULL if none
626 * @param CODE
627 * the error code as declared above
628 */
629 static private void ERR(StringId err, StringId suberr, int CODE,
630 Object... subvalues) {
631 if (suberr == null)
632 System.err.println(trans(err));
633 else
634 System.err.println(trans(err, trans(suberr, subvalues)));
635
59597d59 636 System.err.flush();
88eb8122
NR
637 System.exit(CODE);
638 }
7da41ecd 639}