main.m

#import "SquirrelApplicationDelegate.h"
#import <Cocoa/Cocoa.h>
#import <InputMethodKit/InputMethodKit.h>
#import <rime_api.h>
#import <string.h>

void RegisterInputSource();
void ActivateInputSource();
void DeactivateInputSource();
// BOOL IsInputSourceActive();

// Each input method needs a unique connection name.
// Note that periods and spaces are not allowed in the connection name.
const NSString *kConnectionName = @"ThoanTaigi_1_Connection";

int main(int argc, char *argv[]) {
  if (argc > 1 && !strcmp("--quit", argv[1])) {
    NSString *bundleId = [NSBundle mainBundle].bundleIdentifier;
    NSArray *runningSquirrels =
        [NSRunningApplication runningApplicationsWithBundleIdentifier:bundleId];
    for (NSRunningApplication *squirrelApp in runningSquirrels) {
      [squirrelApp terminate];
    }
    return 0;
  }

  if (argc > 1 && !strcmp("--reload", argv[1])) {
    [[NSDistributedNotificationCenter defaultCenter]
        postNotificationName:@"SquirrelReloadNotification"
                      object:nil];
    return 0;
  }

  if (argc > 1 && !strcmp("--install", argv[1])) {
    // register and enable Squirrel
    RegisterInputSource();
    DeactivateInputSource();
    ActivateInputSource();
    return 0;
  }

  if (argc > 1 && !strcmp("--build", argv[1])) {
    // notification
    show_message("deploy_update", "deploy");
    // build all schemas in current directory
    RIME_STRUCT(RimeTraits, builder_traits);
    builder_traits.app_name = "rime.squirrel-builder";
    rime_get_api()->setup(&builder_traits);
    rime_get_api()->deployer_initialize(NULL);
    return rime_get_api()->deploy() ? 0 : 1;
  }

  @autoreleasepool {
    // find the bundle identifier and then initialize the input method server
    IMKServer *server __unused =
        [[IMKServer alloc] initWithName:(NSString *)kConnectionName
                       bundleIdentifier:[NSBundle mainBundle].bundleIdentifier];

    // load the bundle explicitly because in this case the input method is a
    // background only application
    [NSBundle loadNibNamed:@"MainMenu" owner:[NSApplication sharedApplication]];

    // opencc will be configured with relative dictionary paths
    [[NSFileManager defaultManager]
        changeCurrentDirectoryPath:[NSBundle mainBundle].sharedSupportPath];

    if (NSApp.squirrelAppDelegate.problematicLaunchDetected) {
      NSLog(@"Problematic launch detected!");
      NSArray *args = @[
        @"Problematic launch detected! \
                       Squirrel may be suffering a crash due to imporper configuration. \
                       Revert previous modifications to see if the problem recurs."
      ];
      [NSTask launchedTaskWithLaunchPath:@"/usr/bin/say" arguments:args];
    } else {
      [NSApp.squirrelAppDelegate setupRime];
      [NSApp.squirrelAppDelegate startRimeWithFullCheck:NO];
      [NSApp.squirrelAppDelegate loadSettings];
      NSLog(@"Squirrel reporting!");
    }

    // finally run everything
    [[NSApplication sharedApplication] run];

    NSLog(@"Squirrel is quitting...");
    rime_get_api()->finalize();
  }
  return 0;
}