EShopServlet.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef __WINNT__
#include <windows.h>
#endif
 
#include <dcl/Charset.h>
#include <dcl/Files.h>
#include <dcl/IniFile.h>
 
//#include <dcl/net/HttpCollection.h>
#include <dcl/Html.h>
 
#include "EShopServlet.h"
#include "EShopSession.h"
#include "VisitorBook.h"
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("dcl/eshop/EShopServlet.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
HTTP_SERVLET_INSTANCE(EShopServlet, L"온라인 쇼핑몰")
 
IMPLEMENT_CLASSINFO(EShopServlet, HttpServletEx)
 
EShopServlet::EShopServlet()
{
    __pSQLConnPool = NULL;
    __pHead = NULL;
    __pFoot = NULL;
    __pViewOrder = NULL;
    __pViewProduct = NULL;
    __pViewShoppingBasket = NULL;
}
 
void EShopServlet::onInitialize()
{
    HttpServlet::onInitialize();   // 항상 true
 
#ifdef __WINNT__
#define INI_FILENAME L"HSAEShop_w.ini"
#else
#define INI_FILENAME L"HSAEShop.ini"
#endif
 
    String strIniFile = getIniFileName(INI_FILENAME);
    if (strIniFile.isEmpty()) {
        __DCL_TRACE1(L"not found : %ls\n", strIniFile.data());
        return;
    }
 
    String strSQLDriverName;
    String strSQLConnectionString;
 
    try {
        IniFile ini(strIniFile);
 
#ifdef __DCL_DEBUG
        String strEnableDebugOut = ini.getString(L"DEBUG", L"ENABLE_DEBUG_OUT");
        if (strEnableDebugOut.compareNoCase(L"FALSE") == 0)
            HttpServletEx::__enableDebugOut = false;
#endif
 
        strSQLDriverName = ini.getString(L"DATABASE", L"DRIVER");
        strSQLConnectionString = ini.getString(L"DATABASE", L"CONNECT");
 
        __DCL_TRACE3(L"%ls, %ls, %ls\n",
            strIniFile.data(),
            strSQLDriverName.data(),
            strSQLConnectionString.data()
        );
 
        __strTemplateDir = ini.getString(L"HTML_TEMPLATE", L"DIR");
 
        __pSQLConnPool = new SQLConnectionPool(
            strSQLConnectionString,
            strSQLDriverName
        );
 
        __pHead = new TextTemplate(readTemplate(L"header.html"));
        __pFoot = new TextTemplate(readTemplate(L"footer.html"));
        __pViewProduct = new TextTemplate(readTemplate(L"view_product.html"));
        __pViewShoppingBasket = new TextTemplate(readTemplate(L"view_shopping_basket.html"));
        __pViewOrder = new TextTemplate(readTemplate(L"view_order.html"));
    }
    catch (Exception* e) {
        __DCL_TRACE1(L"Init Failed %ls\n", e->toStringAll().data());
        e->destroy();
 
        onCleanup();
    }
}
 
void EShopServlet::onCleanup()
{
    if (__pSQLConnPool) {
        delete __pSQLConnPool;
        __pSQLConnPool = NULL;
    }
 
    if (__pHead) {
        delete __pHead;
        __pHead = NULL;
    }
 
    if (__pFoot) {
        delete __pFoot;
        __pFoot = NULL;
    }
 
    if (__pViewOrder) {
        delete __pViewOrder;
        __pViewOrder = NULL;
    }
 
    if (__pViewProduct) {
        delete __pViewProduct;
        __pViewProduct = NULL;
    }
 
    if (__pViewShoppingBasket) {
        delete __pViewShoppingBasket;
        __pViewShoppingBasket = NULL;
    }   
 
    HttpServlet::onCleanup();
    __DCL_TRACE0(L"onCleanup()\n");
}
 
void EShopServlet::onService(HttpServletContextEx& ctx)
{
    if (!(ctx.methodNo() == HTTP_METHOD_GET
        || ctx.methodNo() == HTTP_METHOD_POST)
    ) {
        // GET나 POST 아니면
        ctx.setStatusCode(HTTP_STATUS_METHOD_NOT_ALLOWED);
        return;
    }
 
    __DCL_ASSERT(__pSQLConnPool != NULL);
    __DCL_ASSERT(__pHead != NULL);
    __DCL_ASSERT(__pFoot != NULL);
    __DCL_ASSERT(__pViewProduct != NULL);
 
    SQLConnection* pSQLConn = __pSQLConnPool->getConnection();
    if (!pSQLConn) {
        printError(ctx, L"데이터베이스 서버에 연결할 수 없습니다.");
        return;
    }
 
    try {
        if (pSQLConn->canTransact()) {
            pSQLConn->startTrans();
        }
 
        Writer& out = ctx.writer();
        EShopSession session(ctx, pSQLConn);
        switch(session.command()) {
            case CM_LOGIN: {
                if (session.login(ctx.__formMap))
                    session.setCommand(CM_VIEW_PRODUCT);
                else
                    session.setCommand(CM_VIEW_LOGIN_FORM);
                break;
            }
            case CM_LOGOUT: {
                session.logout();
                session.setCommand(CM_ABOUT);
            }
        }
 
        // 복사본을 만들어 템플릿의 메크로들을 완성한다.
 
        printPageHeader(out, session);
 
        switch(session.command()) {
            case CM_VIEW_PRODUCT: {
                onViewProduct(ctx, session);
                break;
            }
            case CM_VIEW_LOGIN_FORM: {
                printLoginForm(out);
                break;
            }
            case CM_VIEW_SHOPPING_BASKET: {
                onViewShoppingBasket(ctx, session);
                break;
            }
            case CM_ORDER :
            case CM_VIEW_ORDER: {
                onViewOrder(ctx, session);
                break;
            }
            case CM_VIEW_QNA: {
                VisitorBook vb(
                    COMMAND_STR,
                    CM_VIEW_QNA,
                    L"vcm",
                    *this);
                if (!vb.onVisitorBook(ctx, out, pSQLConn)) {
                    printFile(out, L"notfound.html");
                }
                break;
            }
            case CM_SALES_REPORT: {
                if (session.isAdmin())
                    onViewSalesReport(ctx, session);
                else
                    printLoginForm(out);
                break;
            }
            case CM_SESSION_MANAGEMENT: {
                session.onViewSessions(
                    ctx.__queryMap,
                    out,
                    COMMAND_STR,
                    CM_SESSION_MANAGEMENT,
                    L"vses",
                    *this
                );
                break;
            }
            case CM_ABOUT: {
                printFile(out, L"about.html");
                break;
            }
            default: {
                printFile(out, L"notfound.html");
            }
        }
 
        if (pSQLConn->inTransaction()) {
            pSQLConn->commitTrans();
        }
        __pSQLConnPool->release(pSQLConn);
        pSQLConn = NULL;
 
        TextTemplate tplFooter = *__pFoot;
        out << tplFooter;
 
        ctx.setContentType(L"text/html");
        ctx.setStatusCode(HTTP_STATUS_OK);    // HTTP_OK
    }
    catch (Exception* _e) {
        if (pSQLConn) {
            if (pSQLConn->inTransaction()) {
                pSQLConn->rollbackTrans();
            }
            __pSQLConnPool->release(pSQLConn);
        }
        throw _e;
    }
}
 
void EShopServlet::printError(HttpServletContextEx& ctx, const String str)
{
    Writer& out = ctx.writer();
    out << L""
        "<!DOCTYPE html>\n"
        "<html lang=\"ko-KR\">\n"
        "<head>\n"
        "<meta charset=\"UTF-8\">\n"
        "<title>에러!</title>\n"
        "</head>\n"
        "<body>\n"
        "<h3>에러가 발생하여 요청을 완료할 수 없습니다.</h3>\n"
        "<hr>\n<p>"
        << Html::format(str, 4, L"<p>", L"</p>\n")
        << L"</p>\n"
        << L"</body>\n</html>\n";
 
    ctx.setContentType(L"text/html", L"utf-8");
    ctx.setStatusCode(HTTP_STATUS_OK);    // HTTP_OK
}
 
void EShopServlet::printLoginForm(
    Writer& out
)
{
    TextTemplate tpl(readTemplate(L"login_form.html"));
    String strHREF = String::format(L"?%ls=%d", COMMAND_STR, CM_LOGIN);
    tpl.assign(L"HREF_LOGIN_ACTION", strHREF);
 
    out << tpl;
}
 
struct MAIN_MENU
{
    bool            bAdminOnly;
    COMMAND_ID        nCmd;
    const wchar_t*  psz;
};
 
static MAIN_MENU menus[] =
{
    { false, CM_VIEW_PRODUCT,            L"상품정보" },
    { false, CM_VIEW_SHOPPING_BASKET,    L"장바구니" },
    { false, CM_VIEW_ORDER,                L"주문정보" },
    { false, CM_VIEW_QNA,                L"Q&A"  },
    { true,  CM_PRODUCT_MANAGEMENT,        L"상품관리" },
    { true,  CM_SALES_REPORT,            L"매출현황" },
    { true,  CM_SESSION_MANAGEMENT,        L"세션관리" },
    { false, CM_ABOUT,                  L"About"        },
//  { false, CM_VIEW_LOGIN_FORM,        L"로그인"  }
};
 
void EShopServlet::printPageHeader(
    Writer& out,
    const EShopSession& session
)
{
    TextTemplate tplHeader = *__pHead;
 
    if (!session.isGuest()) {
        StringBuilder str = L"<strong>" + session.userName();
        if (session.isAdmin()) {
            str += L"<span style=\"color:red\">";
            str += L"(관리자 권한)</span>";
        }
        str += L"</strong> 님 로그인 하셨습니다";
        tplHeader.assign(L"USER_HELLO", str);
    }
    else
        tplHeader.assign(L"USER_HELLO", L"&nbsp");
 
    // main menu
    String strTitle;
    for(unsigned int i = 0; i < (sizeof(menus) / sizeof(MAIN_MENU)); i++) {
        if (menus[i].bAdminOnly) {
            if (!session.isAdmin())
                continue;
        }
 
        if (i) {
            tplHeader.append(L"HREF", L"<span style=\"color: red\"> | </span>");
        }
 
        if (menus[i].nCmd == session.command()) {
            StringBuilder strHREF = L"<span style=\"color:blue\">";
            strHREF += menus[i].psz;
            strHREF += L"</span>\n";
 
            tplHeader.append(L"HREF", strHREF);
            strTitle = menus[i].psz;
        }
        else {
            String strHREF = String::format(L"<a class=\"menu\" href=\"?%ls=%d\">%ls</a>\n",
                COMMAND_STR, menus[i].nCmd, menus[i].psz);
            tplHeader.append(L"HREF", strHREF);
        }
    }
 
    tplHeader.append(L"HREF", L"<span style=\"color: red\"> | </span>");
 
    if (session.isGuest()) {
        if (session.command() == CM_VIEW_LOGIN_FORM) {
            StringBuilder strHREF = L"<span style=\"color:blue\">";
            strHREF += L"로그인";
            strHREF += L"</span>\n";
            tplHeader.append(L"HREF", strHREF);
            strTitle = L"로그인";
        }
        else {
            String strHREF = String::format(L"<a class=\"menu\" href=\"?%ls=%d\">%ls</a>\n",
                COMMAND_STR, CM_VIEW_LOGIN_FORM, L"로그인");
            tplHeader.append(L"HREF", strHREF);
        }
    }
    else {  
        String strHREF = String::format(L"<a class=\"menu\" href=\"?%ls=%d\">%ls</a>",
            COMMAND_STR, CM_LOGOUT, L"로그아웃");
        tplHeader.append(L"HREF", strHREF);
    }
 
    if (strTitle.isEmpty())
        tplHeader.erase(L"TITLE");
    else
        tplHeader.assign(L"TITLE", strTitle);
 
    out << tplHeader;
}
 
void EShopServlet::printFile(
    Writer& out,
    const wchar_t* _filename
)
{
    out << readTemplate(_filename);
}
 
String EShopServlet::readTemplate(const wchar_t* _filename) const
    __DCL_THROWS1(IOException*)
{
    String path = __strTemplateDir;
    if (!path.endsWith(L"/")) {
        path = path + L"/";
    }
    path = path + _filename;
 
    __DCL_TRACE1(L"readTemplate [%ls]\n", path.data());
 
    UTF8Decoder dec;
    return Files::readText(path, dec);
}
 
__DCL_END_NAMESPACE
 
#ifdef __WINNT_NEW_DELETE_OVERRIDE
#undef new
__WINNT_NEW_DELETE_OVERRIDE
#endif