#define INCL_DOS
#define INCL_WIN

#include <os2.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include "dll\fm3dll.h"
#include "dll\fm3dlg.h"
#include "dirsize.h"


MRESULT EXPENTRY DirMainProc (HWND hwnd,ULONG msg,MPARAM mp1,MPARAM mp2) {

  static CHAR curdir[4];

  switch(msg) {
    case WM_INITDLG:
      *curdir = 0;
      WinSendMsg(hwnd,UM_UNDO,MPVOID,MPVOID);
      break;

    case UM_UNDO:
      WinSendDlgItemMsg(hwnd,DIRSIZE_LISTBOX,LM_DELETEALL,MPVOID,MPVOID);
      {
        ULONG x,ulDriveMap,ulDriveNum;
        CHAR  dirname[] = " :\\";
        BOOL  first = TRUE;

        DosError(FERR_DISABLEHARDERR);
        DosQCurDisk(&ulDriveNum,&ulDriveMap);
        for(x = 2;x < 26;x++) {
          if((ulDriveMap & (1L << x)) && !(driveflags[x] & DRIVE_IGNORE)) {
            *dirname = (CHAR)x + 'A';
            WinSendDlgItemMsg(hwnd,DIRSIZE_LISTBOX,LM_INSERTITEM,
                              MPFROM2SHORT(LIT_END,0),MPFROMP(dirname));
            if(first) {
              WinSendDlgItemMsg(hwnd,DIRSIZE_LISTBOX,LM_SELECTITEM,
                                MPFROMSHORT(0),MPFROMSHORT(TRUE));
              first = FALSE;
            }
          }
        }
      }
      return 0;

    case UM_RESCAN:
      {
        CHAR        FileSystem[CCHMAXPATH],s[CCHMAXPATH * 2];
        FSALLOCATE  fsa;
        ULONG       type;
        USHORT      percentused,percentfree;
        struct {
          ULONG serial;
          CHAR  volumelength;
          CHAR  volumelabel[CCHMAXPATH];
        }           volser;
        INT         removable;

        WinSetDlgItemText(hwnd,DIRSIZE_LABEL,"");
        WinSetDlgItemText(hwnd,DIRSIZE_UNITSFREE,"");
        WinSetDlgItemText(hwnd,DIRSIZE_UNITSUSED,"");
        WinSetDlgItemText(hwnd,DIRSIZE_UNITSIZE,"");
        WinSetDlgItemText(hwnd,DIRSIZE_PERCENT,"");
        WinSetDlgItemText(hwnd,DIRSIZE_BYTESUSED,"");
        WinSetDlgItemText(hwnd,DIRSIZE_BYTESFREE,"");
        WinSetDlgItemText(hwnd,DIRSIZE_IFS,"");
        WinSetDlgItemText(hwnd,DIRSIZE_LOCAL,"");
        WinSendDlgItemMsg(hwnd,DIRSIZE_SLIDER,SLM_SETSLIDERINFO,
                          MPFROM2SHORT(SMA_SLIDERARMPOSITION,
                                       SMA_INCREMENTVALUE),
                          MPFROMSHORT(0));
        removable = CheckDrive(toupper(*curdir),FileSystem,&type);
        if(removable != -1) {
          if(type & DRIVE_ZIPSTREAM)
            WinSetDlgItemText(hwnd,DIRSIZE_LOCAL,"Zipstream drive");
          else if(type & DRIVE_REMOTE)
            WinSetDlgItemText(hwnd,DIRSIZE_LOCAL,"Remote drive");
          else {
            sprintf(s,"Local drive%s",(removable) ? " (removable)" : "");
            WinSetDlgItemText(hwnd,DIRSIZE_LOCAL,s);
          }
          sprintf(s,"IFS:  %s",FileSystem);
          WinSetDlgItemText(hwnd,DIRSIZE_IFS,s);
          memset(&volser,0,sizeof(volser));
          DosError(FERR_DISABLEHARDERR);
          if(!DosQueryFSInfo(toupper(*curdir) - '@',FSIL_VOLSER,
                             &volser,(ULONG)sizeof(volser))) {
            sprintf(s,"Label:  %s",volser.volumelabel);
            WinSetDlgItemText(hwnd,DIRSIZE_LABEL,s);
          }
          if(!DosQueryFSInfo(toupper(*curdir) - '@',
             FSIL_ALLOC,&fsa,sizeof(FSALLOCATE))) {
            percentfree = (fsa.cUnitAvail && fsa.cUnit) ?
                           (fsa.cUnitAvail * 100) / fsa.cUnit :
                           0;
            if(!percentfree && fsa.cUnitAvail)
              percentfree = 1;
            percentused = 100 - percentfree;
            sprintf(s,"Units free:  %lu",fsa.cUnitAvail);
            WinSetDlgItemText(hwnd,DIRSIZE_UNITSFREE,s);
            sprintf(s,"Unit size:  %lu x %u = %lu",fsa.cSectorUnit,
                    fsa.cbSector,fsa.cSectorUnit * fsa.cbSector);
            WinSetDlgItemText(hwnd,DIRSIZE_UNITSIZE,s);
            sprintf(s,"Units used:  %lu",fsa.cUnit - fsa.cUnitAvail);
            WinSetDlgItemText(hwnd,DIRSIZE_UNITSUSED,s);
            sprintf(s,"Bytes free:  %lu",fsa.cUnitAvail *
                                           (fsa.cSectorUnit * fsa.cbSector));
            WinSetDlgItemText(hwnd,DIRSIZE_BYTESFREE,s);
            sprintf(s,"Bytes used:  %lu",(fsa.cUnit - fsa.cUnitAvail) *
                                           (fsa.cSectorUnit * fsa.cbSector));
            WinSetDlgItemText(hwnd,DIRSIZE_BYTESUSED,s);
            sprintf(s,"Percent used:  %u%%",percentused);
            WinSetDlgItemText(hwnd,DIRSIZE_PERCENT,s);
            WinSendDlgItemMsg(hwnd,DIRSIZE_SLIDER,SLM_SETSLIDERINFO,
                              MPFROM2SHORT(SMA_SLIDERARMPOSITION,
                                           SMA_INCREMENTVALUE),
                              MPFROMSHORT(percentused));
            WinShowWindow(WinWindowFromID(hwnd,DIRSIZE_SLIDER),TRUE);
          }
        }
        else {
          WinSetDlgItemText(hwnd,DIRSIZE_LOCAL,"Drive not ready.");
          WinShowWindow(WinWindowFromID(hwnd,DIRSIZE_SLIDER),FALSE);
        }
      }
      return 0;

    case WM_CONTROL:
      switch(SHORT1FROMMP(mp1)) {
        case DIRSIZE_LISTBOX:
          switch(SHORT2FROMMP(mp1)) {
            case LN_ENTER:
              WinPostMsg(hwnd,WM_COMMAND,MPFROM2SHORT(DID_OK,0),MPVOID);
              break;
            case LN_SELECT:
              {
                SHORT x;

                x = (SHORT)WinSendDlgItemMsg(hwnd,DIRSIZE_LISTBOX,
                                             LM_QUERYSELECTION,
                                             MPFROMSHORT(LIT_FIRST),
                                             MPVOID);
                if(x >= 0) {
                  WinSendDlgItemMsg(hwnd,DIRSIZE_LISTBOX,
                                    LM_QUERYITEMTEXT,
                                    MPFROM2SHORT(x,sizeof(curdir)),
                                    MPFROMP(curdir));
                  WinPostMsg(hwnd,UM_RESCAN,MPVOID,MPVOID);
                }
              }
              break;
          }
          break;
      }
      return 0;

    case WM_COMMAND:
      switch(SHORT1FROMMP(mp1)) {
        case DID_CANCEL:
          WinDismissDlg(hwnd,0);
          break;

        case DID_OK:
          if(*curdir) {
            WinSetWindowPos(hwnd,HWND_TOP,0,0,0,0,SWP_HIDE);
            WinDlgBox(HWND_DESKTOP,hwnd,
                      DirSizeProc,FM3ModHandle,
                      DSZ_FRAME,curdir);
            WinSetWindowPos(hwnd,HWND_TOP,0,0,0,0,SWP_SHOW);
          }
          else
            DosBeep(50,100);
          break;
      }
      return 0;
  }
  return WinDefDlgProc(hwnd,msg,mp1,mp2);
}


#ifdef NEVER

VOID APIENTRY deinit (ULONG why) {

  if(fmprof)
    PrfCloseProfile(fmprof);
  fmprof = (HINI)0;

  flushall();

  DosExitList(EXLST_REMOVE,deinit);
}

#endif


int main (int argc,char *argv[]) {

  HAB         hab;
  HMQ         hmq;
  static CHAR fullname[CCHMAXPATH];
  INT         x;

  DosError(FERR_DISABLEHARDERR);
  *fullname = 0;
  for(x = 1;x < argc;x++) {
    if(!strchr("/;,`\'",*argv[x]) &&
       !*fullname &&
       (IsRoot(argv[x]) ||
        IsFile(argv[x]) == 0)) {
      if(IsRoot(argv[x]))
        strcpy(fullname,argv[x]);
      else if(DosQueryPathInfo(argv[x],
                               FIL_QUERYFULLNAME,
                               fullname,
                               sizeof(fullname)))
        *fullname = 0;
    }
  }
#ifdef NEVER
  DosExitList(EXLST_ADD,deinit);
#endif
  hab = WinInitialize(0);
  if(hab) {
    hmq = WinCreateMsgQueue(hab,384);
    if(hmq) {
      if(InitFM3DLL(hab,argc,argv)) {
        if(!*fullname)
          WinDlgBox(HWND_DESKTOP,
                    HWND_DESKTOP,
                    DirMainProc,
                    0,
                    DIRSIZE_FRAME,
                    NULL);
        else
          WinDlgBox(HWND_DESKTOP,
                    HWND_DESKTOP,
                    DirSizeProc,
                    FM3ModHandle,
                    DSZ_FRAME,
                    fullname);
      }
      WinDestroyMsgQueue(hmq);
    }
    WinTerminate(hab);
  }
  return 0;
}