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

**

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

** All rights reserved.

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

**

** This file is part of the QtGui module 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$

**

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

#include "qfiledialog.h"

#ifndef QT_NO_FILEDIALOG

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

  QFileDialog debug facilities

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

//#define DEBUG_FILEDIALOG_FILTERS

#include <qapplication.h>

#include <private/qapplication_p.h>

#include <private/qfiledialog_p.h>

#include <private/qt_mac_p.h>

#include <private/qt_cocoa_helpers_mac_p.h>

#include <qregexp.h>

#include <qbuffer.h>

#include <qdebug.h>

#include <qstringlist.h>

#include <qaction.h>

#include <qtextcodec.h>

#include <qvarlengtharray.h>

#include <qdesktopwidget.h>

#include <stdlib.h>

#include <qabstracteventdispatcher.h>

#include "ui_qfiledialog.h"

QT_BEGIN_NAMESPACE

extern QStringList qt_make_filter_list(const QString &filter); // qfiledialog.cpp

extern QStringList qt_clean_filter_list(const QString &filter); // qfiledialog.cpp

extern const char *qt_file_dialog_filter_reg_exp; // qfiledialog.cpp

extern bool qt_mac_is_macsheet(const QWidget *w); // qwidget_mac.mm

QT_END_NAMESPACE

QT_FORWARD_DECLARE_CLASS(QFileDialogPrivate)

QT_FORWARD_DECLARE_CLASS(QString)

QT_FORWARD_DECLARE_CLASS(QStringList)

QT_FORWARD_DECLARE_CLASS(QWidget)

QT_FORWARD_DECLARE_CLASS(QAction)

QT_FORWARD_DECLARE_CLASS(QFileInfo)

QT_USE_NAMESPACE

@class QNSOpenSavePanelDelegate;

@interface QNSOpenSavePanelDelegate : NSObject {

    @public

    NSOpenPanel *mOpenPanel;

    NSSavePanel *mSavePanel;

    NSView *mAccessoryView;

    NSPopUpButton *mPopUpButton;

    NSTextField *mTextField;

    QFileDialogPrivate *mPriv;

    NSString *mCurrentDir;

    bool mConfirmOverwrite;

    int mReturnCode;

    QT_PREPEND_NAMESPACE(QFileDialog::AcceptMode) mAcceptMode;

    QT_PREPEND_NAMESPACE(QDir::Filters) *mQDirFilter;

    QT_PREPEND_NAMESPACE(QFileDialog::FileMode) mFileMode;

    QT_PREPEND_NAMESPACE(QFileDialog::Options) *mFileOptions;

    QString *mLastFilterCheckPath;

    QString *mCurrentSelection;

    QStringList *mQDirFilterEntryList;

    QStringList *mNameFilterDropDownList;

    QStringList *mSelectedNameFilter;

}

- (NSString *)strip:(const QString &)label;

- (BOOL)panel:(id)sender shouldShowFilename:(NSString *)filename;

- (void)filterChanged:(id)sender;

- (void)showModelessPanel;

- (BOOL)runApplicationModalPanel;

- (void)showWindowModalSheet:(QWidget *)docWidget;

- (void)updateProperties;

- (QStringList)acceptableExtensionsForSave;

- (QString)removeExtensions:(const QString &)filter;

- (void)createTextField;

- (void)createPopUpButton:(const QString &)selectedFilter hideDetails:(BOOL)hideDetails;

- (void)createAccessory;

@end

@implementation QNSOpenSavePanelDelegate

- (id)initWithAcceptMode:(QT_PREPEND_NAMESPACE(QFileDialog::AcceptMode))acceptMode

    title:(const QString &)title

    nameFilters:(const QStringList &)nameFilters

    selectedNameFilter:(const QString &)selectedNameFilter

    hideNameFilterDetails:(bool)hideNameFilterDetails

    qDirFilter:(QT_PREPEND_NAMESPACE(QDir::Filters))qDirFilter

    fileOptions:(QT_PREPEND_NAMESPACE(QFileDialog::Options))fileOptions

    fileMode:(QT_PREPEND_NAMESPACE(QFileDialog::FileMode))fileMode

    selectFile:(const QString &)selectFile

    confirmOverwrite:(bool)confirm

    priv:(QFileDialogPrivate *)priv

{

    self = [super init];

    mAcceptMode = acceptMode;

    if (mAcceptMode == QT_PREPEND_NAMESPACE(QFileDialog::AcceptOpen)){

        mOpenPanel = [NSOpenPanel openPanel];

        mSavePanel = mOpenPanel;

    } else {

        mSavePanel = [NSSavePanel savePanel];

        mOpenPanel = 0;

    }

    [mSavePanel setLevel:NSModalPanelWindowLevel];

    [mSavePanel setDelegate:self];

    mQDirFilter = new QT_PREPEND_NAMESPACE(QDir::Filters)(qDirFilter);

    mFileOptions = new QT_PREPEND_NAMESPACE(QFileDialog::Options)(fileOptions);

    mFileMode = fileMode;

    mConfirmOverwrite = confirm;

    mReturnCode = -1;

    mPriv = priv;

    mLastFilterCheckPath = new QString;

    mQDirFilterEntryList = new QStringList;

    mNameFilterDropDownList = new QStringList(nameFilters);

    mSelectedNameFilter = new QStringList(qt_clean_filter_list(selectedNameFilter));

    QFileInfo sel(selectFile);

    if (sel.isDir()){

        mCurrentDir = [qt_mac_QStringToNSString(sel.absoluteFilePath()) retain];

        mCurrentSelection = new QString;

    } else {

        mCurrentDir = [qt_mac_QStringToNSString(sel.absolutePath()) retain];

        mCurrentSelection = new QString(sel.absoluteFilePath());

    }

    [mSavePanel setTitle:qt_mac_QStringToNSString(title)];

    [self createPopUpButton:selectedNameFilter hideDetails:hideNameFilterDetails];

    [self createTextField];

    [self createAccessory];

    [mSavePanel setAccessoryView:mNameFilterDropDownList->size() > 1 ? mAccessoryView : nil];

    if (mPriv){

        [mSavePanel setPrompt:[self strip:mPriv->acceptLabel]];

        if (mPriv->fileNameLabelExplicitlySat)

            [mSavePanel setNameFieldLabel:[self strip:mPriv->qFileDialogUi->fileNameLabel->text()]];

    }

    [self updateProperties];

    [mSavePanel retain];

    return self;

}

- (void)dealloc

{

    delete mQDirFilter;

    delete mFileOptions;

    delete mLastFilterCheckPath;

    delete mQDirFilterEntryList;

    delete mNameFilterDropDownList;

    delete mSelectedNameFilter;

    delete mCurrentSelection;

    [mSavePanel orderOut:mSavePanel];

    [mSavePanel setAccessoryView:nil];

    [mPopUpButton release];

    [mTextField release];

    [mAccessoryView release];

    [mSavePanel setDelegate:nil];

    [mSavePanel release];

    [mCurrentDir release];

    [super dealloc];

}

- (NSString *)strip:(const QString &)label

{

    QAction a(label, 0);

    return qt_mac_QStringToNSString(a.iconText());

}

- (void)closePanel

{

    *mCurrentSelection = QT_PREPEND_NAMESPACE(qt_mac_NSStringToQString)([mSavePanel filename]);

    [mSavePanel close];

}

- (void)showModelessPanel

{

    if (mOpenPanel){

        QFileInfo info(*mCurrentSelection);

        NSString *filename = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.fileName());

        NSString *filepath = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.filePath());

        bool selectable = (mAcceptMode == QFileDialog::AcceptSave)

            || [self panel:nil shouldShowFilename:filepath];

        [mOpenPanel 

            beginForDirectory:mCurrentDir

            file:selectable ? filename : nil

            types:nil

            modelessDelegate:self

            didEndSelector:@selector(openPanelDidEnd:returnCode:contextInfo:)

            contextInfo:nil];

    }

}

- (BOOL)runApplicationModalPanel

{

    QFileInfo info(*mCurrentSelection);

    NSString *filename = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.fileName());

    NSString *filepath = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.filePath());

    bool selectable = (mAcceptMode == QFileDialog::AcceptSave)

        || [self panel:nil shouldShowFilename:filepath];

    mReturnCode = [mSavePanel 

        runModalForDirectory:mCurrentDir

        file:selectable ? filename : @"untitled"];

    QAbstractEventDispatcher::instance()->interrupt();

    return (mReturnCode == NSOKButton);

}

- (QT_PREPEND_NAMESPACE(QDialog::DialogCode))dialogResultCode

{

    return (mReturnCode == NSOKButton) ? QT_PREPEND_NAMESPACE(QDialog::Accepted) : QT_PREPEND_NAMESPACE(QDialog::Rejected);

}

- (void)showWindowModalSheet:(QWidget *)docWidget

{

    Q_UNUSED(docWidget);

    QFileInfo info(*mCurrentSelection);

    NSString *filename = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.fileName());

    NSString *filepath = QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(info.filePath());

    bool selectable = (mAcceptMode == QFileDialog::AcceptSave)

        || [self panel:nil shouldShowFilename:filepath];

    [mSavePanel 

        beginSheetForDirectory:mCurrentDir

        file:selectable ? filename : nil

#ifdef QT_MAC_USE_COCOA

        modalForWindow:QT_PREPEND_NAMESPACE(qt_mac_window_for)(docWidget)

#else

        modalForWindow:nil

#endif

        modalDelegate:self

        didEndSelector:@selector(openPanelDidEnd:returnCode:contextInfo:)

        contextInfo:nil];

}

- (BOOL)panel:(id)sender shouldShowFilename:(NSString *)filename

{

    Q_UNUSED(sender);

    if ([filename length] == 0)

        return NO;

    QString qtFileName = QT_PREPEND_NAMESPACE(qt_mac_NSStringToQString)(filename);

    QFileInfo info(qtFileName.normalized(QT_PREPEND_NAMESPACE(QString::NormalizationForm_C)));

    QString path = info.absolutePath();

    if (path != *mLastFilterCheckPath){

        *mLastFilterCheckPath = path;

        *mQDirFilterEntryList = info.dir().entryList(*mQDirFilter);

    }

    // Check if the QDir filter accepts the file:

    if (!mQDirFilterEntryList->contains(info.fileName()))

        return NO;

    // Always accept directories regardless of their names:

    BOOL isDir;

    if ([[NSFileManager defaultManager] fileExistsAtPath:filename isDirectory:&isDir] && isDir)

        return YES;

    // No filter means accept everything

    if (mSelectedNameFilter->isEmpty())

        return YES;

    // Check if the current file name filter accepts the file:

    for (int i=0; i<mSelectedNameFilter->size(); ++i) {

        if (QDir::match(mSelectedNameFilter->at(i), qtFileName))

            return YES;

    }

    return NO;

}

- (NSString *)panel:(id)sender userEnteredFilename:(NSString *)filename confirmed:(BOOL)okFlag

{

    Q_UNUSED(sender);

    if (!okFlag)

        return filename;

    if (mConfirmOverwrite)

        return filename;

    // User has clicked save, and no overwrite confirmation should occur.

    // To get the latter, we need to change the name we return (hence the prefix):

    return [@"___qt_very_unlikely_prefix_" stringByAppendingString:filename];

}

- (void)setNameFilters:(const QStringList &)filters hideDetails:(BOOL)hideDetails

{

    [mPopUpButton removeAllItems];

    *mNameFilterDropDownList = filters;

    if (filters.size() > 0){

        for (int i=0; i<filters.size(); ++i) {

            QString filter = hideDetails ? [self removeExtensions:filters.at(i)] : filters.at(i);

            [mPopUpButton addItemWithTitle:QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(filter)];

        }

        [mPopUpButton selectItemAtIndex:0];

        [mSavePanel setAccessoryView:mAccessoryView];

    } else

        [mSavePanel setAccessoryView:nil];

    [self filterChanged:self];

}

- (void)filterChanged:(id)sender

{

    // This mDelegate function is called when the _name_ filter changes.

    Q_UNUSED(sender);

    QString selection = mNameFilterDropDownList->value([mPopUpButton indexOfSelectedItem]);

    *mSelectedNameFilter = QT_PREPEND_NAMESPACE(qt_clean_filter_list)(selection);

    [mSavePanel validateVisibleColumns];

    [self updateProperties];

    if (mPriv)

        mPriv->QNSOpenSavePanelDelegate_filterSelected([mPopUpButton indexOfSelectedItem]);

}

- (QString)currentNameFilter

{

    return mNameFilterDropDownList->value([mPopUpButton indexOfSelectedItem]);

}

- (QStringList)selectedFiles

{

    if (mOpenPanel)

        return QT_PREPEND_NAMESPACE(qt_mac_NSArrayToQStringList)([mOpenPanel filenames]);

    else{

        QStringList result;

        QString filename = QT_PREPEND_NAMESPACE(qt_mac_NSStringToQString)([mSavePanel filename]);

        result << filename.remove(QLatin1String("___qt_very_unlikely_prefix_"));

        return result;

    }

}

- (void)updateProperties

{

    // Call this functions if mFileMode, mFileOptions,

    // mNameFilterDropDownList or mQDirFilter changes.

    // The savepanel does not contain the neccessary functions for this.

    bool chooseFilesOnly = mFileMode == QT_PREPEND_NAMESPACE(QFileDialog::ExistingFile)

        || mFileMode == QT_PREPEND_NAMESPACE(QFileDialog::ExistingFiles);

    bool chooseDirsOnly = mFileMode == QT_PREPEND_NAMESPACE(QFileDialog::Directory)

        || mFileMode == QT_PREPEND_NAMESPACE(QFileDialog::DirectoryOnly)

        || *mFileOptions & QT_PREPEND_NAMESPACE(QFileDialog::ShowDirsOnly);

    [mOpenPanel setCanChooseFiles:!chooseDirsOnly];

    [mOpenPanel setCanChooseDirectories:!chooseFilesOnly];

    [mSavePanel setCanCreateDirectories:!(*mFileOptions & QT_PREPEND_NAMESPACE(QFileDialog::ReadOnly))];

    [mOpenPanel setAllowsMultipleSelection:(mFileMode == QT_PREPEND_NAMESPACE(QFileDialog::ExistingFiles))];

    [mOpenPanel setResolvesAliases:!(*mFileOptions & QT_PREPEND_NAMESPACE(QFileDialog::DontResolveSymlinks))];

    QStringList ext = [self acceptableExtensionsForSave];

    if (mPriv && !ext.isEmpty() && !mPriv->defaultSuffix.isEmpty())

        ext.prepend(mPriv->defaultSuffix);

    [mSavePanel setAllowedFileTypes:ext.isEmpty() ? nil : QT_PREPEND_NAMESPACE(qt_mac_QStringListToNSMutableArray(ext))];

    if ([mSavePanel isVisible])

        [mOpenPanel validateVisibleColumns];

}

- (void)panelSelectionDidChange:(id)sender

{

    Q_UNUSED(sender);

}

- (void)openPanelDidEnd:(NSOpenPanel *)panel returnCode:(int)returnCode  contextInfo:(void *)contextInfo

{

    Q_UNUSED(panel);

    Q_UNUSED(contextInfo);

    mReturnCode = returnCode;

    if (mPriv)

        mPriv->QNSOpenSavePanelDelegate_panelClosed(returnCode == NSOKButton);

}

- (void)panel:(id)sender directoryDidChange:(NSString *)path

{

    Q_UNUSED(sender);

    if (!mPriv)

        return;

    if ([path isEqualToString:mCurrentDir])

        return;

    [mCurrentDir release];

    mCurrentDir = [path retain];

    mPriv->QNSOpenSavePanelDelegate_directoryEntered(QT_PREPEND_NAMESPACE(qt_mac_NSStringToQString(mCurrentDir)));

}

/*

    Returns a list of extensions (e.g. "png", "jpg", "gif")

    for the current name filter. If a filter do not conform

    to the format *.xyz or * or *.*, an empty list

    is returned meaning accept everything.

*/

- (QStringList)acceptableExtensionsForSave

{

    QStringList result;

    for (int i=0; i<mSelectedNameFilter->count(); ++i) {

        const QString &filter = mSelectedNameFilter->at(i);

        if (filter.startsWith(QLatin1String("*."))

                && !filter.contains(QLatin1Char('?'))

                && filter.count(QLatin1Char('*')) == 1) {

            result += filter.mid(2);

        } else {

            return QStringList(); // Accept everything

        }

    }

    return result;

}

- (QString)removeExtensions:(const QString &)filter

{

    QRegExp regExp(QT_PREPEND_NAMESPACE(QString::fromLatin1)(QT_PREPEND_NAMESPACE(qt_file_dialog_filter_reg_exp)));

    if (regExp.indexIn(filter) != -1)

        return regExp.cap(1).trimmed();

    return filter;

}

- (void)createTextField

{

    NSRect textRect = { { 0.0, 3.0 }, { 100.0, 25.0 } };

    mTextField = [[NSTextField alloc] initWithFrame:textRect];

    [[mTextField cell] setFont:[NSFont systemFontOfSize:

            [NSFont systemFontSizeForControlSize:NSRegularControlSize]]];

    [mTextField setAlignment:NSRightTextAlignment];

    [mTextField setEditable:false];

    [mTextField setSelectable:false];

    [mTextField setBordered:false];

    [mTextField setDrawsBackground:false];

    if (mPriv){

        [mTextField setStringValue:[self strip:mPriv->qFileDialogUi->fileTypeLabel->text()]];

    } else

        [mTextField setStringValue:QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(QT_PREPEND_NAMESPACE(QFileDialog::tr)("Files of type:"))];

}

- (void)createPopUpButton:(const QString &)selectedFilter hideDetails:(BOOL)hideDetails

{

    NSRect popUpRect = { { 100.0, 5.0 }, { 250.0, 25.0 } };

    mPopUpButton = [[NSPopUpButton alloc] initWithFrame:popUpRect pullsDown:NO];

    [mPopUpButton setTarget:self];

    [mPopUpButton setAction:@selector(filterChanged:)];

    QStringList *filters = mNameFilterDropDownList;

    if (filters->size() > 0){

        for (int i=0; i<mNameFilterDropDownList->size(); ++i) {

            QString filter = hideDetails ? [self removeExtensions:filters->at(i)] : filters->at(i);

            [mPopUpButton addItemWithTitle:QT_PREPEND_NAMESPACE(qt_mac_QStringToNSString)(filter)];

            if (filters->at(i) == selectedFilter)

                [mPopUpButton selectItemAtIndex:i];

        }

    }

}

- (void)createAccessory

{

    NSRect accessoryRect = { { 0.0, 0.0 }, { 450.0, 33.0 } };

    mAccessoryView = [[NSView alloc] initWithFrame:accessoryRect];

    [mAccessoryView addSubview:mTextField];

    [mAccessoryView addSubview:mPopUpButton];

}

@end

QT_BEGIN_NAMESPACE

void QFileDialogPrivate::QNSOpenSavePanelDelegate_selectionChanged(const QString &newPath)

{

    emit q_func()->currentChanged(newPath);

}

void QFileDialogPrivate::QNSOpenSavePanelDelegate_panelClosed(bool accepted)

{

    if (accepted)

        q_func()->accept();

    else

        q_func()->reject();

}

void QFileDialogPrivate::QNSOpenSavePanelDelegate_directoryEntered(const QString &newDir)

{

    setLastVisitedDirectory(newDir);

    emit q_func()->directoryEntered(newDir);

}