/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * This file is part of the Collabora Office project.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

#include <swmodeltestbase.hxx>

#include <memory>

#include <com/sun/star/linguistic2/LinguServiceManager.hpp>

#include <IDocumentLayoutAccess.hxx>
#include <doc.hxx>
#include <frame.hxx>
#include <layfrm.hxx>
#include <pagefrm.hxx>
#include <rootfrm.hxx>
#include <txtfrm.hxx>
#include <docsh.hxx>
#include <wrtsh.hxx>
#include <viewopt.hxx>

using namespace css;
using namespace css::uno;

namespace
{
/// Covers sw/source/core/text/widorp.cxx fixes.
class Test : public SwModelTestBase
{
public:
    Test()
        : SwModelTestBase(u"/sw/qa/core/text/data/"_ustr)
    {
    }
};

CPPUNIT_TEST_FIXTURE(Test, testHideWhitespaceWidorp)
{
    // Given a document with 3 paragraphs: 1.5 on page 1, 1.5 on page 2:
    createSwDoc("hide-whitespace-widorp.odt");

    // When hiding whitespace:
    SwDocShell* pDocShell = getSwDocShell();
    SwWrtShell* pWrtShell = pDocShell->GetWrtShell();
    SwViewOption aViewOptions(*pWrtShell->GetViewOptions());
    aViewOptions.SetHideWhitespaceMode(true);
    pWrtShell->ApplyViewOptions(aViewOptions);
    calcLayout();

    // Then make sure that paragraph 2 is still split:
    SwDoc* pDoc = pDocShell->GetDoc();
    SwRootFrame* pLayout = pDoc->getIDocumentLayoutAccess().GetCurrentLayout();
    auto pPage = pLayout->Lower()->DynCastPageFrame();
    CPPUNIT_ASSERT(pPage);
    SwFrame* pBody = pPage->FindBodyCont();
    SwFrame* pPara1 = pBody->GetLower();
    auto pPara2 = pPara1->GetNext()->DynCastTextFrame();
    CPPUNIT_ASSERT(pPara2);
    // Without the accompanying fix in place, this test would have failed, the entire paragraph 2
    // went to page 1, so it was not split anymore.
    CPPUNIT_ASSERT(pPara2->HasFollow());
}

CPPUNIT_TEST_FIXTURE(Test, testFloattableHeadingSplit)
{
    // Given a document which ends with a floating table and a heading paragraph:
    // When loading that document & laying it out:
    createSwDoc("floattable-heading-split.docx");

    // Then make sure that the floating table is on page 1 and the last heading is on page 2:
    SwDocShell* pDocShell = getSwDocShell();
    SwDoc* pDoc = pDocShell->GetDoc();
    SwRootFrame* pLayout = pDoc->getIDocumentLayoutAccess().GetCurrentLayout();
    auto pPage1 = pLayout->Lower()->DynCastPageFrame();
    // Without the accompanying fix in place, this test would have failed, the floating table went
    // to page 2, not to page 1.
    CPPUNIT_ASSERT(pPage1->GetSortedObjs());
    // Make sure that page 2 has no floating table and has the heading on the correct page.
    auto pPage2 = pPage1->GetNext()->DynCastPageFrame();
    CPPUNIT_ASSERT(!pPage2->GetSortedObjs());
    SwLayoutFrame* pBody2 = pPage2->FindBodyCont();
    SwTextFrame* pPage2Para1 = pBody2->ContainsContent()->DynCastTextFrame();
    CPPUNIT_ASSERT_EQUAL(u"page 2"_ustr, pPage2Para1->GetText());
}

CPPUNIT_TEST_FIXTURE(Test, testFloattableHeadingSplitFooter)
{
    // Given a document which ends with a floating table and a heading paragraph:
    // When loading that document & laying it out:
    createSwDoc("floattable-heading-split-footer.docx");

    // Then make sure that the floating table is on page 2 and the last heading + footnote is on
    // page 3:
    SwDocShell* pDocShell = getSwDocShell();
    SwDoc* pDoc = pDocShell->GetDoc();
    SwRootFrame* pLayout = pDoc->getIDocumentLayoutAccess().GetCurrentLayout();
    auto pPage1 = pLayout->Lower()->DynCastPageFrame();
    auto pPage2 = pPage1->GetNext()->DynCastPageFrame();
    // Without the accompanying fix in place, this test would have failed, the floating table went
    // to page 3, not to page 2.
    CPPUNIT_ASSERT(pPage2->GetSortedObjs());
    CPPUNIT_ASSERT(!pPage2->FindFootnoteCont());
    // Make sure that page 3 has no floating table and has the heading on the correct page.
    auto pPage3 = pPage2->GetNext()->DynCastPageFrame();
    CPPUNIT_ASSERT(!pPage3->GetSortedObjs());
    CPPUNIT_ASSERT(pPage3->FindFootnoteCont());
    SwLayoutFrame* pBody3 = pPage3->FindBodyCont();
    SwTextFrame* pPage3Para1 = pBody3->ContainsContent()->DynCastTextFrame();
    CPPUNIT_ASSERT_EQUAL(u"page 3"_ustr, pPage3Para1->GetText());
}

CPPUNIT_TEST_FIXTURE(Test, testFloattableHeadingSplitHyphen)
{
    // Page 3 top paragraph has a hyphenated word and changing that would influence the widow/orphan
    // code, so just avoid this test if the English hyphen patterns are missing.
    uno::Reference<linguistic2::XLinguServiceManager2> xLingu
        = linguistic2::LinguServiceManager::create(m_xContext);
    uno::Reference<linguistic2::XHyphenator> xHyphenator = xLingu->getHyphenator();
    if (!xHyphenator.is())
    {
        return;
    }
    lang::Locale aLocale;
    aLocale.Language = u"en"_ustr;
    aLocale.Country = u"US"_ustr;
    if (!xHyphenator->hasLocale(aLocale))
    {
        return;
    }

    // Given a document with a floating table on page 3:
    // When loading that document & laying it out:
    createSwDoc("floattable-heading-split-hyphen.docx");

    // Then make sure that the floating table is on page 3 and the last heading + footnote is on
    // page 4:
    SwDocShell* pDocShell = getSwDocShell();
    SwDoc* pDoc = pDocShell->GetDoc();
    SwRootFrame* pLayout = pDoc->getIDocumentLayoutAccess().GetCurrentLayout();
    auto pPage1 = pLayout->Lower()->DynCastPageFrame();
    auto pPage2 = pPage1->GetNext()->DynCastPageFrame();
    auto pPage3 = pPage2->GetNext()->DynCastPageFrame();
    // Without the accompanying fix in place, this test would have failed, the floating table went
    // to page 4, not to page 3.
    CPPUNIT_ASSERT(pPage3->GetSortedObjs());
    CPPUNIT_ASSERT(!pPage3->FindFootnoteCont());
    auto pPage4 = pPage3->GetNext()->DynCastPageFrame();
    CPPUNIT_ASSERT(!pPage4->GetSortedObjs());
    CPPUNIT_ASSERT(pPage4->FindFootnoteCont());
}
}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */