/*
 * Copyright (c) 2002 Robert Collins.
 *
 *     This program is free software; you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation; either version 2 of the License, or
 *     (at your option) any later version.
 *
 *     A copy of the GNU General Public License can be found at
 *     http://www.gnu.org/
 *
 * Written by Robert Collins <robertc@hotmail.com>
 *
 */

#include "PickPackageLine.h"
#include "PickView.h"
#include "package_db.h"

const std::string
PickPackageLine::get_text(int col_num) const
{
  if (col_num == pkgname_col)
    {
      return pkg.name;
    }
  else if (col_num == current_col)
    {
      return pkg.installed.Canonical_version ();
    }
  else if (col_num == new_col)
    {
      return pkg.action_caption ();
    }
  else if (col_num == bintick_col)
    {
      const char *bintick = "?";
      if (/* uninstall or skip */ !pkg.desired ||
          /* current version */ pkg.desired == pkg.installed ||
          /* no source */ !pkg.desired.accessible())
        bintick = "n/a";
      else if (pkg.picked())
        bintick = "yes";
      else
        bintick = "no";

      return bintick;
    }
  else if (col_num == srctick_col)
    {
      const char *srctick = "?";
      if ( /* uninstall */ !pkg.desired ||
           /* when no source mirror available */
           !pkg.desired.sourcePackage().accessible())
        srctick = "n/a";
      else if (pkg.srcpicked())
        srctick = "yes";
      else
        srctick = "no";

      return srctick;
    }
  else if (col_num == cat_col)
    {
      return pkg.getReadableCategoryList();
    }
  else if (col_num == size_col)
    {
      int sz = 0;
      packageversion picked;

      /* Find the size of the package.  If user has chosen to upgrade/downgrade
         the package, use that version.  Otherwise use the currently installed
         version, or if not installed then use the version that would be chosen
         based on the current trust level (curr/prev/test).  */
      if (pkg.desired)
        picked = pkg.desired;
      else if (pkg.installed)
        picked = pkg.installed;
      else
        picked = pkg.trustp (false, theView.deftrust);

      /* Include the size of the binary package, and if selected, the source
         package as well.  */
      sz += picked.source()->size;
      if (pkg.srcpicked())
        sz += picked.sourcePackage().source()->size;

      /* If size still 0, size must be unknown.  */
      std::string size = (sz == 0) ? "?" : format_1000s((sz+1023)/1024) + "k";

      return size;
    }
  else if (col_num == pkg_col)
    {
      return pkg.SDesc();
    }

  return "unknown";
}

int
PickPackageLine::do_action(int col_num)
{
  if (col_num == new_col)
    {
      pkg.set_action (theView.deftrust);
      return 1;
    }

  if (col_num == bintick_col)
    {
      if (pkg.desired.accessible ())
        pkg.pick (!pkg.picked ());
    }
  else if (col_num == srctick_col)
    {
      if (pkg.desired.sourcePackage ().accessible ())
        pkg.srcpick (!pkg.srcpicked ());
    }

  /* Unchecking binary while source is unchecked or vice versa is equivalent to
     uninstalling.  It's essential to set desired correctly, otherwise the
     package gets uninstalled without visual feedback to the user.  The package
     will not even show up in the "Pending" view! */
  if ((col_num == bintick_col) || (col_num == srctick_col))
    {
      if (!pkg.picked () && !pkg.srcpicked ())
        pkg.desired = packageversion ();

      return 1;
    }

  return 0;
}