src/chrome/browser/ui/cocoa/bookmarks/bookmark_editor_base_controller.mm

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <stack>
   6
   7 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_editor_base_controller.h"
   8
   9 #include "base/auto_reset.h"
  10 #include "base/logging.h"
  11 #include "base/mac/bundle_locations.h"
  12 #include "base/mac/mac_util.h"
  13 #include "base/strings/sys_string_conversions.h"
  14 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
  15 #include "chrome/browser/profiles/profile.h"
  16 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_all_tabs_controller.h"
  17 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_editor_controller.h"
  18 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_name_folder_controller.h"
  19 #import "chrome/browser/ui/cocoa/bookmarks/bookmark_tree_browser_cell.h"
  20 #import "chrome/browser/ui/cocoa/browser_window_controller.h"
  21 #include "components/bookmarks/core/browser/bookmark_model.h"
  22 #include "grit/generated_resources.h"
  23 #include "ui/base/l10n/l10n_util.h"
  24 #include "ui/base/l10n/l10n_util_mac.h"
  25
  26 @interface BookmarkEditorBaseController ()
  27
  28 // Return the folder tree object for the given path.
  29 - (BookmarkFolderInfo*)folderForIndexPath:(NSIndexPath*)path;
  30
  31 // (Re)build the folder tree from the BookmarkModel's current state.
  32 - (void)buildFolderTree;
  33
  34 // Notifies the controller that the bookmark model has changed.
  35 // |selection| specifies if the current selection should be
  36 // maintained (usually YES).
  37 - (void)modelChangedPreserveSelection:(BOOL)preserve;
  38
  39 // Notifies the controller that a node has been removed.
  40 - (void)nodeRemoved:(const BookmarkNode*)node
  41          fromParent:(const BookmarkNode*)parent;
  42
  43 // Given a folder node, collect an array containing BookmarkFolderInfos
  44 // describing its subchildren which are also folders.
  45 - (NSMutableArray*)addChildFoldersFromNode:(const BookmarkNode*)node;
  46
  47 // Scan the folder tree stemming from the given tree folder and create
  48 // any newly added folders.  Pass down info for the folder which was
  49 // selected before we began creating folders.
  50 - (void)createNewFoldersForFolder:(BookmarkFolderInfo*)treeFolder
  51                selectedFolderInfo:(BookmarkFolderInfo*)selectedFolderInfo;
  52
  53 // Scan the folder tree looking for the given bookmark node and return
  54 // the selection path thereto.
  55 - (NSIndexPath*)selectionPathForNode:(const BookmarkNode*)node;
  56
  57 // Implementation of getExpandedNodes. See description in header for details.
  58 - (void)getExpandedNodes:(BookmarkExpandedStateTracker::Nodes*)nodes
  59                   folder:(BookmarkFolderInfo*)info
  60                     path:(std::vector<NSUInteger>*)path
  61                     root:(id)root;
  62 @end
  63
  64 // static; implemented for each platform.  Update this function for new
  65 // classes derived from BookmarkEditorBaseController.
  66 void BookmarkEditor::Show(gfx::NativeWindow parent_window,
  67                           Profile* profile,
  68                           const EditDetails& details,
  69                           Configuration configuration) {
  70   if (details.type == EditDetails::EXISTING_NODE &&
  71       details.existing_node->is_folder()) {
  72     BookmarkNameFolderController* controller =
  73         [[BookmarkNameFolderController alloc]
  74             initWithParentWindow:parent_window
  75                          profile:profile
  76                             node:details.existing_node];
  77     [controller runAsModalSheet];
  78     return;
  79   }
  80
  81   if (details.type == EditDetails::NEW_FOLDER && details.urls.empty()) {
  82     BookmarkNameFolderController* controller =
  83         [[BookmarkNameFolderController alloc]
  84              initWithParentWindow:parent_window
  85                           profile:profile
  86                            parent:details.parent_node
  87                          newIndex:details.index];
  88      [controller runAsModalSheet];
  89      return;
  90   }
  91
  92   BookmarkEditorBaseController* controller = nil;
  93   if (details.type == EditDetails::NEW_FOLDER) {
  94     controller = [[BookmarkAllTabsController alloc]
  95                   initWithParentWindow:parent_window
  96                                profile:profile
  97                                 parent:details.parent_node
  98                                    url:details.url
  99                                  title:details.title
 100                          configuration:configuration];
 101   } else {
 102     controller = [[BookmarkEditorController alloc]
 103                   initWithParentWindow:parent_window
 104                                profile:profile
 105                                 parent:details.parent_node
 106                                   node:details.existing_node
 107                                    url:details.url
 108                                  title:details.title
 109                          configuration:configuration];
 110   }
 111   [controller runAsModalSheet];
 112 }
 113
 114 // Adapter to tell BookmarkEditorBaseController when bookmarks change.
 115 class BookmarkEditorBaseControllerBridge : public BookmarkModelObserver {
 116  public:
 117   BookmarkEditorBaseControllerBridge(BookmarkEditorBaseController* controller)
 118       : controller_(controller),
 119         importing_(false)
 120   { }
 121
 122   virtual void BookmarkModelLoaded(BookmarkModel* model,
 123                                    bool ids_reassigned) OVERRIDE {
 124     [controller_ modelChangedPreserveSelection:YES];
 125   }
 126
 127   virtual void BookmarkNodeMoved(BookmarkModel* model,
 128                                  const BookmarkNode* old_parent,
 129                                  int old_index,
 130                                  const BookmarkNode* new_parent,
 131                                  int new_index) OVERRIDE {
 132     if (!importing_ && new_parent->GetChild(new_index)->is_folder())
 133       [controller_ modelChangedPreserveSelection:YES];
 134   }
 135
 136   virtual void BookmarkNodeAdded(BookmarkModel* model,
 137                                  const BookmarkNode* parent,
 138                                  int index) OVERRIDE {
 139     if (!importing_ && parent->GetChild(index)->is_folder())
 140       [controller_ modelChangedPreserveSelection:YES];
 141   }
 142
 143   virtual void BookmarkNodeRemoved(
 144       BookmarkModel* model,
 145       const BookmarkNode* parent,
 146       int old_index,
 147       const BookmarkNode* node,
 148       const std::set<GURL>& removed_urls) OVERRIDE {
 149     [controller_ nodeRemoved:node fromParent:parent];
 150     if (node->is_folder())
 151       [controller_ modelChangedPreserveSelection:NO];
 152   }
 153
 154   virtual void BookmarkAllNodesRemoved(
 155       BookmarkModel* model,
 156       const std::set<GURL>& removed_urls) OVERRIDE {
 157     [controller_ modelChangedPreserveSelection:NO];
 158   }
 159
 160   virtual void BookmarkNodeChanged(BookmarkModel* model,
 161                                    const BookmarkNode* node) OVERRIDE {
 162     if (!importing_ && node->is_folder())
 163       [controller_ modelChangedPreserveSelection:YES];
 164   }
 165
 166   virtual void BookmarkNodeChildrenReordered(
 167       BookmarkModel* model,
 168       const BookmarkNode* node) OVERRIDE {
 169     if (!importing_)
 170       [controller_ modelChangedPreserveSelection:YES];
 171   }
 172
 173   virtual void BookmarkNodeFaviconChanged(BookmarkModel* model,
 174                                           const BookmarkNode* node) OVERRIDE {
 175     // I care nothing for these 'favicons': I only show folders.
 176   }
 177
 178   virtual void ExtensiveBookmarkChangesBeginning(
 179       BookmarkModel* model) OVERRIDE {
 180     importing_ = true;
 181   }
 182
 183   // Invoked after a batch import finishes.  This tells observers to update
 184   // themselves if they were waiting for the update to finish.
 185   virtual void ExtensiveBookmarkChangesEnded(BookmarkModel* model) OVERRIDE {
 186     importing_ = false;
 187     [controller_ modelChangedPreserveSelection:YES];
 188   }
 189
 190  private:
 191   BookmarkEditorBaseController* controller_;  // weak
 192   bool importing_;
 193 };
 194
 195
 196 #pragma mark -
 197
 198 @implementation BookmarkEditorBaseController
 199
 200 @synthesize initialName = initialName_;
 201 @synthesize displayName = displayName_;
 202
 203 - (id)initWithParentWindow:(NSWindow*)parentWindow
 204                    nibName:(NSString*)nibName
 205                    profile:(Profile*)profile
 206                     parent:(const BookmarkNode*)parent
 207                        url:(const GURL&)url
 208                      title:(const base::string16&)title
 209              configuration:(BookmarkEditor::Configuration)configuration {
 210   NSString* nibpath = [base::mac::FrameworkBundle()
 211                         pathForResource:nibName
 212                                  ofType:@"nib"];
 213   if ((self = [super initWithWindowNibPath:nibpath owner:self])) {
 214     parentWindow_ = parentWindow;
 215     profile_ = profile;
 216     parentNode_ = parent;
 217     url_ = url;
 218     title_ = title;
 219     configuration_ = configuration;
 220     initialName_ = [@"" retain];
 221     observer_.reset(new BookmarkEditorBaseControllerBridge(self));
 222     [self bookmarkModel]->AddObserver(observer_.get());
 223   }
 224   return self;
 225 }
 226
 227 - (void)dealloc {
 228   [self bookmarkModel]->RemoveObserver(observer_.get());
 229   [initialName_ release];
 230   [displayName_ release];
 231   [super dealloc];
 232 }
 233
 234 - (void)awakeFromNib {
 235   [self setDisplayName:[self initialName]];
 236
 237   if (configuration_ != BookmarkEditor::SHOW_TREE) {
 238     // Remember the tree view's height; we will shrink our frame by that much.
 239     NSRect frame = [[self window] frame];
 240     CGFloat browserHeight = [folderTreeView_ frame].size.height;
 241     frame.size.height -= browserHeight;
 242     frame.origin.y += browserHeight;
 243     // Remove the folder tree and "new folder" button.
 244     [folderTreeView_ removeFromSuperview];
 245     [newFolderButton_ removeFromSuperview];
 246     // Finally, commit the size change.
 247     [[self window] setFrame:frame display:YES];
 248   }
 249
 250   // Build up a tree of the current folder configuration.
 251   [self buildFolderTree];
 252 }
 253
 254 - (void)windowDidLoad {
 255   if (configuration_ == BookmarkEditor::SHOW_TREE) {
 256     [self selectNodeInBrowser:parentNode_];
 257   }
 258 }
 259
 260 /* TODO(jrg):
 261 // Implementing this informal protocol allows us to open the sheet
 262 // somewhere other than at the top of the window.  NOTE: this means
 263 // that I, the controller, am also the window's delegate.
 264 - (NSRect)window:(NSWindow*)window willPositionSheet:(NSWindow*)sheet
 265         usingRect:(NSRect)rect {
 266   // adjust rect.origin.y to be the bottom of the toolbar
 267   return rect;
 268 }
 269 */
 270
 271 // TODO(jrg): consider NSModalSession.
 272 - (void)runAsModalSheet {
 273   // Lock down floating bar when in full-screen mode.  Don't animate
 274   // otherwise the pane will be misplaced.
 275   [[BrowserWindowController browserWindowControllerForWindow:parentWindow_]
 276    lockBarVisibilityForOwner:self withAnimation:NO delay:NO];
 277   [NSApp beginSheet:[self window]
 278      modalForWindow:parentWindow_
 279       modalDelegate:self
 280      didEndSelector:@selector(didEndSheet:returnCode:contextInfo:)
 281         contextInfo:nil];
 282 }
 283
 284 // This constant has to match the name of the method after it.
 285 NSString* const kOkEnabledName = @"okEnabled";
 286 - (BOOL)okEnabled {
 287   return YES;
 288 }
 289
 290 - (IBAction)ok:(id)sender {
 291   NSWindow* window = [self window];
 292   [window makeFirstResponder:window];
 293   // At least one of these two functions should be provided by derived classes.
 294   BOOL hasWillCommit = [self respondsToSelector:@selector(willCommit)];
 295   BOOL hasDidCommit = [self respondsToSelector:@selector(didCommit)];
 296   DCHECK(hasWillCommit || hasDidCommit);
 297   BOOL shouldContinue = YES;
 298   if (hasWillCommit) {
 299     NSNumber* hasWillContinue = [self performSelector:@selector(willCommit)];
 300     if (hasWillContinue && [hasWillContinue isKindOfClass:[NSNumber class]])
 301       shouldContinue = [hasWillContinue boolValue];
 302   }
 303   if (shouldContinue)
 304     [self createNewFolders];
 305   if (hasDidCommit) {
 306     NSNumber* hasDidContinue = [self performSelector:@selector(didCommit)];
 307     if (hasDidContinue && [hasDidContinue isKindOfClass:[NSNumber class]])
 308       shouldContinue = [hasDidContinue boolValue];
 309   }
 310   if (shouldContinue)
 311     [NSApp endSheet:window];
 312 }
 313
 314 - (IBAction)cancel:(id)sender {
 315   [NSApp endSheet:[self window]];
 316 }
 317
 318 - (void)didEndSheet:(NSWindow*)sheet
 319          returnCode:(int)returnCode
 320         contextInfo:(void*)contextInfo {
 321   [sheet close];
 322   [[BrowserWindowController browserWindowControllerForWindow:parentWindow_]
 323    releaseBarVisibilityForOwner:self withAnimation:YES delay:NO];
 324 }
 325
 326 - (void)windowWillClose:(NSNotification*)notification {
 327   [self autorelease];
 328 }
 329
 330 #pragma mark Folder Tree Management
 331
 332 - (BookmarkModel*)bookmarkModel {
 333   return BookmarkModelFactory::GetForProfile(profile_);
 334 }
 335
 336 - (Profile*)profile {
 337   return profile_;
 338 }
 339
 340 - (const BookmarkNode*)parentNode {
 341   return parentNode_;
 342 }
 343
 344 - (const GURL&)url {
 345   return url_;
 346 }
 347
 348 - (const base::string16&)title{
 349   return title_;
 350 }
 351
 352 - (BookmarkFolderInfo*)folderForIndexPath:(NSIndexPath*)indexPath {
 353   NSUInteger pathCount = [indexPath length];
 354   BookmarkFolderInfo* item = nil;
 355   NSArray* treeNode = [self folderTreeArray];
 356   for (NSUInteger i = 0; i < pathCount; ++i) {
 357     item = [treeNode objectAtIndex:[indexPath indexAtPosition:i]];
 358     treeNode = [item children];
 359   }
 360   return item;
 361 }
 362
 363 - (NSIndexPath*)selectedIndexPath {
 364   NSIndexPath* selectedIndexPath = nil;
 365   NSArray* selections = [self tableSelectionPaths];
 366   if ([selections count]) {
 367     DCHECK([selections count] == 1);  // Should be exactly one selection.
 368     selectedIndexPath = [selections objectAtIndex:0];
 369   }
 370   return selectedIndexPath;
 371 }
 372
 373 - (BookmarkFolderInfo*)selectedFolder {
 374   BookmarkFolderInfo* item = nil;
 375   NSIndexPath* selectedIndexPath = [self selectedIndexPath];
 376   if (selectedIndexPath) {
 377     item = [self folderForIndexPath:selectedIndexPath];
 378   }
 379   return item;
 380 }
 381
 382 - (const BookmarkNode*)selectedNode {
 383   const BookmarkNode* selectedNode = NULL;
 384   // Determine a new parent node only if the browser is showing.
 385   if (configuration_ == BookmarkEditor::SHOW_TREE) {
 386     BookmarkFolderInfo* folderInfo = [self selectedFolder];
 387     if (folderInfo)
 388       selectedNode = [folderInfo folderNode];
 389   } else {
 390     // If the tree is not showing then we use the original parent.
 391     selectedNode = parentNode_;
 392   }
 393   return selectedNode;
 394 }
 395
 396 - (void)expandNodes:(const BookmarkExpandedStateTracker::Nodes&)nodes {
 397   id treeControllerRoot = [folderTreeController_ arrangedObjects];
 398   for (BookmarkExpandedStateTracker::Nodes::const_iterator i = nodes.begin();
 399        i != nodes.end(); ++i) {
 400     NSIndexPath* path = [self selectionPathForNode:*i];
 401     id folderNode = [treeControllerRoot descendantNodeAtIndexPath:path];
 402     [folderTreeView_ expandItem:folderNode];
 403   }
 404 }
 405
 406 - (BookmarkExpandedStateTracker::Nodes)getExpandedNodes {
 407   BookmarkExpandedStateTracker::Nodes nodes;
 408   std::vector<NSUInteger> path;
 409   NSArray* folderNodes = [self folderTreeArray];
 410   NSUInteger i = 0;
 411   for (BookmarkFolderInfo* folder in folderNodes) {
 412     path.push_back(i);
 413     [self getExpandedNodes:&nodes
 414                     folder:folder
 415                       path:&path
 416                       root:[folderTreeController_ arrangedObjects]];
 417     path.clear();
 418     ++i;
 419   }
 420   return nodes;
 421 }
 422
 423 - (void)getExpandedNodes:(BookmarkExpandedStateTracker::Nodes*)nodes
 424                   folder:(BookmarkFolderInfo*)folder
 425                     path:(std::vector<NSUInteger>*)path
 426                     root:(id)root {
 427   NSIndexPath* indexPath = [NSIndexPath indexPathWithIndexes:&(path->front())
 428                                                       length:path->size()];
 429   id node = [root descendantNodeAtIndexPath:indexPath];
 430   if (![folderTreeView_ isItemExpanded:node])
 431     return;
 432   nodes->insert([folder folderNode]);
 433   NSArray* children = [folder children];
 434   NSUInteger i = 0;
 435   for (BookmarkFolderInfo* childFolder in children) {
 436     path->push_back(i);
 437     [self getExpandedNodes:nodes folder:childFolder path:path root:root];
 438     path->pop_back();
 439     ++i;
 440   }
 441 }
 442
 443 - (NSArray*)folderTreeArray {
 444   return folderTreeArray_.get();
 445 }
 446
 447 - (NSArray*)tableSelectionPaths {
 448   return tableSelectionPaths_.get();
 449 }
 450
 451 - (void)setTableSelectionPath:(NSIndexPath*)tableSelectionPath {
 452   [self setTableSelectionPaths:[NSArray arrayWithObject:tableSelectionPath]];
 453 }
 454
 455 - (void)setTableSelectionPaths:(NSArray*)tableSelectionPaths {
 456   tableSelectionPaths_.reset([tableSelectionPaths retain]);
 457 }
 458
 459 - (void)selectNodeInBrowser:(const BookmarkNode*)node {
 460   DCHECK(configuration_ == BookmarkEditor::SHOW_TREE);
 461   NSIndexPath* selectionPath = [self selectionPathForNode:node];
 462   [self willChangeValueForKey:kOkEnabledName];
 463   [self setTableSelectionPath:selectionPath];
 464   [self didChangeValueForKey:kOkEnabledName];
 465 }
 466
 467 - (NSIndexPath*)selectionPathForNode:(const BookmarkNode*)desiredNode {
 468   // Back up the parent chaing for desiredNode, building up a stack
 469   // of ancestor nodes.  Then crawl down the folderTreeArray looking
 470   // for each ancestor in order while building up the selectionPath.
 471   std::stack<const BookmarkNode*> nodeStack;
 472   BookmarkModel* model = BookmarkModelFactory::GetForProfile(profile_);
 473   const BookmarkNode* rootNode = model->root_node();
 474   const BookmarkNode* node = desiredNode;
 475   while (node != rootNode) {
 476     DCHECK(node);
 477     nodeStack.push(node);
 478     node = node->parent();
 479   }
 480   NSUInteger stackSize = nodeStack.size();
 481
 482   NSIndexPath* path = nil;
 483   NSArray* folders = [self folderTreeArray];
 484   while (!nodeStack.empty()) {
 485     node = nodeStack.top();
 486     nodeStack.pop();
 487     // Find node in the current folders array.
 488     NSUInteger i = 0;
 489     for (BookmarkFolderInfo *folderInfo in folders) {
 490       const BookmarkNode* testNode = [folderInfo folderNode];
 491       if (testNode == node) {
 492         path = path ? [path indexPathByAddingIndex:i] :
 493         [NSIndexPath indexPathWithIndex:i];
 494         folders = [folderInfo children];
 495         break;
 496       }
 497       ++i;
 498     }
 499   }
 500   DCHECK([path length] == stackSize);
 501   return path;
 502 }
 503
 504 - (NSMutableArray*)addChildFoldersFromNode:(const BookmarkNode*)node {
 505   NSMutableArray* childFolders = nil;
 506   int childCount = node->child_count();
 507   for (int i = 0; i < childCount; ++i) {
 508     const BookmarkNode* childNode = node->GetChild(i);
 509     if (childNode->is_folder() && childNode->IsVisible()) {
 510       NSString* childName = base::SysUTF16ToNSString(childNode->GetTitle());
 511       NSMutableArray* children = [self addChildFoldersFromNode:childNode];
 512       BookmarkFolderInfo* folderInfo =
 513           [BookmarkFolderInfo bookmarkFolderInfoWithFolderName:childName
 514                                                     folderNode:childNode
 515                                                       children:children];
 516       if (!childFolders)
 517         childFolders = [NSMutableArray arrayWithObject:folderInfo];
 518       else
 519         [childFolders addObject:folderInfo];
 520     }
 521   }
 522   return childFolders;
 523 }
 524
 525 - (void)buildFolderTree {
 526   // Build up a tree of the current folder configuration.
 527   BookmarkModel* model = BookmarkModelFactory::GetForProfile(profile_);
 528   const BookmarkNode* rootNode = model->root_node();
 529   NSMutableArray* baseArray = [self addChildFoldersFromNode:rootNode];
 530   DCHECK(baseArray);
 531   [self willChangeValueForKey:@"folderTreeArray"];
 532   folderTreeArray_.reset([baseArray retain]);
 533   [self didChangeValueForKey:@"folderTreeArray"];
 534 }
 535
 536 - (void)modelChangedPreserveSelection:(BOOL)preserve {
 537   if (creatingNewFolders_)
 538     return;
 539   const BookmarkNode* selectedNode = [self selectedNode];
 540   [self buildFolderTree];
 541   if (preserve &&
 542       selectedNode &&
 543       configuration_ == BookmarkEditor::SHOW_TREE)
 544     [self selectNodeInBrowser:selectedNode];
 545 }
 546
 547 - (void)nodeRemoved:(const BookmarkNode*)node
 548          fromParent:(const BookmarkNode*)parent {
 549   if (node->is_folder()) {
 550     if (parentNode_ == node || parentNode_->HasAncestor(node)) {
 551       parentNode_ = [self bookmarkModel]->bookmark_bar_node();
 552       if (configuration_ != BookmarkEditor::SHOW_TREE) {
 553         // The user can't select a different folder, so just close up shop.
 554         [self cancel:self];
 555         return;
 556       }
 557     }
 558
 559     if (configuration_ == BookmarkEditor::SHOW_TREE) {
 560       // For safety's sake, in case deleted node was an ancestor of selection,
 561       // go back to a known safe place.
 562       [self selectNodeInBrowser:parentNode_];
 563     }
 564   }
 565 }
 566
 567 #pragma mark New Folder Handler
 568
 569 - (void)createNewFoldersForFolder:(BookmarkFolderInfo*)folderInfo
 570                selectedFolderInfo:(BookmarkFolderInfo*)selectedFolderInfo {
 571   NSArray* subfolders = [folderInfo children];
 572   const BookmarkNode* parentNode = [folderInfo folderNode];
 573   DCHECK(parentNode);
 574   NSUInteger i = 0;
 575   for (BookmarkFolderInfo* subFolderInfo in subfolders) {
 576     if ([subFolderInfo newFolder]) {
 577       BookmarkModel* model = [self bookmarkModel];
 578       const BookmarkNode* newFolder =
 579           model->AddFolder(parentNode, i,
 580               base::SysNSStringToUTF16([subFolderInfo folderName]));
 581       // Update our dictionary with the actual folder node just created.
 582       [subFolderInfo setFolderNode:newFolder];
 583       [subFolderInfo setNewFolder:NO];
 584     }
 585     [self createNewFoldersForFolder:subFolderInfo
 586                  selectedFolderInfo:selectedFolderInfo];
 587     ++i;
 588   }
 589 }
 590
 591 - (IBAction)newFolder:(id)sender {
 592   // Create a new folder off of the selected folder node.
 593   BookmarkFolderInfo* parentInfo = [self selectedFolder];
 594   if (parentInfo) {
 595     NSIndexPath* selection = [self selectedIndexPath];
 596     NSString* newFolderName =
 597         l10n_util::GetNSStringWithFixup(IDS_BOOKMARK_EDITOR_NEW_FOLDER_NAME);
 598     BookmarkFolderInfo* folderInfo =
 599         [BookmarkFolderInfo bookmarkFolderInfoWithFolderName:newFolderName];
 600     [self willChangeValueForKey:@"folderTreeArray"];
 601     NSMutableArray* children = [parentInfo children];
 602     if (children) {
 603       [children addObject:folderInfo];
 604     } else {
 605       children = [NSMutableArray arrayWithObject:folderInfo];
 606       [parentInfo setChildren:children];
 607     }
 608     [self didChangeValueForKey:@"folderTreeArray"];
 609
 610     // Expose the parent folder children.
 611     [folderTreeView_ expandItem:parentInfo];
 612
 613     // Select the new folder node and put the folder name into edit mode.
 614     selection = [selection indexPathByAddingIndex:[children count] - 1];
 615     [self setTableSelectionPath:selection];
 616     NSInteger row = [folderTreeView_ selectedRow];
 617     DCHECK(row >= 0);
 618
 619     // Put the cell into single-line mode before putting it into edit mode.
 620     // TODO(kushi.p): Remove this when the project hits a 10.6+ only state.
 621     NSCell* folderCell = [folderTreeView_ preparedCellAtColumn:0 row:row];
 622     if ([folderCell
 623           respondsToSelector:@selector(setUsesSingleLineMode:)]) {
 624       [folderCell setUsesSingleLineMode:YES];
 625     }
 626
 627     [folderTreeView_ editColumn:0 row:row withEvent:nil select:YES];
 628   }
 629 }
 630
 631 - (void)createNewFolders {
 632   base::AutoReset<BOOL> creatingNewFoldersSetter(&creatingNewFolders_, YES);
 633   // Scan the tree looking for nodes marked 'newFolder' and create those nodes.
 634   NSArray* folderTreeArray = [self folderTreeArray];
 635   for (BookmarkFolderInfo *folderInfo in folderTreeArray) {
 636     [self createNewFoldersForFolder:folderInfo
 637                  selectedFolderInfo:[self selectedFolder]];
 638   }
 639 }
 640
 641 #pragma mark For Unit Test Use Only
 642
 643 - (BOOL)okButtonEnabled {
 644   return [okButton_ isEnabled];
 645 }
 646
 647 - (void)selectTestNodeInBrowser:(const BookmarkNode*)node {
 648   [self selectNodeInBrowser:node];
 649 }
 650
 651 @end  // BookmarkEditorBaseController
 652
 653 @implementation BookmarkFolderInfo
 654
 655 @synthesize folderName = folderName_;
 656 @synthesize folderNode = folderNode_;
 657 @synthesize children = children_;
 658 @synthesize newFolder = newFolder_;
 659
 660 + (id)bookmarkFolderInfoWithFolderName:(NSString*)folderName
 661                             folderNode:(const BookmarkNode*)folderNode
 662                               children:(NSMutableArray*)children {
 663   return [[[BookmarkFolderInfo alloc] initWithFolderName:folderName
 664                                               folderNode:folderNode
 665                                                 children:children
 666                                                newFolder:NO]
 667           autorelease];
 668 }
 669
 670 + (id)bookmarkFolderInfoWithFolderName:(NSString*)folderName {
 671   return [[[BookmarkFolderInfo alloc] initWithFolderName:folderName
 672                                               folderNode:NULL
 673                                                 children:nil
 674                                                newFolder:YES]
 675           autorelease];
 676 }
 677
 678 - (id)initWithFolderName:(NSString*)folderName
 679               folderNode:(const BookmarkNode*)folderNode
 680                 children:(NSMutableArray*)children
 681                newFolder:(BOOL)newFolder {
 682   if ((self = [super init])) {
 683     // A folderName is always required, and if newFolder is NO then there
 684     // should be a folderNode.  Children is optional.
 685     DCHECK(folderName && (newFolder || folderNode));
 686     if (folderName && (newFolder || folderNode)) {
 687       folderName_ = [folderName copy];
 688       folderNode_ = folderNode;
 689       children_ = [children retain];
 690       newFolder_ = newFolder;
 691     } else {
 692       NOTREACHED();  // Invalid init.
 693       [self release];
 694       self = nil;
 695     }
 696   }
 697   return self;
 698 }
 699
 700 - (id)init {
 701   NOTREACHED();  // Should never be called.
 702   return [self initWithFolderName:nil folderNode:nil children:nil newFolder:NO];
 703 }
 704
 705 - (void)dealloc {
 706   [folderName_ release];
 707   [children_ release];
 708   [super dealloc];
 709 }
 710
 711 // Implementing isEqual: allows the NSTreeController to preserve the selection
 712 // and open/shut state of outline items when the data changes.
 713 - (BOOL)isEqual:(id)other {
 714   return [other isKindOfClass:[BookmarkFolderInfo class]] &&
 715       folderNode_ == [(BookmarkFolderInfo*)other folderNode];
 716 }
 717
 718 @end