Diff for /chimera/BookmarksService.mm between versions 1.1 and 1.29

version 1.1, 2002/02/09 00:00:47 version 1.29, 2002/04/20 00:02:55
Line 1 Line 1
/*/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 *  BookmarksService.cpp/* ***** BEGIN LICENSE BLOCK *****
 *  Chimera * Version: NPL 1.1/GPL 2.0/LGPL 2.1
  *   *
 *  Created by David Hyatt on Thu Feb 07 2002. * The contents of this file are subject to the Netscape Public License
 *  Copyright (c) 2001 __MyCompanyName__. All rights reserved. * Version 1.1 (the "License"); you may not use this file except in
  * compliance with the License. You may obtain a copy of the License at
  * http://www.mozilla.org/NPL/
  *   *
 */ * Software distributed under the License is distributed on an "AS IS" basis,
  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  * for the specific language governing rights and limitations under the
  * License.
  *
  * The Original Code is mozilla.org code.
  *
  * The Initial Developer of the Original Code is 
  * Netscape Communications Corporation.
  * Portions created by the Initial Developer are Copyright (C) 2002
  * the Initial Developer. All Rights Reserved.
  *
  * Contributor(s):
  *
  * Alternatively, the contents of this file may be used under the terms of
  * either the GNU General Public License Version 2 or later (the "GPL"), or 
  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  * in which case the provisions of the GPL or the LGPL are applicable instead
  * of those above. If you wish to allow use of your version of this file only
  * under the terms of either the GPL or the LGPL, and not to allow others to
  * use your version of this file under the terms of the NPL, indicate your
  * decision by deleting the provisions above and replace them with the notice
  * and other provisions required by the GPL or the LGPL. If you do not delete
  * the provisions above, a recipient may use your version of this file under
  * the terms of any one of the NPL, the GPL or the LGPL.
  *
  * ***** END LICENSE BLOCK ***** */
   
 #import "NSBrowserView.h"  #import "NSBrowserView.h"
 #include "BookmarksService.h"  #include "BookmarksService.h"
Line 13 Line 41
 #include "nsIContent.h"  #include "nsIContent.h"
 #include "nsIAtom.h"  #include "nsIAtom.h"
 #include "nsITextContent.h"  #include "nsITextContent.h"
#include "nsIDOMDocument.h"#include "nsIDOMWindow.h"
 #include "nsIDOMHTMLDocument.h"
 #include "nsIDOMElement.h"  #include "nsIDOMElement.h"
 #include "nsString.h"  #include "nsString.h"
 #include "nsIFile.h"  #include "nsIFile.h"
 #include "nsAppDirectoryServiceDefs.h"  #include "nsAppDirectoryServiceDefs.h"
 #include "nsIXMLHttpRequest.h"  #include "nsIXMLHttpRequest.h"
   #include "nsIDOMSerializer.h"
 #include "nsNetUtil.h"  #include "nsNetUtil.h"
 #include "nsINamespaceManager.h"  #include "nsINamespaceManager.h"
 #include "nsIXBLService.h"  #include "nsIXBLService.h"
   #include "nsIWebBrowser.h"
   
 @implementation BookmarksDataSource  @implementation BookmarksDataSource
   
Line 29 Line 60
 {  {
     [super init];      [super init];
     mBookmarks = nsnull;      mBookmarks = nsnull;
       mCachedHref = nil;
     return self;      return self;
 }  }
   
 -(void) dealloc  -(void) dealloc
 {  {
    [super dealloc];  [super dealloc];
    if (mBookmarks)}
        BookmarksService::RemoveObserver(nsnull);
 -(void) windowClosing
 {
   if (mBookmarks) {
     mBookmarks->RemoveObserver();
     delete mBookmarks;
   }
 }  }
   
 -(void) ensureBookmarks  -(void) ensureBookmarks
Line 44 Line 82
     if (mBookmarks)      if (mBookmarks)
         return;          return;
           
    mBookmarks = BookmarksService::AddObserver(nsnull);    mBookmarks = new BookmarksService(self);
     mBookmarks->AddObserver();
           
     [mOutlineView setTarget: self];      [mOutlineView setTarget: self];
     [mOutlineView setDoubleAction: @selector(openBookmark:)];      [mOutlineView setDoubleAction: @selector(openBookmark:)];
       [mOutlineView setDeleteAction: @selector(deleteBookmarks:)];
     [mOutlineView reloadData];      [mOutlineView reloadData];
 }  }
   
-(IBAction)openBookmark: (id)aSender-(IBAction)addBookmark:(id)aSender
 {  {
    printf("Opening bookmark!\n");  [self addBookmark: aSender useSelection: YES isFolder: NO];
    }
 
 -(IBAction)addFolder:(id)aSender
 {
   [self addBookmark: aSender useSelection: YES isFolder: YES];
 }
 
 -(void)addBookmark:(id)aSender useSelection:(BOOL)aUseSel isFolder:(BOOL)aIsFolder
 {
   if (!mBookmarks)
     return;
 
   // We use the selected item to determine the parent only if aUseSel is YES.
   BookmarkItem* item = nil;
   if (aUseSel && ([mOutlineView numberOfSelectedRows] == 1)) {
     // There is only one selected row.  If it is a folder, use it as our parent.
     // Otherwise, use our parent,
     int index = [mOutlineView selectedRow];      int index = [mOutlineView selectedRow];
    if (index == -1)    item = [mOutlineView itemAtRow: index];
        return;    if (![mOutlineView isExpandable: item]) {
          // We can't be used as the parent.  Try our parent.
    id item = [mOutlineView itemAtRow: index];      nsIContent* content = [item contentNode];
    if (!item)      nsCOMPtr<nsIContent> parentContent;
        return;      content->GetParent(*getter_AddRefs(parentContent));
              nsCOMPtr<nsIContent> root;
    if ([mOutlineView isExpandable: item]) {      mBookmarks->GetRootContent(getter_AddRefs(root));
        if ([mOutlineView isItemExpanded: item])      
            [mOutlineView collapseItem: item];      // The root has no item, so we don't need to do a lookup unless we
        else      // aren't the root.
            [mOutlineView expandItem: item];      if (parentContent != root) {
         PRUint32 contentID;
         parentContent->GetContentID(&contentID);
         item = [(BookmarksService::gDictionary) objectForKey: [NSNumber numberWithInt: contentID]];
       }
     }      }
    else {  }
        nsIContent* content = [item contentNode];
        nsAutoString href;  nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(mBookmarks->gBookmarks));
        content->GetAttr(kNameSpaceID_None, BookmarksService::gHrefAtom, href);  
        if (!href.IsEmpty()) {  // Fetch the title of the current page and the URL.
            nsCAutoString cstr; cstr.AssignWithConversion(href);  nsAutoString title, href;
            NSString* url = [NSString stringWithCString: cstr.get()];  if (!aIsFolder) {
            [[mBrowserView getBrowserView] loadURI:[NSURL URLWithString: url] flags:NSLoadFlagsNone];    BookmarksService::GetTitleAndHrefForBrowserView([[mBrowserWindowController getMyBrowserView] getBrowserView],
        }                                                    title, href);
    } 
     mCachedHref = [NSString stringWithCharacters: href.get() length: nsCRT::strlen(href.get())];
     [mCachedHref retain];
   }
   else {
     mCachedHref = nil;
     title = NS_LITERAL_STRING("New Folder");
   }
   
   NSTextField* textField = [mBrowserWindowController getAddBookmarkTitle];
   [textField setStringValue: [NSString stringWithCharacters: title.get() length: nsCRT::strlen(title.get())]];
 
   [mBrowserWindowController cacheBookmarkDS: self];
 
   // Show/hide the bookmark all tabs checkbox as appropriate.
   NSTabView* tabView = [mBrowserWindowController getTabBrowser];
   id checkbox = [mBrowserWindowController getAddBookmarkCheckbox];
   BOOL hasSuperview = [checkbox superview] != nil;
   if (aIsFolder && hasSuperview) {
     // Just don't show it at all.
     [checkbox removeFromSuperview];
     [checkbox retain];
   }
   else if (!aIsFolder && !hasSuperview) {
     // Put it back in.
     [[[mBrowserWindowController getAddBookmarkSheetWindow] contentView] addSubview: checkbox];
     [checkbox autorelease];
   }
 
   // Enable the bookmark all tabs checkbox if appropriate.
   if (!aIsFolder)
     [[mBrowserWindowController getAddBookmarkCheckbox] setEnabled: ([tabView numberOfTabViewItems] > 1)];
   
   // Build up the folder list.
   NSPopUpButton* popup = [mBrowserWindowController getAddBookmarkFolder];
   BookmarksService::ConstructAddBookmarkFolderList(popup, item);
   
   [NSApp beginSheet:    [mBrowserWindowController getAddBookmarkSheetWindow]
      modalForWindow:    [mBrowserWindowController window]
       modalDelegate:    nil //self
      didEndSelector:    nil //@selector(sheetDidEnd:)
         contextInfo:    nil];
 }
 
 -(void)endAddBookmark: (int)aCode
 {
   if (aCode == 0)
     return;
 
   BOOL isGroup = NO;
   id checkbox = [mBrowserWindowController getAddBookmarkCheckbox];
   if (([checkbox superview] != nil) && [checkbox isEnabled] && ([checkbox state] == NSOnState)) {
     mCachedHref = nil;
     isGroup = YES;
   }
   
   const char* titleC = [[[mBrowserWindowController getAddBookmarkTitle] stringValue] cString];
   nsAutoString title; title.AssignWithConversion(titleC);
 
   nsAutoString tagName;
   if (mCachedHref)
     tagName = NS_LITERAL_STRING("bookmark");
   else
     tagName = NS_LITERAL_STRING("folder");
   
   nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(mBookmarks->gBookmarks));
   nsCOMPtr<nsIDOMElement> elt;
   domDoc->CreateElementNS(NS_LITERAL_STRING("http://chimera.mozdev.org/bookmarks/"),
                           tagName,
                           getter_AddRefs(elt));
 
   elt->SetAttribute(NS_LITERAL_STRING("name"), title);
 
   if (mCachedHref) {
     nsAutoString href; href.AssignWithConversion([mCachedHref cString]);
     [mCachedHref release];
     elt->SetAttribute(NS_LITERAL_STRING("href"), href);
   }
 
   if (isGroup) {
     // We have to iterate over each tab and create content nodes using the
     // title/href of all the pages.  They are inserted underneath the parent.
     elt->SetAttribute(NS_LITERAL_STRING("group"), NS_LITERAL_STRING("true"));
     id tabBrowser = [mBrowserWindowController getTabBrowser];
     int count = [tabBrowser numberOfTabViewItems];
     for (int i = 0; i < count; i++) {
       id browserView = [[[tabBrowser tabViewItemAtIndex: i] view] getBrowserView];
       nsAutoString title, href;
       BookmarksService::GetTitleAndHrefForBrowserView(browserView, title, href);
       nsCOMPtr<nsIDOMElement> childElt;
       domDoc->CreateElementNS(NS_LITERAL_STRING("http://chimera.mozdev.org/bookmarks/"),
                               NS_LITERAL_STRING("bookmark"),
                               getter_AddRefs(childElt));
       childElt->SetAttribute(NS_LITERAL_STRING("name"), title);
       childElt->SetAttribute(NS_LITERAL_STRING("href"), href);
       nsCOMPtr<nsIDOMNode> dummy;
       elt->AppendChild(childElt, getter_AddRefs(dummy));
     }
   }
   
   // Figure out the parent element.
   nsCOMPtr<nsIDOMElement> parentElt;
   nsCOMPtr<nsIContent> parentContent;
   NSPopUpButton* popup = [mBrowserWindowController getAddBookmarkFolder];
   NSMenuItem* selectedItem = [popup selectedItem];
   int tag = [selectedItem tag];
   if (tag == -1) {
     mBookmarks->GetRootContent(getter_AddRefs(parentContent));
     parentElt = do_QueryInterface(parentContent);
   }
   else {
     BookmarkItem* item = [(BookmarksService::gDictionary) objectForKey: [NSNumber numberWithInt: tag]];
     // Get the content node.
     parentContent = [item contentNode];
     parentElt = do_QueryInterface(parentContent);
   }
   
   nsCOMPtr<nsIDOMNode> dummy;
   parentElt->AppendChild(elt, getter_AddRefs(dummy));
 
   nsCOMPtr<nsIContent> childContent(do_QueryInterface(elt));
   mBookmarks->BookmarkAdded(parentContent, childContent);
 }
 
 -(IBAction)deleteBookmarks: (id)aSender
 {
   if (!mBookmarks)
     return;
 
   int index = [mOutlineView selectedRow];
   if (index == -1)
     return;
   if ([mOutlineView numberOfSelectedRows] == 1) {
     BookmarkItem* item = [mOutlineView itemAtRow: index];
     [self deleteBookmark: item];
     int total = [mOutlineView numberOfRows];
     if (index == total)
       index--;
     [mOutlineView selectRow: index byExtendingSelection: NO];
   }
   else {
     NSMutableArray* itemsToDelete = [[[NSMutableArray alloc] init] autorelease];
     NSEnumerator* selRows = [mOutlineView selectedRowEnumerator];
     for (NSNumber* currIndex = [selRows nextObject];
          currIndex != nil;
          currIndex = [selRows nextObject]) {
       index = [currIndex intValue];
       BookmarkItem* item = [mOutlineView itemAtRow: index];
       [itemsToDelete addObject: item];
     }
 
     int count = [itemsToDelete count];
     for (int i = 0; i < count; i++) {
       BookmarkItem* item = [itemsToDelete objectAtIndex: i];
       [self deleteBookmark: item];      
     }
   }
 }       
 
 -(void)deleteBookmark:(id)aItem
 {
   nsCOMPtr<nsIContent> content = [aItem contentNode];
   nsCOMPtr<nsIDOMElement> child(do_QueryInterface(content));
   if (child == BookmarksService::gToolbarRoot)
     return; // Don't allow the personal toolbar to be deleted.
   
   nsCOMPtr<nsIDOMNode> parent;
   child->GetParentNode(getter_AddRefs(parent));
   nsCOMPtr<nsIContent> parentContent(do_QueryInterface(parent));
   nsCOMPtr<nsIDOMNode> dummy;
   parent->RemoveChild(child, getter_AddRefs(dummy));
   mBookmarks->BookmarkRemoved(parentContent, content);
 }
 
 -(IBAction)openBookmark: (id)aSender
 {
   int index = [mOutlineView selectedRow];
   if (index == -1)
     return;
 
   id item = [mOutlineView itemAtRow: index];
   if (!item)
     return;
 
   nsIContent* content = [item contentNode];
   nsCOMPtr<nsIDOMElement> elt(do_QueryInterface(content));
   nsAutoString group;
   elt->GetAttribute(NS_LITERAL_STRING("group"), group);
   if (!group.IsEmpty())
     mBookmarks->OpenBookmarkGroup([mBrowserWindowController getTabBrowser], elt);
   else if ([mOutlineView isExpandable: item]) {
     if ([mOutlineView isItemExpanded: item])
       [mOutlineView collapseItem: item];
     else
       [mOutlineView expandItem: item];
   }
   else {
     nsAutoString href;
     content->GetAttr(kNameSpaceID_None, BookmarksService::gHrefAtom, href);
     if (!href.IsEmpty()) {
       nsCAutoString cstr; cstr.AssignWithConversion(href);
       NSString* url = [NSString stringWithCString: cstr.get()];
       [[[mBrowserWindowController getMyBrowserView] getBrowserView] loadURI:[NSURL URLWithString: url] flags:                      NSLoadFlagsNone];
       // Focus and activate our content area.
       [[[mBrowserWindowController getMyBrowserView] getBrowserView] setActive: YES];
     }
   }
 }
 
 - (BOOL)outlineView:(NSOutlineView *)outlineView shouldEditTableColumn:(NSTableColumn *)tableColumn item:(id)item
 {
     return NO;
 }  }
   
 - (id)outlineView:(NSOutlineView *)outlineView child:(int)index ofItem:(id)item  - (id)outlineView:(NSOutlineView *)outlineView child:(int)index ofItem:(id)item
Line 86 Line 357
     if (!mBookmarks)      if (!mBookmarks)
         return nil;          return nil;
                 
     printf("Begin item obtain\n");  
     nsCOMPtr<nsIContent> content;      nsCOMPtr<nsIContent> content;
     if (!item)      if (!item)
         mBookmarks->GetRootContent(getter_AddRefs(content));          mBookmarks->GetRootContent(getter_AddRefs(content));
Line 105 Line 375
           
     if (!item)      if (!item)
         return YES; // The root node is always open.          return YES; // The root node is always open.
           
     printf("Begin expandable check.\n");  
           
     nsCOMPtr<nsIAtom> tagName;      nsCOMPtr<nsIAtom> tagName;
     nsIContent* content = [item contentNode];      nsIContent* content = [item contentNode];
Line 119 Line 387
 {  {
     if (!mBookmarks)      if (!mBookmarks)
         return 0;          return 0;
   
    printf("Begin child count.\n"); 
     
     nsCOMPtr<nsIContent> content;      nsCOMPtr<nsIContent> content;
     if (!item)      if (!item)
         mBookmarks->GetRootContent(getter_AddRefs(content));          mBookmarks->GetRootContent(getter_AddRefs(content));
     else       else 
         content = [item contentNode];          content = [item contentNode];
           
     if (!content)  
         printf("OH NO!!!!!\n");  
           
     PRInt32 childCount;      PRInt32 childCount;
     content->ChildCount(childCount);      content->ChildCount(childCount);
           
     printf("Child count is %d\n", childCount);  
       
     return childCount;      return childCount;
 }  }
   
 - (id)outlineView:(NSOutlineView *)outlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item  - (id)outlineView:(NSOutlineView *)outlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
 {  {
    if (!item)    NSString                                    *columnName = [tableColumn identifier];
        return nil;    NSMutableAttributedString       *cellValue = [[NSMutableAttributedString alloc] init];
        NSFileWrapper                               *fileWrapper = [[NSFileWrapper alloc] initRegularFileWithContents:nil];
    printf("Obtaining string.\n");    NSTextAttachment                    *textAttachment = [[NSTextAttachment alloc] initWithFileWrapper:fileWrapper];
        NSMutableAttributedString   *attachmentAttrString = nil;
    NSString* columnName = [tableColumn identifier];    NSCell                                              *attachmentAttrStringCell;
    if (columnName)
      printf("It has a name!\n"); 
       
     if ([columnName isEqualToString: @"name"]) {      if ([columnName isEqualToString: @"name"]) {
         nsIContent* content = [item contentNode];          nsIContent* content = [item contentNode];
         nsAutoString nameAttr;          nsAutoString nameAttr;
         content->GetAttr(kNameSpaceID_None, BookmarksService::gNameAtom, nameAttr);          content->GetAttr(kNameSpaceID_None, BookmarksService::gNameAtom, nameAttr);
         nsCAutoString cStr; cStr.AssignWithConversion(nameAttr);          nsCAutoString cStr; cStr.AssignWithConversion(nameAttr);
        printf("Well? %s\n", cStr.get());        
        return [NSString stringWithCString: cStr.get()];        //Set cell's textual contents
         [cellValue replaceCharactersInRange:NSMakeRange(0, [cellValue length])
                                  withString:[NSString stringWithCString: cStr.get()]];
         
         //Create an attributed string to hold the empty attachment, then release the components.
         attachmentAttrString = [[NSMutableAttributedString attributedStringWithAttachment:textAttachment] retain];
         [textAttachment release];
         [fileWrapper release];
 
         //Get the cell of the text attachment.
         attachmentAttrStringCell = (NSCell *)[(NSTextAttachment *)[attachmentAttrString attribute:NSAttachmentAttributeName
                                                                                           atIndex:0
                                                                                    effectiveRange:nil] attachmentCell];
         //Figure out which image to add, and set the cell's image.
         if ( [self outlineView:outlineView isItemExpandable:item] ) {
             [attachmentAttrStringCell setImage:[NSImage imageNamed:@"folder"]];
         } else {
             [attachmentAttrStringCell setImage:[NSImage imageNamed:@"smallbookmark"]];
         }
         //Insert the image
         [cellValue replaceCharactersInRange:NSMakeRange(0, 0) withAttributedString:attachmentAttrString];
         
         //Tweak the baseline to vertically center the text.
         [cellValue addAttribute:NSBaselineOffsetAttributeName
                           value:[NSNumber numberWithFloat:-3.0]
                           range:NSMakeRange(0, 1)];
     }      }
        return cellValue;
    return nil; 
 }  }
   
 - (void)outlineView:(NSOutlineView *)outlineView setObjectValue:(id)object forTableColumn:(NSTableColumn *)tableColumn byItem:(id)item  - (void)outlineView:(NSOutlineView *)outlineView setObjectValue:(id)object forTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
 {  {
 }  }
   
   - (void)reloadDataForItem:(id)item reloadChildren: (BOOL)aReloadChildren
   {
       printf("Reloading?\n");
       if (!item)
           [mOutlineView reloadData];
       else if ([mOutlineView isItemExpanded: item])
           [mOutlineView reloadItem: item reloadChildren: aReloadChildren];
   }
   
 @end  @end
   
 @implementation BookmarkItem  @implementation BookmarkItem
Line 178 Line 470
 {  {
     mContentNode = aContentNode;      mContentNode = aContentNode;
 }  }
   
   - (id)copyWithZone:(NSZone *)aZone
   {
       BookmarkItem* copy = [[[self class] allocWithZone: aZone] init];
       [copy setContentNode: mContentNode];
       return copy;
   }
   
 @end  @end
   
 // Helper for stripping whitespace  // Helper for stripping whitespace
Line 206  StripWhitespaceNodes(nsIContent* aElemen Line 506  StripWhitespaceNodes(nsIContent* aElemen
 }  }
   
 PRUint32 BookmarksService::gRefCnt = 0;  PRUint32 BookmarksService::gRefCnt = 0;
BookmarksService* BookmarksService::gSingleton = nsnull;nsIDocument* BookmarksService::gBookmarks = nsnull;
 NSMutableDictionary* BookmarksService::gDictionary = nil;
 MainController* BookmarksService::gMainController = nil;
 NSMenu* BookmarksService::gBookmarksMenu = nil;
 nsIDOMElement* BookmarksService::gToolbarRoot = nsnull;
 nsIAtom* BookmarksService::gFolderAtom = nsnull;  nsIAtom* BookmarksService::gFolderAtom = nsnull;
 nsIAtom* BookmarksService::gBookmarkAtom = nsnull;  nsIAtom* BookmarksService::gBookmarkAtom = nsnull;
 nsIAtom* BookmarksService::gHrefAtom = nsnull;  nsIAtom* BookmarksService::gHrefAtom = nsnull;
 nsIAtom* BookmarksService::gNameAtom = nsnull;  nsIAtom* BookmarksService::gNameAtom = nsnull;
   nsVoidArray* BookmarksService::gInstances = nsnull;
   
BookmarksService::BookmarksService()BookmarksService::BookmarksService(BookmarksDataSource* aDataSource)
 {  {
    mDictionary = nil;  mDataSource = aDataSource;
    nsCOMPtr<nsIFile> profileDir;  mToolbar = nil;
    NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(profileDir));}
    profileDir->Append("bookmarks.xml"); 
   
    nsXPIDLCString bookmarksFileURL;BookmarksService::BookmarksService(BookmarksToolbar* aToolbar)
    NS_GetURLSpecFromFile(profileDir, getter_Copies(bookmarksFileURL));{
      mDataSource = nil;
    nsCOMPtr<nsIURI> uri;  mToolbar = aToolbar;
    NS_NewURI(getter_AddRefs(uri), bookmarksFileURL.get()); 
       
    nsCOMPtr<nsIXBLService> xblService(do_GetService("@mozilla.org/xbl;1")); 
     
    xblService->FetchSyncXMLDocument(uri, getter_AddRefs(mBookmarks)); 
     
    nsCOMPtr<nsIContent> rootNode; 
    GetRootContent(getter_AddRefs(rootNode)); 
     
    if (!rootNode) 
        printf("OH SHIT\n"); 
         
    StripWhitespaceNodes(rootNode); 
 }  }
   
 BookmarksService::~BookmarksService()  BookmarksService::~BookmarksService()
 {  {
     [mDictionary release];  
 }  }
   
 void  void
 BookmarksService::GetRootContent(nsIContent** aResult)  BookmarksService::GetRootContent(nsIContent** aResult)
 {  {
     *aResult = nsnull;      *aResult = nsnull;
    if (mBookmarks) {    if (gBookmarks) {
         nsCOMPtr<nsIDOMElement> elt;          nsCOMPtr<nsIDOMElement> elt;
        nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(mBookmarks));        nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(gBookmarks));
         domDoc->GetDocumentElement(getter_AddRefs(elt));          domDoc->GetDocumentElement(getter_AddRefs(elt));
         elt->QueryInterface(NS_GET_IID(nsIContent), (void**)aResult); // Addref happens here.          elt->QueryInterface(NS_GET_IID(nsIContent), (void**)aResult); // Addref happens here.
     }      }
Line 258  BookmarksService::GetRootContent(nsICont Line 548  BookmarksService::GetRootContent(nsICont
 BookmarkItem*  BookmarkItem*
 BookmarksService::GetWrapperFor(nsIContent* aContent)  BookmarksService::GetWrapperFor(nsIContent* aContent)
 {  {
    if (!mDictionary)    if (!gDictionary)
        mDictionary = [[NSMutableDictionary alloc] initWithCapacity: 30];        gDictionary = [[NSMutableDictionary alloc] initWithCapacity: 30];
           
     PRUint32 contentID;      PRUint32 contentID;
     aContent->GetContentID(&contentID);      aContent->GetContentID(&contentID);
           
    BookmarkItem* item = [mDictionary objectForKey: [NSNumber numberWithInt: contentID]];    BookmarkItem* item = [gDictionary objectForKey: [NSNumber numberWithInt: contentID]];
     if (item)      if (item)
         return item;          return item;
     else {      else {
         // Create an item.          // Create an item.
         item = [[[BookmarkItem alloc] init] autorelease]; // The dictionary retains us.          item = [[[BookmarkItem alloc] init] autorelease]; // The dictionary retains us.
         [item setContentNode: aContent];          [item setContentNode: aContent];
        [mDictionary setObject: item forKey: [NSNumber numberWithInt: contentID]];        [gDictionary setObject: item forKey: [NSNumber numberWithInt: contentID]];
     }      }
     return item;      return item;
 }  }
   
BookmarksService*NSMenu*
BookmarksService::AddObserver(nsIDocumentObserver* aObserver)BookmarksService::LocateMenu(nsIContent* aContent)
 {
   nsCOMPtr<nsIContent> parent;
   aContent->GetParent(*getter_AddRefs(parent));
   if (!parent) {
     return BookmarksService::gBookmarksMenu;
   }
   
   NSMenu* parentMenu = LocateMenu(parent);
   
   PRUint32 contentID;
   aContent->GetContentID(&contentID);
 
   NSMenuItem* childMenu = [parentMenu itemWithTag: contentID];
   return [childMenu submenu];
 }
 
 void
 BookmarksService::BookmarkAdded(nsIContent* aContainer, nsIContent* aChild)
 {
   if (!gInstances || !gDictionary)
     return;
 
   PRInt32 count = gInstances->Count();
   for (PRInt32 i = 0; i < count; i++) {
     BookmarksService* instance = (BookmarksService*)gInstances->ElementAt(i);
 
     if (instance->mDataSource) {
       // We're a tree view.
       nsCOMPtr<nsIContent> parent;
       aContainer->GetParent(*getter_AddRefs(parent));
 
       BookmarkItem* item = nil;
       if (parent)
         // We're not the root.
         item = GetWrapperFor(aContainer);
 
       [(instance->mDataSource) reloadDataForItem: item reloadChildren: YES];
     }
     else if (instance->mToolbar) {
       // We're a personal toolbar.
       nsCOMPtr<nsIDOMElement> parentElt(do_QueryInterface(aContainer));
       if (parentElt == gToolbarRoot) {
         // We only care about changes that occur to the personal toolbar's immediate
         // children.
         PRInt32 index = -1;
         aContainer->IndexOf(aChild, index);
         nsCOMPtr<nsIDOMElement> elt(do_QueryInterface(aChild));
         [(instance->mToolbar) addButton: elt atIndex: index];
       }
     }
     else {
       // We're the menu.
       PRInt32 index = -1;
       aContainer->IndexOf(aChild, index);
       NSMenu* menu = LocateMenu(aContainer);
       AddMenuBookmark(menu, aContainer, aChild, index);
     }
   }
   
   FlushBookmarks();  
 }
 
 void
 BookmarksService::BookmarkChanged(nsIContent* aItem)
 {
   if (!gInstances || !gDictionary)
     return;
 
   PRInt32 count = gInstances->Count();
   for (PRInt32 i = 0; i < count; i++) {
     BookmarksService* instance = (BookmarksService*)gInstances->ElementAt(i);
    
     if (instance->mDataSource) {
       BookmarkItem* item = GetWrapperFor(aItem);
       [(instance->mDataSource) reloadDataForItem: item reloadChildren: NO];
     }
   }
 
   FlushBookmarks();  
 }
 
 void
 BookmarksService::BookmarkRemoved(nsIContent* aContainer, nsIContent* aChild)
 {
   if (!gInstances)
     return;
 
   PRInt32 count = gInstances->Count();
   for (PRInt32 i = 0; i < count; i++) {
     BookmarksService* instance = (BookmarksService*)gInstances->ElementAt(i);
 
     if (instance->mDataSource) {
       // We're a tree view.
       nsCOMPtr<nsIContent> parent;
       aContainer->GetParent(*getter_AddRefs(parent));
 
       BookmarkItem* item = nil;
       if (parent)
         // We're not the root.
         item = GetWrapperFor(aContainer);
 
       [(instance->mDataSource) reloadDataForItem: item reloadChildren: YES];
     }
     else if (instance->mToolbar) {
       // We're a personal toolbar.
       nsCOMPtr<nsIDOMElement> parentElt(do_QueryInterface(aContainer));
       if (parentElt == gToolbarRoot) {
         // We only care about changes that occur to the personal toolbar's immediate
         // children.
         nsCOMPtr<nsIDOMElement> childElt(do_QueryInterface(aChild));
         [(instance->mToolbar) removeButton: childElt];
       }
     }    
     else {
       // We're the menu.
       NSMenu* menu = LocateMenu(aContainer);
       PRUint32 contentID;
       aChild->GetContentID(&contentID);
       NSMenuItem* childItem = [menu itemWithTag: contentID];
       [menu removeItem: childItem];
     }
   }
 
   FlushBookmarks(); 
 }
 
 void
 BookmarksService::AddObserver()
 {  {
     gRefCnt++;      gRefCnt++;
     if (gRefCnt == 1) {      if (gRefCnt == 1) {
         gSingleton = new BookmarksService();  
         gBookmarkAtom = NS_NewAtom("bookmark");          gBookmarkAtom = NS_NewAtom("bookmark");
         gFolderAtom = NS_NewAtom("folder");          gFolderAtom = NS_NewAtom("folder");
         gNameAtom = NS_NewAtom("name");          gNameAtom = NS_NewAtom("name");
         gHrefAtom = NS_NewAtom("href");          gHrefAtom = NS_NewAtom("href");
           gInstances = new nsVoidArray();
                   
           nsCOMPtr<nsIFile> profileDir;
           NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(profileDir));
           profileDir->Append("bookmarks.xml");
       
           nsCAutoString bookmarksFileURL;
           NS_GetURLSpecFromFile(profileDir, bookmarksFileURL);
           
           nsCOMPtr<nsIURI> uri;
           NS_NewURI(getter_AddRefs(uri), bookmarksFileURL.get());
       
           nsCOMPtr<nsIXBLService> xblService(do_GetService("@mozilla.org/xbl;1"));    
           xblService->FetchSyncXMLDocument(uri, &gBookmarks); // The addref is here.
           
           nsCOMPtr<nsIContent> rootNode;
           GetRootContent(getter_AddRefs(rootNode));
           StripWhitespaceNodes(rootNode);
     }      }
           
    return gSingleton;    gInstances->AppendElement(this);
 }  }
   
 void  void
BookmarksService::RemoveObserver(nsIDocumentObserver* aObserver)BookmarksService::RemoveObserver()
 {  {
     if (gRefCnt == 0)      if (gRefCnt == 0)
         return;          return;
         
     gInstances->RemoveElement(this);
      
     gRefCnt--;      gRefCnt--;
     if (gRefCnt == 0) {      if (gRefCnt == 0) {
        delete gSingleton;        NS_IF_RELEASE(gBookmarks);
         NS_RELEASE(gBookmarkAtom);          NS_RELEASE(gBookmarkAtom);
         NS_RELEASE(gFolderAtom);          NS_RELEASE(gFolderAtom);
         NS_RELEASE(gNameAtom);          NS_RELEASE(gNameAtom);
         NS_RELEASE(gHrefAtom);          NS_RELEASE(gHrefAtom);
           [gDictionary release];
       }
   }
   
   void
   BookmarksService::FlushBookmarks()
   {
       nsCOMPtr<nsIFile> bookmarksFile;
       NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(bookmarksFile));
       bookmarksFile->Append("bookmarks.xml");
   
       nsCOMPtr<nsIOutputStream> outputStream;
       NS_NewLocalFileOutputStream(getter_AddRefs(outputStream), bookmarksFile);
   
       nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(gBookmarks));
       
       nsCOMPtr<nsIDOMSerializer> domSerializer(do_CreateInstance(NS_XMLSERIALIZER_CONTRACTID));
       domSerializer->SerializeToStream(domDoc, outputStream, nsnull);
   }
   
   void BookmarksService::EnsureToolbarRoot()
   {
     if (gToolbarRoot)
       return;
   
     nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(gBookmarks));
     nsCOMPtr<nsIDOMElement> rootElt;
     domDoc->GetDocumentElement(getter_AddRefs(rootElt));
     
     nsCOMPtr<nsIDOMNode> child;
     rootElt->GetFirstChild(getter_AddRefs(child));
     nsAutoString typeValue;
     while (child) {
       nsCOMPtr<nsIDOMElement> childElt(do_QueryInterface(child));
       if (childElt) {
         childElt->GetAttribute(NS_LITERAL_STRING("type"), typeValue);
         if (typeValue.Equals(NS_LITERAL_STRING("toolbar")))
           gToolbarRoot = childElt;
       }
       
       nsCOMPtr<nsIDOMNode> temp;
       child->GetNextSibling(getter_AddRefs(temp));
       child = temp;
     }
   
     if (!gToolbarRoot) {
       printf("Repairing personal toolbar.\n");
       nsCOMPtr<nsIDOMElement> elt;
       domDoc->CreateElementNS(NS_LITERAL_STRING("http://chimera.mozdev.org/bookmarks/"),
                               NS_LITERAL_STRING("folder"),
                               getter_AddRefs(elt));
   
       elt->SetAttribute(NS_LITERAL_STRING("name"), NS_LITERAL_STRING("Toolbar Bookmarks"));
       elt->SetAttribute(NS_LITERAL_STRING("type"), NS_LITERAL_STRING("toolbar"));
   
       nsCOMPtr<nsIDOMNode> dummy;
       rootElt->AppendChild(elt, getter_AddRefs(dummy));
       gToolbarRoot = elt;
     }
   }
   
   static
   void RecursiveAddBookmarkConstruct(NSPopUpButton* aPopup, NSMenu* aMenu, int aTagToMatch)
   {
     // Get the menu item children.
     NSArray* children = [aMenu itemArray];
     int startPosition = 0;
     if (aMenu == BookmarksService::gBookmarksMenu)
       startPosition = 3;
   
     int count = [children count];
     for (int i = startPosition; i < count; i++) {
       NSMenuItem* menuItem = [children objectAtIndex: i];
       NSMenu* submenu = [menuItem submenu];
       if (submenu) {
         // This is a folder.  Add it to our list and then recur.
         [aPopup addItemWithTitle: [menuItem title]];
         NSMenuItem* lastItem = [aPopup lastItem];
         if ([menuItem tag] == aTagToMatch)
           [aPopup selectItem: lastItem];
         
         [lastItem setTag: [menuItem tag]];
         RecursiveAddBookmarkConstruct(aPopup, submenu, aTagToMatch);
       }
     }
   }
   
   void
   BookmarksService::ConstructAddBookmarkFolderList(NSPopUpButton* aPopup, BookmarkItem* aItem)
   {
     [aPopup removeAllItems];
     [aPopup addItemWithTitle: [gBookmarksMenu title]];
     NSMenuItem* lastItem = [aPopup lastItem];
     [lastItem setTag: -1];
     int tag = -1;
     if (aItem) {
       nsIContent* content = [aItem contentNode];
       PRUint32 utag;
       content->GetContentID(&utag);
       tag = (int)utag;
     }
     RecursiveAddBookmarkConstruct(aPopup, gBookmarksMenu, tag);
   }
   
   void
   BookmarksService::GetTitleAndHrefForBrowserView(id aBrowserView, nsString& aTitle, nsString& aHref)
   {
     nsCOMPtr<nsIWebBrowser> webBrowser = getter_AddRefs([aBrowserView getWebBrowser]);
     nsCOMPtr<nsIDOMWindow> window;
     webBrowser->GetContentDOMWindow(getter_AddRefs(window));
     nsCOMPtr<nsIDOMDocument> htmlDoc;
     window->GetDocument(getter_AddRefs(htmlDoc));
     nsCOMPtr<nsIDocument> pageDoc(do_QueryInterface(htmlDoc));
   
     if (pageDoc) {
       nsCOMPtr<nsIURI> url;
       pageDoc->GetDocumentURL(getter_AddRefs(url));
       nsCAutoString spec;
       url->GetSpec(spec);
       aHref.AssignWithConversion(spec.get());
     }
   
     nsCOMPtr<nsIDOMHTMLDocument> htmlDocument(do_QueryInterface(htmlDoc));
     if (htmlDocument)
       htmlDocument->GetTitle(aTitle);
     if (aTitle.IsEmpty())
       aTitle = aHref;  
   }
   
   void
   BookmarksService::ConstructBookmarksMenu(NSMenu* aMenu, nsIContent* aContent)
   {
       nsCOMPtr<nsIContent> content = aContent;
       if (!content) {
           GetRootContent(getter_AddRefs(content));
           GetWrapperFor(content);
           gBookmarksMenu = aMenu;
       }
       
       // Now walk our children, and for folders also recur into them.
       PRInt32 childCount;
       content->ChildCount(childCount);
       
       for (PRInt32 i = 0; i < childCount; i++) {
         nsCOMPtr<nsIContent> child;
         content->ChildAt(i, *getter_AddRefs(child));
         AddMenuBookmark(aMenu, content, child, -1);
     }      }
 }  }
   
   void
   BookmarksService::AddMenuBookmark(NSMenu* aMenu, nsIContent* aParent, nsIContent* aChild, PRInt32 aIndex)
   {
     nsAutoString name;
     aChild->GetAttr(kNameSpaceID_None, gNameAtom, name);
     nsCAutoString nameCStr; nameCStr.AssignWithConversion(name);
     NSString* title = [NSString stringWithCString: nameCStr.get()];
   
     // Create a menu or menu item for the child.
     NSMenuItem* menuItem = [[[NSMenuItem alloc] initWithTitle: title action: NULL keyEquivalent: @""] autorelease];
     GetWrapperFor(aChild);
   
     if (aIndex == -1)
       [aMenu addItem: menuItem];
     else
       [aMenu insertItem: menuItem atIndex: aIndex];
     
     nsCOMPtr<nsIAtom> tagName;
     aChild->GetTag(*getter_AddRefs(tagName));
   
     if (tagName == gFolderAtom) {
       NSMenu* menu = [[[NSMenu alloc] initWithTitle: title] autorelease];
       [aMenu setSubmenu: menu forItem: menuItem];
       [menu setAutoenablesItems: NO];
       ConstructBookmarksMenu(menu, aChild);
     }
     else {
       [menuItem setTarget: gMainController];
       [menuItem setAction: @selector(openMenuBookmark:)];
     }
   
     PRUint32 contentID;
     aChild->GetContentID(&contentID);
     [menuItem setTag: contentID];
   }
   
   void 
   BookmarksService::OpenMenuBookmark(BrowserWindowController* aController, id aMenuItem)
   {
       // Get the corresponding bookmark item.
       BookmarkItem* item = [gDictionary objectForKey: [NSNumber numberWithInt: [aMenuItem tag]]];
           
       // Get the content node.
       nsIContent* content = [item contentNode];
           
       // Get the href attribute.  This is the URL we want to load.
       nsAutoString href;
       content->GetAttr(kNameSpaceID_None, gHrefAtom, href);
       nsCAutoString cref; cref.AssignWithConversion(href);
       if (cref.IsEmpty())
           return;
           
       NSString* url = [NSString stringWithCString: cref.get()];
       
       // Now load the URL in the window.
       [aController loadURL:[NSURL URLWithString: url]];
       
       // Focus and activate our content area.
       [[[aController getMyBrowserView] getBrowserView] setActive: YES];
   }
   
   void
   BookmarksService::OpenBookmarkGroup(id aTabView, nsIDOMElement* aFolder)
   {
     // We might conceivably have to make new tabs in order to load all
     // the items in the group.
     int currentIndex = 0;
     int total = [aTabView numberOfTabViewItems];
     nsCOMPtr<nsIDOMNode> child;
     aFolder->GetFirstChild(getter_AddRefs(child));
     while (child) {
       nsCOMPtr<nsIDOMElement> elt(do_QueryInterface(child));
       if (elt) {
         nsAutoString href;
         elt->GetAttribute(NS_LITERAL_STRING("href"), href);
         if (!href.IsEmpty()) {
           nsCAutoString cref; cref.AssignWithConversion(href);
           NSString* url = [NSString stringWithCString: cref.get()];
           NSTabViewItem* tabViewItem = nil;
           if (currentIndex >= total) {
             // We need to make a new tab.
             tabViewItem = [[[NSTabViewItem alloc] initWithIdentifier: nil] autorelease];
             MyBrowserView* newView = [[[MyBrowserView alloc] initWithTab: tabViewItem andWindow: [aTabView window]] autorelease];
             [tabViewItem setLabel: @"Untitled"];
             [tabViewItem setView: newView];
             [aTabView addTabViewItem: tabViewItem];
           }
           else
             tabViewItem = [aTabView tabViewItemAtIndex: currentIndex];
   
           [[[tabViewItem view] getBrowserView] loadURI:[NSURL URLWithString: url]
                                                  flags: NSLoadFlagsNone];
         }
       }
       
       nsCOMPtr<nsIDOMNode> temp = child;
       temp->GetNextSibling(getter_AddRefs(child));
       currentIndex++;
     }
   
     // XXXdwh Select and activate the first tab.
   }

Removed from v.1.1  
changed lines
  Added in v.1.29


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>