view/aaTreeViewFlow.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
  33 /* Project includes */
  34 #include "aaIEntity.h"
  35 #include "aaIFlow.h"
  36 #include "aaBaseKeys.h"
  37 #include "aaBaseLoaders.h"
  38 #include "aaISqlRequest.h"
  39 #include "aaTreeViewFlow.h"
  40
  41 aaTreeViewFlow::aaTreeViewFlow()
  42 {
  43   mDataLoader = do_CreateInstance(AA_LOADFLOW_CONTRACT);
  44 }
  45
  46 aaTreeViewFlow::~aaTreeViewFlow()
  47 {
  48 }
  49
  50 NS_IMPL_ISUPPORTS_INHERITED0(aaTreeViewFlow, aaTreeView)
  51
  52 /* Private functions */
  53 nsresult
  54 aaTreeViewFlow::postLoadHandler()
  55 {
  56   if (!mFlags)
  57     return NS_OK;
  58
  59   nsresult rv;
  60   nsCOMPtr<aaILoadObserver> resultSet = do_QueryInterface(mDataSet, &rv);
  61   NS_ENSURE_SUCCESS(rv, rv);
  62
  63   nsCOMPtr<aaIFlow> pnl = do_CreateInstance(AA_INCOMEFLOW_CONTRACT, &rv);
  64   NS_ENSURE_SUCCESS(rv, rv);
  65
  66   rv = resultSet->ObserveLoad(pnl);
  67   NS_ENSURE_SUCCESS(rv, rv);
  68
  69   return NS_OK;
  70 }
  71
  72 nsresult
  73 aaTreeViewFlow::doGetCellText(PRInt32 row, nsITreeColumn *col,
  74     nsAString & _retval)
  75 {
  76   nsresult rv;
  77   NS_ENSURE_TRUE(col, NS_ERROR_INVALID_ARG);
  78
  79   PRInt32 ci;
  80   rv = col->GetIndex(&ci);
  81   NS_ENSURE_SUCCESS(rv, rv);
  82   _retval.Truncate();
  83   if (ci > 1) {
  84     return NS_OK;
  85   }
  86
  87   nsCOMPtr<aaIFlow> flow(do_QueryElementAt(mDataSet, row, &rv));
  88   NS_ENSURE_SUCCESS(rv, rv);
  89   if (! flow)
  90     return NS_OK;
  91
  92   switch (ci) {
  93   case 0:
  94     return flow->GetTag(_retval);
  95     break;
  96   case 1:
  97     {
  98       nsCOMPtr<aaIEntity> entity;
  99       flow->GetEntity(getter_AddRefs( entity ));
 100       if (! entity)
 101         return NS_OK;
 102       return entity->GetTag(_retval);
 103     }
 104     break;
 105   }
 106   return NS_OK;
 107 }
 108
 109 NS_IMETHODIMP
 110   aaTreeViewFlow::convertSortColumnToString(nsITreeColumn* col, nsACString& name)
 111 {
 112   nsresult rc = NS_OK;
 113   PRInt32 idx = -1;
 114
 115   NS_ENSURE_ARG_POINTER(col);
 116
 117   name.Truncate();
 118
 119   rc = col->GetIndex(&idx);
 120   NS_ENSURE_SUCCESS(rc, rc);
 121
 122   switch (idx) {
 123   case 0:
 124     name = NS_LITERAL_CSTRING("flow.tag");
 125     break;
 126   case 1:
 127     name = NS_LITERAL_CSTRING("flow.entity.tag");
 128     break;
 129   default:
 130     name.Truncate();
 131     return NS_ERROR_INVALID_ARG;
 132   }
 133
 134   return rc;
 135 }