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