#include "SRModuleView.h"
#include "SRFramework/WCString.h"
#include "SRMainFrame.h"
#include <string.h>

DWORD WINAPI LoadTextThread(LPVOID arg)
{
    SRModuleView *view = reinterpret_cast<SRModuleView*>(arg);
    if(!view)
        return 1;
        
    view->m_fThreadRunning = TRUE;
    view->m_wStatusBarHeight = SR_STATUS_BAR_HEIGHT;
    view->LoadTextView(true);
    view->m_wStatusBarHeight = 0;
    if(!view->m_fAbortThread){
        view->RefreshWindow();
    }
    view->m_fThreadRunning = FALSE;
    view->m_fAbortThread = FALSE;

    return 0;
}

SRModuleView::SRModuleView(SRViewType viewType)
: SRTextView()
, m_pModule(NULL)
, m_fChapterChanged(FALSE)
, m_fSwordInit(FALSE)
, m_fThreadRunning(FALSE)
, m_fAbortThread(FALSE)
, m_viewType(viewType)
, m_verse("Gen 1:1")
{
}

SRModuleView::~SRModuleView(void)
{
    if(m_hLoadTextThread)
        CloseHandle(m_hLoadTextThread);
}

BOOL SRModuleView::OnCommand(WORD wNotifyCode, WORD wID, HWND hWndCtl)
{
    if(m_pParentWnd){
        return m_pParentWnd->OnCommand(wNotifyCode, wID, hWndCtl);
    }
    return FALSE;
}

VOID SRModuleView::SetModule(SWModule *pModule)
{
    if(m_pModule == pModule)
        return;
    AbortThread();
    m_pModule = pModule;
    m_fChapterChanged = TRUE;
    LoadTextView();
}

VOID SRModuleView::AbortThread()
{
    DWORD dwTimeout = 200;
    INT rc;
    if(m_fThreadRunning){
        m_fAbortThread = TRUE;
        rc = WaitForSingleObject(m_hLoadTextThread, dwTimeout);
        while(rc == WAIT_FAILED){
            ::Sleep(20);// Let's sleep for a bit and give the thread a chance to spawn.
            rc = WaitForSingleObject(m_hLoadTextThread, dwTimeout);
        }
        if(rc == WAIT_TIMEOUT){
            // Sometimes the call to ::UpdateWindow never returns. 
            TerminateThread(m_hLoadTextThread, -1);
            // Clean up from the rogue thread
            m_fThreadRunning = FALSE;
            m_fAbortThread = FALSE;
            m_wStatusBarHeight = 0;
        }
    }
}

VOID SRModuleView::SetKey(const VerseKey &verse)
{
    if(!m_fChapterChanged && m_verse == verse)
        return;
    AbortThread();
    if(m_viewType == BibleView){
        if( (verse.Book() != m_verse.Book()) || ((m_verse.Book() == verse.Book()) && (m_verse.Chapter() != verse.Chapter()))){
            m_fChapterChanged = true;
            m_verse = verse;
            LoadTextView();
        }else{
            ScrollToVerse(verse.Verse());
        }
    }else if(m_viewType == CommView){
        if(m_verse != verse){
            m_fChapterChanged = true;
            m_verse = verse;
            LoadTextView();
        }
    }
}

VOID SRModuleView::SetBook(INT nBook)
{
    if(nBook > BIBLE_OT_BOOKS){
        if( m_verse.Testament() != 2 || m_verse.Book() + BIBLE_OT_BOOKS != nBook){
            m_verse.Testament(SWORD_NEW_TESTAMENT);
            m_verse.Book(nBook - BIBLE_OT_BOOKS);
            m_verse.Chapter(1);
            m_verse.setVerse(1);
        }
    }else{
        if( m_verse.Testament() != 1 || m_verse.Book() != nBook){
            m_verse.Testament(SWORD_OLD_TESTAMENT);
            m_verse.Book(nBook);
            m_verse.Chapter(1);
            m_verse.setVerse(1);
        }
    }
    m_fChapterChanged = true;
}

VOID SRModuleView::SetChapter(INT nChapter)
{
    if(m_verse.Chapter() != nChapter){
        m_verse.Chapter(nChapter);
        m_verse.setVerse(1);
        m_fChapterChanged = true;
    }
}

VOID SRModuleView::SetVerse(INT nVerse, BOOL fScroll)
{
    if(nVerse < 0)
        return; // This isn't a valid verse.
    if(nVerse == 0)
        nVerse++; // This chapter starts with a heading or some other pretext, so we normalize the verse number here.

    if(m_viewType == CommView && m_verse.Verse() == nVerse)
        return;
    
    m_verse.setVerse(nVerse);
    if(fScroll)
        ScrollToVerse(m_verse.Verse());
}

VOID SRModuleView::Show()
{
    SRTextView::Show();
    if(m_fLoadTextDelayed)
        LoadTextView(FALSE);
}

VOID SRModuleView::LoadTextView(BOOL fInThread)
{
	WCString buf, text, s;
    BOOL done = FALSE;
    char strNum[10];
    

    if(!IsVisible()){
        m_fLoadTextDelayed = TRUE;
        return;
    }else
        m_fLoadTextDelayed = FALSE;
    
    if(!m_fSwordInit)
        return;

    if(!fInThread){
        AbortThread();
    }
    
    VerseKey keyCur(m_verse);
    
    if(!fInThread){ // This text should still be valid...
        if(!m_fChapterChanged)
            return;
        Clear();
        if(m_viewType == BibleView)
            keyCur.setVerse(1);
    }else{
        m_fChapterChanged = false;
        keyCur.setVerse(m_nLoadedTill + 1);
    }
    
    buf = "<html><body>";
    AddText(buf.w_str(), buf.length()); // a <style> section presumably gets skipped
    
    int i = 0;
    if(fInThread)
        i++;

    while (!done && (!fInThread || !m_fAbortThread)) {
        if(fInThread)
            i++;
        if(!m_pModule){
            buf = "<b>There are no modules of this type to be loaded.<br></b>Visit http://www.crosswire.org for information on available modules.";
            AddText(buf.w_str(), buf.length());
            break;
        }

        m_pModule->SetKey(keyCur);
        int pvHeading = 0;
        text = (TCHAR *)m_pModule->RenderText();
        if(m_viewType == CommView && text.length() == 0){
            text = "<b>This commentary (";
            text +=  m_pModule->Name();
            text += ") does not have an entry for this verse.</b>";
        }
        
        s = "";
		do {
			sprintf(strNum, "%i", pvHeading++);
            SWBuf preverseHeading = m_pModule->getEntryAttributes()["Heading"]["Preverse"][strNum].c_str();
			if (preverseHeading.length()) {
                if(keyCur.Verse() != 1)
                    s +=  "<br /><br /><b>";
                else
                    s += "<b>";
                s += (TCHAR *)m_pModule->RenderText(preverseHeading);
                s += "</b><br /><br />";
			}
			else break;
		} while (!fInThread || !m_fAbortThread);
        if(m_viewType == BibleView){
            if(text.c_str()[0] == '¶')
                s += "<br><br>";
            s += GetVerseHeader(keyCur);
            s += text + " ";
        }else if(m_viewType == CommView){
            s += GetVerseHeader(keyCur) + "<br>";
            s += text;
        }
		
        AddText(s.w_str(), s.length());
        if(m_viewType == BibleView){
            if (keyCur.Verse() == keyCur.getVerseMax() ){
    			done = TRUE;
                break;
            }else if(!fInThread && CurrentPageFilled()){
                done = FALSE;
                break;
            }else if(fInThread){
                DrawLoadingProgress(keyCur);
            }
        }else if(m_viewType == CommView){
            done = TRUE;
            break;
        }
        keyCur.increment(1);
	}

    buf = "</body></html>";
    AddText(buf.w_str(), buf.length());

    if(!fInThread || !m_fAbortThread)
        RefreshWindow();

    if(done){
        ScrollToVerse(m_verse.Verse());
    }else if(!fInThread && m_pModule){ // Overkill but, I am a paranoid guy.
        m_nLoadedTill = keyCur.Verse();
        CreateLoadTextThread();
    }
}

VOID SRModuleView::DrawLoadingProgress(const VerseKey& keyCur)
{
    const   INT BUFSIZE = 64;
    TCHAR   szStat[BUFSIZE] = {0};
    RECT    r = {0};
    float   fpPercent;
    RECT clientRect = GetBounds();
    HBRUSH  brushBG, brushPBAR;
    HGDIOBJ oldBrush, oldFont;
    LOGFONT lfFont;
    HFONT   font;
    INT nLen;
    HDC hdc; 
    COLORREF oldColor;

    r.left = 0;
    r.right = clientRect.right;
    r.top = clientRect.bottom - 12;
    r.bottom = clientRect.bottom;

    hdc = GetDC(m_hWnd);
    PreRenderBuff(hdc); // Allows the text to show as soon as it is loaded in.
    fpPercent = 100.f * (float)keyCur.Verse()/(float)keyCur.getVerseMax();
    swprintf(szStat, L"%2.1f%%", fpPercent);
    nLen = wcsnlen(szStat, BUFSIZE);

    brushBG =  CreateSolidBrush((COLORREF)BUTTON_BACKGROUND);
    brushPBAR =  CreateSolidBrush((COLORREF)PROGRESS_BACKGROUND);
    // Draw the status bar...
    oldBrush = SelectObject(hdc, GetStockObject(LTGRAY_BRUSH));
    Rectangle(hdc, r.left, r.top, r.right, r.bottom);
    // Draw the progress bar...
    r.right = (LONG)((r.right - r.left) * (fpPercent/100.0f));
    SelectObject(hdc, GetStockObject(HOLLOW_BRUSH));
    FillRect(hdc, &r, brushPBAR);
    Rectangle(hdc, r.left, r.top, r.right, r.bottom);
    // Set the font...
    SetBkColor(hdc, (COLORREF)PROGRESS_BACKGROUND);
    memset(&lfFont, 0, sizeof(lfFont));
    lfFont.lfHeight = -9;
    lfFont.lfWeight = FW_NORMAL;
    lfFont.lfQuality = CLEARTYPE_QUALITY;
    font = CreateFontIndirect(&lfFont);
    oldFont = SelectObject(hdc, font);
    oldColor = SetTextColor(hdc, (COLORREF)PROGRESS_TEXT_COLOR);
    // Draw the percent done...
    DrawText(hdc, &szStat[0], nLen, &r, DT_VCENTER | DT_CENTER | DT_SINGLELINE);
    // Restore sanity...
    SelectObject(hdc, oldFont);
    SetTextColor(hdc, oldColor);
    SelectObject(hdc, oldBrush);
    DeleteObject(font);
    DeleteObject(brushBG);
    DeleteObject(brushPBAR);

    ReleaseDC(m_hWnd, hdc);
}

VOID SRModuleView::CreateLoadTextThread()
{
    if(m_hLoadTextThread)
        ::CloseHandle(m_hLoadTextThread);
    m_fThreadRunning = TRUE;
    m_hLoadTextThread = ::CreateThread(NULL, 0, LoadTextThread, (LPVOID) this, 0, &m_dwLoadTextThreadID);
    if(m_hLoadTextThread)
        ::SetThreadPriority(m_hLoadTextThread, THREAD_PRIORITY_BELOW_NORMAL);
}

WCString SRModuleView::GetVerseHeader(VerseKey key) const
{
    wchar_t wverse[16] = {0};
    wchar_t wchap[16] = {0};
    _itow(key.Verse(), wverse, 10);
    _itow(key.Chapter(), wchap, 10);

	WCString result = "\n<b><a name=\"";
	result += wverse;
	result += "\">";
    if (key.Verse() == 1) {
		result += wchap;
		result += ":";
	}
	result += wverse;
	result += "</a></b>&nbsp;";

	return result;
}
    
const VerseKey& SRModuleView::GetVerseKey() const
{
    return m_verse;
}

VOID SRModuleView::RefreshScreen(BOOL fReloadText)
{
    if(fReloadText)
        m_fChapterChanged = TRUE;
    LoadTextView();
}


BOOL SRModuleView::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags)
{
    INT a = 0;
    RECT clientRect = GetBounds();
    switch(nChar)
    {
    case VK_UP:
        ScrollFullPage(1);
        if(GetVerseNum() > -1)
            m_verse.setVerse(GetVerseNum());
        break;
    case VK_DOWN:
        ScrollFullPage(-1);
        if(GetVerseNum() > -1)
            m_verse.setVerse(GetVerseNum());
        break;
    case VK_LEFT:
        if(m_viewType == BibleView){
            if(m_verse.Verse() > 1){
                m_verse.setVerse(1);
                ScrollToVerse(1);
            }else{
                m_verse.decrement(1);
                m_verse.setVerse(1);
                m_fChapterChanged = true;
                LoadTextView();
            }
        }else if(m_viewType == CommView){
            if(-m_nTop > (clientRect.bottom - clientRect.top)){
                ScrollToPosition(0,FALSE);
            }else{
                m_verse.decrement(1);
                m_fChapterChanged = true;
                LoadTextView();
            }
        }
        break;
    case VK_RIGHT:

        if(m_viewType == BibleView){
            m_verse.Chapter(m_verse.Chapter() + 1);
            m_fChapterChanged = true;
            LoadTextView();
        }else if(m_viewType == CommView){
            m_verse.increment(1);
            m_fChapterChanged = true;
            LoadTextView();
        }
        break;
    default:
        return FALSE;
    }
    
    return TRUE;
}

WCString SRModuleView::GetWindowTitle()
{    
    wchar_t wverse[16] = {0};
    wchar_t wchapt[16] = {0};
    if(!m_fSwordInit || !m_pModule)
        return "";
    WCString title =  
        WCString(m_pModule->Name()) + " " + 
        SRMainFrame::GetBookNames()[(m_verse.Book() - 1) + BIBLE_OT_BOOKS*(m_verse.Testament() - 1)] + WCString(" ") + 
        WCString(_itow(m_verse.Chapter(),wchapt, 10)) + WCString(":") + 
        WCString(_itow(m_verse.Verse(), wverse, 10));
    
    return title;
}