New Utility: New+ (#33136)

Co-authored-by: Jaime Bernardo <jaime@janeasystems.com>
Co-authored-by: Stefan Markovic <stefan@janeasystems.com>

src/modules/NewPlus/NewShellExtensionContextMenu/template_item.cpp

@@ -0,0 +1,147 @@
+
+
+#include "pch.h"
+#include "template_item.h"
+#include <shellapi.h>
+#include "new_utilities.h"
+#include <cassert>
+#include <thread>
+#include <shellapi.h>
+#include <shlobj_core.h>
+
+using namespace Microsoft::WRL;
+using namespace newplus;
+
+template_item::template_item(const std::filesystem::path entry)
+{
+    path = entry;
+}
+
+std::wstring template_item::get_menu_title(const bool show_extension, const bool show_starting_digits) const
+{
+    std::wstring title = path.filename();
+
+    if (!show_starting_digits)
+    {
+        // Hide starting digits, spaces, and .
+        title = remove_starting_digits_from_filename(title);
+    }
+
+    if (show_extension || !path.has_extension())
+    {
+        return title;
+    }
+
+    std::wstring ext = path.extension();
+    title = title.substr(0, title.length() - ext.length());
+
+    return title;
+}
+
+std::wstring template_item::get_target_filename(const bool include_starting_digits) const
+{
+    std::wstring filename = path.filename();
+
+    if (!include_starting_digits)
+    {
+        // Remove starting digits, spaces, and .
+        filename = remove_starting_digits_from_filename(filename);
+    }
+
+    return filename;
+}
+
+std::wstring template_item::remove_starting_digits_from_filename(std::wstring filename) const
+{
+    filename.erase(0, min(filename.find_first_not_of(L"0123456789 ."), filename.size()));
+
+    return filename;
+}
+
+std::wstring template_item::get_explorer_icon() const
+{
+    return utilities::get_explorer_icon(path);
+}
+
+std::filesystem::path template_item::copy_object_to(const HWND window_handle, const std::filesystem::path destination) const
+{
+    // SHFILEOPSTRUCT wants the from and to paths to be terminated with two NULLs,
+    wchar_t double_terminated_path_from[MAX_PATH + 1] = { 0 };
+    wcsncpy_s(double_terminated_path_from, this->path.c_str(), this->path.string().length());
+    double_terminated_path_from[this->path.string().length() + 1] = 0;
+
+    wchar_t double_terminated_path_to[MAX_PATH + 1] = { 0 };
+    wcsncpy_s(double_terminated_path_to, destination.c_str(), destination.string().length());
+    double_terminated_path_to[destination.string().length() + 1] = 0;
+
+    SHFILEOPSTRUCT file_operation_params = { 0 };
+    file_operation_params.wFunc = FO_COPY;
+    file_operation_params.hwnd = window_handle;
+    file_operation_params.pFrom = double_terminated_path_from;
+    file_operation_params.pTo = double_terminated_path_to;
+    file_operation_params.fFlags = FOF_RENAMEONCOLLISION | FOF_ALLOWUNDO | FOF_NOCONFIRMMKDIR | FOF_NOCOPYSECURITYATTRIBS | FOF_WANTMAPPINGHANDLE;
+
+    const int result = SHFileOperation(&file_operation_params);
+
+    if (!file_operation_params.hNameMappings)
+    {
+        // No file name collision on copy
+        return destination;
+    }
+
+    struct file_operation_collision_mapping
+    {
+        int index;
+        SHNAMEMAPPING* mapping;
+    };
+
+    file_operation_collision_mapping* mapping = static_cast<file_operation_collision_mapping*>(file_operation_params.hNameMappings);
+    SHNAMEMAPPING* map = &mapping->mapping[0];
+    std::wstring final_path(map->pszNewPath);
+
+    SHFreeNameMappings(file_operation_params.hNameMappings);
+
+    return final_path;
+}
+
+void template_item::enter_rename_mode(const ComPtr<IUnknown> site, const std::filesystem::path target_fullpath) const
+{
+    std::thread thread_for_renaming_workaround(rename_on_other_thread_workaround, site, target_fullpath);
+    thread_for_renaming_workaround.detach();
+}
+
+void template_item::rename_on_other_thread_workaround(const ComPtr<IUnknown> site, const std::filesystem::path target_fullpath)
+{
+    // Have been unable to have Windows Explorer Shell enter rename mode from the main thread
+    // Sleep for a bit to only enter rename mode when icon has been drawn. Not strictly needed.
+    const std::chrono::milliseconds approx_wait_for_icon_redraw_not_needed{ 350 };
+    std::this_thread::sleep_for(std::chrono::milliseconds(approx_wait_for_icon_redraw_not_needed));
+
+    const std::wstring filename = target_fullpath.filename();
+
+    ComPtr<IServiceProvider> service_provider;
+    site->QueryInterface(IID_PPV_ARGS(&service_provider));
+    ComPtr<IFolderView> folder_view;
+    service_provider->QueryService(__uuidof(IFolderView), IID_PPV_ARGS(&folder_view));
+
+    int count = 0;
+    folder_view->ItemCount(SVGIO_ALLVIEW, &count);
+
+    for (int i = 0; i < count; ++i)
+    {
+        std::wstring path_of_item(MAX_PATH, 0);
+        LPITEMIDLIST pidl;
+
+        folder_view->Item(i, &pidl);
+        SHGetPathFromIDList(pidl, &path_of_item[0]);
+        CoTaskMemFree(pidl);
+
+        std::wstring current_filename = std::filesystem::path(path_of_item.c_str()).filename();
+
+        if (utilities::wstring_same_when_comparing_ignore_case(filename, current_filename))
+        {
+            folder_view->SelectItem(i, SVSI_EDIT | SVSI_SELECT | SVSI_DESELECTOTHERS | SVSI_ENSUREVISIBLE | SVSI_FOCUSED);
+            break;
+        }
+    }
+}