// TortoiseSVN - a Windows shell extension for easy version control

// Copyright (C) 2017, 2021 - TortoiseSVN

// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//
#pragma once

#include <string>
#include <vector>
#include <ShlDisp.h>
#include <ShObjIdl.h>

#include "RegHistory.h"

/**
* Helper class for the CAutoComplete class: implements the string enumerator.
*/
class CAutoCompleteEnum : public IEnumString
{
public:
    CAutoCompleteEnum(const std::vector<std::wstring*>& vec);
    CAutoCompleteEnum(const std::vector<std::wstring>& vec);
    virtual ~CAutoCompleteEnum() {}
    //IUnknown members
    HRESULT STDMETHODCALLTYPE QueryInterface(REFIID, void**) override;
    ULONG STDMETHODCALLTYPE   AddRef() override;
    ULONG STDMETHODCALLTYPE   Release() override;

    //IEnumString members
    HRESULT STDMETHODCALLTYPE Next(ULONG, LPOLESTR*, ULONG*) override;
    HRESULT STDMETHODCALLTYPE Skip(ULONG) override;
    HRESULT STDMETHODCALLTYPE Reset() override;
    HRESULT STDMETHODCALLTYPE Clone(IEnumString**) override;

    void Init(const std::vector<std::wstring*>& vec);
    void Init(const std::vector<std::wstring>& vec);

private:
    std::vector<std::wstring> m_vecStrings;
    ULONG                     m_cRefCount;
    size_t                    m_iCur;
};

class CAutoComplete : public CRegHistory
{
public:
    CAutoComplete();
    ~CAutoComplete() override;

    bool Init(HWND hEdit);
    bool Enable(bool bEnable) const;
    bool AddEntry(LPCWSTR szText) override;

    template <typename T>
    void SetEntries(T t);

    template <typename T>
    void AddEntries(T t);

    bool  RemoveSelected();
    void  SetOptions(DWORD dwFlags) const;
    DWORD GetOptions() const;

private:
    CAutoCompleteEnum*     m_pcacs;
    IAutoComplete2*        m_pac;
    IAutoCompleteDropDown* m_pdrop;
};

template <typename T>
void CAutoComplete::SetEntries(T t)
{
    m_arEntries.clear();
    AddEntries(t);
}

template <typename T>
void CAutoComplete::AddEntries(T t)
{
    m_arEntries.insert(m_arEntries.begin(), t.begin(), t.end());
    if (m_pcacs)
        m_pcacs->Init(m_arEntries);
    if (m_pdrop)
        m_pdrop->ResetEnumerator();
}