source: gsdl/trunk/src/recpt/dynamicclassifieraction.cpp@ 15940

/**********************************************************************
 *
 * dynamicclassifieraction.cpp -- 
 * Copyright (C) 2008  DL Consulting Ltd
 *
 * A component of the Greenstone digital library software
 * from the New Zealand Digital Library Project at the
 * University of Waikato, New Zealand.
 *
 * 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.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 *********************************************************************/

#include "dynamicclassifieraction.h"
#include "recptprototools.h"


dynamicclassifieraction::dynamicclassifieraction ()
{
  recpt = NULL;

  cgiarginfo arg_ainfo;
  arg_ainfo.shortname = "dcl";
  arg_ainfo.longname = "dynamic classifier ID";
  arg_ainfo.multiplechar = true;
  arg_ainfo.defaultstatus = cgiarginfo::weak;
  arg_ainfo.argdefault = "";
  arg_ainfo.savedarginfo = cgiarginfo::must;
  argsinfo.addarginfo (NULL, arg_ainfo);

  arg_ainfo.shortname = "dcn";
  arg_ainfo.longname = "dynamic classifier node";
  arg_ainfo.multiplechar = true;
  arg_ainfo.defaultstatus = cgiarginfo::weak;
  arg_ainfo.argdefault = "";
  arg_ainfo.savedarginfo = cgiarginfo::must;
  argsinfo.addarginfo (NULL, arg_ainfo);
}


dynamicclassifieraction::~dynamicclassifieraction()
{
}


bool dynamicclassifieraction::check_cgiargs (cgiargsinfoclass &argsinfo, cgiargsclass &args, 
                         recptprotolistclass *protos, ostream &logout)
{
  return true;
}


void dynamicclassifieraction::get_cgihead_info (cgiargsclass &args, recptprotolistclass *protos,
                        response_t &response,text_t &response_data, 
                        ostream &logout)
{
  response = content;
  response_data = "text/html";
}


// define all the macros which might be used by other actions to produce pages. 
void dynamicclassifieraction::define_external_macros (displayclass &disp, cgiargsclass &args, 
                              recptprotolistclass *protos, ostream &logout)
{
  // A valid collection server is vital
  recptproto *collectproto = protos->getrecptproto (args["c"], logout);
  if (collectproto == NULL)
  {
    logout << "dynamicclassifieraction::define_external_macros called with NULL collectproto\n";
    return;
  }

  // Define _dynamicclassifiernavbarentries_ to add buttons to the navigation bar for the dynamic classifiers
  text_t navigation_bar_entries = "";
  ColInfoResponse_t *cinfo = recpt->get_collectinfo_ptr (collectproto, args["c"], logout);
  text_tmap::iterator dynamic_classifier_iterator = cinfo->dynamic_classifiers.begin();
  while (dynamic_classifier_iterator != cinfo->dynamic_classifiers.end())
  {
    text_t dynamic_classifier_id = (*dynamic_classifier_iterator).first;
    navigation_bar_entries += "_navbarspacer_";
    navigation_bar_entries += "_navtab_(_gwcgi_?c=" + args["c"] + "&amp;a=dc&amp;dcl=" + dynamic_classifier_id + "," + dynamic_classifier_id;
    if (args["a"] == "dc" && args["dcl"] == dynamic_classifier_id)
    {
      navigation_bar_entries += ",selected";
    }
    navigation_bar_entries += ")";
    dynamic_classifier_iterator++;
  }

  disp.setmacro("dynamicclassifiernavbarentries", displayclass::defaultpackage, navigation_bar_entries);
}


// define all the macros which are related to pages generated
// by this action. we also load up the formatinfo structure 
// here (it's used in do_action as well as here)
void dynamicclassifieraction::define_internal_macros (displayclass &disp, cgiargsclass &args, 
                              recptprotolistclass *protos, ostream &logout)
{
  // define_internal_macros sets the following macros:
}


bool dynamicclassifieraction::do_action(cgiargsclass &args, recptprotolistclass *protos, 
                    browsermapclass *browsers, displayclass &disp, 
                    outconvertclass &outconvert, ostream &textout, 
                    ostream &logout)
{
  // A valid collection server is vital
  recptproto *collectproto = protos->getrecptproto (args["c"], logout);
  if (collectproto == NULL)
  {
    logout << "dynamicclassifieraction::do_action called with NULL collectproto\n";
    return false;
  }

  textout << outconvert << disp << "_document:header_\n";
  textout << outconvert << disp << "_document:content_\n";

  // Check a dynamic classifier ID has been specified
  text_t arg_dcl = args["dcl"];
  if (arg_dcl.empty())
  {
    textout << outconvert << disp << "Error: Missing dcl argument.\n";
    textout << outconvert << disp << "_document:footer_\n";
    return true;
  }

  // Check the dynamic classifier ID is valid (ie. there is an entry in the collect.cfg file for it)
  ColInfoResponse_t *cinfo = recpt->get_collectinfo_ptr (collectproto, args["c"], logout);
  if (cinfo->dynamic_classifiers.find(arg_dcl) == cinfo->dynamic_classifiers.end())
  {
    textout << outconvert << disp << "Error: Invalid dcl value \"" << arg_dcl << "\".\n";
    textout << outconvert << disp << "_document:footer_\n";
    return true;
  }

  text_t metadata_element_name = cinfo->dynamic_classifiers[arg_dcl];
  text_t classifier_type = "VList";
  browserclass *bptr = browsers->getbrowser (classifier_type);

  // Get the formatstring if there is one, or use the browser's default otherwise
  text_t formatstring;
  if (!get_formatstring (arg_dcl, classifier_type, cinfo->format, formatstring))
  {
    formatstring = bptr->get_default_formatstring();
  }
  format_t *formatlistptr = new format_t();
  text_tset metadata;
  bool getParents = false;
  parse_formatstring (formatstring, formatlistptr, metadata, getParents);
  bool use_table = is_table_content (formatlistptr);

  // Check if a dynamic classifier node has been specified
  text_t arg_dcn = args["dcn"];
  if (arg_dcn.empty())
  {
    // No, so display the top-level classifier page, containing all the metadata values for the specified element
    FilterResponse_t metadata_values_response;
    get_metadata_values (metadata_element_name, args["c"], collectproto, metadata_values_response, logout);

    // Deal with hierarchical metadata values
    map<text_t, int, lttext_t> classifier_nodes;
    ResultDocInfo_tarray::iterator metadata_value_iterator = metadata_values_response.docInfo.begin();
    while (metadata_value_iterator != metadata_values_response.docInfo.end())
    {
      text_t metadata_value = (*metadata_value_iterator).OID;

      // Is this metadata value hierarchical?
      text_t::iterator hierarchy_split_position = findchar(metadata_value.begin(), metadata_value.end(), '|');
      if (hierarchy_split_position != metadata_value.end())
      {
    // Yes, so split off the first part of the hierarchy for the classifier node
    metadata_value = substr(metadata_value.begin(), hierarchy_split_position);
      }

      // Create a node for this metadata value if we haven't seen it before
      if (classifier_nodes.find(metadata_value) == classifier_nodes.end())
      {
    classifier_nodes[metadata_value] = 0;
      }

      // Increment the occurrence count
      classifier_nodes[metadata_value] += (*metadata_value_iterator).result_num;

      metadata_value_iterator++;
    }

    // Create the structure containing the classifer nodes to pass to output_section_group(), with the right info
    FilterResponse_t classifier_nodes_response;
    map<text_t, int, lttext_t>::iterator classifier_nodes_iterator = classifier_nodes.begin();
    while (classifier_nodes_iterator != classifier_nodes.end())
    {
      ResultDocInfo_t classifier_node;
      classifier_node.OID = (*classifier_nodes_iterator).first;

      // Add metadata necessary for output_section_group() to display the results as classifier nodes
      classifier_node.metadata["doctype"].values.push_back ("classify");
      classifier_node.metadata["haschildren"].values.push_back ("1");
      classifier_node.metadata["numleafdocs"].values.push_back ((*classifier_nodes_iterator).second);
      classifier_node.metadata["Title"].values.push_back ((*classifier_nodes_iterator).first);

      classifier_nodes_response.docInfo.push_back (classifier_node);
      classifier_nodes_iterator++;
    }

    // Display the classifier nodes
    bptr->output_section_group (classifier_nodes_response, args, args["c"], 0, formatlistptr, use_table, metadata, getParents, collectproto, disp, outconvert, textout, logout);
  }
  else
  {
    // Yes, so get all the documents that fall under this node
    text_t metadata_value = arg_dcn;
    FilterResponse_t document_OIDs_response;
    get_documents_with_metadata_value (metadata_element_name, metadata_value, "dls.Title", args["c"], collectproto, document_OIDs_response, logout);

    // Check the metadata value is valid
    if (document_OIDs_response.docInfo.empty())
    {
      textout << outconvert << disp << "Error: No documents have metadata value \"" << arg_dcn << "\".\n";
      textout << outconvert << disp << "_document:footer_\n";
      return true;
    }

    // Make an array of matching document OIDs
    text_tarray document_OIDs;
    ResultDocInfo_tarray::iterator document_OID_iterator = document_OIDs_response.docInfo.begin();
    while (document_OID_iterator != document_OIDs_response.docInfo.end())
    {
      document_OIDs.push_back ((*document_OID_iterator).OID);
      document_OID_iterator++;
    }

    // Display the classifier node
    ResultDocInfo_t classifier_node;
    classifier_node.OID = arg_dcn;
    classifier_node.metadata["doctype"].values.push_back("classify");
    classifier_node.metadata["haschildren"].values.push_back("1");
    classifier_node.metadata["numleafdocs"].values.push_back(document_OIDs.size());
    classifier_node.metadata["Title"].values.push_back(classifier_node.OID);
    bptr->output_section_group (classifier_node, args, args["c"], 0, formatlistptr, use_table, metadata, getParents, collectproto, disp, outconvert, textout, logout);

    // Request the necessary metadata for the documents under this node
    FilterResponse_t document_info_response;
    get_info (document_OIDs, args["c"], args["l"], metadata, getParents, collectproto, document_info_response, logout);

    // Display the document nodes
    bptr->output_section_group (document_info_response, args, args["c"], 1, formatlistptr, use_table, metadata, getParents, collectproto, disp, outconvert, textout, logout);
  }

  textout << outconvert << disp << "_document:footer_\n";
  return true;
}