#include "qsmb.h"
#include <qpushbutton.h>
#include <qpe/qpeapplication.h>

#include <string.h>
#include <qstring.h>
#include <qstringlist.h>
#include <qdir.h>
#include <qfileinfo.h>

#include <qpe/process.h>
#include <qlabel.h>
#include <qlineedit.h>
#include <qcombobox.h>
#include <qstringlist.h>
#include <qcheckbox.h>
#include <qtextview.h>
#include <qmessagebox.h>
#include <qtextstream.h>


#include <pthread.h>
#include <signal.h>
#include <ctype.h>


#include <netinet/in.h>
#include <arpa/inet.h>

#include <rpc/clnt.h>


#include <opie2/odebug.h>
using namespace Opie::Core;


Qsmb::Qsmb( QWidget* parent,  const char* name, WFlags fl )
   : FormQPESMBBase( parent, name, fl )
{
   connect(CBHost, SIGNAL(activated(int)), this, SLOT(hostSelected(int)));
   connect(DoItBtn, SIGNAL(clicked()), this, SLOT(DoItClicked()));
   connect(BtnScan, SIGNAL(clicked()), this, SLOT(scanClicked()));
   connect(BtnClear, SIGNAL(clicked()), this, SLOT(clear()));

   mountpt->insertItem("/mnt/samba1",-1);
   mountpt->insertItem("/mnt/samba2",-1);
   mountpt->insertItem("/mnt/samba3",-1);

   top_element = NULL;
   scanning = false;
}

Qsmb::~Qsmb()
{
}

void Qsmb::clear() 
{
   if (scanning) return;
   ListViewScan->clear();	
   TextViewOutput->setText("");
   top_element = NULL;	
}

void Qsmb::scanClicked() 
{
   if (scanning) return;
   pthread_create(&tpid, NULL, runit, (void *)this);
}

void Qsmb::DoItClicked() 
{

   if(! ListViewScan->selectedItem()) {
      QMessageBox::warning(this, tr("Error"),tr("<p>No share selected!</p>"));
      return;
   }
   if (scanning) return;
   pthread_create(&tpid, NULL, runitm, (void *)this);
}

void* runit(void* arg) 
{
   Qsmb* caller = (Qsmb*)arg;
   caller->scan();
   return(0);
}

void* runitm(void* arg) 
{
   Qsmb* caller = (Qsmb*)arg;
   caller->DoIt();
   return(0);
}

void Qsmb::scan() 
{
   int i;

   if (scanning) return;
   scanning = true;

   QString match;
   QString cmd;
   LScan->setText("Scanning...");
   qApp->processEvents();

   sockaddr_in my_addr;
   get_myaddress( &my_addr);

   QString ip = inet_ntoa( my_addr.sin_addr);
   owarn << "IP Address : " << ip<< oendl;

   match = ip.left(5);

   QStringList ccmd;
   TextViewOutput->append("smbfind");

   QFile lmhosts("/etc/samba/lmhosts");
   QTextStream lms(&lmhosts);
   lmhosts.open(IO_WriteOnly);
   lms << "127.0.0.1 localhost\n";
    
   /* parse output and display in ListViewScan */
   ccmd = "smbfind";
   owarn <<"cmd: " << ccmd << oendl;
	 runCommand(ccmd);

   QTextStream s(&out, IO_ReadOnly);

   while ( !s.atEnd() ) {
      QString ip_addr, host, output;
      QString tmp = s.readLine();
      bool ok;
      tmp.left(1).toInt( &ok, 10 );
      if(ok) {
         QStringList token = QStringList::split(' ',  tmp );
         ip_addr =  token[0];
         host = token[1];
         CBHost->insertItem( host, -1);
         lms <<  ip_addr+" "+host+"\n";
      }
   }
   lmhosts.close();

   TextViewOutput->append("\n\n============================================\n");
   LScan->setText("");
   scanning = false;
}

void Qsmb::hostSelected(int index)
{
   owarn << "hostSelected" << oendl;
   int  i;

   QListViewItem *element;
   QListViewItem *parent;

   QString text = CBHost->currentText();

   if (scanning) return;
   scanning = true;

   QString cmd;
   char result[256];

   FILE *pipe;

   LScan->setText("Scanning...");

   if(username->text().isEmpty())
      cmd = "/usr/bin/smbclient -L //"+CBHost->currentText()+" -N 2>&1 |grep Disk";
   else
      cmd = "/usr/bin/smbclient -L //"+CBHost->currentText()+" -N -U "+username->text()+"\%"+password->text()+" 2>&1 |grep Disk";

//    for(i = 0; i < 512; i++) {
//       if(cmd[i]==':') {
//          cmd[i]='%';
//          break;
//       }
//        if(cmd[i]=='\0')
//           break;
//     }

   owarn << "i="<< index << "cmd:" <<  cmd << oendl;

   TextViewOutput->append(cmd);

   /* run smbclient & read output */
   if ((pipe = popen(cmd.latin1(), "r")) == NULL) {
      snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
//      cmd = "Error: Can't run "+cmd;
      TextViewOutput->append(result);
      return;
   }

   /* parse output and display in ListViewScan */
   while(fgets(result, 256, pipe) != NULL) {
      /* put result into TextViewOutput */
      TextViewOutput->append(result);

      if( strchr(result, '$') == NULL ) { 
         char share[256], *ptr1;

         strcpy(share,result);
         ptr1 = strchr(share,' ');
         share[ptr1 - share]='\0';

         owarn<< "add share: " << share << oendl;

         if(top_element != NULL) {
            bool found = false;
            element = top_element;

            while(element != NULL && !found) {
               if(strcmp( element->text(0).ascii(), share)==0) {
                  parent = element;
                  found = true;
               }
               element = element->nextSibling();
            }

            if(!found) {
               element = new QListViewItem(ListViewScan,share);
               element->setOpen(true);
               parent=element;
            }
         } else {
            element = new QListViewItem(ListViewScan,share);
            element->setOpen(true);
            top_element = element;
            parent = element;
         }
      }
   }

   TextViewOutput->append("\n\n============================================\n");
   LScan->setText("");
   scanning = false;
}

void Qsmb::DoIt()
{
   QListViewItem *element;
   element = ListViewScan->selectedItem();
   if(!element) {
      QMessageBox::warning(this, tr("Error!!"),tr("<p><b>No</b> share selected!!</p>"));
      return;
   }

   if (scanning) return;
   scanning = true;

   int i;
   bool err = false;

   char share[512];
   QString cmd;
   QString cmd2;
   char result[256];
//   QString result;
   QString text = mountpt->currentText();

   FILE *pipe,*pipe2;

   LScan->setText("Mounting...");
   qApp->processEvents();


   if(! QFileInfo(text).exists()) {
//       /* make sure mount exists! */

//       cmd = "mkdir -p "+ text;
//       owarn<<"cmd: "<< cmd << oendl;
//       if ((pipe2 = popen(cmd.latin1(), "r")) == NULL)  {
//          snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
//          TextViewOutput->append(result);
//          return;
//       }
//       while(fgets(result, 256, pipe2) != NULL) {
//          /* put result into TextViewOutput */
//          TextViewOutput->append(result);
//       }
      QStringList ccmd;
      ccmd << "mkdir";
      ccmd << "-p";
      ccmd << text;

      owarn<<"cmd: "<< ccmd << oendl;
      runCommand(ccmd);
   }


   strcpy(share,(const char *)element->text(0));

   for(i = 0; i < 256; i++) {
      if(isalpha( share[i]))  {
         strcpy( share, share + i);
         break;
      }
   }

   cmd = "/usr/bin/smbmount  //"+CBHost->currentText()+"/"+share+" "+mountpt->currentText()+" -o username="+username->text()+",password="+password->text();

   owarn << "cmd: " << cmd << oendl;
   TextViewOutput->append(cmd.latin1());


   if(onbootBtn->isChecked()) {
      owarn << "Saving Setting permanently..." << oendl;
      cmd2 = "echo '" + cmd + "'>/opt/QtPalmtop/etc/samba.env";
      /* run command & read output */
      if ((pipe = popen(cmd2.latin1(), "r")) == NULL) {
         snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
         return;
      }
      /* parse output and display in ListViewScan */
      while(fgets(result, 256, pipe) != NULL) {
      }
   }
	

   /* run command & read output */
   if ((pipe = popen(cmd.latin1(), "r")) == NULL) {
      snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
      TextViewOutput->append(result);
      return;
   }

   /* parse output and display in ListViewScan */
   while(fgets(result, 256, pipe) != NULL) {
      /* put result into TextViewOutput */
      TextViewOutput->append(result);
   }

   
   TextViewOutput->append("\n\n================CheckMounts==================\n");
   LScan->setText("");

   cmd = "/bin/mount 2>&1";
   owarn << "cmd: " << cmd << oendl;

   if ((pipe2 = popen(cmd.latin1(), "r")) == NULL) {

      snprintf(result, 256, "Error: Can't run %s", cmd.latin1());
//         result = "Error: Can't run "+ cmd;

      TextViewOutput->append(result);
      return;
   }
   /* parse output and display in ListViewScan */
   while(fgets(result, 256, pipe2) != NULL) {
      /* put result into TextViewOutput */
      TextViewOutput->append(result);
   }

   TextViewOutput->append("\n\n============================================\n");
   scanning = false;
}

bool Qsmb::runCommand(const QStringList & command) {
   owarn << "runCommand " << command.join(" ") << oendl;
   out = "";
   Process ipkg_status( command);
   bool r = ipkg_status.exec("",out);

   if(!r) {
      QMessageBox::warning(this, tr("Error!!"),tr("<p>"+out+"</p>"));
   }

   owarn << "Output " << out << oendl;
   TextViewOutput->append(out.latin1());
   return r;
}