// ffiledialog.cpp
// class FFileDialog

#include "ffiledialog.h"

//----------------------------------------------------------------------
static bool sortByName (const dir_entry &lhs, const dir_entry &rhs)
{
  // lhs < rhs
  return bool(strcasecmp(lhs.name, rhs.name) < 0);
}

//----------------------------------------------------------------------
static bool sortDirFirst (const dir_entry &lhs, const dir_entry &rhs)
{
  // sort directories first
  if ( lhs.type == DT_DIR && rhs.type != DT_DIR )
    return true;
  else
    return false;
}


//----------------------------------------------------------------------
// class FFileDialog
//----------------------------------------------------------------------

// constructors and destructor
//----------------------------------------------------------------------
FFileDialog::FFileDialog(FWidget* parent) : FDialog(parent)
{
  dlg_type = FFileDialog::Open;
  init();
}

//----------------------------------------------------------------------
FFileDialog::FFileDialog ( const FString& dirname,
                           const FString& filter,
                           DialogType type,
                           FWidget* parent ) : FDialog(parent)
{
  if ( dirname )
    setPath(dirname);
  if ( filter )
    setFilter(filter);
  dlg_type = type;
  init();
}

//----------------------------------------------------------------------
FFileDialog::~FFileDialog()  // destructor
{
  delete open;
  delete cancel;
  delete hidden;
  delete filebrowser;
  delete filename;
  clear();
}


// private methods of FFileDialog
//----------------------------------------------------------------------
void FFileDialog::init()
{
  int x, y;
  show_hidden = false;
  height = 15;
  width = 42;
  if ( width < 15 )
    width = 15;
  if ( width < 20 )
    width = 20;
  x = 1 + int((parentWidget()->getWidth()-width)/2);
  y = 1 + int((parentWidget()->getHeight()-height)/3);

  if ( dlg_type == FFileDialog::Save )
    FDialog::setText("Save file");
  else
    FDialog::setText("Open file");

  filename = new FLineEdit(this);
  filename->setLabelText("File&name:");
  filename->setText(filter_pattern);
  filename->setGeometry(11, 1, 28, 1);
  filename->setShadow();
  filename->setFocus();

  filebrowser = new FListBox(this);
  filebrowser->setGeometry(2, 3, 38, 6);
  printPath(directory);

  hidden = new FCheckBox("&hidden files", this);
  hidden->setGeometry(2, 10, 16, 1);

  cancel = new FButton("&Cancel", this);
  cancel->setGeometry(19, 10, 9, 1);
  cancel->setShadow();

  if ( dlg_type == FFileDialog::Save )
    open = new FButton("&Save",this);
  else
    open = new FButton("&Open",this);
  open->setGeometry(30, 10, 9, 1);
  open->setShadow();

  setGeometry (x, y, width, height);

  filename->addCallback
  (
    "activate",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processActivate),
    null
  );
  filebrowser->addCallback
  (
    "row-changed",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processRowChanged),
    null
  );
  filebrowser->addCallback
  (
    "clicked",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processClicked),
    null
  );
  hidden->addCallback
  (
    "toggled",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processShowHidden),
    null
  );
  cancel->addCallback
  (
    "clicked",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processCancel),
    null
  );
  open->addCallback
  (
    "clicked",
    this,
    reinterpret_cast<FWidget::FMemberCallback>(&FFileDialog::cb_processOpen),
    null
  );
  setModal();
  setTransparentShadow();

  readDir();
}

//----------------------------------------------------------------------
char* FFileDialog::getHomeDir()
{
  struct passwd* pwd;
  pwd = getpwuid( geteuid() );

  if ( pwd == 0 )
    return const_cast<char*>("");
  else
  {
    pwd = getpwnam(pwd->pw_name);

    if ( pwd == 0 )
      return const_cast<char*>("");
    else
      return pwd->pw_dir;
  }
}

//----------------------------------------------------------------------
void FFileDialog::draw()
{
  FDialog::draw();

  if ( Encoding == fc::VT100 )
    unsetVT100altChar();

}

//----------------------------------------------------------------------
inline bool FFileDialog::pattern_match ( const char* pattern,
                                         const char* fname )
{
  char search[128];
  if ( show_hidden && fname[0] == '.' && fname[1] != '\0' )  // hidden files
  {
    search[0] = '.';
    search[1] = '\0';
    strncat(search, pattern, sizeof(search) - strlen(search) - 1);
  }
  else
    strncpy(search, pattern, sizeof(search));

  if ( fnmatch (search, fname, FNM_PERIOD) == 0 )
    return true;
  else
    return false;
}

//----------------------------------------------------------------------
void FFileDialog::clear()
{
  std::vector<dir_entry>::const_iterator iter, end;

  if ( dir_entries.empty() )
    return;

  // delete all directory entries;
  iter = dir_entries.begin();
  end = dir_entries.end();

  while ( iter != end )
  {
    free ((*iter).name);
    ++iter;
  }

  dir_entries.clear();
}

//----------------------------------------------------------------------
int FFileDialog::numOfDirs()
{
  int n = 0;
  if ( ! dir_entries.empty() )
  {
    std::vector<dir_entry>::const_iterator iter, end;
    iter = dir_entries.begin();
    end = dir_entries.end();

    while ( iter != end )
    {
      if ( (*iter).type == DT_DIR && strcmp((*iter).name, ".") != 0 )
        n++;
      ++iter;
    }
  }
  return n;
}

//----------------------------------------------------------------------
int FFileDialog::changeDir (const FString& dirname)
{
  FString lastdir = directory;
  FString newdir = dirname;

  if ( newdir.includes('~') )
    newdir = newdir.replace('~', getHomeDir());

  if ( newdir[0] == '/' )
    setPath(newdir);
  else
    setPath(directory + newdir);

  switch ( readDir() )
  {
    case -1:
      setPath(lastdir);
      return -1;

    case -2:
      setPath(lastdir);
      readDir();
      return -2;

    case 0:
      if ( newdir == FString("..") )
      {
        if ( lastdir == FString('/') )
          filename->setText('/');
        else if ( ! dir_entries.empty() )
        {
          int i = 1;
          std::vector<dir_entry>::const_iterator iter, end;
          const char* baseName = basename(const_cast<char*>(lastdir.c_str()));

          iter = dir_entries.begin();
          end = dir_entries.end();

          while ( iter != end )
          {
            if ( strcmp((*iter).name, baseName) == 0 )
            {
              filebrowser->setCurrentItem(i);
              filename->setText(FString(baseName) + '/');
              break;
            }
            i++;
            ++iter;
          }
        }
      }
      else
      {
        FString firstname = dir_entries[0].name;
        if ( dir_entries[0].type == DT_DIR )
          filename->setText(firstname + '/');
        else
          filename->setText(firstname);
      }
      printPath(directory);
      filename->redraw();
      filebrowser->redraw();

    default:
      return 0;
  }
}

//----------------------------------------------------------------------
void FFileDialog::printPath (const FString& txt)
{
  FString path = txt;
  uInt max_width = uInt(filebrowser->getWidth()) - 4;

  if ( path.getLength() > max_width )
    filebrowser->setText(".." + path.right(max_width-2));
  else
    filebrowser->setText(path);
}

//----------------------------------------------------------------------
void FFileDialog::cb_processActivate (FWidget*, void*)
{
  if (  filename->getText().includes('*')
     || filename->getText().includes('?') )
  {
    setFilter(filename->getText());
    readDir();
    filebrowser->redraw();
  }
  else if ( filename->getText().getLength() == 0 )
  {
    setFilter("*");
    readDir();
    filebrowser->redraw();
  }
  else if (  filename->getText().trim() == FString("..")
          || filename->getText().includes('/')
          || filename->getText().includes('~') )
  {
    changeDir(filename->getText().trim());
  }
  else
  {
    bool found = false;
    if ( ! dir_entries.empty() )
    {
      std::vector<dir_entry>::const_iterator iter, end;
      FString input = filename->getText().trim();

      iter = dir_entries.begin();
      end = dir_entries.end();

      while ( iter != end )
      {
        if ( strcmp((*iter).name, input) == 0 && (*iter).type == DT_DIR )
        {
          found = true;
          changeDir(input);
          break;
        }
        ++iter;
      }
    }
    if ( ! found )
      done (FDialog::Accept);
  }
}

//----------------------------------------------------------------------
void FFileDialog::cb_processRowChanged (FWidget*, void*)
{
  int n = filebrowser->currentItem();
  if ( n == 0 )
    return;
  FString name = dir_entries[uLong(n-1)].name;
  if ( dir_entries[uLong(n-1)].type == DT_DIR )
    filename->setText( name + '/' );
  else
    filename->setText( name );
  filename->redraw();
}

//----------------------------------------------------------------------
void FFileDialog::cb_processClicked (FWidget*, void*)
{
  uLong n = uLong(filebrowser->currentItem() - 1);

  if ( dir_entries[n].type == DT_DIR )
    changeDir(dir_entries[n].name);
  else
    done (FDialog::Accept);
}

//----------------------------------------------------------------------
void FFileDialog::cb_processCancel (FWidget*, void*)
{
  done (FDialog::Reject);
}

//----------------------------------------------------------------------
void FFileDialog::cb_processOpen (FWidget*, void*)
{
  done (FDialog::Accept);
}

//----------------------------------------------------------------------
void FFileDialog::cb_processShowHidden (FWidget*, void*)
{
  setShowHiddenFiles(not show_hidden);
}


// protected methods of FFileDialog
//----------------------------------------------------------------------
void FFileDialog::adjustSize()
{
  int X, Y;
  int max_height = getRootWidget()->getClientHeight();
  int max_width = getRootWidget()->getClientWidth();
  int h = max_height - 6;
  if ( h < 15 )  // minimum
    h = 15;
  if ( h > 30 )  // maximum
    h = 30;
  setHeight (h, false);
  X = 1 + int((max_width-width)/2);
  Y = 1 + int((max_height-height)/3);
  setX(X, false);
  setY(Y, false);
  filebrowser->setHeight(h-8, false);
  hidden->setY(h-4, false);
  cancel->setY(h-4, false);
  open->setY(h-4, false);
  FDialog::adjustSize();
  printPath(directory);
}


// public methods of FFileDialog
//----------------------------------------------------------------------
void FFileDialog::onKeyPress (FKeyEvent* ev)
{
  if ( ! isEnabled() )
    return;

  FDialog::onKeyPress (ev);

  if ( ! filebrowser->hasFocus() )
    return;

  int key = ev->key();
  switch ( key )
  {
    case fc::Fkey_erase:
    case fc::Fkey_backspace:
      changeDir("..");
      ev->accept();
      break;

    default:
      break;
  }

}

//----------------------------------------------------------------------
void FFileDialog::setPath (const FString& dir)
{
  const char* dirname = dir.c_str();
  char resolved_path[MAXPATHLEN];
  FString r_dir;
  struct stat sb;

  stat(dirname, &sb);
  if ( S_ISLNK(sb.st_mode) )
  {
    lstat(dirname, &sb);
  }
  if ( ! S_ISDIR(sb.st_mode) )
  {
    directory = '/';
    return;
  }

  if ( realpath(dir.c_str(), resolved_path) != 0 )
    r_dir = resolved_path;
  else
    r_dir = dir;

  if ( r_dir[r_dir.getLength()-1] != '/' )
    directory = r_dir + "/";
  else
    directory = r_dir;
}

//----------------------------------------------------------------------
void FFileDialog::setFilter (const FString& filter)
{
  filter_pattern = filter;
}

//----------------------------------------------------------------------
const FString FFileDialog::getSelectedFile() const
{
  uLong n = uLong(filebrowser->currentItem() - 1);

  if ( dir_entries[n].type == DT_DIR )
    return FString("");
  else
    return FString(dir_entries[n].name);
}

//----------------------------------------------------------------------
int FFileDialog::readDir()
{
  const char* dir = directory.c_str();
  const char* filter = filter_pattern.c_str();
  errno = 0;
  directory_stream = opendir(dir);
  if ( ! directory_stream )
  {
    FMessageBox::error (this, "Can't open directory\n" + directory);
    return -1;
  }

  clear();

  while ( true )
  {
    errno = 0;
    struct dirent* next = readdir (directory_stream);
    if ( next )
    {
      if ( next->d_name[0] == '.' && next->d_name[1] == '\0' )
        continue;
      if (  ! show_hidden
         && next->d_name[0] == '.'
         && next->d_name[1] != '\0'
         && next->d_name[1] != '.' )
        continue;
      if ( dir[0] == '/' && dir[1] == '\0' && strcmp(next->d_name, "..") == 0  )
        continue;
      dir_entry entry;
      entry.name = strdup(next->d_name);
      entry.type = next->d_type;

      if ( next->d_type == DT_LNK )  // symbolic link
      {
        char resolved_path[MAXPATHLEN];
        char symLink[MAXPATHLEN];
        strncpy(symLink, dir, sizeof(symLink));
        strncat(symLink, next->d_name, sizeof(symLink) - strlen(symLink) - 1);

        if ( realpath(symLink, resolved_path) != 0 )  // follow link
        {
          struct stat sb;
          lstat(resolved_path, &sb);
          if ( S_ISDIR(sb.st_mode) )
            entry.type = DT_DIR;
        }
      }
      if ( entry.type == DT_DIR )
        dir_entries.push_back (entry);
      else if ( pattern_match(filter, entry.name) )
        dir_entries.push_back (entry);
      else
        free(entry.name);
    }
    else if (errno != 0)
    {
      FMessageBox::error (this, "Reading directory\n" + directory);
      if ( errno != EOVERFLOW )
        break;
    }
    else
      break;
  }  // end while

  if ( closedir (directory_stream) != 0 )
  {
    FMessageBox::error (this, "Closing directory\n" + directory);
    return -2;
  }

  int start;
  if ( strcmp((*dir_entries.begin()).name, "..") == 0 )
    start=1;
  else
    start=0;
  int dir_num = numOfDirs();

  // directories first
  std::sort(dir_entries.begin()+start, dir_entries.end(), sortDirFirst);

  // sort directories by name
  std::sort(dir_entries.begin()+start, dir_entries.begin()+dir_num, sortByName);

  // sort files by name
  std::sort(dir_entries.begin()+dir_num, dir_entries.end(), sortByName);

  // fill list with directory entries
  filebrowser->clear();
  if ( ! dir_entries.empty() )
  {
    std::vector<dir_entry>::const_iterator iter, end;
    iter = dir_entries.begin();
    end = dir_entries.end();

    while ( iter != end )
    {
      if ( (*iter).type == DT_DIR )
        filebrowser->insert(FString((*iter).name), fc::SquareBrackets);
      else
        filebrowser->insert(FString((*iter).name));
      ++iter;
    }
  }
  return 0;
}

//----------------------------------------------------------------------
bool FFileDialog::setShowHiddenFiles (bool on)
{
  if ( on == show_hidden )
    return show_hidden;

  show_hidden = on;
  readDir();
  filebrowser->redraw();
  return show_hidden;
}

//----------------------------------------------------------------------
FString FFileDialog::fileOpenChooser ( FWidget* parent,
                                       const FString& dirname,
                                       const FString& filter )
{
  FString ret;
  FString path = dirname;
  FString file_filter = filter;
  if ( path.isNull() )
    path = getHomeDir();
  if ( file_filter.isNull() )
    file_filter = FString("*");
  FFileDialog* fileopen = new FFileDialog ( path,
                                            file_filter,
                                            FFileDialog::Open,
                                            parent );
  if ( fileopen->exec() == FDialog::Accept )
    ret = fileopen->getPath() + fileopen->getSelectedFile();
  else
    ret = FString();
  delete fileopen;
  return ret;
}

//----------------------------------------------------------------------
FString FFileDialog::fileSaveChooser ( FWidget* parent,
                                       const FString& dirname,
                                       const FString& filter )
{
  FString ret;
  FString path = dirname;
  FString file_filter = filter;
  if ( path.isNull() )
    path = getHomeDir();
  if ( file_filter.isNull() )
    file_filter = FString("*");
  FFileDialog* fileopen = new FFileDialog ( path,
                                            file_filter,
                                            FFileDialog::Save,
                                            parent );
  if ( fileopen->exec() == FDialog::Accept )
    ret = fileopen->getPath() + fileopen->getSelectedFile();
  else
    ret = FString();
  delete fileopen;
  return ret;
}