source: trunk/dll/objcnr.c@ 613

/***********************************************************************

  $Id: objcnr.c 574 2007-03-23 22:37:07Z gyoung $

  Object containers

  Copyright (c) 1993-98 M. Kimes
  Copyright (c) 2005, 2007 Steven H. Levine

  24 May 05 SHL Rework for CNRITEM.szSubject
  13 Jul 06 SHL Use Runtime_Error
  01 Sep 06 SHL Do not complain for normal cancel
  19 Oct 06 SHL Correct . and .. detect
  03 Nov 06 SHL Renames
  22 Mar 07 GKY Use QWL_USER

***********************************************************************/

#define INCL_DOS
#define INCL_WIN
#include <os2.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "fm3dll.h"
#include "fm3dlg.h"
#include "fm3str.h"

typedef struct
{
  CHAR *filename;
  HWND hwndCnr;
  CHAR *stopflag;
}
DIRSIZE;

typedef struct
{
  CHAR *dirname;
  CHAR stopflag;
  BOOL dying;
  BOOL working;
}
TEMP;

#pragma data_seg(DATA1)

static PSZ pszSrcFile = __FILE__;

static HWND objcnrwnd;

#pragma alloc_text(OBJCNR,ProcessDir,FillCnrsThread,ObjCnrDlgProc)

static VOID ProcessDir(HWND hwndCnr, CHAR * filename, PCNRITEM pciParent,
                       CHAR * stopflag)
{
  CHAR maskstr[CCHMAXPATH], *endpath, *p;
  ULONG nm, ulM;
  HDIR hdir;
  FILEFINDBUF3 *ffb, *fft;
  APIRET rc;
  RECORDINSERT ri;
  PCNRITEM pciP;

  ffb = xmalloc(sizeof(FILEFINDBUF3), pszSrcFile, __LINE__);
  if (!ffb)
    return;
  strcpy(maskstr, filename);
  if (maskstr[strlen(maskstr) - 1] != '\\')
    strcat(maskstr, "\\");
  endpath = &maskstr[strlen(maskstr)];
  strcat(maskstr, "*");
  hdir = HDIR_CREATE;
  nm = 1L;
  rc = DosFindFirst(filename, &hdir,
                    FILE_NORMAL | FILE_READONLY | FILE_ARCHIVED |
                    FILE_SYSTEM | FILE_HIDDEN | MUST_HAVE_DIRECTORY,
                    ffb, sizeof(FILEFINDBUF3), &nm, FIL_STANDARD);
  if (!rc)
    DosFindClose(hdir);

  if (rc) {                             /* work around furshluginer FAT bug... */
    if (IsRoot(filename))
      rc = 0;
  }

  if ((!rc && (ffb->attrFile & FILE_DIRECTORY))) {
    pciP =
      WinSendMsg(hwndCnr, CM_ALLOCRECORD, MPFROMLONG(EXTRA_RECORD_BYTES2),
                 MPFROMLONG(1L));
    if (!pciP) {
      free(ffb);
      return;
    }
    strcpy(pciP->szFileName, filename);
    pciP->pszDispAttr = pciP->szDispAttr;
    pciP->pszSubject = pciP->szSubject;
    pciP->pszLongname = pciP->szLongname;
    pciP->pszDispAttr = pciP->szDispAttr;
    *pciP->szDispAttr = *pciP->szLongname = *pciP->szSubject = 0;
    if (strlen(filename) < 4)
      pciP->pszFileName = pciP->szFileName;
    else {
      p = strrchr(pciP->szFileName, '\\');
      if (!p)
        pciP->pszFileName = pciP->szFileName;
      else if (*(p + 1))
        p++;
      pciP->pszFileName = p;
    }
    pciP->rc.pszIcon = pciP->pszFileName;
    if (fForceUpper)
      strupr(pciP->szFileName);
    else if (fForceLower)
      strlwr(pciP->szFileName);
    pciP->rc.flRecordAttr |= CRA_RECORDREADONLY;
  }
  else {
    free(ffb);
    Dos_Error(MB_ENTER,
              rc,
              HWND_DESKTOP,
              pszSrcFile,
              __LINE__, GetPString(IDS_CANTFINDDIRTEXT), filename);
    return;
  }
  {
    HPOINTER hptr;

    hptr = WinLoadFileIcon(pciP->szFileName, FALSE);
    if (hptr)
      pciP->rc.hptrIcon = hptr;
  }
  if (!pciP->rc.hptrIcon || pciP->rc.hptrIcon == hptrFile)      /* OS/2 bug bug bug bug */
    pciP->rc.hptrIcon = hptrDir;
  memset(&ri, 0, sizeof(RECORDINSERT));
  ri.cb = sizeof(RECORDINSERT);
  ri.pRecordOrder = (PRECORDCORE) CMA_END;
  ri.pRecordParent = (PRECORDCORE) pciParent;
  ri.zOrder = (USHORT) CMA_TOP;
  ri.cRecordsInsert = 1L;
  ri.fInvalidateRecord = TRUE;
  if (!WinSendMsg(hwndCnr, CM_INSERTRECORD, MPFROMP(pciP), MPFROMP(&ri))) {
    free(ffb);
    return;
  }
  hdir = HDIR_CREATE;
  if (!isalpha(*maskstr) || maskstr[1] != ':' || maskstr[2] != '\\' ||
      ((driveflags[toupper(*maskstr) - 'A'] & DRIVE_REMOTE) && fRemoteBug))
    ulM = 1L;
  else
    ulM = min(FilesToGet, 225);
  if (ulM > 1L) {
    fft = xrealloc(ffb, sizeof(FILEFINDBUF3) * ulM, pszSrcFile, __LINE__);
    if (!fft)
      ulM = 1L;
    else
      ffb = fft;
  }
  nm = ulM;
  rc = DosFindFirst(maskstr, &hdir,
                    FILE_NORMAL | FILE_READONLY | FILE_ARCHIVED |
                    FILE_SYSTEM | FILE_HIDDEN | MUST_HAVE_DIRECTORY, ffb,
                    sizeof(FILEFINDBUF3) * ulM, &nm, FIL_STANDARD);
  if (!rc) {

    register PBYTE fb = (PBYTE) ffb;
    FILEFINDBUF3 *pffbFile;
    ULONG x;

    while (!rc) {
      for (x = 0L; x < nm; x++) {
        pffbFile = (FILEFINDBUF3 *) fb;
        if (*stopflag)
          break;
        if ((pffbFile->attrFile & FILE_DIRECTORY) &&
            // Skip . and ..
            (pffbFile->achName[0] != '.' ||
             (pffbFile->achName[1] &&
              (pffbFile->achName[1] != '.' || pffbFile->achName[2])))) {
          strcpy(endpath, pffbFile->achName);
          ProcessDir(hwndCnr, maskstr, pciP, stopflag);
        }
        if (!pffbFile->oNextEntryOffset)
          break;
        fb += pffbFile->oNextEntryOffset;
      }
      DosSleep(0L);
      if (*stopflag)
        break;
      nm = ulM;
      rc = DosFindNext(hdir, ffb, sizeof(FILEFINDBUF3) * ulM, &nm);
    }
    DosFindClose(hdir);
  }
  free(ffb);
  WinSendMsg(hwndCnr, CM_INVALIDATERECORD, MPFROMP(&pciP),
             MPFROM2SHORT(1, 0));
}

static VOID FillCnrsThread(VOID * args)
{
  HAB hab;
  HMQ hmq;
  DIRSIZE *dirsize = (DIRSIZE *) args;

  if (!dirsize)
    return;

  DosError(FERR_DISABLEHARDERR);

  hab = WinInitialize(0);
  if (hab) {
    hmq = WinCreateMsgQueue(hab, 0);
    if (hmq) {
      WinCancelShutdown(hmq, TRUE);
      ProcessDir(dirsize->hwndCnr, dirsize->filename, (PCNRITEM) NULL,
                 dirsize->stopflag);
      DosPostEventSem(CompactSem);
      WinDestroyMsgQueue(hmq);
    }
    WinTerminate(hab);
  }
  PostMsg(WinQueryWindow(dirsize->hwndCnr, QW_PARENT), UM_CONTAINER_FILLED,
          MPVOID, MPVOID);
  free(dirsize);
}

MRESULT EXPENTRY ObjCnrDlgProc(HWND hwnd, ULONG msg, MPARAM mp1, MPARAM mp2)
{
  TEMP *data;

  switch (msg) {
  case WM_INITDLG:
    if (objcnrwnd) {
      Runtime_Error(pszSrcFile, __LINE__, "objcnrwnd set");
      WinSetWindowPos(objcnrwnd, HWND_TOP, 0, 0, 0, 0,
                      SWP_RESTORE | SWP_SHOW | SWP_ACTIVATE | SWP_ZORDER);
      WinDismissDlg(hwnd, 0);
      break;
    }
    if (!mp2) {
      Runtime_Error(pszSrcFile, __LINE__, "mp2 NULL");
      WinDismissDlg(hwnd, 0);
      break;
    }
    objcnrwnd = hwnd;
    data = xmallocz(sizeof(TEMP), pszSrcFile, __LINE__);
    if (!data) {
      WinDismissDlg(hwnd, 0);
      break;
    }
    data->dirname = (CHAR *) mp2;
    WinSetWindowPtr(hwnd, QWL_USER, (PVOID) data);
    if (*data->dirname)
      WinSetDlgItemText(hwnd, OBJCNR_DIR, data->dirname);
    {
      DIRSIZE *dirsize;

      dirsize = xmalloc(sizeof(DIRSIZE), pszSrcFile, __LINE__);
      if (!dirsize) {
        WinDismissDlg(hwnd, 0);
        break;
      }
      dirsize->stopflag = (CHAR *) & data->stopflag;
      dirsize->filename = data->dirname;
      dirsize->hwndCnr = WinWindowFromID(hwnd, OBJCNR_CNR);
      if (_beginthread(FillCnrsThread, NULL, 65536 * 8, (PVOID) dirsize) ==
          -1) {
        Runtime_Error(pszSrcFile, __LINE__,
                      GetPString(IDS_COULDNTSTARTTHREADTEXT));
        free(dirsize);
        WinDismissDlg(hwnd, 0);
        break;
      }
      else
        data->working = TRUE;
    }
    PostMsg(hwnd, UM_SETUP, MPVOID, MPVOID);
    break;

  case UM_SETUP:
    // WinEnableWindowUpdate(WinWindowFromID(hwnd,OBJCNR_CNR),FALSE);
    {
      CNRINFO cnri;

      memset(&cnri, 0, sizeof(CNRINFO));
      cnri.cb = sizeof(CNRINFO);
      WinSendDlgItemMsg(hwnd, OBJCNR_CNR, CM_QUERYCNRINFO,
                        MPFROMP(&cnri), MPFROMLONG(sizeof(CNRINFO)));
      cnri.cyLineSpacing = 0;
      cnri.cxTreeIndent = 12L;
      cnri.pszCnrTitle = GetPString(IDS_WORKINGTEXT);
      cnri.flWindowAttr = CV_TREE | CV_FLOW |
        CA_CONTAINERTITLE | CA_TITLESEPARATOR | CA_TREELINE;
      if (WinQueryWindowUShort(hwnd, QWS_ID) == QTREE_FRAME)
        cnri.flWindowAttr |= CV_MINI;
      WinSendDlgItemMsg(hwnd, OBJCNR_CNR, CM_SETCNRINFO, MPFROMP(&cnri),
                        MPFROMLONG(CMA_FLWINDOWATTR | CMA_LINESPACING |
                                   CMA_CXTREEINDENT));
    }
    return 0;

  case UM_CONTAINER_FILLED:
    WinSetDlgItemText(hwnd, OBJCNR_NOTE, NullStr);
//      WinEnableWindowUpdate(WinWindowFromID(hwnd,OBJCNR_CNR),TRUE);
    WinSendDlgItemMsg(hwnd, OBJCNR_CNR, CM_INVALIDATERECORD, MPVOID,
                      MPFROM2SHORT(0, CMA_ERASE | CMA_INVALIDATE));
    data = INSTDATA(hwnd);
    if (data) {
      data->working = FALSE;
      if (data->dying)
        WinDismissDlg(hwnd, 0);
      {
        PCNRITEM pci;
        USHORT id;

        id = WinQueryWindowUShort(hwnd, QWS_ID);
        pci = (PCNRITEM) WinSendDlgItemMsg(hwnd, OBJCNR_CNR,
                                           CM_QUERYRECORD,
                                           MPVOID,
                                           MPFROM2SHORT(CMA_FIRST,
                                                        CMA_ITEMORDER));
        if (pci && (INT) pci != -1) {
          ExpandAll(WinWindowFromID(hwnd, OBJCNR_CNR), TRUE, pci);
          if (id == QTREE_FRAME)
            pci = (PCNRITEM) WinSendDlgItemMsg(hwnd, OBJCNR_CNR,
                                               CM_QUERYRECORD,
                                               MPFROMP(pci),
                                               MPFROM2SHORT(CMA_FIRSTCHILD,
                                                            CMA_ITEMORDER));
        }
        if ((!pci || (INT) pci == -1) && id == QTREE_FRAME) {
          Notify(GetPString(IDS_NODIRSUNDERTEXT));
          WinDismissDlg(hwnd, 0);
          break;
        }
      }
    }
    return 0;

  case WM_CONTROL:
    switch (SHORT1FROMMP(mp1)) {
    case OBJCNR_CNR:
      if (SHORT2FROMMP(mp1) == CN_ENTER) {

        PCNRITEM pci = (PCNRITEM) ((PNOTIFYRECORDENTER) mp2)->pRecord;

        if (pci && (INT) pci != -1)
          WinSendDlgItemMsg(hwnd, DID_OK, BM_CLICK, MPVOID, MPVOID);
      }
      break;
    }
    return 0;

  case WM_COMMAND:
    switch (SHORT1FROMMP(mp1)) {
    case IDM_HELP:
      if (hwndHelp) {

        USHORT id;

        id = WinQueryWindowUShort(hwnd, QWS_ID);

        if (id == QTREE_FRAME)
          WinSendMsg(hwndHelp, HM_DISPLAY_HELP,
                     MPFROM2SHORT(HELP_QUICKTREE, 0),
                     MPFROMSHORT(HM_RESOURCEID));
        else
          WinSendMsg(hwndHelp, HM_DISPLAY_HELP,
                     MPFROM2SHORT(HELP_OBJECTPATH, 0),
                     MPFROMSHORT(HM_RESOURCEID));
      }
      break;

    case OBJCNR_DESKTOP:
    case DID_OK:
      data = INSTDATA(hwnd);
      if (data) {

        PCNRITEM pci;

        if (data->working) {
          Runtime_Error(pszSrcFile, __LINE__, "working unexpected");
          break;
        }
        if (SHORT1FROMMP(mp1) == OBJCNR_DESKTOP) {
          WinDismissDlg(hwnd, 2);
          break;
        }
        pci = (PCNRITEM) WinSendDlgItemMsg(hwnd, OBJCNR_CNR,
                                           CM_QUERYRECORDEMPHASIS,
                                           MPFROMLONG(CMA_FIRST),
                                           MPFROMSHORT(CRA_CURSORED));
        if (pci && (INT) pci != -1)
          strcpy(data->dirname, pci->szFileName);
        WinDismissDlg(hwnd, 1);
      }
      break;

    case DID_CANCEL:
      data = INSTDATA(hwnd);
      if (data) {
        if (data->working) {
          data->dying = TRUE;
          data->stopflag = 0xff;
          break;
        }
        WinDismissDlg(hwnd, 0);
      }
      break;
    }
    return 0;

  case WM_DESTROY:
    objcnrwnd = (HWND) 0;
    data = INSTDATA(hwnd);
    if (data)
      free(data);
    break;
  }
  return WinDefDlgProc(hwnd, msg, mp1, mp2);
}