#include <map>
#include <iostream>
#include <pqxx/connection>
#include <pqxx/transaction>
#include <pqxx/result>
#include <Wt/WMenu>
#include <Wt/WStackedWidget>
#include <Wt/WText>
#include "navbar.h"

using namespace pqxx;

void navbar::get_elements(const char* pg_con)
{
  result R;
  connection *conn = new connection(pg_con);
  work T(*conn);
  R = T.exec("SELECT name,link,id,parent_id FROM navbar");
  T.commit();
  int arr;
  Menu_data[][4] =0;
  for(result::const_iterator query = R.begin(); query != R.end(); ++query){
      Menu_data[arr][1]= query["id"].as<std::string>();
      Menu_data[arr][2]= query["name"].as<std::string>();
      Menu_data[arr][3]= query["link"].as<std::string>();
      Menu_data[arr][4]= query["parent_id"].as<std::string>();
      arr++;
  }
  conn->disconnect();
  delete conn;
  return true;
}

bool navbar::create_navbar(Wt::WStackedWidget *contents,const char* m_type)
{
  if (get_elements("host=localhost user=tuxist password=test dbname=tuxist")!=true){
    return false;
  }

  Wt::WMenu *menu;
  if(m_type=="Vertical"){
    menu = new Wt::WMenu(contents, Wt::Vertical);
  }else{
    menu = new Wt::WMenu(contents, Wt::Horizontal);
  }
    for(int pos = 0; pos <= sizeof(Menu_data)/sizeof(int); pos++){
      if(Menu_data[pos][4]=="0"){
        menu->addItem(Menu_data[pos][2],new Wt::WText(("<a href=\"",Menu_data[pos][3],"\" >",Menu_data[pos][2],"</a>")));
      }
    }
  contents ->addWidget(menu);
  return true;
}