filewnd.cpp

/*
 * filewnd.cpp - Small File Manager for DOS w/ archive support
 * Copyright (c) 2002, 2006 Simon Peter <dn.tlp@gmx.net>
 *
 * This software is provided 'as-is', without any express or implied
 * warranty.  In no event will the authors be held liable for any damages
 * arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must not
 *    claim that you wrote the original software. If you use this software
 *    in a product, an acknowledgment in the product documentation would be
 *    appreciated but is not required.
 * 2. Altered source versions must be plainly marked as such, and must not be
 *    misrepresented as being the original software.
 * 3. This notice may not be removed or altered from any source distribution.
 */

#include <stdlib.h>
#include <dos.h>
#include <conio.h>
#include <string.h>
#include <adplug/adplug.h>

#ifdef DJGPP
#	include <unistd.h>
#	include <dirent.h>
#endif

#include "filewnd.h"
#include "arcfile.h"

#define MAXFCOLORS      8

unsigned char FileWnd::fc[MAXFCOLORS] = {0x70,7,0x70,7,0x70,7,0x70,7};

FileWnd::FileWnd()
  : CListWnd(), sortby(SortByName), err(None), arcmode(0)
{
  *dirprefix = '\0';
}

FileWnd::~FileWnd()
{
}

void FileWnd::setfilecolor(FColor c, unsigned char v)
{
  fc[c] = v;
}

void FileWnd::refresh()
{
  FileItem *item = new FileItem;

  err = None; removeall();        // clear list
  listdrives();
  if(arcmode) listarc(arc); else listfiles();
  item->settext(".."); item->attr = FileItem::Directory;
  item->setcolor(Item::Selected,fc[DirSel]);
  item->setcolor(Item::Unselected,fc[DirUnsel]);
  additem(item);
}

FileWnd::Error FileWnd::geterror()
{
  return err;
}

char *FileWnd::getfilename(char *fn)
{
  err = None;
  if(!arcmode || !(*dirprefix))
    strcpy(fn,getselection()->gettext());
  else {
    strcpy(fn,dirprefix);
    strcat(fn,getselection()->gettext());
  }

  return fn;
}

bool FileWnd::select()
  /* Returns false, if this method overtakes all actions.
   * Returns true, if user selected a file to be used.
   */
{
  unsigned int dummy,drive;
  struct diskfree_t dummy2;
  FileItem *f = (FileItem *)getselection();
  char *fname = f->gettext();

  err = None;

  // Disk drive selected?
  if(f->attr == FileItem::Drive && (drive = drivenum(fname))) {
    // Try to switch to drive
    if(!_dos_getdiskfree(drive,&dummy2)) {
      _dos_setdrive(drive,&dummy);
      arcmode = 0; *dirprefix = '\0'; // Bail out of archive mode
      return false;
    } else {  // Drive not ready error
      err = Drive_NotReady;
      return false;
    }
  }

  // Currently displaying an archive?
  if(arcmode) {
    // ".." selected
    if(!strcmp(fname,"..")) {
      if(*dirprefix) { // Maybe just up a directory
	char *tmpstr = strrchr(dirprefix,'/'); *tmpstr = '\0';
	if((tmpstr = strrchr(dirprefix,'/')))
	  *(tmpstr+1) = '\0';
	else
	  *dirprefix = '\0';
      } else  // Exit the archive
	arcmode--;
      return false;
    }

    // Directory within archive selected?
    if(f->attr == FileItem::Directory) {
      strcat(dirprefix,fname);
      strcat(dirprefix,"/");
      return false;
    }

    // File in archive selected
    return true;
  }

  // If directory selected, switch to it
  if(f->attr == FileItem::Directory) {
    chdir(fname);
    return false;
  }

  // If archive selected, open it
  if(arc.open(fname)) {
    arcmode++;
    return false;
  }

  return true;   // Ordinary file selected
}

bool FileWnd::inarchive()
{
  err = None;
  return arcmode;
}

archive *FileWnd::getarchive()
{
  err = None;
  return &arc;
}

bool FileWnd::supported(const char *filename)
  /* Checks if a filename is supported by AdPlug. */
{
  CPlayers::const_iterator	i;
  unsigned int			j;

  for(i = CAdPlug::players.begin(); i != CAdPlug::players.end(); i++)
    for(j = 0; (*i)->get_extension(j); j++)
      if(CFileProvider::extension(filename, (*i)->get_extension(j)))
	return true;

  return false;
}

bool FileWnd::is_subdir(struct dirent *direntp)
{
  bool retval;

#if defined(DJGPP)
  unsigned int attr;

  _dos_getfileattr(direntp->d_name, &attr);
  retval = attr & _A_SUBDIR ? true : false;
#else
  retval = false;
#endif

  return retval;
}

void FileWnd::listfiles()
  /* Rebuilds ourself (our CListWnd, actually) with a filelist of the current
   * working directory.
   */
{
  DIR *dirp;
  struct dirent *direntp;
  char *wd;
  FileItem *f;

  // add files and directories
  wd = getcwd(NULL, PATH_MAX);
  dirp = opendir(wd);
  while((direntp = readdir(dirp))) {	// add files
    if(!is_subdir(direntp)) {
      f = new FileItem;
      f->settext(direntp->d_name);
      f->attr = FileItem::File;
      if(supported(direntp->d_name)) {
	f->setcolor(Item::Selected,fc[SupportedSel]);
	f->setcolor(Item::Unselected,fc[SupportedUnsel]);
      } else {
	f->setcolor(Item::Selected,fc[FileSel]);
	f->setcolor(Item::Unselected,fc[FileUnsel]);
      }
      sortinsert(f);
    }
  }
  rewinddir(dirp);
  while((direntp = readdir(dirp)))	// add directories
    if(is_subdir(direntp) && strcmp(direntp->d_name,".")
       && strcmp(direntp->d_name,"..")) {
      f = new FileItem;
      f->settext(direntp->d_name);
      f->attr = FileItem::Directory;
      f->setcolor(Item::Selected,fc[DirSel]);
      f->setcolor(Item::Unselected,fc[DirUnsel]);
      sortinsert(f);
    }
  closedir(dirp);
  free(wd);
}

void FileWnd::sortinsert(FileItem *newitem)
  /* Inserts the given FileItem sorted into the listbox. Assumes that the
   * drive letters are already added and sorts files before drives and
   * directories before files.
   */
{
  unsigned int i = 0;
  FileItem *f;
  char *ext1, *ext2;

  while((f = (FileItem *)getitem(i))) {
    // at "end of list" (in respect to the item to be inserted)?
    if(f->attr == FileItem::Drive ||
       (newitem->attr == FileItem::Directory &&
	f->attr == FileItem::File)) {
      if(i) insertitem(newitem,i-1); else additem(newitem);
      return;
    }

    // sort into list
    switch(sortby) {
    case SortByExtension:	// Sort first by extension, then by name
      ext1 = strrchr(newitem->gettext(), '.');
      ext2 = strrchr(f->gettext(), '.');

      if(strcmp(ext1, ext2) > 0)	// still before
	break;
      if(strcmp(ext1, ext2) < 0) {	// already after
	if(i) insertitem(newitem,i-1); else additem(newitem);
	return;
      }

      // Fall through...

    case SortByName:	// Sort by full filename
      if(strcmp(newitem->gettext(),f->gettext()) < 0) {
	if(i) insertitem(newitem,i-1); else additem(newitem);
	return;
      }
      break;
    }

    i++;
  }

  // List is empty, just add item...
  additem(newitem);
}

void FileWnd::listdrives()
{
  char drvstr[3] = "A:";
  unsigned int drives,drive,i;
  FileItem *f;

  // add drives
  _dos_getdrive(&drive);		// save current drive
  _dos_setdrive(drive,&drives);     // get number of drives
  for(i=drives;i>=1;i--) {
    unsigned int d;

    _dos_setdrive(i,&drives);
    _dos_getdrive(&d);
    if(d == i) {  // drive exists?
      *drvstr = 'A'+i-1;      // build drive string
      f = new FileItem;
      f->settext(drvstr);
      f->attr = FileItem::Drive;
      f->setcolor(Item::Selected,fc[DriveSel]);
      f->setcolor(Item::Unselected,fc[DriveUnsel]);
      additem(f);     // add drive name
    }
  }
  _dos_setdrive(drive,&drives);   // restore current drive
}

void FileWnd::listarc(archive &a)
{
  unsigned int i;
  ArcFile *f;
  char *basename;
  FileItem *item;

  // Add files of current directory
  for(i=0;i<a.getfiles();i++) {
    f = a.getfile(i);

    // In our current archive's directory?
    if(strncmp(f->name,dirprefix,strlen(dirprefix))) continue;
    basename = f->name + strlen(dirprefix);
    if(strchr(basename,'/')) continue;

    if(!(f->attr & ArcFile::Directory)) {
      item = new FileItem;
      item->settext(basename);
      item->attr = FileItem::File;
      item->setcolor(Item::Selected,fc[FileSel]);
      item->setcolor(Item::Unselected,fc[FileUnsel]);
      sortinsert(item);
    }
  }

  // Add Subdirectories
  for(i=0;i<a.getfiles();i++) {
    f = a.getfile(i);

    // In our current archive's directory?
    if(strncmp(f->name,dirprefix,strlen(dirprefix))) continue;
    basename = f->name + strlen(dirprefix);

    if(f->attr & ArcFile::Directory && !strchr(basename,'/')) {
      item = new FileItem;
      item->settext(basename);
      item->attr = FileItem::Directory;
      item->setcolor(Item::Selected,fc[DirSel]);
      item->setcolor(Item::Unselected,fc[DirUnsel]);
      sortinsert(item);
    }
  }
}

unsigned int FileWnd::drivenum(const char *fname)
{
  char tmpstr[] = "A:";
  unsigned int a_as_uint = 'A';
  for(*tmpstr=char('A');*tmpstr<=char('Z');(*tmpstr)++)
    if(!strcmp(fname,tmpstr))
      return ((*tmpstr)-a_as_uint+1);

  return 0;
}