Added AppleScript support

chrome/browser/cocoa/applescript/bookmark_folder_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/bookmark_folder_applescript.h"
+
+#import "base/scoped_nsobject.h"
+#include "base/sys_string_conversions.h"
+#include "chrome/browser/bookmarks/bookmark_model.h"
+#import "chrome/browser/cocoa/applescript/bookmark_item_applescript.h"
+#import "chrome/browser/cocoa/applescript/constants_applescript.h"
+#include "chrome/browser/cocoa/applescript/error_applescript.h"
+#include "googleurl/src/gurl.h"
+
+@implementation BookmarkFolderAppleScript
+
+- (NSArray*)bookmarkFolders {
+  NSMutableArray* bookmarkFolders = [NSMutableArray
+      arrayWithCapacity:bookmarkNode_->GetChildCount()];
+
+  for (int i = 0; i < bookmarkNode_->GetChildCount(); ++i) {
+    const BookmarkNode* node = bookmarkNode_->GetChild(i);
+
+    if (!node->is_folder())
+      continue;
+    scoped_nsobject<BookmarkFolderAppleScript> bookmarkFolder(
+        [[BookmarkFolderAppleScript alloc]
+            initWithBookmarkNode:node]);
+    [bookmarkFolder setContainer:self
+                        property:AppleScript::kBookmarkFoldersProperty];
+    [bookmarkFolders addObject:bookmarkFolder];
+  }
+
+  return bookmarkFolders;
+}
+
+- (void)insertInBookmarkFolders:(id)aBookmarkFolder {
+  // This method gets called when a new bookmark folder is created so
+  // the container and property are set here.
+  [aBookmarkFolder setContainer:self
+                       property:AppleScript::kBookmarkFoldersProperty];
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  const BookmarkNode* node = model->AddGroup(bookmarkNode_,
+                                             bookmarkNode_->GetChildCount(),
+                                             std::wstring());
+  if (!node) {
+    AppleScript::SetError(AppleScript::errCreateBookmarkFolder);
+    return;
+  }
+
+  [aBookmarkFolder setBookmarkNode:node];
+}
+
+- (void)insertInBookmarkFolders:(id)aBookmarkFolder atIndex:(int)index {
+  // This method gets called when a new bookmark folder is created so
+  // the container and property are set here.
+  [aBookmarkFolder setContainer:self
+                       property:AppleScript::kBookmarkFoldersProperty];
+  int position = [self calculatePositionOfBookmarkFolderAt:index];
+
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  const BookmarkNode* node = model->AddGroup(bookmarkNode_,
+                                             position,
+                                             std::wstring());
+  if (!node) {
+    AppleScript::SetError(AppleScript::errCreateBookmarkFolder);
+    return;
+  }
+
+  [aBookmarkFolder setBookmarkNode:node];
+}
+
+- (void)removeFromBookmarkFoldersAtIndex:(int)index {
+  int position = [self calculatePositionOfBookmarkFolderAt:index];
+
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  model->Remove(bookmarkNode_, position);
+}
+
+- (NSArray*)bookmarkItems {
+  NSMutableArray* bookmarkItems = [NSMutableArray
+      arrayWithCapacity:bookmarkNode_->GetChildCount()];
+
+  for (int i = 0; i < bookmarkNode_->GetChildCount(); ++i) {
+    const BookmarkNode* node = bookmarkNode_->GetChild(i);
+
+    if (!node->is_url())
+      continue;
+    scoped_nsobject<BookmarkFolderAppleScript> bookmarkItem(
+        [[BookmarkItemAppleScript alloc]
+            initWithBookmarkNode:node]);
+    [bookmarkItem setContainer:self
+                      property:AppleScript::kBookmarkItemsProperty];
+    [bookmarkItems addObject:bookmarkItem];
+  }
+
+  return bookmarkItems;
+}
+
+- (void)insertInBookmarkItems:(BookmarkItemAppleScript*)aBookmarkItem {
+  // This method gets called when a new bookmark item is created so
+  // the container and property are set here.
+  [aBookmarkItem setContainer:self
+                     property:AppleScript::kBookmarkItemsProperty];
+
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  GURL url = GURL(base::SysNSStringToUTF8([aBookmarkItem URL]));
+  if (!url.is_valid()) {
+    AppleScript::SetError(AppleScript::errInvalidURL);
+    return;
+  }
+
+  const BookmarkNode* node = model->AddURL(bookmarkNode_,
+                                           bookmarkNode_->GetChildCount(),
+                                           std::wstring(),
+                                           url);
+  if (!node) {
+    AppleScript::SetError(AppleScript::errCreateBookmarkItem);
+    return;
+  }
+
+  [aBookmarkItem setBookmarkNode:node];
+}
+
+- (void)insertInBookmarkItems:(BookmarkItemAppleScript*)aBookmarkItem
+                      atIndex:(int)index {
+  // This method gets called when a new bookmark item is created so
+  // the container and property are set here.
+  [aBookmarkItem setContainer:self
+                     property:AppleScript::kBookmarkItemsProperty];
+  int position = [self calculatePositionOfBookmarkItemAt:index];
+
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  GURL url(base::SysNSStringToUTF8([aBookmarkItem URL]));
+  if (!url.is_valid()) {
+    AppleScript::SetError(AppleScript::errInvalidURL);
+    return;
+  }
+
+  const BookmarkNode* node = model->AddURL(bookmarkNode_,
+                                           position,
+                                           std::wstring(),
+                                           url);
+  if (!node) {
+    AppleScript::SetError(AppleScript::errCreateBookmarkItem);
+    return;
+  }
+
+  [aBookmarkItem setBookmarkNode:node];
+}
+
+- (void)removeFromBookmarkItemsAtIndex:(int)index {
+  int position = [self calculatePositionOfBookmarkItemAt:index];
+
+  BookmarkModel* model = [self bookmarkModel];
+  if (!model)
+    return;
+
+  model->Remove(bookmarkNode_, position);
+}
+
+- (int)calculatePositionOfBookmarkFolderAt:(int)index {
+  // Traverse through all the child nodes till the required node is found and
+  // return its position.
+  // AppleScript is 1-based therefore index is incremented by 1.
+  ++index;
+  int count = -1;
+  while (index) {
+    if (bookmarkNode_->GetChild(++count)->is_folder())
+      --index;
+  }
+  return count;
+}
+
+- (int)calculatePositionOfBookmarkItemAt:(int)index {
+  // Traverse through all the child nodes till the required node is found and
+  // return its position.
+  // AppleScript is 1-based therefore index is incremented by 1.
+  ++index;
+  int count = -1;
+  while (index) {
+    if (bookmarkNode_->GetChild(++count)->is_url())
+      --index;
+  }
+  return count;
+}
+
+@end