view/aaTreeViewResource.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 "nsServiceManagerUtils.h"
  27 #include "nsStringAPI.h"
  28
  29 /* Unfrozen API */
  30 #include "nsIArray.h"
  31 #include "nsArrayUtils.h"
  32 #include "nsITreeColumns.h"
  33 #include "nsIStringBundle.h"
  34
  35 /* Project includes */
  36 #include "aaIResource.h"
  37 #include "aaISqlRequest.h"
  38 #include "aaBaseLoaders.h"
  39 #include "aaIOrderCondition.h"
  40 #include "aaConditionKeys.h"
  41 #include "aaISqlFilter.h"
  42 #include "aaTreeViewResource.h"
  43
  44 aaTreeViewResource::aaTreeViewResource()
  45 {
  46   mDataLoader = do_CreateInstance(AA_LOADRESOURCE_CONTRACT);
  47 }
  48
  49 aaTreeViewResource::~aaTreeViewResource()
  50 {
  51 }
  52
  53 NS_IMPL_ISUPPORTS_INHERITED0(aaTreeViewResource, aaTreeView)
  54
  55 /* Private functions */
  56 nsresult
  57 aaTreeViewResource::doGetCellText(PRInt32 row, nsITreeColumn *col,
  58     nsAString & _retval)
  59 {
  60   _retval.Truncate();
  61   nsresult rv;
  62   NS_ENSURE_TRUE(col, NS_ERROR_INVALID_ARG);
  63
  64   PRInt32 ci;
  65   rv = col->GetIndex(&ci);
  66   NS_ENSURE_SUCCESS(rv, rv);
  67   NS_ENSURE_SUCCESS(ci <= 1, NS_OK);
  68
  69   nsCOMPtr<aaIResource> resource(do_QueryElementAt(mDataSet, row, &rv));
  70   NS_ENSURE_SUCCESS(rv, rv);
  71   if (!resource)
  72     return NS_OK;
  73
  74   switch (ci) {
  75   case 0:
  76     return getTypeColumnText(resource, _retval);
  77   case 1:
  78     return resource->GetTag(_retval);
  79   default:
  80     break;
  81   }
  82
  83   return NS_OK;
  84 }
  85
  86 nsresult
  87 aaTreeViewResource::getTypeColumnText(aaIResource *aResource,
  88     nsAString & _retval)
  89 {
  90   nsresult rv;
  91   rv = ensureState();
  92   NS_ENSURE_SUCCESS(rv, rv);
  93
  94   nsAutoString w, key;
  95   switch (aResource->PickType()) {
  96   case aaIResource::TYPE_MONEY:
  97     key = NS_LITERAL_STRING("resource.type.money");
  98     break;
  99   case aaIResource::TYPE_ASSET:
 100     key = NS_LITERAL_STRING("resource.type.asset");
 101     break;
 102   default:
 103     break;
 104   }
 105
 106   rv = mBundle->GetStringFromName(key.get(), getter_Copies(w));
 107   NS_ENSURE_SUCCESS(rv, rv);
 108
 109   _retval.Assign(w);
 110   return NS_OK;
 111 }
 112
 113 nsresult
 114 aaTreeViewResource::ensureState()
 115 {
 116   nsresult rv;
 117   if (!mBundle) {
 118     nsCOMPtr<nsIStringBundleService> heap = do_GetService(
 119         NS_STRINGBUNDLE_CONTRACTID, &rv);
 120     NS_ENSURE_SUCCESS(rv, rv);
 121
 122     rv = heap->CreateBundle("chrome://abstract/locale/resource.properties",
 123         getter_AddRefs(mBundle));
 124     NS_ENSURE_SUCCESS(rv, rv);
 125   }
 126
 127   return NS_OK;
 128 }
 129
 130 NS_IMETHODIMP
 131   aaTreeViewResource::convertSortColumnToString(nsITreeColumn* col, nsACString& name)
 132 {
 133   nsresult rc = NS_OK;
 134   PRInt32 idx = -1;
 135
 136   NS_ENSURE_ARG_POINTER(col);
 137
 138   name.Truncate();
 139
 140   rc = col->GetIndex(&idx);
 141   NS_ENSURE_SUCCESS(rc, rc);
 142
 143   switch (idx) {
 144   case 0:
 145     name = NS_LITERAL_CSTRING("resource.type");
 146     break;
 147   case 1:
 148     name = NS_LITERAL_CSTRING("resource.tag");
 149     break;
 150   default:
 151     name.Truncate();
 152     return NS_ERROR_INVALID_ARG;
 153   }
 154   return rc;
 155 }