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 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 <abstract/aacore.h>
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsStringAPI.h"
  26 #include "nsEmbedString.h"
  27 #include "nsIDOMElement.h"
  28 #include "nsIDOMNode.h"
  29 #include "nsIDOMDocument.h"
  30 #include "unstable/nsIDOMXULElement.h"
  31 #include "nsIInterfaceRequestorUtils.h"
  32 #include "nsISHistory.h"
  33 #include "nsServiceManagerUtils.h"
  34 #include "nsIObserverService.h"
  35
  36 /* Unfrozen API */
  37 #include "unstable/nsITreeSelection.h"
  38 #include "unstable/nsITreeColumns.h"
  39 #include "unstable/nsITreeBoxObject.h"
  40 #include "unstable/nsIArray.h"
  41 #include "unstable/nsArrayUtils.h"
  42
  43 /* Project includes */
  44 #include <abstract/storage/aaISession.h>
  45 #include <abstract/storage/aaILoadQuery.h>
  46 #include "aaTreeView.h"
  47
  48 aaTreeView::aaTreeView()
  49   :mFlags(0)
  50 {
  51 }
  52
  53 aaTreeView::~aaTreeView()
  54 {
  55 }
  56
  57 NS_IMPL_ISUPPORTS2(aaTreeView,
  58                    aaIDataTreeView,
  59                    nsITreeView)
  60
  61 NS_IMETHODIMP
  62 aaTreeView::GetRowCount(PRInt32 *aRowCount)
  63 {
  64   NS_ENSURE_ARG_POINTER(aRowCount);
  65   *aRowCount = 0;
  66   if (! mDataSet)
  67     return NS_OK;
  68
  69   return mDataSet->GetLength((PRUint32 *) aRowCount);
  70 }
  71
  72 NS_IMETHODIMP
  73 aaTreeView::GetSelection(nsITreeSelection * *aSelection)
  74 {
  75   NS_ENSURE_ARG_POINTER( aSelection );
  76   NS_IF_ADDREF(*aSelection = mSelection);
  77   return NS_OK;
  78 }
  79 NS_IMETHODIMP
  80 aaTreeView::SetSelection(nsITreeSelection * aSelection)
  81 {
  82   mSelection = aSelection;
  83
  84   return NS_OK;
  85 }
  86
  87 NS_IMETHODIMP
  88 aaTreeView::GetRowProperties(PRInt32 index, nsISupportsArray *properties)
  89 {
  90     return NS_ERROR_NOT_IMPLEMENTED;
  91 }
  92
  93 NS_IMETHODIMP
  94 aaTreeView::GetCellProperties(PRInt32 row, nsITreeColumn *col, nsISupportsArray *properties)
  95 {
  96   return NS_ERROR_NOT_IMPLEMENTED;
  97 }
  98
  99 NS_IMETHODIMP
 100 aaTreeView::GetColumnProperties(nsITreeColumn *col, nsISupportsArray *properties)
 101 {
 102     return NS_ERROR_NOT_IMPLEMENTED;
 103 }
 104
 105 NS_IMETHODIMP
 106 aaTreeView::IsContainer(PRInt32 index, PRBool *_retval)
 107 {
 108   NS_ENSURE_ARG_POINTER(_retval);
 109   *_retval = PR_FALSE;
 110   return NS_OK;
 111 }
 112
 113 NS_IMETHODIMP
 114 aaTreeView::IsContainerOpen(PRInt32 index, PRBool *_retval)
 115 {
 116     return NS_ERROR_NOT_IMPLEMENTED;
 117 }
 118
 119 NS_IMETHODIMP
 120 aaTreeView::IsContainerEmpty(PRInt32 index, PRBool *_retval)
 121 {
 122     return NS_ERROR_NOT_IMPLEMENTED;
 123 }
 124
 125 NS_IMETHODIMP
 126 aaTreeView::IsSeparator(PRInt32 index, PRBool *_retval)
 127 {
 128   NS_ENSURE_ARG_POINTER(_retval);
 129   *_retval = PR_FALSE;
 130   return NS_OK;
 131 }
 132
 133 NS_IMETHODIMP
 134 aaTreeView::IsSorted(PRBool *_retval)
 135 {
 136     return NS_ERROR_NOT_IMPLEMENTED;
 137 }
 138
 139 NS_IMETHODIMP
 140 aaTreeView::CanDrop(PRInt32 index, PRInt32 orientation, PRBool *_retval)
 141 {
 142     return NS_ERROR_NOT_IMPLEMENTED;
 143 }
 144
 145 NS_IMETHODIMP
 146 aaTreeView::Drop(PRInt32 row, PRInt32 orientation)
 147 {
 148     return NS_ERROR_NOT_IMPLEMENTED;
 149 }
 150
 151 NS_IMETHODIMP
 152 aaTreeView::GetParentIndex(PRInt32 rowIndex, PRInt32 *_retval)
 153 {
 154     return NS_ERROR_NOT_IMPLEMENTED;
 155 }
 156
 157 NS_IMETHODIMP
 158 aaTreeView::HasNextSibling(PRInt32 rowIndex, PRInt32 afterIndex, PRBool *_retval)
 159 {
 160   NS_ENSURE_ARG_POINTER(_retval);
 161   nsresult rv;
 162   PRInt32 count;
 163   rv = GetRowCount(&count);
 164   NS_ENSURE_SUCCESS(rv, rv);
 165   if (rowIndex == afterIndex && rowIndex < count) {
 166     *_retval = PR_TRUE;
 167   } else {
 168     *_retval = PR_FALSE;
 169   }
 170   return NS_OK;
 171 }
 172
 173 NS_IMETHODIMP
 174 aaTreeView::GetLevel(PRInt32 index, PRInt32 *_retval)
 175 {
 176   NS_ENSURE_ARG_POINTER(_retval);
 177   *_retval = 0;
 178   return NS_OK;
 179 }
 180
 181 NS_IMETHODIMP
 182 aaTreeView::GetImageSrc(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 183 {
 184     return NS_ERROR_NOT_IMPLEMENTED;
 185 }
 186
 187 NS_IMETHODIMP
 188 aaTreeView::GetProgressMode(PRInt32 row, nsITreeColumn *col, PRInt32 *_retval)
 189 {
 190     return NS_ERROR_NOT_IMPLEMENTED;
 191 }
 192
 193 NS_IMETHODIMP
 194 aaTreeView::GetCellValue(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 195 {
 196     return NS_ERROR_NOT_IMPLEMENTED;
 197 }
 198
 199 NS_IMETHODIMP
 200 aaTreeView::GetCellText(PRInt32 row, nsITreeColumn *col, nsAString & _retval)
 201 {
 202   /**
 203    * There may be a bug in <tree> implementation. This function is
 204    * occasionally called after mTree is detached by SetTree(nsnull);
 205    */
 206   NS_ENSURE_TRUE(mDataSet, NS_ERROR_NOT_INITIALIZED);
 207   NS_ENSURE_TRUE(row >= 0, NS_ERROR_INVALID_ARG);
 208   return doGetCellText(row, col, _retval);
 209 }
 210
 211 NS_IMETHODIMP
 212 aaTreeView::SetTree(nsITreeBoxObject *tree)
 213 {
 214   mTree = tree;
 215   return NS_OK;
 216 }
 217
 218 NS_IMETHODIMP
 219 aaTreeView::ToggleOpenState(PRInt32 index)
 220 {
 221     return NS_ERROR_NOT_IMPLEMENTED;
 222 }
 223
 224 NS_IMETHODIMP
 225 aaTreeView::CycleHeader(nsITreeColumn *col)
 226 {
 227     return NS_ERROR_NOT_IMPLEMENTED;
 228 }
 229
 230 NS_IMETHODIMP
 231 aaTreeView::SelectionChanged()
 232 {
 233   return NS_OK;
 234 }
 235
 236 NS_IMETHODIMP
 237 aaTreeView::CycleCell(PRInt32 row, nsITreeColumn *col)
 238 {
 239     return NS_ERROR_NOT_IMPLEMENTED;
 240 }
 241
 242 NS_IMETHODIMP
 243 aaTreeView::IsEditable(PRInt32 row, nsITreeColumn *col, PRBool *_retval)
 244 {
 245     return NS_ERROR_NOT_IMPLEMENTED;
 246 }
 247
 248 NS_IMETHODIMP
 249 aaTreeView::IsSelectable(PRInt32 row, nsITreeColumn *col, PRBool *_retval)
 250 {
 251   NS_ENSURE_ARG_POINTER(_retval);
 252   *_retval = PR_TRUE;
 253   return NS_OK;
 254 }
 255
 256 NS_IMETHODIMP
 257 aaTreeView::SetCellValue(PRInt32 row, nsITreeColumn *col, const nsAString & value)
 258 {
 259     return NS_ERROR_NOT_IMPLEMENTED;
 260 }
 261
 262 NS_IMETHODIMP
 263 aaTreeView::SetCellText(PRInt32 row, nsITreeColumn *col, const nsAString & value)
 264 {
 265     return NS_ERROR_NOT_IMPLEMENTED;
 266 }
 267
 268 NS_IMETHODIMP
 269 aaTreeView::PerformAction(const PRUnichar *action)
 270 {
 271     return NS_ERROR_NOT_IMPLEMENTED;
 272 }
 273
 274 NS_IMETHODIMP
 275 aaTreeView::PerformActionOnRow(const PRUnichar *action, PRInt32 row)
 276 {
 277     return NS_ERROR_NOT_IMPLEMENTED;
 278 }
 279
 280 NS_IMETHODIMP
 281 aaTreeView::PerformActionOnCell(const PRUnichar *action, PRInt32 row, nsITreeColumn *col)
 282 {
 283     return NS_ERROR_NOT_IMPLEMENTED;
 284 }
 285
 286 /* aaIDataTreeView */
 287 NS_IMETHODIMP
 288 aaTreeView::GetDataSet(aaILoadQuery ** aDataSet)
 289 {
 290   NS_ENSURE_ARG_POINTER(aDataSet);
 291   NS_IF_ADDREF(*aDataSet = mDataSet);
 292   return NS_OK;
 293 }
 294 NS_IMETHODIMP
 295 aaTreeView::SetDataSet(aaILoadQuery * aDataSet)
 296 {
 297   mDataSet = aDataSet;
 298   return NS_OK;
 299 }
 300
 301 NS_IMETHODIMP
 302 aaTreeView::Update()
 303 {
 304   nsresult rv;
 305   NS_ENSURE_TRUE(mDataSet, NS_ERROR_NOT_INITIALIZED);
 306
 307   if (mTree) {
 308     PRInt32 oldCount;
 309     GetRowCount(&oldCount);
 310     mTree->RowCountChanged(0, - oldCount);
 311   }
 312
 313   rv = mDataSet->Load();
 314   NS_ENSURE_SUCCESS(rv, rv);
 315
 316   if (mTree) {
 317     PRInt32 newCount;
 318     GetRowCount(&newCount);
 319     mTree->RowCountChanged(0, newCount);
 320   }
 321
 322   nsCOMPtr<nsIObserverService> broadcaster = do_GetService(
 323       "@mozilla.org/observer-service;1");
 324   NS_ENSURE_TRUE(broadcaster, NS_OK);
 325   broadcaster->NotifyObservers(this, "tree-view-load", nsnull);
 326
 327   return NS_OK;
 328 }
 329
 330 NS_IMETHODIMP
 331 aaTreeView::GetFlags(PRUint32 *aFlags)
 332 {
 333   NS_ENSURE_ARG_POINTER(aFlags);
 334   *aFlags = mFlags;
 335   return NS_OK;
 336 }
 337 NS_IMETHODIMP
 338 aaTreeView::SetFlags(PRUint32 aFlags)
 339 {
 340   PRBool needUpdate = (aFlags != mFlags);
 341   mFlags = aFlags;
 342   if (NS_LIKELY( needUpdate && mTree )) {
 343     mTree->Invalidate();
 344     nsCOMPtr<nsIObserverService> broadcaster = do_GetService(
 345         "@mozilla.org/observer-service;1");
 346     NS_ENSURE_TRUE(broadcaster, NS_OK);
 347     broadcaster->NotifyObservers(this, "tree-flags-change", nsnull);
 348   }
 349   return NS_OK;
 350 }
 351
 352 /* Protected methods */
 353 PRInt32
 354 aaTreeView::getColumnIndex(nsITreeColumn *col, PRUint32 colPtrSz)
 355 {
 356   nsresult rv;
 357   NS_ENSURE_TRUE(col, -1);
 358
 359   PRInt32 ci;
 360   rv = col->GetIndex(&ci);
 361   NS_ENSURE_SUCCESS(rv, -1);
 362
 363   NS_ENSURE_TRUE(ci >= 0 && ci < (PRInt32) (colPtrSz / sizeof(int *)), -1);
 364
 365   return ci;
 366 }