#include "Client/Browser/BrowserClientSite.h"
#include "Client/Browser/WebBrowser.h"

#include <new>

// OFFSET: 0x00AB2AB0
// FUNCTION: fn_17417
bool CWebBrowser::AttachSite() {
  this->site = new (std::nothrow) BrowserClientSite(this->host, this->browser);
  IOleObject* object;
  HRESULT result = this->browser->QueryInterface(IID_IOleObject, reinterpret_cast<void**>(&object));
  if (FAILED(result)) {
    return false;
  }

  result = object->SetClientSite(this->site);
  if (SUCCEEDED(result)) {
    OleSetContainedObject(object, TRUE);
    RECT rectangle;
    GetClientRect(this->host, &rectangle);
    result = object->DoVerb(OLEIVERB_INPLACEACTIVATE, nullptr, this->site, 0, this->host, &rectangle);
    if (SUCCEEDED(result)) {
      const int width = rectangle.right - rectangle.left;
      const int height = rectangle.bottom - rectangle.top;
      this->Resize(width, height);
      result = AdviseBrowserEvents(object, this->site, DIID_DWebBrowserEvents2, &this->cookie);
    }
  }

  object->Release();
  return SUCCEEDED(result);
}

// OFFSET: 0x00AB2430
// FUNCTION: fn_45431
void CWebBrowser::DetachSite() {
  IOleObject* object;
  if (SUCCEEDED(this->browser->QueryInterface(IID_IOleObject, reinterpret_cast<void**>(&object)))) {
    UnadviseBrowserEvents(object, DIID_DWebBrowserEvents2, this->cookie);
    object->Release();
  }

  delete this->site;
  this->site = nullptr;
}