home *** CD-ROM | disk | FTP | other *** search
/ PC World 1999 July / PCWorld_1999-07_cd.bin / 602 / WBPERSON / data1.cab / SDK_Files / wbwizard / sources / WBMFCSSA.CPP < prev    next >
C/C++ Source or Header  |  1999-06-09  |  10KB  |  340 lines
  1. // wbmfcssa.cpp : implementation file
  2. //
  3.  
  4. #include "stdafx.h"
  5. #include "resource.h"
  6. #include "wbmfc.h"
  7. #include "wbmfcssa.h"
  8.  
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  14.  
  15. BOOL WBSelSrvrAppl(const char *UserName, const char *Password, const char *Appl)
  16. {
  17.     CSelSrvrApplDlg Dlg(UserName, Password, Appl);
  18.     Dlg.DoModal();
  19.     return(TRUE);
  20. }
  21. /////////////////////////////////////////////////////////////////////////////
  22. // CSelSrvrApplDlg dialog
  23.  
  24.  
  25. CSelSrvrApplDlg::CSelSrvrApplDlg(const char *UserName, const char *Password, const char *Appl, CWnd* pParent /*=NULL*/)
  26.     : CDialog(CSelSrvrApplDlg::IDD, pParent)
  27. {
  28.     //{{AFX_DATA_INIT(CSelSrvrApplDlg)
  29.         // NOTE: the ClassWizard will add member initialization here
  30.     //}}AFX_DATA_INIT
  31.     m_SelServer = NULL;
  32.     m_UserName  = UserName;
  33.     m_Password  = Password;
  34.     m_Appl      = Appl;
  35.     m_NoAppl    = UserName == WBAPPI_NOLOGIN || Appl != NULL;
  36. }
  37.  
  38.  
  39. BEGIN_MESSAGE_MAP(CSelSrvrApplDlg, CDialog)
  40.     //{{AFX_MSG_MAP(CSelSrvrApplDlg)
  41.     ON_NOTIFY(TVN_SELCHANGED, IDC_SRVRTREE, OnSelchanged)
  42.     ON_NOTIFY(NM_SETFOCUS, IDC_SRVRTREE, OnSelchanged)
  43.     ON_NOTIFY(TVN_ITEMEXPANDING, IDC_SRVRTREE, OnItemexpanding)
  44.     ON_NOTIFY(NM_DBLCLK, IDC_SRVRTREE, OnDblclk)
  45.     ON_EN_CHANGE(IDC_APPLEDIT, OnChangeEdit)
  46.     ON_EN_CHANGE(IDC_SRVREDIT, OnChangeEdit)
  47.     ON_WM_DESTROY()
  48.     //}}AFX_MSG_MAP
  49. END_MESSAGE_MAP()
  50.  
  51. /////////////////////////////////////////////////////////////////////////////
  52. // CSelSrvrApplDlg message handlers
  53.  
  54. BOOL CSelSrvrApplDlg::OnInitDialog() 
  55. {
  56.     CDialog::OnInitDialog();
  57.     
  58.     // TODO: Add extra initialization here
  59.     m_SrvrTree  = (CTreeCtrl *)GetDlgItem(IDC_SRVRTREE);
  60.     GetDlgItem(IDOK)->EnableWindow(FALSE);
  61.     if (!m_SrvrTree)
  62.         return FALSE;
  63.     FillSrvrTree();    
  64.     if (m_NoAppl)
  65.     {
  66.         CWnd *hEdit = GetDlgItem(IDC_APPLEDIT);
  67.         hEdit->EnableWindow(FALSE);
  68.         hEdit->ShowWindow(SW_HIDE);
  69.         GetDlgItem(IDC_APPLSTATIC)->ShowWindow(FALSE);
  70.         CRect Rect;
  71.         GetDlgItem(IDC_SRVREDIT)->GetWindowRect(&Rect);
  72.         int NewyExt = Rect.bottom + 20;
  73.         GetWindowRect(&Rect);
  74.         Rect.bottom = NewyExt;
  75.         MoveWindow(&Rect, FALSE);
  76.         SetWindowLong(*m_SrvrTree, GWL_STYLE, GetWindowLong(*m_SrvrTree, GWL_STYLE) & ~TVS_HASBUTTONS);
  77.     }
  78.     return TRUE;  // return TRUE unless you set the focus to a control
  79.                   // EXCEPTION: OCX Property Pages should return FALSE
  80. }
  81.  
  82. #define IND_APPL     0
  83. #define IND_SRVLOCAL 1
  84. #define IND_SRVREMNR 2
  85. #define IND_SRVRAS   3 
  86.  
  87. BOOL CSelSrvrApplDlg::FillSrvrTree()
  88. { 
  89.     HKEY hKey, hSubKey;    
  90.     char Path[MAX_PATH];
  91.     int  ind = 0;
  92.     tobjname Server;  
  93.  
  94.     LONG Err = RegOpenKeyEx(HKEY_LOCAL_MACHINE, "SOFTWARE\\Software602\\WinBase602\\Database", 0, KEY_READ, &hKey);
  95.     if (Err != ERROR_SUCCESS) 
  96.         return(FALSE);
  97.  
  98.     CBitmap bm;
  99.     bm.LoadBitmap(IDR_SRVRAPPL);
  100.     CImageList *il = new CImageList;
  101.     il->Create(16, 16, ILC_COLOR | ILC_MASK, 4, 0);
  102.     il->Add(&bm, RGB(0, 128, 128)); 
  103.     m_SrvrTree->SetImageList(il, TVSIL_NORMAL);
  104.     bm.DeleteObject();
  105.  
  106.     do
  107.     { 
  108.         DWORD KeyLen = sizeof(Server);
  109.         if (RegEnumKey(hKey, ind++, Server, KeyLen) != ERROR_SUCCESS)
  110.             break;
  111.         // get server info from the registry:
  112.         BOOL local = FALSE, verok = FALSE, ras = FALSE;
  113.         if (RegOpenKeyEx(hKey, Server, 0, KEY_READ, &hSubKey) == ERROR_SUCCESS)
  114.         { 
  115.             KeyLen = MAX_PATH;  
  116.             *Path  = 0;
  117.             RegQueryValueEx(hSubKey, "Path",  NULL, NULL, (BYTE *)Path, &KeyLen);
  118.             local  = *Path != 0;
  119.             KeyLen = MAX_PATH;  
  120.             *Path  = 0;
  121.             RegQueryValueEx(hSubKey, "Version", NULL, NULL, (BYTE *)Path, &KeyLen);
  122.             verok  = !lstrcmp(Path, "5");
  123.             KeyLen = MAX_PATH;  
  124.             *Path  = 0;
  125.             RegQueryValueEx(hSubKey, "ConnectionName", NULL, NULL, (BYTE *)Path, &KeyLen);
  126.             if (*Path) 
  127.                 ras = TRUE;
  128.             KeyLen = MAX_PATH;  
  129.             *(BOOL *)Path = FALSE;
  130.             RegQueryValueEx(hSubKey, "Through Firewall", NULL, NULL, (BYTE *)Path, &KeyLen);
  131.             if (*(BOOL *)Path) 
  132.                 ras = TRUE;
  133.             RegCloseKey(hSubKey);
  134.         }
  135.         if (!verok) 
  136.             continue;
  137.         // select the proper image:
  138.         int iImage;
  139.         if (local)
  140.             iImage  = IND_SRVLOCAL;
  141.         else
  142.               iImage  = ras ? IND_SRVRAS : IND_SRVREMNR;
  143.         HTREEITEM it = m_SrvrTree->InsertItem
  144.                        (
  145.                            TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM,
  146.                            Server,
  147.                            iImage,
  148.                            iImage,
  149.                            0, 0, 
  150.                            0,
  151.                            TVI_ROOT,
  152.                            TVI_LAST
  153.                        );
  154.         m_SrvrTree->InsertItem
  155.         (
  156.             TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM,
  157.             NULL,
  158.             IND_APPL,
  159.             IND_APPL,
  160.             0, 0, 
  161.             1,
  162.             it,
  163.             TVI_LAST
  164.         );
  165. // fictive branch, used to expand the item:
  166.               //is.hParent= anItem;
  167.             //TreeView_InsertItem(hTree, &is);
  168.     } 
  169.     while (TRUE);
  170.     RegCloseKey(hKey);
  171.     return(TRUE);
  172. }
  173.  
  174. void CSelSrvrApplDlg::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
  175. {
  176.     // TODO: Add your control notification handler code here
  177.     *pResult = 0;
  178.     HTREEITEM hItem = m_SrvrTree->GetSelectedItem();
  179.     if (!hItem)
  180.         return;
  181.  
  182.     if (IsAppl(hItem))
  183.     {
  184.         SetDlgItemText(IDC_APPLEDIT, m_SrvrTree->GetItemText(hItem));
  185.     }
  186.     else
  187.     {
  188.         TV_ITEM ti;
  189.         ti.mask       = TVIF_TEXT;
  190.         ti.hItem      = hItem;
  191.         ti.pszText    = m_SelServerName;
  192.         ti.cchTextMax = sizeof(m_SelServerName);
  193.         m_SrvrTree->GetItem(&ti);
  194.         SetDlgItemText(IDC_SRVREDIT, m_SelServerName);
  195.     }
  196. }
  197.  
  198. void CSelSrvrApplDlg::OnItemexpanding(NMHDR* pNMHDR, LRESULT* pResult) 
  199. {
  200.     NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  201.     // TODO: Add your control notification handler code here
  202.     *pResult = 0;
  203.     if (m_NoAppl)
  204.     {
  205.         *pResult = 1;
  206.         return;
  207.     }
  208.     if (pNMTreeView->action == TVE_EXPAND)
  209.     {
  210.         //m_SrvrTree->SetRedraw(FALSE);
  211.         CWBApp *WBApp   = (CWBApp *)AfxGetApp();
  212.         if (m_SelServer)
  213.         {
  214.             WBApp->Logout();
  215.             m_SrvrTree->Expand(m_SelServer, TVE_COLLAPSE);
  216.             m_SelServer = NULL;
  217.         }
  218.         HTREEITEM hItem = pNMTreeView->itemNew.hItem;
  219.         m_SrvrTree->SelectItem(hItem);
  220.         CString Server  = m_SrvrTree->GetItemText(hItem);
  221.         if (!WBApp->WBAppInit(NULL, m_UserName, m_Password, Server))
  222.         {
  223.                *pResult = TRUE;
  224.             return;
  225.         }
  226.         HTREEITEM delItem;
  227.         while ((delItem = m_SrvrTree->GetChildItem(hItem)))
  228.             m_SrvrTree->DeleteItem(delItem);
  229.         m_SelServer = hItem;
  230.         cdp_t cdp = WBApp->GetCdp();
  231.         tcursnum Curs;
  232.         if (cd_Open_cursor_direct(cdp, "SELECT OBJ_NAME FROM OBJTAB WHERE CATEGORY=Chr(7)", &Curs))
  233.         {
  234.             Signalize();
  235.             return;
  236.         }
  237.         tobjname Appl;
  238.         trecnum  Pos = 0;
  239.         while (!cd_Read_ind(cdp, Curs, Pos++, 1, NOINDEX, Appl))
  240.         {
  241.             small_string(Appl, TRUE);
  242.             m_SrvrTree->InsertItem
  243.             (
  244.                 TVIF_TEXT | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM,
  245.                 Appl,
  246.                 IND_APPL,
  247.                 IND_APPL,
  248.                 0, 0, 
  249.                 1,
  250.                 hItem,
  251.                 TVI_LAST
  252.             );
  253.         }
  254.         if (cd_Sz_error(cdp) != OUT_OF_TABLE)
  255.             Signalize();
  256.         Close_cursor(Curs);
  257. //redraw:
  258.         //m_SrvrTree->SetRedraw(TRUE);
  259.     }
  260.     else if (pNMTreeView->action == TVE_COLLAPSE)
  261.     {
  262.         ((CWBApp *)AfxGetApp())->Logout();
  263.         m_SelServer = NULL;
  264.     }
  265. }
  266.  
  267. void CSelSrvrApplDlg::OnDblclk(NMHDR* pNMHDR, LRESULT* pResult) 
  268. {
  269.     // TODO: Add your control notification handler code here
  270.     *pResult = 0;
  271.     if (m_NoAppl || IsAppl())
  272.         OnOK();
  273. }
  274.  
  275. void CSelSrvrApplDlg::OnChangeEdit() 
  276. {
  277.     // TODO: Add your control notification handler code here
  278.     tobjname ServerName;
  279.     CWnd *Edit = GetDlgItem(IDC_SRVREDIT);
  280.     int sz = Edit->GetWindowText(ServerName, sizeof(ServerName));
  281.     if (lstrcmp(ServerName, m_SelServerName))
  282.     {
  283.         if (m_SelServer)
  284.         {
  285.             m_SrvrTree->Expand(m_SelServer, TVE_COLLAPSE);
  286.             m_SelServer = NULL;
  287.         }
  288.         ((CWBApp *)AfxGetApp())->Logout();
  289.     }
  290.     Edit = GetDlgItem(IDC_APPLEDIT);
  291.     GetDlgItem(IDOK)->EnableWindow(sz && (m_NoAppl || Edit->GetWindowTextLength()));
  292. }
  293.  
  294. void CSelSrvrApplDlg::OnOK() 
  295. {
  296.     // TODO: Add extra validation here
  297.     tobjname SrvrAppl;
  298.     CWBApp *WBApp = (CWBApp *)AfxGetApp();
  299.     if (!WBApp->Linked())
  300.     {
  301.         if (!GetDlgItemText(IDC_SRVREDIT, SrvrAppl, sizeof(SrvrAppl)))
  302.             return;
  303.         if (!WBApp->WBAppInit(NULL, m_UserName, m_Password, SrvrAppl))
  304.             return;
  305.     }
  306.     if (!m_NoAppl)
  307.     {    
  308.         if (!m_Appl)
  309.         {
  310.             if (!GetDlgItemText(IDC_APPLEDIT, SrvrAppl, sizeof(SrvrAppl)))
  311.                 return;
  312.             m_Appl = SrvrAppl;
  313.         }
  314.         if (!WBApp->SetApplication(m_Appl))
  315.             return;
  316.     }
  317.     CDialog::OnOK();
  318. }
  319.  
  320. BOOL CSelSrvrApplDlg::IsAppl(HTREEITEM hItem)
  321. {
  322.     TV_ITEM   ti;
  323.     ti.mask  = TVIF_PARAM;
  324.     ti.hItem = hItem;
  325.     if (!hItem)
  326.         ti.hItem = m_SrvrTree->GetSelectedItem();
  327.     m_SrvrTree->GetItem(&ti);
  328.     return ti.lParam;
  329. }
  330.  
  331. void CSelSrvrApplDlg::OnDestroy() 
  332. {
  333.     CDialog::OnDestroy();
  334.     
  335.     // TODO: Add your message handler code here
  336.     CImageList *il = m_SrvrTree->GetImageList(TVSIL_NORMAL);
  337.     delete il;
  338. }
  339.  
  340.