view/aaTreeViewEntity.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 "nsComponentManagerUtils.h"
  26 #include "nsStringAPI.h"
  27
  28 /* Unfrozen API */
  29 #include "nsIArray.h"
  30 #include "nsArrayUtils.h"
  31 #include "nsITreeColumns.h"
  32 #include "nsISupportsPrimitives.h"
  33
  34 /* Project includes */
  35 #include "aaIEntity.h"
  36 #include "aaISqlRequest.h"
  37 #include "aaBaseLoaders.h"
  38 #include "aaTreeViewEntity.h"
  39 #include "aaISqlFilter.h"
  40 #include "aaConditionKeys.h"
  41 #include "aaIOrderCondition.h"
  42
  43 #include "aaNotify.h"
  44
  45 #define AA_ENTITY_MSG_SORT_TAG "tv.sort.entity.tag"
  46
  47 aaTreeViewEntity::aaTreeViewEntity()
  48 {
  49   mDataLoader = do_CreateInstance(AA_LOADENTITY_CONTRACT);
  50 }
  51
  52 aaTreeViewEntity::~aaTreeViewEntity()
  53 {
  54 }
  55
  56 NS_IMPL_ISUPPORTS_INHERITED0(aaTreeViewEntity, aaTreeView)
  57
  58 /* Private functions */
  59 nsresult
  60 aaTreeViewEntity::doGetCellText(PRInt32 row, nsITreeColumn *col,
  61     nsAString & _retval)
  62 {
  63   nsresult rv;
  64   NS_ENSURE_TRUE(col, NS_ERROR_INVALID_ARG);
  65
  66   PRInt32 ci;
  67   rv = col->GetIndex(&ci);
  68   NS_ENSURE_SUCCESS(rv, rv);
  69   if (ci > 0) {
  70     _retval.Truncate();
  71     return NS_OK;
  72   }
  73
  74   nsCOMPtr<aaIEntity> entity(do_QueryElementAt(mDataSet, row, &rv));
  75   NS_ENSURE_SUCCESS(rv, rv);
  76   if (entity)
  77     return entity->GetTag(_retval);
  78
  79   _retval.Truncate();
  80   return NS_OK;
  81 }
  82
  83 NS_IMETHODIMP
  84   aaTreeViewEntity::convertSortColumnToString(nsITreeColumn* col, nsACString& name)
  85 {
  86   nsresult rc = NS_OK;
  87   PRInt32 idx = -1;
  88
  89   NS_ENSURE_ARG_POINTER(col);
  90
  91   name.Truncate();
  92
  93   rc = col->GetIndex(&idx);
  94   NS_ENSURE_SUCCESS(rc, rc);
  95
  96   switch (idx) {
  97   case 0:
  98     name = NS_LITERAL_CSTRING("entity.tag");
  99     break;
 100   default:
 101     name.Truncate();
 102     return NS_ERROR_INVALID_ARG;
 103   }
 104
 105   return rc;
 106 }
 107
 108 NS_IMETHODIMP
 109   aaTreeViewEntity::dataLoadSuccess()
 110 {
 111   return aaTreeView::dataLoadSuccess();
 112 }