src/be/nikiroo/jvcard/launcher/Main.java

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