Diff for /chimera/BookmarksService.mm between versions 1.3 and 1.5

version 1.3, 2002/02/09 01:25:26 version 1.5, 2002/02/09 08:12:33
Line 13 Line 13
 #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"
Line 22 Line 23
 #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 35 Line 37
 -(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 52
     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:)];
Line 55 Line 64
 {  {
     if (!mBookmarks)      if (!mBookmarks)
         return;          return;
            
     nsCOMPtr<nsIContent> content;      nsCOMPtr<nsIContent> content;
     int index = [mOutlineView selectedRow];      int index = [mOutlineView selectedRow];
       
     if (index > 0) {      if (index > 0) {
         BookmarkItem* item = [mOutlineView itemAtRow: index];          BookmarkItem* item = [mOutlineView itemAtRow: index];
        if (![mOutlineView isExpandable: item])        if ([mOutlineView isExpandable: item]) 
             content = [item contentNode];              content = [item contentNode];
     }      }
           
     if (!content)      if (!content)
         mBookmarks->GetRootContent(getter_AddRefs(content));          mBookmarks->GetRootContent(getter_AddRefs(content));
                   
    nsCOMPtr<nsIDocument> doc;    nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(mBookmarks->gBookmarks));
    content->GetDocument(*getter_AddRefs(doc)); 
    nsCOMPtr<nsIDOMDocument> domDoc(do_QueryInterface(doc)); 
     nsCOMPtr<nsIDOMElement> elt;      nsCOMPtr<nsIDOMElement> elt;
     domDoc->CreateElementNS(NS_LITERAL_STRING("http://chimera.mozdev.org/bookmarks"),       domDoc->CreateElementNS(NS_LITERAL_STRING("http://chimera.mozdev.org/bookmarks"), 
                             NS_LITERAL_STRING("bookmark"),                               NS_LITERAL_STRING("bookmark"), 
                             getter_AddRefs(elt));                              getter_AddRefs(elt));
           
    elt->SetAttribute(NS_LITERAL_STRING("title"), NS_LITERAL_STRING("MozillaZine"));    // Fetch the title of the current page and the URL.
    elt->SetAttribute(NS_LITERAL_STRING("href"), NS_LITERAL_STRING("http://www.mozillazine.org/"));    nsCOMPtr<nsIWebBrowser> webBrowser = getter_AddRefs([[mBrowserView getBrowserView] getWebBrowser]);
     nsCOMPtr<nsIDOMWindow> window;
     webBrowser->GetContentDOMWindow(getter_AddRefs(window));
     nsCOMPtr<nsIDOMDocument> htmlDoc;
     window->GetDocument(getter_AddRefs(htmlDoc));
     nsCOMPtr<nsIDocument> pageDoc(do_QueryInterface(htmlDoc));
     
     nsAutoString href; 
     if (pageDoc) {
         nsCOMPtr<nsIURI> url;
         pageDoc->GetDocumentURL(getter_AddRefs(url));
         nsXPIDLCString spec;
         url->GetSpec(getter_Copies(spec));
         href.AssignWithConversion(spec.get());
     }
     
     nsAutoString title;
     nsCOMPtr<nsIDOMHTMLDocument> htmlDocument(do_QueryInterface(htmlDoc));
     if (htmlDocument)
         htmlDocument->GetTitle(title);
     if (title.IsEmpty())
         title = href;
         
     elt->SetAttribute(NS_LITERAL_STRING("name"), title);
     elt->SetAttribute(NS_LITERAL_STRING("href"), href);    
           
     nsCOMPtr<nsIDOMElement> parent(do_QueryInterface(content));      nsCOMPtr<nsIDOMElement> parent(do_QueryInterface(content));
     nsCOMPtr<nsIDOMNode> dummy;      nsCOMPtr<nsIDOMNode> dummy;
     parent->AppendChild(elt, getter_AddRefs(dummy));      parent->AppendChild(elt, getter_AddRefs(dummy));
   
       mBookmarks->NotifyObservers(content, PR_TRUE);
 }  }
   
 -(IBAction)deleteBookmark: (id)aSender  -(IBAction)deleteBookmark: (id)aSender
Line 186 Line 219
 {  {
 }  }
   
   - (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 226  StripWhitespaceNodes(nsIContent* aElemen Line 268  StripWhitespaceNodes(nsIContent* aElemen
 }  }
   
 PRUint32 BookmarksService::gRefCnt = 0;  PRUint32 BookmarksService::gRefCnt = 0;
BookmarksService* BookmarksService::gSingleton = nsnull;nsIDocument* BookmarksService::gBookmarks = nsnull;
 NSMutableDictionary* BookmarksService::gDictionary = nil;
 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; 
    NS_GetSpecialDirectory(NS_APP_USER_PROFILE_50_DIR, getter_AddRefs(profileDir)); 
    profileDir->Append("bookmarks.xml"); 
 
    nsXPIDLCString bookmarksFileURL; 
    NS_GetURLSpecFromFile(profileDir, getter_Copies(bookmarksFileURL)); 
     
    nsCOMPtr<nsIURI> uri; 
    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)); 
    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 273  BookmarksService::GetRootContent(nsICont Line 300  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*void
BookmarksService::AddObserver(nsIDocumentObserver* aObserver)BookmarksService::NotifyObservers(nsIContent* aContainer, PRBool aReloadChildren)
 {
     if (!gInstances)
         return;
     
     PRInt32 count = gInstances->Count();
     for (PRInt32 i = 0; i < count; i++) {
         BookmarksService* instance = (BookmarksService*)gInstances->ElementAt(i);
         instance->NotifyObserver(aContainer, aReloadChildren);
     }
 }
 
 
 void
 BookmarksService::NotifyObserver(nsIContent* aContainer, PRBool aReloadChildren)
 {
     if (!gDictionary)
         return;
        
     printf("uh. appended!\n");
     
     nsCOMPtr<nsIContent> parent;
     aContainer->GetParent(*getter_AddRefs(parent));
     
     BookmarkItem* item = nil;
     if (parent)
         // We're not the root.
         item = GetWrapperFor(aContainer);
     
     [mDataSource reloadDataForItem: item reloadChildren: aReloadChildren];
 }
 
 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");
       
           nsXPIDLCString bookmarksFileURL;
           NS_GetURLSpecFromFile(profileDir, getter_Copies(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];
     }      }
 }  }
   

Removed from v.1.3  
changed lines
  Added in v.1.5


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