Added AppleScript support

chrome/browser/cocoa/applescript/window_applescript.mm

+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#import "chrome/browser/cocoa/applescript/window_applescript.h"
+
+#include "base/logging.h"
+#import "base/scoped_nsobject.h"
+#include "base/scoped_ptr.h"
+#include "base/time.h"
+#import "chrome/browser/app_controller_mac.h"
+#include "chrome/browser/browser.h"
+#include "chrome/browser/browser_list.h"
+#include "chrome/browser/browser_window.h"
+#import "chrome/browser/chrome_browser_application_mac.h"
+#include "chrome/browser/cocoa/applescript/constants_applescript.h"
+#include "chrome/browser/cocoa/applescript/error_applescript.h"
+#import "chrome/browser/cocoa/applescript/tab_applescript.h"
+#include "chrome/browser/profile.h"
+#include "chrome/browser/tab_contents/tab_contents.h"
+#include "chrome/common/url_constants.h"
+
+@interface WindowAppleScript(WindowAppleScriptPrivateMethods)
+// The NSWindow that corresponds to this window.
+- (NSWindow*)nativeHandle;
+@end
+
+@implementation WindowAppleScript
+
+- (id)init {
+  // Check which mode to open a new window.
+  NSScriptCommand* command = [NSScriptCommand currentCommand];
+  NSString* mode = [[[command evaluatedArguments]
+      objectForKey:@"KeyDictionary"] objectForKey:@"mode"];
+  AppController* appDelegate = [NSApp delegate];
+
+  Profile* defaultProfile = [appDelegate defaultProfile];
+
+  if (!defaultProfile) {
+    AppleScript::SetError(AppleScript::errGetProfile);
+    return nil;
+  }
+
+  Profile* profile;
+  if ([mode isEqualToString:AppleScript::kIncognitoWindowMode]) {
+    profile = defaultProfile->GetOffTheRecordProfile();
+  }
+  else if ([mode isEqualToString:AppleScript::kNormalWindowMode] || !mode) {
+    profile = defaultProfile;
+  } else {
+    // Mode cannot be anything else
+    AppleScript::SetError(AppleScript::errInvalidMode);
+    return nil;
+  }
+  // Set the mode to nil, to ensure that it is not set once more.
+  [[[command evaluatedArguments] objectForKey:@"KeyDictionary"]
+      setValue:nil forKey:@"mode"];
+  return [self initWithProfile:profile];
+}
+
+- (id)initWithProfile:(Profile*)aProfile {
+  if (!aProfile) {
+    [self release];
+    return nil;
+  }
+
+  if ((self = [super init])) {
+    browser_ = Browser::Create(aProfile);
+    browser_->NewTab();
+    browser_->window()->Show();
+    scoped_nsobject<NSNumber> numID(
+        [[NSNumber alloc] initWithInt:browser_->session_id().id()]);
+    [self setUniqueID:numID];
+    [self setContainer:
+      (BrowserCrApplication*)[BrowserCrApplication sharedApplication]
+              property:AppleScript::kWindowsProperty];
+  }
+  return self;
+}
+
+- (id)initWithBrowser:(Browser*)aBrowser {
+  if (!aBrowser) {
+    [self release];
+    return nil;
+  }
+
+  if ((self = [super init])) {
+    // It is safe to be weak, if a window goes away (eg user closing a window)
+    // the applescript runtime calls appleScriptWindows in
+    // BrowserCrApplication and this particular window is never returned.
+    browser_ = aBrowser;
+    scoped_nsobject<NSNumber> numID(
+        [[NSNumber alloc] initWithInt:browser_->session_id().id()]);
+    [self setUniqueID:numID];
+    [self setContainer:NSApp
+              property:AppleScript::kWindowsProperty];
+  }
+  return self;
+}
+
+- (NSWindow*)nativeHandle {
+  // window() can be NULL during startup.
+  if (browser_->window())
+    return browser_->window()->GetNativeHandle();
+  return nil;
+}
+
+- (NSNumber*)activeTabIndex {
+  // Note: applescript is 1-based, that is lists begin with index 1.
+  int activeTabIndex = browser_->selected_index() + 1;
+  if (!activeTabIndex) {
+    return nil;
+  }
+  return [NSNumber numberWithInt:activeTabIndex];
+}
+
+- (void)setActiveTabIndex:(NSNumber*)anActiveTabIndex {
+  // Note: applescript is 1-based, that is lists begin with index 1.
+  int atIndex = [anActiveTabIndex intValue] - 1;
+  if (atIndex >= 0 && atIndex < browser_->tab_count())
+    browser_->SelectTabContentsAt(atIndex, true);
+  else
+    AppleScript::SetError(AppleScript::errInvalidTabIndex);
+}
+
+- (NSString*)mode {
+  Profile* profile = browser_->profile();
+  if (profile->IsOffTheRecord())
+    return AppleScript::kIncognitoWindowMode;
+  return AppleScript::kNormalWindowMode;
+}
+
+- (void)setMode:(NSString*)theMode {
+  // cannot set mode after window is created.
+  if (theMode) {
+    AppleScript::SetError(AppleScript::errSetMode);
+  }
+}
+
+- (TabAppleScript*)activeTab {
+  TabAppleScript* currentTab = [[[TabAppleScript alloc]
+      initWithTabContent:browser_->GetSelectedTabContents()] autorelease];
+  [currentTab setContainer:self
+                  property:AppleScript::kTabsProperty];
+  return currentTab;
+}
+
+- (NSArray*)tabs {
+  NSMutableArray* tabs = [NSMutableArray
+      arrayWithCapacity:browser_->tab_count()];
+
+  for (int i = 0; i < browser_->tab_count(); ++i) {
+    // Check to see if tab is closing.
+    if (browser_->GetTabContentsAt(i)->is_being_destroyed()) {
+      continue;
+    }
+
+    scoped_nsobject<TabAppleScript> tab(
+        [[TabAppleScript alloc]
+            initWithTabContent:(browser_->GetTabContentsAt(i))]);
+    [tab setContainer:self
+             property:AppleScript::kTabsProperty];
+    [tabs addObject:tab];
+  }
+  return tabs;
+}
+
+- (void)insertInTabs:(TabAppleScript*)aTab {
+  // This method gets called when a new tab is created so
+  // the container and property are set here.
+  [aTab setContainer:self
+            property:AppleScript::kTabsProperty];
+
+  // Set how long it takes a tab to be created.
+  base::TimeTicks newTabStartTime = base::TimeTicks::Now();
+  TabContents* tabContents = browser_->AddTabWithURL(
+                        GURL(chrome::kChromeUINewTabURL),
+                        GURL(),
+                        PageTransition::TYPED,
+                        -1, //  To indicate tab is inserted at end.
+                        TabStripModel::ADD_SELECTED,
+                        NULL,
+                        std::string());
+  tabContents->set_new_tab_start_time(newTabStartTime);
+
+  [aTab setTabContent:tabContents];
+}
+
+- (void)insertInTabs:(TabAppleScript*)aTab atIndex:(int)index {
+  // This method gets called when a new tab is created so
+  // the container and property are set here.
+  [aTab setContainer:self
+            property:AppleScript::kTabsProperty];
+
+  // Set how long it takes a tab to be created.
+  base::TimeTicks newTabStartTime = base::TimeTicks::Now();
+  TabContents* tabContents = browser_->AddTabWithURL(
+                        GURL(chrome::kChromeUINewTabURL),
+                        GURL(),
+                        PageTransition::TYPED,
+                        index,
+                        TabStripModel::ADD_SELECTED,
+                        NULL,
+                        std::string());
+  tabContents->set_new_tab_start_time(newTabStartTime);
+
+  [aTab setTabContent:tabContents];
+}
+
+- (void)removeFromTabsAtIndex:(int)index {
+  browser_->tabstrip_model()->DetachTabContentsAt(index);
+}
+
+- (NSNumber*)orderedIndex{
+  return [NSNumber numberWithInt:[[self nativeHandle] orderedIndex]];
+}
+
+- (void)setOrderedIndex:(NSNumber*)anIndex {
+  int index = [anIndex intValue] - 1;
+  if (index < 0 || index >= (int)BrowserList::size()) {
+    AppleScript::SetError(AppleScript::errWrongIndex);
+    return;
+  }
+  [[self nativeHandle] setOrderedIndex:index];
+}
+
+- (NSComparisonResult)windowComparator:(WindowAppleScript*)otherWindow {
+  int thisIndex = [[self orderedIndex] intValue];
+  int otherIndex = [[otherWindow orderedIndex] intValue];
+  if (thisIndex < otherIndex)
+    return NSOrderedAscending;
+  else if (thisIndex > otherIndex)
+    return NSOrderedDescending;
+  // Indexes can never be same.
+  NOTREACHED();
+  return NSOrderedSame;
+}
+
+// Get and set values from the associated NSWindow.
+- (id)valueForUndefinedKey:(NSString*)key {
+  return [[self nativeHandle] valueForKey:key];
+}
+
+- (void)setValue:(id)value forUndefinedKey:(NSString*)key {
+  [[self nativeHandle] setValue:(id)value forKey:key];
+}
+
+- (void)handlesCloseScriptCommand:(NSCloseCommand*)command {
+  // window() can be NULL during startup.
+  if (browser_->window())
+    browser_->window()->Close();
+}
+
+@end