FileDialog.h

#pragma once
 
#include "EnumBit.h"
#include "IDialog.h"
#include "List.h"
#include "Property.h"
#include <windows.h>
 
namespace sw
{
    enum class FileDialogFlags : DWORD {
        AllowMultiSelect = 0x00000200,
 
        CreatePrompt = 0x00002000,
 
        DontAddTorecent = 0x02000000,
 
        EnableHook = 0x00000020,
 
        EnableIncludeNotify = 0x00400000,
 
        EnableSizing = 0x00800000,
 
        EnableTemplate = 0x00000040,
 
        EnableTemplateHandle = 0x00000080,
 
        Explorer = 0x00080000,
 
        ExtensionDifferent = 0x00000400,
 
        FileMustExist = 0x00001000,
 
        ForceShowHidden = 0x10000000,
 
        HideReadOnly = 0x00000004,
 
        LongNames = 0x00200000,
 
        NoChangeDir = 0x00000008,
 
        NoDereferenceLinks = 0x00100000,
 
        NoLongNames = 0x00040000,
 
        NoNetworkButton = 0x00020000,
 
        NoReadOnlyReturn = 0x00008000,
 
        NoTestFileCreate = 0x00010000,
 
        NoValidate = 0x00000100,
 
        OverwritePrompt = 0x00000002,
 
        PathMustExist = 0x00000800,
 
        ReadOnly = 0x00000001,
 
        ShareAware = 0x00004000,
 
        ShowHelp = 0x00000010,
    };
 
    _SW_ENUM_ENABLE_BIT_OPERATIONS(FileDialogFlags);
 
    struct FileFilterItem {
        std::wstring name;
 
        std::wstring filter;
 
        std::wstring defaultExt;
    };
 
    class FileFilter
    {
    private:
        std::vector<wchar_t> _buffer;
 
        std::vector<std::wstring> _defaultExts;
 
    public:
        FileFilter() = default;
 
        FileFilter(std::initializer_list<FileFilterItem> filters);
 
        bool AddFilter(const std::wstring &name, const std::wstring &filter, const std::wstring &defaultExt = L"");
 
        int SetFilter(std::initializer_list<FileFilterItem> filters);
 
        void Clear();
 
        wchar_t *GetFilterStr();
 
        const wchar_t *GetDefaultExt(int index);
    };
 
    class FileDialog : public IDialog
    {
    private:
        OPENFILENAMEW _ofn{};
 
        std::vector<wchar_t> _buffer;
 
        std::wstring _title;
 
        std::wstring _initialDir;
 
        FileFilter _filter;
 
    public:
        const Property<int> BufferSize;
 
        const Property<FileDialogFlags> Flags;
 
        const Property<std::wstring> Title;
 
        const Property<std::wstring> InitialDir;
 
        const ReadOnlyProperty<FileFilter *> Filter;
 
        const Property<int> FilterIndex;
 
        const ReadOnlyProperty<std::wstring> FileName;
 
        const Property<bool> MultiSelect;
 
        const ReadOnlyProperty<sw::List<std::wstring>> FileNames;
 
    public:
        FileDialog();
 
        void SetFilter(const FileFilter &filter);
 
        virtual void Close() override;
 
        virtual void Show() override;
 
        virtual int ShowDialog(Window *owner = nullptr) override = 0;
 
        virtual int ShowDialog(Window &owner) = 0;
 
    protected:
        OPENFILENAMEW *GetOFN();
 
        wchar_t *GetBuffer();
 
        void ClearBuffer();
 
        virtual void ProcessFileName(std::wstring &fileName);
    };
 
    class OpenFileDialog : public FileDialog
    {
    public:
        OpenFileDialog();
 
        virtual int ShowDialog(Window *owner = nullptr) override;
 
        virtual int ShowDialog(Window &owner) override;
    };
 
    class SaveFileDialog : public FileDialog
    {
    private:
        std::wstring _initialFileName;
 
    public:
        const Property<std::wstring> InitialFileName;
 
        SaveFileDialog();
 
        virtual int ShowDialog(Window *owner = nullptr) override;
 
        virtual int ShowDialog(Window &owner) override;
 
    protected:
        virtual void ProcessFileName(std::wstring &fileName) override;
 
    private:
        void _SetInitialFileName();
    };
}