MessageWriteForm.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#ifdef __WINNT__
#include <windows.h>
#endif
 
#include <dcl/DateTime.h>
#include <dcl/Files.h>
#include <dcl/Dir.h>
#include <dcl/Regex.h>
 
#include <dcl/Html.h>
#include <dcl/URI.h>
 
#include "ServletMain.h"
#include "MessageTree.h"
#include "HtmlPage.h"
#include "MessageView.h"
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("fastpage/MessageWriteForm.cpp");
#endif
 
__DCL_BEGIN_NAMESPACE
 
inline String __GetLocalFileName(
    const String& strAttachDir,
    const String& strDsID,
    const String& strMessageID,
    const String& strAttachNo
)
{
    return strAttachDir
        + strDsID + L"/"
        + strMessageID + L"-"
        + strAttachNo + L".dat";
}
 
IMPLEMENT_CLASSINFO_EX(MessageWriteForm, MessageView)
 
MessageWriteForm::MessageWriteForm(HtmlPage* pPage)
    : MessageView(pPage)
{
    __nThumbnailImageSize = 0;
}
 
void MessageWriteForm::init(SQLFields& fields)
    __DCL_THROWS1(Exception*)
{
    MessageView::init(fields);
 
    __mapParams.lookup(L"LIST", __strListPage);
    __mapParams.lookup(L"DETAIL", __strDetailPage);
 
    String rValue;
    if (__mapParams.lookup(L"THUMBNAIL_SIZE", rValue)) {
        __nThumbnailImageSize = Integer::parse(rValue, 10, 110);
        if (__nThumbnailImageSize <= 0)
            __nThumbnailImageSize = 110;
    }
}
 
static String __GetLinkInsertedString(const wchar_t* psz)
{
    // "http://"로 시작하고 공백이 아닌 문자열
    // 즉, L"http://"로 시작하고 space( \r\n) 이전까지
    Regex re(
        L"http://[^ \t\r\n\"\'<>]+",
//      "(http://[^ \t\r\n]+)|"
//          "(mailto:[a-z0-9_-]+@[a-z0-9_-]+(.[a-z0-9_-]+){1,2})",
//      "http://[^ \t\r\n]+|mailto:[a-z0-9_-]+@[a-z0-9._-]+",
        Regex::ICASE
    );
 
    const wchar_t* begin = psz;
    const wchar_t* end = psz + String::length(psz);
    StringBuilder r;
 
    Regex::MatchResults m;
    while (begin < end && re.search(begin, end, m)) {
        __DCL_ASSERT(m.size() > 0);
 
        if (begin < m[0].first) {
            r.append(begin, m[0].first);
        }
        r.append(L"<a href=\"")
            .append(m[0].first, m[0].second)
            .append(L"\" target=\"_new\">")
            .append(m[0].first, m[0].second)
            .append(L"</a>");
 
        begin = m[0].second;
    }
 
    if (begin < end) {
        r.append(begin, end);
    }
 
    return r;
}
 
void MessageWriteForm::onPrint(Session& session)
    __DCL_THROWS1(Exception*)
{
    // GET 일 때만 사용
    ListedStringToStringArrayMap& mapQuery = session.__ctx.__queryMap;
    SQLQuery& q = session.__query;
 
    String strTableID = String::valueOf(__nTableID);
    String strDsID = String::valueOf(__nDsID);
 
    String strMessageID = getDefault(mapQuery, L"modify");
    if (strMessageID.isEmpty()) {
        // 새글
        assign(L"BODY_TYPE", L"-1");
        assign(L"BODY_VALUE", L"");
        erase(L"ATTACH");
        assign(L"WRITER", session.__strUserName);
        /*
        ListedStringToStringArrayMap::ConstIterator itMap = mapQuery.find(L"msg");
        if (itMap != mapQuery.end())
            mapQuery[L"reply"] = (*itMap).value;
        */
        strMessageID = getDefault(mapQuery, L"reply");
        if (strMessageID.isEmpty() || strMessageID == L"0") {
            // 시작글
            if (session.isSysAdmin()) {
                TextTemplate& IS_ADMIN = (*this)[L"IS_ADMIN"];
                IS_ADMIN.assign(L"VNO", VNO());
                assign(L"IS_ADMIN", IS_ADMIN);
            }
            else
                erase(L"IS_ADMIN");
 
            assign(L"SUBJECT", String());
        }
        else {
            // 답글
            erase(L"IS_ADMIN");
            assign(L"ACTION_EX", L"reply=" + strMessageID);
 
            StringBuilder strRe = L"Re: ";
            q.execute(L""
                "SELECT SUBJECT"
                "\n FROM DCL_MESSAGE_" + strTableID + L""
                "\n WHERE DS_ID = " + strDsID + L""
                "\n  AND MESSAGE_ID = " + strMessageID
            );
            q.fetch();
            if (!q.eof() && !q.fields()[0].isNull())
                strRe += q.fields()[0].asString();
 
            assign(L"SUBJECT", strRe);
        }
    }
    else {
        // 수정
        erase(L"IS_ADMIN");
        TextTemplate& IS_MODIFY = (*this)[L"IS_MODIFY"];
        IS_MODIFY.assign(L"VNO", VNO());
        assign(L"IS_MODIFY", IS_MODIFY);
        assign(L"ACTION_EX", L"modify=" + strMessageID);
 
        q.execute(L""
            "SELECT M.TYPE AS BODY_TYPE, M.SUBJECT, M.BODY_ORG,"
            "\n  U.USER_NAME AS WRITER,"
            "\n  A.NO, A.FILENAME, CEILING(A.SIZE / 1024) AS KSIZE"
            "\n FROM (DCL_MESSAGE_" + strTableID + L" AS M"
            "\n  INNER JOIN DCLWC_USER AS U"
            "\n    ON(M.USER_ID = U.USER_ID))"
            "\n   LEFT OUTER JOIN DCL_MESSAGE_A_" + strTableID + L" AS A"
            "\n     ON (M.DS_ID = A.DS_ID AND M.MESSAGE_ID = A.MESSAGE_ID)"
            "\n WHERE M.DS_ID = " + strDsID + L""
            "\n  AND M.MESSAGE_ID = " + strMessageID + L""
            "\n ORDER BY A.NO ASC"
        );
        q.fetch();
        if (!q.eof()) {
            assign(q.fields(), L"");
            SQLField& fBODY_ORG = q.fields().byName(L"BODY_ORG");
            String strBodyValue;
            if (!fBODY_ORG.isNull()) {
                strBodyValue = fBODY_ORG.asString();
                strBodyValue = String::escape(strBodyValue, strBodyValue.length());
            }
            assign(L"BODY_VALUE", strBodyValue);
 
            SQLField& AttachNo = q.fields().byName(L"NO");
            if (AttachNo.isNull())
                erase(L"ATTACH");
            else {
                TextTemplate& ATTACH = (*this)[L"ATTACH"];
                TextTemplate& LIST = ATTACH[L"LIST"];
                do {
                    LIST.assign(L"VNO", VNO());
                    LIST.assign(L"MESSAGE_ID", strMessageID);
                    LIST.assign(L"NO", q.fields().byName(L"NO").asString());
                    LIST.assign(L"FILENAME", q.fields().byName(L"FILENAME").asString());
                    LIST.assign(L"KSIZE", q.fields().byName(L"KSIZE").asString());
                    ATTACH.append(L"LIST", LIST);
                    q.fetch();
                } while(!q.eof());
 
                assign(L"ATTACH", ATTACH);
            }
        }
    }
 
    MessageView::onPrint(session);
}
 
#if 0
bool GenThumbnailImage(
        const char* pszSrcFile,
        const char* pszContentType,
        const char* pszThumbnailFile,
        int nSize // = 110
    );
#endif
 
void MessageWriteForm::onPost(Session& session)
    __DCL_THROWS1(Exception*)
{
//  HtmlView::dump(*(__pPage->session()));
//  return ;
 
    ListedStringToStringArrayMap& mapQuery = session.__ctx.__queryMap;
    ListedStringToStringArrayMap& mapForm = session.__ctx.__formMap;
    SQLQuery& q = session.__query;
 
    String strTableID = String::valueOf(__nTableID);
    String strDsID = String::valueOf(__nDsID);
 
    String strListPage = getDefault(mapQuery, L"list");
 
    if (session.isSysGuest()) {
        StringBuilder strRefresh = L"?page=";
        if (!strListPage.isEmpty())
            strRefresh += strListPage;
        else
            strRefresh += __strListPage;
        __pPage->refresh(session, strRefresh);
        return;
    }
 
#define CM_UPDATE           0
#define CM_INSERT_START     1
#define CM_INSERT_REPLY     2
 
    int nCM = CM_UPDATE;
    bool bNotice = false;
    int nBodyType = BODY_TYPE_PLAIN_BR;
 
    String strMessageID = getDefault(mapQuery, L"modify");
    if (strMessageID.isEmpty()) {
        strMessageID = getDefault(mapQuery, L"reply");
        if (strMessageID.isEmpty() || strMessageID == L"0") // 새로운 시작글
            nCM = CM_INSERT_START;
        else // 답글
            nCM = CM_INSERT_REPLY;
    }
    else // 수정
        nCM = CM_UPDATE;
 
    nBodyType = Integer::parse(
        getDefault(mapForm, L"BODY_TYPE"),
        10,
        nBodyType
    );
    if (nBodyType > BODY_TYPE_MAX)
        nBodyType = BODY_TYPE_PLAIN_BR;
 
    if (mapForm.find(L"NOTICE") != mapForm.end()) {
        bNotice = true;
 
        // mapForm의 action을 잘못 기술했다. 
        // 공지사항은 항상 시작글이고 답글은 없으며 수정할 때에도 이 값은 없어야 한다.
        __DCL_ASSERT(nCM == CM_INSERT_START);
    }
 
    String strSubject = getDefault(mapForm, L"SUBJECT");
    String strBodyOrg = getDefault(mapForm, L"BODY_VALUE");
    String strBodyPlain = Html::strip(strBodyOrg, NULL); // 모든 html element
    String strBodyHtml;
 
    if (!strSubject.trim().isEmpty()) {
        strSubject = Html::escape(strSubject, L"<>");
    }
 
    switch(nBodyType) {
        case BODY_TYPE_PLAIN_BR :
            strBodyHtml = 
                __GetLinkInsertedString(
                    Html::format(
                        strBodyOrg,
                        4,          // TAB 문자를 4개의 ' '로 변환
                        String(),   // 빈 문자열
                        L"<br>"
                    )
                );
            break;
        case BODY_TYPE_PLAIN_P :
            strBodyHtml =
                __GetLinkInsertedString(
                    Html::format(
                        strBodyOrg,
                        4,          // TAB 문자를 4개의 ' '로 변환
                        L"<p>",
                        L"</p>"
                    )
                );
            break;
        case BODY_TYPE_HTML :
            strBodyHtml = strBodyOrg.replace_r(L"<[ \t\r\n]*body", L"<div", true);
            strBodyHtml = strBodyHtml.replace_r(L"/[ \t\r\n]*body", L"/div", true);
            strBodyHtml = Html::strip(
                strBodyHtml,
                L"!,--,html,head,title,meta,link,script,style,body"
                ",mapForm,input,button,textarea,select,option"
            );
            break;
    }
 
    int nAttachNo = 1;
    if (nCM == CM_UPDATE) {
        q.execute(L""
            "SELECT MESSAGE_ID"
            "\n FROM DCL_MESSAGE_" + strTableID + L""
            "\n WHERE DS_ID = " + strDsID + L""
            "\n  AND MESSAGE_ID = " + strMessageID + L""
            "\n  AND USER_ID = " + String::valueOf(session.__nUserID)
        );
        q.fetch();
        if (!q.eof()) {
            q.prepare(L""
                "UPDATE DCL_MESSAGE_" + strTableID + L""
                "\n SET TYPE = :TYPE, SUBJECT = :SUBJECT"
                    ", BODY_ORG = :BODY_ORG, BODY_PLAIN = :BODY_PLAIN"
                    ", BODY_HTML = :BODY_HTML, UPDATE_TIME = CURRENT_TIMESTAMP"
                "\n WHERE DS_ID = " + strDsID + L""
                    " AND MESSAGE_ID = " + strMessageID
            );
            SQLParams& params = q.params();
            params.byName(L"TYPE").setValue(nBodyType);
            params.byName(L"SUBJECT").setValue(strSubject);
            params.byName(L"BODY_ORG").setValue(strBodyOrg);
            params.byName(L"BODY_PLAIN").setValue(strBodyPlain);
            params.byName(L"BODY_HTML").setValue(strBodyHtml);
            q.execute();
 
            // 삭제할 이전 첨부파일을 삭제한다.
            StringArray& vDeleteAttach = mapForm[L"DELETE_ATTACH"];
            String strEraseAttach = String::join(vDeleteAttach, L',');
            if (!strEraseAttach.isEmpty()) {
                q.execute(L""
                    "DELETE FROM DCL_MESSAGE_A_" + strTableID + L""
                    "\n WHERE DS_ID = " + strDsID + L""
                    "\n  AND MESSAGE_ID = " + strMessageID + L""
                    "\n  AND NO IN (L" + strEraseAttach + L")"
                );
                if (q.affectedRows()) {
                    for(size_t i = 0; i < vDeleteAttach.size(); i++) {
                        String strFileName = __GetLocalFileName(
                            __pPage->site()->strAttachmentDir,
                            strDsID,
                            strMessageID,
                            vDeleteAttach[i]
                            );
                        if (Files::exists(strFileName))
                            Files::unlink(strFileName);
                    }
                }
            }
 
            // 첨부파일이 있는 경우 새로운 첨부파일의 시작번호를 얻어온다.
            q.execute(L""
                "SELECT MAX(NO) FROM DCL_MESSAGE_A_" + strTableID + L""
                "\n WHERE DS_ID = " + strDsID + L""
                "\n  AND MESSAGE_ID = " + strMessageID
            );
            q.fetch();
            if (!q.fields()[0].isNull())
                nAttachNo = q.fields()[0].asInteger() + 1;
        }
        /*
        else
        {
            EOF인 경우 메시지가 없거나 세션 사용자의 글이 아닌 경우이다. 무시한다.
        }
        */
 
    }
    else {
        // INSERT
        int nParentID = 0;
        int nPrevID = 0;
        int nMessageID, nStartID, nChild = 0;
        int64_t nReplyID = 1;
 
        q.execute(L"SET AUTOCOMMIT = 0");
        q.execute(L"COMMIT");
        q.execute(L""
            "SELECT MESSAGE_ID, NOTICE_ID"
            "\n FROM DCLWC_DATA_SOURCE"
            "\n WHERE DS_ID = " + strDsID + L""
            "\n FOR UPDATE"
            );
        q.fetch();
        if (bNotice)
            nMessageID = nStartID = q.fields()[1].asInteger();
        else
            nMessageID = nStartID = q.fields()[0].asInteger();
 
        if (nCM == CM_INSERT_REPLY) {
            MessageTree mt(__nReplyDepth);
 
            q.execute(L""
                "SELECT MESSAGE_ID, START_ID, REPLY_ID, NCHILD"
                "\n FROM DCL_MESSAGE_" + strTableID + L""
                "\n WHERE DS_ID = " + strDsID + L""
                "\n  AND MESSAGE_ID = " + strMessageID
            );
            q.fetch();
            if (q.eof()) {
                // 상위 메시지가 없음
            }
 
            nParentID = q.fields()[0].asInteger();
            nStartID = q.fields()[1].asInteger();
            nReplyID = q.fields()[2].asInt64();
            nChild = q.fields()[3].asInteger();
 
            MessageTree::Position pos;
            mt.getPosition(nReplyID, pos);
            if (nChild == mt.width()
                || pos.nDepthIndex == (mt.depth() - 1)
            ) {
                // 부모노드가
                // 최대 답글 개수에 도달했거나 최대 깊이에 도달했기 때문에
                // 더이상 답글을 달 수 없다.
 
            }
 
            // 답글의 REPLY_ID를 구한다.
            // 내림차순은 자식노드의 마지막부터, 오름차순은 첫번째부터 구한다.
            int64_t nPrevReplyID = 0;
            if (__nSequence < 0) {
                int64_t _r = nReplyID;
                nReplyID = mt.getChildID(nReplyID, mt.width() - nChild - 1);
                if (nChild > 0)
                    nPrevReplyID = mt.getChildID(_r, mt.width() - nChild);
            }
            else {
                int64_t _r = nReplyID;
                nReplyID = mt.getChildID(nReplyID, nChild);
                if (nChild > 0)
                    nPrevReplyID = mt.getChildID(_r, nChild - 1);
            }
 
            if (nChild > 0) {
                __DCL_TRACE2(L"%d, %ls\n", nChild, Int64::toString(nPrevReplyID).data());
                q.execute(L""
                    "SELECT MESSAGE_ID FROM DCL_MESSAGE_" + strTableID + L""
                    "\n WHERE DS_ID = " + strDsID + L""
                    "\n  AND START_ID = " + String::valueOf(nStartID) + L""
                    "\n  AND REPLY_ID = " + Int64::toString(nPrevReplyID)
                    );
                q.fetch();
                __DCL_ASSERT(!q.eof());
                nPrevID = q.fields()[0].asInteger();
            }
        }
        else {
            StringBuilder strSQL = L""
                "SELECT MAX(MESSAGE_ID) FROM DCL_MESSAGE_" + strTableID + L""
                "\n WHERE DS_ID = " + strDsID + L""
                "\n  AND PARENT_ID = 0";
            if (!bNotice)
                strSQL += L" AND MESSAGE_ID <= 2000000000";
 
            TRACE0(L"000\n");
            q.execute(strSQL);
            q.fetch();
            __DCL_ASSERT(!q.eof());
            if (!q.fields()[0].isNull())
                nPrevID = q.fields()[0].asInteger();
            TRACE0(L"111\n");
        }
 
        q.prepare(L""
            "INSERT INTO DCL_MESSAGE_" + strTableID + L""
            "\n (DS_ID, MESSAGE_ID, PREV_ID, NEXT_ID, PARENT_ID, START_ID, REPLY_ID,"
                " TYPE, USER_ID, SUBJECT, BODY_ORG, BODY_PLAIN, BODY_HTML,"
                " UPDATE_TIME)"
            "\n VALUES"
            "\n (:DS_ID, :MESSAGE_ID, :PREV_ID, :NEXT_ID, :PARENT_ID, :START_ID, :REPLY_ID,"
                " :BODY_TYPE, :USER_ID, :SUBJECT, :BODY_ORG, :BODY_PLAIN, :BODY_HTML,"
                " CURRENT_TIMESTAMP)"
        );
        SQLParams& params = q.params();
        params.byName(L"DS_ID").setValue(__nDsID);
        params.byName(L"MESSAGE_ID").setValue(nMessageID);
        if (__nSequence < 0) {
            params.byName(L"PREV_ID").setValue(0);
            params.byName(L"NEXT_ID").setValue(nPrevID);
        }
        else {
            params.byName(L"PREV_ID").setValue(nPrevID);
            params.byName(L"NEXT_ID").setValue(0);
        }
        params.byName(L"PARENT_ID").setValue(nParentID);
        params.byName(L"START_ID").setValue(nStartID);
        params.byName(L"REPLY_ID").setValue(nReplyID);
        params.byName(L"BODY_TYPE").setValue(nBodyType);
        params.byName(L"USER_ID").setValue(session.__nUserID);
        params.byName(L"SUBJECT").setValue(strSubject);
        params.byName(L"BODY_ORG").setValue(strBodyOrg);
        params.byName(L"BODY_PLAIN").setValue(strBodyPlain);
        params.byName(L"BODY_HTML").setValue(strBodyHtml);
        q.execute();
 
        if (nPrevID > 0) {
            if (__nSequence < 0)
                q.execute(L""
                    "UPDATE DCL_MESSAGE_" + strTableID + L""
                    "\n SET PREV_ID = " + String::valueOf(nMessageID) + L""
                    "\n WHERE DS_ID = " + strDsID + L""
                    "\n  AND MESSAGE_ID = " + String::valueOf(nPrevID)
                );
            else
                q.execute(L""
                    "UPDATE DCL_MESSAGE_" + strTableID + L""
                    "\n SET NEXT_ID = " + String::valueOf(nMessageID) + L""
                    "\n WHERE DS_ID = " + strDsID + L""
                    "\n  AND MESSAGE_ID = " + String::valueOf(nPrevID)
                );
        }
 
        if (nCM == CM_INSERT_REPLY) {
            String strChildID;
            if (nChild == 0)
                strChildID = L", CHILD_ID = " + String::valueOf(nMessageID);
 
            q.execute(L""
                "UPDATE DCL_MESSAGE_" + strTableID + L""
                "\n SET NCHILD = NCHILD + 1" + strChildID + L""
                "\n WHERE DS_ID = " + strDsID + L""
                "\n  AND MESSAGE_ID = " + strMessageID
            );
        }
 
        if (bNotice)
            q.execute(L""
                "UPDATE DCLWC_DATA_SOURCE"
                "\n SET NOTICE_ID = NOTICE_ID + 1"
                "\n WHERE DS_ID = " + strDsID
            );
        else
            q.execute(L""
                "UPDATE DCLWC_DATA_SOURCE"
                "\n SET MESSAGE_ID = MESSAGE_ID + 1"
                "\n WHERE DS_ID = " + strDsID
            );
        q.execute(L"COMMIT");
        q.execute(L"SET AUTOCOMMIT = 1");
        strMessageID = String::valueOf(nMessageID);
    }
 
    // 첨부파일이 있으면 삽입한다.
    StoredHttpFormData& formFile = session.__ctx.__formFileMap;
    if (!formFile.isEmpty()) {
        String strDir = __pPage->site()->strAttachmentDir + strDsID;
        if (!Files::exists(strDir))
            Files::mkdir(strDir);
 
        if (__nThumbnailImageSize > 0) {
            strDir = __pPage->site()->strThumbnailDir + strDsID;
            if (!Files::exists(strDir))
                Files::mkdir(strDir);
        }
 
        q.prepare(L""
            "INSERT INTO DCL_MESSAGE_A_" + strTableID + L""
            "\n  (DS_ID, MESSAGE_ID, NO, FILENAME, TYPE)"
            "\n VALUES"
            "\n  (L" + strDsID + L", L" + strMessageID + L""
                ", :NO, :FILENAME, :TYPE)"
        );
        for(size_t i = 0; i < formFile.size(); i++) {
            StoredHttpFormData::FileInfoArray& v = formFile[i];
            for(size_t j = 0; j < v.size(); j++)
            {
                StoredHttpFormData::FileInfo& info = v[j];
 
                String strLocalFileName = __GetLocalFileName(
                        __pPage->site()->strAttachmentDir,
                        strDsID,
                        strMessageID,
                        String::valueOf(nAttachNo)
                    );
                Files::rename(
                    info.tempFilename,
                    strLocalFileName
                );
                if (__nThumbnailImageSize > 0) {
#if 0
                    GenThumbnailImage(
                        strLocalFileName,
                        info.strContentType,
                        __GetLocalFileName(
                            __pPage->site()->strThumbnailDir,
                            strDsID,
                            strMessageID,
                            String::valueOf(nAttachNo)
                            ) + L".jpg",
                        __nThumbnailImageSize
                    );
#endif
                }
                q.params()[0].setValue(nAttachNo);
                q.params()[1].setValue(info.filename);
                q.params()[2].setValue(info.contentType);
                q.execute();
 
                nAttachNo++;
            }
        }
    }
 
    StringBuilder strNext = getDefault(mapQuery, L"next");
    if (!strNext.isEmpty())
        strNext = L"?page=" + strNext.toString();
    else {
        strNext = L"?page=" + __strDetailPage;
        strNext += L"&msg=" + strMessageID;
    }
 
    if (!strListPage.isEmpty())
        strNext += L"&list=" + URLEncoder::encode(strListPage);
 
    __pPage->refresh(session, strNext);
}
 
__DCL_END_NAMESPACE