From d8a89b3d30a6b0466e838e9a5a9819c897b75017 Mon Sep 17 00:00:00 2001
From: "benjamin%smedbergs.us" <benjamin%smedbergs.us>
Date: Tue, 18 Jul 2006 16:50:42 +0000
Subject: [PATCH] Implemented removeAnnotation() Added getAnnotationInt64() and
 setAnnotationInt64() to make dealing with dates and times easier. r=bryner
 b=319769

Original committer: annie.sullivan%gmail.com
Original revision: 1.7
Original date: 2005/12/10 07:25:13
---
 .../places/src/nsAnnotationService.cpp        | 81 ++++++++++++++++++-
 1 file changed, 79 insertions(+), 2 deletions(-)

diff --git a/toolkit/components/places/src/nsAnnotationService.cpp b/toolkit/components/places/src/nsAnnotationService.cpp
index ba82abefdf8..aa4c00c702b 100644
--- a/toolkit/components/places/src/nsAnnotationService.cpp
+++ b/toolkit/components/places/src/nsAnnotationService.cpp
@@ -108,6 +108,9 @@ nsAnnotationService::Init()
   rv = mDBConn->CreateStatement(NS_LITERAL_CSTRING("INSERT INTO moz_anno (page, name, mime_type, content, flags, expiration) VALUES (?2, ?3, ?4, ?5, ?6, ?7)"),
                                 getter_AddRefs(mDBAddAnnotation));
   NS_ENSURE_SUCCESS(rv, rv);
+  rv = mDBConn->CreateStatement(NS_LITERAL_CSTRING("DELETE FROM moz_anno WHERE page = ?1 AND name = ?2"),
+                                getter_AddRefs(mDBRemoveAnnotation));
+  NS_ENSURE_SUCCESS(rv, rv);
 
   return NS_OK;
 }
@@ -194,6 +197,38 @@ nsAnnotationService::SetAnnotationInt32(nsIURI* aURI,
 }
 
 
+// nsAnnotationService::SetAnnotationInt64
+
+NS_IMETHODIMP
+nsAnnotationService::SetAnnotationInt64(nsIURI* aURI,
+                                        const nsACString& aName,
+                                        PRInt64 aValue,
+                                        PRInt32 aFlags, PRInt32 aExpiration)
+{
+  mozStorageTransaction transaction(mDBConn, PR_FALSE);
+  mozIStorageStatement* statement; // class var, not owned by this function
+  nsresult rv = StartSetAnnotation(aURI, aName, aFlags, aExpiration, &statement);
+  NS_ENSURE_SUCCESS(rv, rv);
+  mozStorageStatementScoper statementResetter(statement);
+
+  rv = statement->BindInt64Parameter(kAnnoIndex_Content, aValue);
+  NS_ENSURE_SUCCESS(rv, rv);
+  rv = statement->BindNullParameter(kAnnoIndex_MimeType);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  rv = statement->Execute();
+  NS_ENSURE_SUCCESS(rv, rv);
+  transaction.Commit();
+
+  // should reset the statement; observers may call our service back to get
+  // annotation values!
+  statement->Reset();
+  statementResetter.Abandon();
+  CallSetObservers(aURI, aName);
+  return NS_OK;
+}
+
+
 // nsAnnotationService::SetAnnotationBinary
 
 NS_IMETHODIMP
@@ -286,6 +321,22 @@ nsAnnotationService::GetAnnotationInt32(nsIURI* aURI,
 }
 
 
+// nsAnnotationService::GetAnnotationInt64
+
+NS_IMETHODIMP
+nsAnnotationService::GetAnnotationInt64(nsIURI* aURI,
+                                        const nsACString& aName,
+                                        PRInt64 *_retval)
+{
+  nsresult rv = StartGetAnnotationFromURI(aURI, aName);
+  if (NS_FAILED(rv))
+    return rv;
+  *_retval = mDBGetAnnotationFromURI->AsInt64(kAnnoIndex_Content);
+  mDBGetAnnotationFromURI->Reset();
+  return NS_OK;
+}
+
+
 // nsAnnotationService::GetAnnotationBinary
 
 NS_IMETHODIMP
@@ -409,8 +460,34 @@ NS_IMETHODIMP
 nsAnnotationService::RemoveAnnotation(nsIURI* aURI,
                                       const nsACString& aName)
 {
-  // FIXME
-  return NS_ERROR_NOT_IMPLEMENTED;
+  nsresult rv;
+  nsNavHistory* history = nsNavHistory::GetHistoryService();
+  NS_ENSURE_TRUE(history, NS_ERROR_FAILURE);
+
+  PRInt64 uriID;
+  rv = history->GetUrlIdFor(aURI, &uriID, PR_FALSE);
+  NS_ENSURE_SUCCESS(rv, rv);
+  if (uriID == 0) // Check if URI exists.
+    return NS_OK;
+  
+  mozStorageStatementScoper resetter(mDBRemoveAnnotation);
+
+  rv = mDBRemoveAnnotation->BindInt64Parameter(0, uriID);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  rv = mDBRemoveAnnotation->BindUTF8StringParameter(1, aName);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  rv = mDBRemoveAnnotation->Execute();
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  resetter.Abandon();
+  
+  // Update observers
+  for (PRInt32 i = 0; i < mObservers.Count(); i ++)
+    mObservers[i]->OnAnnotationRemoved(aURI, aName);
+
+  return NS_OK;
 }