view/aaTreeView.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
   3 /*
   4  * Copyright (C) 2007,2008 Sergey Yanovich <ynvich@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public
  17  * License along with this program; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include "xpcom-config.h"
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsStringAPI.h"
  26 #include "nsIDOMElement.h"
  27 #include "nsIDOMNode.h"
  28 #include "nsIDOMDocument.h"
  29 #include "nsIDOMXULElement.h"
  30 #include "nsIInterfaceRequestorUtils.h"
  31 #include "nsISHistory.h"
  32 #include "nsServiceManagerUtils.h"
  33 #include "nsIObserverService.h"
  34
  35 /* Unfrozen API */
  36 #include "nsITreeSelection.h"
  37 #include "nsITreeColumns.h"
  38 #include "nsITreeBoxObject.h"
  39 #include "nsIArray.h"
  40 #include "nsArrayUtils.h"
  41
  42 /* Project includes */
  43 #include "aaISession.h"
  44 #include "aaISqlTransaction.h"
  45 #include "aaISqlRequest.h"
  46 #include "aaTreeView.h"
  47
  48 aaTreeView::aaTreeView()
  49   :mFlags(0)
  50 {
  51 }
  52
  53 aaTreeView::~aaTreeView()
  54 {
  55 }
  56
  57 nsresult
  58 aaTreeView::Init(nsISupports *aParam)
  59 {
  60   nsresult rv;
  61   mSession = do_QueryInterface(aParam, &rv);
  62   NS_ENSURE_SUCCESS(rv, rv);
  63
  64   return NS_OK;
  65 }
  66
  67 NS_IMPL_ISUPPORTS2(aaTreeView,
  68                    aaIDataTreeView,
  69                    nsITreeView)
  70
  71 NS_IMETHODIMP
  72 aaTreeView::GetRowCount(PRInt32 *aRowCount)
  73 {
  74   NS_ENSURE_ARG_POINTER(aRowCount);
  75   *aRowCount = 0;
  76   if (! mDataSet)
  77     return NS_OK;
  78
  79   return mDataSet->GetLength((PRUint32 *) aRowCount);
  80 }
  81
  82 NS_IMETHODIMP
  83 aaTreeView::GetSelection(nsITreeSelection * *aSelection)
  84 {
  85   NS_ENSURE_ARG_POINTER( aSelection );
  86   NS_IF_ADDREF(*aSelection = mSelection);
  87   return NS_OK;
  88 }
  89 NS_IMETHODIMP
  90 aaTreeView::SetSelection(nsITreeSelection * aSelection)
  91 {
  92   mSelection = aSelection;
  93
  94   return NS_OK;
  95 }
  96
  97 NS_IMETHODIMP
  98 aaTreeView::GetRowProperties(PRInt32 index, nsISupportsArray *properties)
  99 {
 100     return NS_ERROR_NOT_IMPLEMENTED;
 101 }
 102
 103 NS_IMETHODIMP
 104 aaTreeView::GetCellProperties(PRInt32 row, nsITreeColumn *col, nsISupportsArray *properties)
 105 {
 106   return NS_ERROR_NOT_IMPLEMENTED;
 107 }
 108
 109 NS_IMETHODIMP
 110 aaTreeView::GetColumnProperties(nsITreeColumn *col, nsISupportsArray *properties)
 111 {
 112     return NS_ERROR_NOT_IMPLEMENTED;
 113 }
 114
 115 NS_IMETHODIMP
 116 aaTreeView::IsContainer(PRInt32 index, PRBool *_retval)
 117 {
 118   NS_ENSURE_ARG_POINTER(_retval);
 119   *_retval = PR_FALSE;
 120   return NS_OK;
 121 }
 122
 123 NS_IMETHODIMP
 124 aaTreeView::IsContainerOpen(PRInt32 index, PRBool *_retval)
 125 {
 126     return NS_ERROR_NOT_IMPLEMENTED;
 127 }
 128
 129 NS_IMETHODIMP
 130 aaTreeView::IsContainerEmpty(PRInt32 index, PRBool *_retval)
 131 {
 132     return NS_ERROR_NOT_IMPLEMENTED;
 133 }
 134
 135 NS_IMETHODIMP
 136 aaTreeView::IsSeparator(PRInt32 index, PRBool *_retval)
 137 {
 138   NS_ENSURE_ARG_POINTER(_retval);
 139   *_retval = PR_FALSE;
 140   return NS_OK;
 141 }
 142
 143 NS_IMETHODIMP
 144 aaTreeView::IsSorted(PRBool *_retval)
 145 {
 146     return NS_ERROR_NOT_IMPLEMENTED;
 147 }
 148
 149 NS_IMETHODIMP
 150 aaTreeView::CanDrop(PRInt32 index, PRInt32 orientation, PRBool *_retval)
 151 {
 152     return NS_ERROR_NOT_IMPLEMENTED;
 153 }
 154
 155 NS_IMETHODIMP
 156 aaTreeView::Drop(PRInt32 row, PRInt32 orientation)
 157 {
 158     return NS_ERROR_NOT_IMPLEMENTED;
 159 }
 160
 161 NS_IMETHODIMP
 162 aaTreeView::GetParentIndex(PRInt32 rowIndex, PRInt32 *_retval)
 163 {
 164     return NS_ERROR_NOT_IMPLEMENTED;
 165 }
 166
 167 NS_IMETHODIMP
 168 aaTreeView::HasNextSibling(PRInt32 rowIndex, PRInt32 afterIndex, PRBool *_retval)
 169 {
 170   NS_ENSURE_ARG_POINTER(_retval);
 171   nsresult rv;
 172   PRInt32 count;
 173   rv = GetRowCount(&count);
 174   NS_ENSURE_SUCCESS(rv, rv);
 175   if (rowIndex == afterIndex && rowIndex < count) {
 176     *_retval = PR_TRUE;
 177   } else {
 178     *_retval = PR_FALSE;
 179   }
 180   return NS_OK;
 181 }
 182
 183 NS_IMETHODIMP
 184 aaTreeView::GetLevel(PRInt32 index, PRInt32 *_retval)
 185 {
 186   NS_ENSURE_ARG_POINTER(_retval);
 187   *_retval = 0;
 188   return NS_OK;
 189 }
 190
 191 NS_IMETHODIMP
 192 aaTreeView::GetImageSrc(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 193 {
 194     return NS_ERROR_NOT_IMPLEMENTED;
 195 }
 196
 197 NS_IMETHODIMP
 198 aaTreeView::GetProgressMode(PRInt32 row, nsITreeColumn *col, PRInt32 *_retval)
 199 {
 200     return NS_ERROR_NOT_IMPLEMENTED;
 201 }
 202
 203 NS_IMETHODIMP
 204 aaTreeView::GetCellValue(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 205 {
 206     return NS_ERROR_NOT_IMPLEMENTED;
 207 }
 208
 209 NS_IMETHODIMP
 210 aaTreeView::GetCellText(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 211 {
 212   /**
 213    * There may be a bug in <tree> implementation. This function is
 214    * occasionally called after mTree is detached by SetTree(nsnull);
 215    */
 216   NS_ENSURE_TRUE(mDataSet, NS_ERROR_NOT_INITIALIZED);
 217   NS_ENSURE_TRUE(row >= 0, NS_ERROR_INVALID_ARG);
 218   return doGetCellText(row, col, _retval);
 219 }
 220
 221 NS_IMETHODIMP
 222 aaTreeView::SetTree(nsITreeBoxObject *tree)
 223 {
 224   mTree = tree;
 225   return NS_OK;
 226 }
 227
 228 NS_IMETHODIMP
 229 aaTreeView::ToggleOpenState(PRInt32 index)
 230 {
 231     return NS_ERROR_NOT_IMPLEMENTED;
 232 }
 233
 234 NS_IMETHODIMP
 235 aaTreeView::CycleHeader(nsITreeColumn *col)
 236 {
 237     return NS_ERROR_NOT_IMPLEMENTED;
 238 }
 239
 240 NS_IMETHODIMP
 241 aaTreeView::SelectionChanged()
 242 {
 243   return NS_OK;
 244 }
 245
 246 NS_IMETHODIMP
 247 aaTreeView::CycleCell(PRInt32 row, nsITreeColumn *col)
 248 {
 249     return NS_ERROR_NOT_IMPLEMENTED;
 250 }
 251
 252 NS_IMETHODIMP
 253 aaTreeView::IsEditable(PRInt32 row, nsITreeColumn *col, PRBool *_retval)
 254 {
 255     return NS_ERROR_NOT_IMPLEMENTED;
 256 }
 257
 258 NS_IMETHODIMP
 259 aaTreeView::IsSelectable(PRInt32 row, nsITreeColumn *col, PRBool *_retval)
 260 {
 261   NS_ENSURE_ARG_POINTER(_retval);
 262   *_retval = PR_TRUE;
 263   return NS_OK;
 264 }
 265
 266 NS_IMETHODIMP
 267 aaTreeView::SetCellValue(PRInt32 row, nsITreeColumn *col, const nsAString & value)
 268 {
 269     return NS_ERROR_NOT_IMPLEMENTED;
 270 }
 271
 272 NS_IMETHODIMP
 273 aaTreeView::SetCellText(PRInt32 row, nsITreeColumn *col, const nsAString & value)
 274 {
 275     return NS_ERROR_NOT_IMPLEMENTED;
 276 }
 277
 278 NS_IMETHODIMP
 279 aaTreeView::PerformAction(const PRUnichar *action)
 280 {
 281     return NS_ERROR_NOT_IMPLEMENTED;
 282 }
 283
 284 NS_IMETHODIMP
 285 aaTreeView::PerformActionOnRow(const PRUnichar *action, PRInt32 row)
 286 {
 287     return NS_ERROR_NOT_IMPLEMENTED;
 288 }
 289
 290 NS_IMETHODIMP
 291 aaTreeView::PerformActionOnCell(const PRUnichar *action, PRInt32 row, nsITreeColumn *col)
 292 {
 293     return NS_ERROR_NOT_IMPLEMENTED;
 294 }
 295
 296 /* aaIDataTreeView */
 297 NS_IMETHODIMP
 298 aaTreeView::GetDataSet(nsIArray * * aDataSet)
 299 {
 300   NS_ENSURE_ARG_POINTER(aDataSet);
 301   NS_IF_ADDREF(*aDataSet = mDataSet);
 302   return NS_OK;
 303 }
 304
 305 NS_IMETHODIMP
 306 aaTreeView::Update(nsISupports *aParam = 0)
 307 {
 308   NS_ENSURE_TRUE(mDataLoader || mComplexLoader, NS_ERROR_NOT_INITIALIZED);
 309   nsresult rv;
 310
 311   if (mTree) {
 312     PRInt32 oldCount;
 313     rv = GetRowCount(&oldCount);
 314     NS_ENSURE_SUCCESS(rv, rv);
 315     mTree->RowCountChanged(0, - oldCount);
 316   }
 317
 318   if (mDataLoader) {
 319     rv = mSession->Load(mDataLoader, getter_AddRefs(mDataSet));
 320     NS_ENSURE_SUCCESS(rv, rv);
 321
 322     rv = postLoadHandler();
 323     NS_ENSURE_SUCCESS(rv, rv);
 324   }
 325   else {
 326     if (aParam) {
 327       rv = mComplexLoader->SetParam(aParam);
 328       NS_ENSURE_SUCCESS(rv, rv);
 329     }
 330
 331     rv = mSession->Execute(mComplexLoader, getter_AddRefs(mDataSet));
 332     NS_ENSURE_SUCCESS(rv, rv);
 333   }
 334
 335   if (mTree) {
 336     PRInt32 newCount;
 337     rv = GetRowCount(&newCount);
 338     NS_ENSURE_SUCCESS(rv, rv);
 339     mTree->RowCountChanged(0, newCount);
 340   }
 341
 342   nsCOMPtr<nsIObserverService> broadcaster = do_GetService(
 343       "@mozilla.org/observer-service;1");
 344   NS_ENSURE_TRUE(broadcaster, NS_OK);
 345   broadcaster->NotifyObservers(this, "tree-view-load", nsnull);
 346
 347   return NS_OK;
 348 }
 349
 350 NS_IMETHODIMP
 351 aaTreeView::GetFlags(PRUint32 *aFlags)
 352 {
 353   NS_ENSURE_ARG_POINTER(aFlags);
 354   *aFlags = mFlags;
 355   return NS_OK;
 356 }
 357 NS_IMETHODIMP
 358 aaTreeView::SetFlags(PRUint32 aFlags)
 359 {
 360   PRBool needUpdate = (aFlags != mFlags);
 361   mFlags = aFlags;
 362   if (NS_LIKELY( needUpdate && mTree )) {
 363     mTree->Invalidate();
 364     nsCOMPtr<nsIObserverService> broadcaster = do_GetService(
 365         "@mozilla.org/observer-service;1");
 366     NS_ENSURE_TRUE(broadcaster, NS_OK);
 367     broadcaster->NotifyObservers(this, "tree-flags-change", nsnull);
 368   }
 369   return NS_OK;
 370 }
 371
 372 /* Protected methods */
 373 nsresult
 374 aaTreeView::postLoadHandler()
 375 {
 376   return NS_OK;
 377 }
 378
 379 PRInt32
 380 aaTreeView::getColumnIndex(nsITreeColumn *col, PRUint32 colPtrSz)
 381 {
 382   nsresult rv;
 383   NS_ENSURE_TRUE(col, -1);
 384
 385   PRInt32 ci;
 386   rv = col->GetIndex(&ci);
 387   NS_ENSURE_SUCCESS(rv, -1);
 388
 389   NS_ENSURE_TRUE(ci >= 0 && ci < (PRInt32) (colPtrSz / sizeof(int *)), -1);
 390
 391   return ci;
 392 }