#pragma once

#include <WebView2.h>
#include <WebView2EnvironmentOptions.h>
#include <wil/com.h>

#include <functional>
#include <string>
#include <vector>

#include "platform/webview_platform.h"
#include "rendering/graphics_context.h"
#include "webview/webview.h"
#include <windows.ui.composition.h>

namespace coreframe_flutter_windows {

struct WebviewCreationError {
  HRESULT hr;
  std::string message;

  explicit WebviewCreationError(HRESULT hr, std::string message)
      : hr(hr), message(message) {}

  static std::unique_ptr<WebviewCreationError>
  create(HRESULT hr, const std::string message) {
    return std::make_unique<WebviewCreationError>(hr, message);
  }
};

class WebviewHost {
public:
  typedef std::function<void(std::unique_ptr<Webview>,
                             std::unique_ptr<WebviewCreationError>)>
      WebviewCreationCallback;
  typedef std::function<void(wil::com_ptr<ICoreWebView2CompositionController>,
                             std::unique_ptr<WebviewCreationError>)>
      CompositionControllerCreationCallback;
  typedef std::function<void(wil::com_ptr<ICoreWebView2PointerInfo>,
                             std::unique_ptr<WebviewCreationError>)>
      PointerInfoCreationCallback;

  static std::unique_ptr<WebviewHost>
  Create(WebviewPlatform *platform,
         std::optional<std::wstring> user_data_directory = std::nullopt,
         std::optional<std::wstring> browser_exe_path = std::nullopt,
         std::optional<std::string> arguments = std::nullopt);

  void CreateWebview(HWND hwnd, bool offscreen_only, bool owns_window,
                     WebviewCreationCallback callback);

  void CreateWebViewPointerInfo(PointerInfoCreationCallback cb);

  wil::com_ptr<ICoreWebView2WebResourceRequest>
  CreateWebResourceRequest(const std::string &url, const std::string &method,
                           const std::string &headers,
                           const std::vector<uint8_t> *body);
  wil::com_ptr<ICoreWebView2WebResourceResponse>
  CreateWebResourceResponse(const std::vector<uint8_t> &body, int status,
                            const std::string &reason,
                            const std::string &headers);

  winrt::com_ptr<ABI::Windows::UI::Composition::ICompositor>
  compositor() const {
    return compositor_;
  }

private:
  winrt::com_ptr<ABI::Windows::UI::Composition::ICompositor> compositor_;
  wil::com_ptr<ICoreWebView2Environment3> webview_env_;

  WebviewHost(WebviewPlatform *platform,
              wil::com_ptr<ICoreWebView2Environment3> webview_env);
  void
  CreateWebViewCompositionController(HWND hwnd,
                                     CompositionControllerCreationCallback cb);
};

} // namespace coreframe_flutter_windows