/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the Qt Linguist of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

/*  TRANSLATOR MainWindow

  This is the application's main window.

*/

#include "mainwindow.h"

#include "batchtranslationdialog.h"

#include "errorsview.h"

#include "finddialog.h"

#include "formpreviewview.h"

#include "globals.h"

#include "messageeditor.h"

#include "messagemodel.h"

#include "phrasebookbox.h"

#include "phrasemodel.h"

#include "phraseview.h"

#include "printout.h"

#include "sourcecodeview.h"

#include "statistics.h"

#include "translatedialog.h"

#include "translationsettingsdialog.h"

#include <QAction>

#include <QApplication>

#include <QBitmap>

#include <QCloseEvent>

#include <QDebug>

#include <QDesktopWidget>

#include <QDockWidget>

#include <QFile>

#include <QFileDialog>

#include <QFileInfo>

#include <QHeaderView>

#include <QInputDialog>

#include <QItemDelegate>

#include <QLabel>

#include <QLayout>

#include <QLibraryInfo>

#include <QMenu>

#include <QMenuBar>

#include <QMessageBox>

#include <QPrintDialog>

#include <QPrinter>

#include <QProcess>

#include <QRegExp>

#include <QSettings>

#include <QSortFilterProxyModel>

#include <QStackedWidget>

#include <QStatusBar>

#include <QTextStream>

#include <QToolBar>

#include <QUrl>

#include <QWhatsThis>

#include <ctype.h>

QT_BEGIN_NAMESPACE

static const int MessageMS = 2500;

enum Ending {

    End_None,

    End_FullStop,

    End_Interrobang,

    End_Colon,

    End_Ellipsis

};

static bool hasFormPreview(const QString &fileName)

{

    return fileName.endsWith(QLatin1String(".ui"))

      || fileName.endsWith(QLatin1String(".jui"));

}

static Ending ending(QString str, QLocale::Language lang)

{

    str = str.simplified();

    if (str.isEmpty())

        return End_None;

    switch (str.at(str.length() - 1).unicode()) {

    case 0x002e: // full stop

        if (str.endsWith(QLatin1String("...")))

            return End_Ellipsis;

        else

            return End_FullStop;

    case 0x0589: // armenian full stop

    case 0x06d4: // arabic full stop

    case 0x3002: // ideographic full stop

        return End_FullStop;

    case 0x0021: // exclamation mark

    case 0x003f: // question mark

    case 0x00a1: // inverted exclamation mark

    case 0x00bf: // inverted question mark

    case 0x01c3: // latin letter retroflex click

    case 0x037e: // greek question mark

    case 0x061f: // arabic question mark

    case 0x203c: // double exclamation mark

    case 0x203d: // interrobang

    case 0x2048: // question exclamation mark

    case 0x2049: // exclamation question mark

    case 0x2762: // heavy exclamation mark ornament

        return End_Interrobang;

    case 0x003b: // greek 'compatibility' questionmark

        return lang == QLocale::Greek ? End_Interrobang : End_None;

    case 0x003a: // colon

        return End_Colon;

    case 0x2026: // horizontal ellipsis

        return End_Ellipsis;

    default:

        return End_None;

    }

}

class ContextItemDelegate : public QItemDelegate

{

public:

    ContextItemDelegate(QObject *parent, MultiDataModel *model) : QItemDelegate(parent), m_dataModel(model) {}

    void paint(QPainter *painter, const QStyleOptionViewItem &option,

        const QModelIndex &index) const

    {

        const QAbstractItemModel *model = index.model();

        Q_ASSERT(model);

        if (!model->parent(index).isValid()) {

            if (index.column() - 1 == m_dataModel->modelCount()) {

                QStyleOptionViewItem opt = option;

                opt.font.setBold(true);

                QItemDelegate::paint(painter, opt, index);

                return;

            }

        }

        QItemDelegate::paint(painter, option, index);

    }

private:

    MultiDataModel *m_dataModel;

};

static const QVariant &pxObsolete()

{

    static const QVariant v =

        qVariantFromValue(QPixmap(QLatin1String(":/images/s_check_obsolete.png")));

    return v;

}

class SortedMessagesModel : public QSortFilterProxyModel

{

public:

    SortedMessagesModel(QObject *parent, MultiDataModel *model) : QSortFilterProxyModel(parent), m_dataModel(model) {}

    QVariant headerData(int section, Qt::Orientation orientation, int role) const

    {

        if (role == Qt::DisplayRole && orientation == Qt::Horizontal)

            switch (section - m_dataModel->modelCount()) {

                case 0: return QString();

                case 1: return MainWindow::tr("Source text");

                case 2: return MainWindow::tr("Index");

            }

        if (role == Qt::DecorationRole && orientation == Qt::Horizontal && section - 1 < m_dataModel->modelCount())

            return pxObsolete();

        return QVariant();

    }

private:

    MultiDataModel *m_dataModel;

};

class SortedContextsModel : public QSortFilterProxyModel

{

public:

    SortedContextsModel(QObject *parent, MultiDataModel *model) : QSortFilterProxyModel(parent), m_dataModel(model) {}

    QVariant headerData(int section, Qt::Orientation orientation, int role) const

    {

        if (role == Qt::DisplayRole && orientation == Qt::Horizontal)

            switch (section - m_dataModel->modelCount()) {

                case 0: return QString();

                case 1: return MainWindow::tr("Context");

                case 2: return MainWindow::tr("Items");

                case 3: return MainWindow::tr("Index");

            }

        if (role == Qt::DecorationRole && orientation == Qt::Horizontal && section - 1 < m_dataModel->modelCount())

            return pxObsolete();

        return QVariant();

    }

private:

    MultiDataModel *m_dataModel;

};

class FocusWatcher : public QObject

{

public:

    FocusWatcher(MessageEditor *msgedit, QObject *parent) : QObject(parent), m_messageEditor(msgedit) {}

protected:

    bool eventFilter(QObject *object, QEvent *event);

private:

    MessageEditor *m_messageEditor;

};

bool FocusWatcher::eventFilter(QObject *, QEvent *event)

{

    if (event->type() == QEvent::FocusIn)

        m_messageEditor->setEditorFocus(-1);

    return false;

}

MainWindow::MainWindow()

    : QMainWindow(0, Qt::Window),

      m_assistantProcess(0),

      m_printer(0),

      m_findMatchCase(Qt::CaseInsensitive),

      m_findIgnoreAccelerators(true),

      m_findWhere(DataModel::NoLocation),

      m_foundWhere(DataModel::NoLocation),

      m_translationSettingsDialog(0),

      m_settingCurrentMessage(false),

      m_fileActiveModel(-1),

      m_editActiveModel(-1),

      m_statistics(0)

{

    setUnifiedTitleAndToolBarOnMac(true);

    m_ui.setupUi(this);

#ifndef Q_WS_MAC

    setWindowIcon(QPixmap(QLatin1String(":/images/appicon.png") ));

#endif

    m_dataModel = new MultiDataModel(this);

    m_messageModel = new MessageModel(this, m_dataModel);

    // Set up the context dock widget

    m_contextDock = new QDockWidget(this);

    m_contextDock->setObjectName(QLatin1String("ContextDockWidget"));

    m_contextDock->setAllowedAreas(Qt::AllDockWidgetAreas);

    m_contextDock->setFeatures(QDockWidget::AllDockWidgetFeatures);

    m_contextDock->setWindowTitle(tr("Context"));

    m_contextDock->setAcceptDrops(true);

    m_contextDock->installEventFilter(this);

    m_sortedContextsModel = new SortedContextsModel(this, m_dataModel);

    m_sortedContextsModel->setSortRole(MessageModel::SortRole);

    m_sortedContextsModel->setSortCaseSensitivity(Qt::CaseInsensitive);

    m_sortedContextsModel->setSourceModel(m_messageModel);

    m_contextView = new QTreeView(this);

    m_contextView->setRootIsDecorated(false);

    m_contextView->setItemsExpandable(false);

    m_contextView->setUniformRowHeights(true);

    m_contextView->setAlternatingRowColors(true);

    m_contextView->setAllColumnsShowFocus(true);

    m_contextView->setItemDelegate(new ContextItemDelegate(this, m_dataModel));

    m_contextView->setSortingEnabled(true);

    m_contextView->setWhatsThis(tr("This panel lists the source contexts."));

    m_contextView->setModel(m_sortedContextsModel);

    m_contextView->header()->setMovable(false);

    m_contextView->setColumnHidden(0, true);

    m_contextView->header()->setStretchLastSection(false);

    m_contextDock->setWidget(m_contextView);

    // Set up the messages dock widget

    m_messagesDock = new QDockWidget(this);

    m_messagesDock->setObjectName(QLatin1String("StringsDockWidget"));

    m_messagesDock->setAllowedAreas(Qt::AllDockWidgetAreas);

    m_messagesDock->setFeatures(QDockWidget::AllDockWidgetFeatures);

    m_messagesDock->setWindowTitle(tr("Strings"));

    m_messagesDock->setAcceptDrops(true);

    m_messagesDock->installEventFilter(this);

    m_sortedMessagesModel = new SortedMessagesModel(this, m_dataModel);

    m_sortedMessagesModel->setSortRole(MessageModel::SortRole);

    m_sortedMessagesModel->setSortCaseSensitivity(Qt::CaseInsensitive);

    m_sortedMessagesModel->setSortLocaleAware(true);

    m_sortedMessagesModel->setSourceModel(m_messageModel);

    m_messageView = new QTreeView(m_messagesDock);

    m_messageView->setSortingEnabled(true);

    m_messageView->setRootIsDecorated(false);

    m_messageView->setUniformRowHeights(true);

    m_messageView->setAllColumnsShowFocus(true);

    m_messageView->setItemsExpandable(false);

    m_messageView->setModel(m_sortedMessagesModel);

    m_messageView->header()->setMovable(false);

    m_messageView->setColumnHidden(0, true);

    m_messagesDock->setWidget(m_messageView);

    // Set up main message view

    m_messageEditor = new MessageEditor(m_dataModel, this);

    m_messageEditor->setAcceptDrops(true);

    m_messageEditor->installEventFilter(this);

    // We can't call setCentralWidget(m_messageEditor), since it is already called in m_ui.setupUi()

    QBoxLayout *lout = new QBoxLayout(QBoxLayout::TopToBottom, m_ui.centralwidget);

    lout->addWidget(m_messageEditor);

    lout->setMargin(0);

    m_ui.centralwidget->setLayout(lout);

    // Set up the phrases & guesses dock widget

    m_phrasesDock = new QDockWidget(this);

    m_phrasesDock->setObjectName(QLatin1String("PhrasesDockwidget"));

    m_phrasesDock->setAllowedAreas(Qt::AllDockWidgetAreas);

    m_phrasesDock->setFeatures(QDockWidget::AllDockWidgetFeatures);

    m_phrasesDock->setWindowTitle(tr("Phrases and guesses"));

    m_phraseView = new PhraseView(m_dataModel, &m_phraseDict, this);

    m_phrasesDock->setWidget(m_phraseView);

    // Set up source code and form preview dock widget

    m_sourceAndFormDock = new QDockWidget(this);

    m_sourceAndFormDock->setObjectName(QLatin1String("SourceAndFormDock"));

    m_sourceAndFormDock->setAllowedAreas(Qt::AllDockWidgetAreas);

    m_sourceAndFormDock->setFeatures(QDockWidget::AllDockWidgetFeatures);

    m_sourceAndFormDock->setWindowTitle(tr("Sources and Forms"));

    m_sourceAndFormView = new QStackedWidget(this);

    m_sourceAndFormDock->setWidget(m_sourceAndFormView);

    //connect(m_sourceAndDock, SIGNAL(visibilityChanged(bool)),

    //    m_sourceCodeView, SLOT(setActivated(bool)));

    m_formPreviewView = new FormPreviewView(0, m_dataModel);

    m_sourceCodeView = new SourceCodeView(0);

    m_sourceAndFormView->addWidget(m_sourceCodeView);

    m_sourceAndFormView->addWidget(m_formPreviewView);

    // Set up errors dock widget

    m_errorsDock = new QDockWidget(this);

    m_errorsDock->setObjectName(QLatin1String("ErrorsDockWidget"));

    m_errorsDock->setAllowedAreas(Qt::AllDockWidgetAreas);

    m_errorsDock->setFeatures(QDockWidget::AllDockWidgetFeatures);

    m_errorsDock->setWindowTitle(tr("Warnings"));

    m_errorsView = new ErrorsView(m_dataModel, this);

    m_errorsDock->setWidget(m_errorsView);

    // Arrange dock widgets

    setDockNestingEnabled(true);

    setCorner(Qt::TopLeftCorner, Qt::LeftDockWidgetArea);

    setCorner(Qt::TopRightCorner, Qt::RightDockWidgetArea);

    setCorner(Qt::BottomLeftCorner, Qt::LeftDockWidgetArea);

    setCorner(Qt::BottomRightCorner, Qt::RightDockWidgetArea);

    addDockWidget(Qt::LeftDockWidgetArea, m_contextDock);

    addDockWidget(Qt::TopDockWidgetArea, m_messagesDock);

    addDockWidget(Qt::BottomDockWidgetArea, m_phrasesDock);

    addDockWidget(Qt::TopDockWidgetArea, m_sourceAndFormDock);

    addDockWidget(Qt::BottomDockWidgetArea, m_errorsDock);

    //tabifyDockWidget(m_errorsDock, m_sourceAndFormDock);

    //tabifyDockWidget(m_sourceCodeDock, m_phrasesDock);

    // Allow phrases doc to intercept guesses shortcuts

    m_messageEditor->installEventFilter(m_phraseView);

    // Set up shortcuts for the dock widgets

    QShortcut *contextShortcut = new QShortcut(QKeySequence(Qt::Key_F6), this);

    connect(contextShortcut, SIGNAL(activated()), this, SLOT(showContextDock()));

    QShortcut *messagesShortcut = new QShortcut(QKeySequence(Qt::Key_F7), this);

    connect(messagesShortcut, SIGNAL(activated()), this, SLOT(showMessagesDock()));

    QShortcut *errorsShortcut = new QShortcut(QKeySequence(Qt::Key_F8), this);

    connect(errorsShortcut, SIGNAL(activated()), this, SLOT(showErrorDock()));

    QShortcut *sourceCodeShortcut = new QShortcut(QKeySequence(Qt::Key_F9), this);

    connect(sourceCodeShortcut, SIGNAL(activated()), this, SLOT(showSourceCodeDock()));

    QShortcut *phrasesShortcut = new QShortcut(QKeySequence(Qt::Key_F10), this);

    connect(phrasesShortcut, SIGNAL(activated()), this, SLOT(showPhrasesDock()));

    connect(m_phraseView, SIGNAL(phraseSelected(int,QString)),

            m_messageEditor, SLOT(setTranslation(int,QString)));

    connect(m_contextView->selectionModel(),

            SIGNAL(currentRowChanged(QModelIndex,QModelIndex)),

            this, SLOT(selectedContextChanged(QModelIndex,QModelIndex)));

    connect(m_messageView->selectionModel(),

            SIGNAL(currentRowChanged(QModelIndex,QModelIndex)),

            this, SLOT(selectedMessageChanged(QModelIndex,QModelIndex)));

    connect(m_contextView->selectionModel(),

            SIGNAL(currentColumnChanged(QModelIndex,QModelIndex)),

            SLOT(updateLatestModel(QModelIndex)));

    connect(m_messageView->selectionModel(),

            SIGNAL(currentColumnChanged(QModelIndex,QModelIndex)),

            SLOT(updateLatestModel(QModelIndex)));

    connect(m_messageEditor, SIGNAL(activeModelChanged(int)), SLOT(updateActiveModel(int)));

    m_translateDialog = new TranslateDialog(this);

    m_batchTranslateDialog = new BatchTranslationDialog(m_dataModel, this);

    m_findDialog = new FindDialog(this);

    setupMenuBar();

    setupToolBars();

    m_progressLabel = new QLabel();

    statusBar()->addPermanentWidget(m_progressLabel);

    m_modifiedLabel = new QLabel(tr(" MOD ", "status bar: file(s) modified"));

    statusBar()->addPermanentWidget(m_modifiedLabel);

    modelCountChanged();

    initViewHeaders();

    resetSorting();

    connect(m_dataModel, SIGNAL(modifiedChanged(bool)),

            this, SLOT(setWindowModified(bool)));

    connect(m_dataModel, SIGNAL(modifiedChanged(bool)),

            m_modifiedLabel, SLOT(setVisible(bool)));

    connect(m_dataModel, SIGNAL(multiContextDataChanged(MultiDataIndex)),

            SLOT(updateProgress()));

    connect(m_dataModel, SIGNAL(messageDataChanged(MultiDataIndex)),

            SLOT(maybeUpdateStatistics(MultiDataIndex)));

    connect(m_dataModel, SIGNAL(translationChanged(MultiDataIndex)),

            SLOT(translationChanged(MultiDataIndex)));

    connect(m_dataModel, SIGNAL(languageChanged(int)),

            SLOT(updatePhraseDict(int)));

    setWindowModified(m_dataModel->isModified());

    m_modifiedLabel->setVisible(m_dataModel->isModified());

    connect(m_messageView, SIGNAL(clicked(QModelIndex)),

            this, SLOT(toggleFinished(QModelIndex)));

    connect(m_messageView, SIGNAL(activated(QModelIndex)),

            m_messageEditor, SLOT(setEditorFocus()));

    connect(m_contextView, SIGNAL(activated(QModelIndex)),

            m_messageView, SLOT(setFocus()));

    connect(m_messageEditor, SIGNAL(translationChanged(QStringList)),

            this, SLOT(updateTranslation(QStringList)));

    connect(m_messageEditor, SIGNAL(translatorCommentChanged(QString)),

            this, SLOT(updateTranslatorComment(QString)));

    connect(m_findDialog, SIGNAL(findNext(QString,DataModel::FindLocation,bool,bool)),

            this, SLOT(findNext(QString,DataModel::FindLocation,bool,bool)));

    connect(m_translateDialog, SIGNAL(requestMatchUpdate(bool&)), SLOT(updateTranslateHit(bool&)));

    connect(m_translateDialog, SIGNAL(activated(int)), SLOT(translate(int)));

    QSize as(qApp->desktop()->size());

    as -= QSize(30, 30);

    resize(QSize(1000, 800).boundedTo(as));

    show();

    readConfig();

    m_statistics = 0;

    connect(m_ui.actionLengthVariants, SIGNAL(toggled(bool)),

            m_messageEditor, SLOT(setLengthVariants(bool)));

    m_messageEditor->setLengthVariants(m_ui.actionLengthVariants->isChecked());

    m_focusWatcher = new FocusWatcher(m_messageEditor, this);

    m_contextView->installEventFilter(m_focusWatcher);

    m_messageView->installEventFilter(m_focusWatcher);

    m_messageEditor->installEventFilter(m_focusWatcher);

    m_sourceAndFormView->installEventFilter(m_focusWatcher);

    m_phraseView->installEventFilter(m_focusWatcher);

    m_errorsView->installEventFilter(m_focusWatcher);

}

MainWindow::~MainWindow()

{

    writeConfig();

    if (m_assistantProcess && m_assistantProcess->state() == QProcess::Running) {

        m_assistantProcess->terminate();

        m_assistantProcess->waitForFinished(3000);

    }

    qDeleteAll(m_phraseBooks);

    delete m_dataModel;

    delete m_statistics;

    delete m_printer;

}

void MainWindow::initViewHeaders()

{

    m_contextView->header()->setResizeMode(1, QHeaderView::Stretch);

    m_contextView->header()->setResizeMode(2, QHeaderView::ResizeToContents);

    m_messageView->setColumnHidden(2, true);

    // last visible column auto-stretches

}

void MainWindow::modelCountChanged()

{

    int mc = m_dataModel->modelCount();

    for (int i = 0; i < mc; ++i) {

        m_contextView->header()->setResizeMode(i + 1, QHeaderView::Fixed);

        m_contextView->header()->resizeSection(i + 1, 24);