/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

#include "pch.h"
#include "strutils.h"
#include "BrowserPage.h"
#include "BrowserPage.g.cpp"
#include "Bookmark.g.cpp"
#include "ConsoleLog.g.cpp"
#include "Devtools/Client.h"

using namespace std::placeholders;
using namespace winrt::Windows::Foundation;
using namespace winrt::Windows::UI::Xaml;
using namespace winrt::Windows::UI::Core;
using namespace winrt::Windows::UI::ViewManagement;
using namespace winrt::Windows::ApplicationModel::Core;
using namespace winrt::Windows::ApplicationModel::Resources;
using namespace winrt::Windows::ApplicationModel::Resources::Core;
using namespace winrt::Windows::UI::Notifications;
using namespace winrt::Windows::Data::Json;
using namespace winrt::Windows::Data::Xml::Dom;
using namespace winrt::Windows::Storage;
using namespace winrt::servo;

namespace winrt::ServoApp::implementation {

BrowserPage::BrowserPage() {
  InitializeComponent();
  BindServoEvents();
  mLogs = winrt::single_threaded_observable_vector<IInspectable>();

  auto ctx = ResourceContext::GetForCurrentView();
  auto current = ResourceManager::Current();
  auto tree = current.MainResourceMap().GetSubtree(L"PromotedPrefs");
  for (auto s : tree) {
    hstring k = s.Key();
    std::wstring wk = k.c_str();
    std::replace(wk.begin(), wk.end(), '/', '.');
    hstring v = s.Value().Resolve(ctx).ValueAsString();
    mPromotedPrefs.insert(std::pair(wk, v));
  }
}

void BrowserPage::BindServoEvents() {
  servoView().OnURLChanged([=](const auto &, hstring url) {
    mCurrentUrl = url;
    urlTextbox().Text(url);
    UpdateBookmarkPanel();
  });
  servoView().OnTitleChanged([=](const auto &, hstring title) {
    if (title.size() > 0) {
      mCurrentTitle = {title};
    } else {
      mCurrentTitle = {};
    }
    UpdateBookmarkPanel();
  });
  servoView().OnHistoryChanged([=](bool back, bool forward) {
    backButton().IsEnabled(back);
    forwardButton().IsEnabled(forward);
  });
  servoView().OnServoPanic([=](const auto &, hstring /*message*/) {
    mPanicking = true;
    CheckCrashReport();
  });
  servoView().OnLoadStarted([=] {
    mCurrentUrl = {};
    mCurrentTitle = {};
    urlbarLoadingIndicator().IsActive(true);
    transientLoadingIndicator().IsIndeterminate(true);
    reloadButton().IsEnabled(false);
    reloadButton().Visibility(Visibility::Collapsed);
    stopButton().IsEnabled(true);
    stopButton().Visibility(Visibility::Visible);
    devtoolsButton().IsEnabled(true);
    CheckCrashReport();
    UpdateBookmarkPanel();
  });
  servoView().OnLoadEnded([=] {
    urlbarLoadingIndicator().IsActive(false);
    transientLoadingIndicator().IsIndeterminate(false);
    reloadButton().IsEnabled(true);
    reloadButton().Visibility(Visibility::Visible);
    stopButton().IsEnabled(false);
    stopButton().Visibility(Visibility::Collapsed);
  });
  bookmarkPanel().Opening([=](const auto &, const auto &) {
    if (!mCurrentUrl.has_value()) {
      return;
    }
    hstring url = *mCurrentUrl;
    auto resourceLoader = ResourceLoader::GetForCurrentView();
    if (!mBookmarks.Contains(url)) {
      auto label = resourceLoader.GetString(L"bookmarkPanel/addedTitle");
      bookmarkPanelLabel().Text(label);
      mBookmarks.Set(url, bookmarkPanelTitle().Text());
    } else {
      auto label = resourceLoader.GetString(L"bookmarkPanel/editTitle");
      bookmarkPanelLabel().Text(label);
    }
    bookmarkPanelTitle().SelectAll();
  });
  servoView().OnCaptureGesturesStarted([=] {
    servoView().Focus(FocusState::Programmatic);
    navigationBar().IsHitTestVisible(false);
  });
  servoView().OnCaptureGesturesEnded(
      [=] { navigationBar().IsHitTestVisible(true); });
  urlTextbox().GotFocus(std::bind(&BrowserPage::OnURLFocused, this, _1));
  servoView().OnMediaSessionMetadata(
      [=](hstring /*title*/, hstring /*artist*/, hstring /*album*/) {});
  servoView().OnMediaSessionPosition(
      [=](double /*duration*/, double /*position*/, double /*rate*/) {});
  servoView().OnMediaSessionPlaybackStateChange([=](const auto &, int state) {
    if (state == Servo::MediaSessionPlaybackState::None) {
      mediaControls().Visibility(Visibility::Collapsed);
      return;
    }
    mediaControls().Visibility(Visibility::Visible);
    playButton().Visibility(state == Servo::MediaSessionPlaybackState::Paused
                                ? Visibility::Visible
                                : Visibility::Collapsed);
    pauseButton().Visibility(state == Servo::MediaSessionPlaybackState::Paused
                                 ? Visibility::Collapsed
                                 : Visibility::Visible);
  });
  servoView().OnDevtoolsStatusChanged(
      [=](DevtoolsStatus status, unsigned int port, hstring token) {
        mDevtoolsStatus = status;
        mDevtoolsPort = port;
        mDevtoolsToken = token;
      });
  Window::Current().VisibilityChanged(
      [=](const auto &, const VisibilityChangedEventArgs &args) {
        servoView().ChangeVisibility(args.Visible());
      });

  auto obsBM =
      mBookmarks.TemplateSource().as<IObservableVector<IInspectable>>();
  obsBM.VectorChanged(std::bind(&BrowserPage::OnBookmarkDBChanged, this));
}

void BrowserPage::OnURLFocused(IInspectable const &) {
  urlTextbox().SelectAll();
}

void BrowserPage::OnURLKeyboardAccelerator(
    IInspectable const &, Input::KeyboardAcceleratorInvokedEventArgs const &) {
  urlTextbox().Focus(FocusState::Programmatic);
}

void BrowserPage::LoadFXRURI(Uri uri) {
  auto scheme = uri.SchemeName();
  std::wstring raw{uri.RawUri()};
  if (scheme == FXR_SCHEME) {
    auto raw2 = raw.substr(FXR_SCHEME_SLASH_SLASH.size());
    servoView().LoadURIOrSearch(raw2);
    SetTransientMode(false);
  } else if (scheme == FXRMIN_SCHEME) {
    auto raw2 = raw.substr(FXRMIN_SCHEME_SLASH_SLASH.size());
    servoView().LoadURIOrSearch(raw2);
    SetTransientMode(true);
  } else {
    log(L"Unexpected URL: ", uri.RawUri().c_str());
  }
}

void BrowserPage::SetTransientMode(bool transient) {
  servoView().SetTransientMode(transient);
  navigationBar().Visibility(transient ? Visibility::Collapsed
                                       : Visibility::Visible);
  transientLoadingIndicator().Visibility(transient ? Visibility::Visible
                                                   : Visibility::Collapsed);
}

void BrowserPage::SetArgs(hstring args) { servoView().SetArgs(args); }

void BrowserPage::Shutdown() { servoView().Shutdown(); }

/**** USER INTERACTIONS WITH UI ****/

void BrowserPage::OnBackButtonClicked(IInspectable const &,
                                      RoutedEventArgs const &) {
  servoView().GoBack();
}

void BrowserPage::OnForwardButtonClicked(IInspectable const &,
                                         RoutedEventArgs const &) {
  servoView().GoForward();
}

void BrowserPage::OnReloadButtonClicked(IInspectable const &,
                                        RoutedEventArgs const &) {
  servoView().Reload();
}

void BrowserPage::OnStopButtonClicked(IInspectable const &,
                                      RoutedEventArgs const &) {
  servoView().Stop();
}

void BrowserPage::OnHomeButtonClicked(IInspectable const &,
                                      RoutedEventArgs const &) {
  servoView().GoHome();
}

// Given a pref, update its associated UI control.
void BrowserPage::UpdatePref(ServoApp::Pref pref, Controls::Control ctrl) {
  auto value = pref.Value();
  auto type = value.as<IPropertyValue>().Type();
  if (type == PropertyType::Boolean) {
    ctrl.as<Controls::CheckBox>().IsChecked(unbox_value<bool>(value));
  } else if (type == PropertyType::Double) {
    ctrl.as<Microsoft::UI::Xaml::Controls::NumberBox>().Value(
        unbox_value<double>(value));
  } else if (type == PropertyType::Int64) {
    ctrl.as<Microsoft::UI::Xaml::Controls::NumberBox>().Value(
        (double)unbox_value<int64_t>(value));
  } else if (type == PropertyType::String) {
    ctrl.as<Controls::TextBox>().Text(unbox_value<hstring>(value));
  }
  auto stack = ctrl.Parent().as<Controls::StackPanel>();
  auto font = winrt::Windows::UI::Text::FontWeights::Normal();
  if (!pref.IsDefault()) {
    font = winrt::Windows::UI::Text::FontWeights::Bold();
  }
  stack.Children().GetAt(0).as<Controls::TextBlock>().FontWeight(font);
  stack.Children().GetAt(2).as<Controls::Button>().IsEnabled(!pref.IsDefault());
}

void BrowserPage::OnSeeAllPrefClicked(IInspectable const &,
                                      RoutedEventArgs const &) {
  BuildPrefList();
}

// Retrieve the preference list from Servo and build the preference table.
void BrowserPage::BuildPrefList() {
  prefList().Children().Clear();
  bool promoted = !seeAllPrefCheckBox().IsChecked().GetBoolean();
  preferenceSearchbox().Visibility(promoted ? Visibility::Collapsed
                                            : Visibility::Visible);
  preferenceSearchbox().Text(L"");
  // It would be better to use a template and bindings, but the
  // <ListView> takes too long to generate all the items, and
  // it's pretty difficiult to have different controls depending
  // on the pref type.
  auto resourceLoader = ResourceLoader::GetForCurrentView();
  auto resetStr =
      resourceLoader.GetString(L"devtoolsPreferenceResetButton/Content");
  for (auto pref : servoView().Preferences()) {
    std::optional<hstring> description = {};
    if (promoted) {
      auto search = mPromotedPrefs.find(pref.Key());
      if (search == mPromotedPrefs.end()) {
        continue;
      }
      description = {search->second};
    }
    auto value = pref.Value();
    auto type = value.as<IPropertyValue>().Type();
    std::optional<Controls::Control> ctrl;
    if (type == PropertyType::Boolean) {
      auto checkbox = Controls::CheckBox();
      checkbox.IsChecked(unbox_value<bool>(value));
      checkbox.Click([=](const auto &, auto const &) {
        auto upref = servoView().SetBoolPref(pref.Key(),
                                             checkbox.IsChecked().GetBoolean());
        UpdatePref(upref, checkbox);
      });
      ctrl = checkbox;
    } else if (type == PropertyType::String) {
      auto textbox = Controls::TextBox();
      textbox.Text(unbox_value<hstring>(value));
      textbox.KeyUp([=](const auto &, Input::KeyRoutedEventArgs const &e) {
        if (e.Key() == Windows::System::VirtualKey::Enter) {
          auto upref = servoView().SetStringPref(pref.Key(), textbox.Text());
          UpdatePref(upref, textbox);
        }
      });
      ctrl = textbox;
    } else if (type == PropertyType::Int64) {
      // Note: These are *not* under Windows::UI:Xaml namespace.
      auto nbox = Microsoft::UI::Xaml::Controls::NumberBox();
      nbox.Value((double)unbox_value<int64_t>(value));
      nbox.SpinButtonPlacementMode(
          Microsoft::UI::Xaml::Controls::NumberBoxSpinButtonPlacementMode::
              Inline);
      nbox.ValueChanged([=](const auto &, const auto &) {
        int val = (int)nbox.Value();
        auto upref = servoView().SetIntPref(pref.Key(), val);
        UpdatePref(upref, nbox);
      });
      ctrl = nbox;
    } else if (type == PropertyType::Double) {
      auto nbox = Microsoft::UI::Xaml::Controls::NumberBox();
      nbox.Value(unbox_value<double>(value));
      nbox.ValueChanged([=](const auto &, const auto &) {
        auto upref = servoView().SetIntPref(pref.Key(), (int64_t)nbox.Value());
        UpdatePref(upref, (Controls::Control &)nbox);
      });
      ctrl = nbox;
    }
    if (ctrl.has_value()) {
      auto stack = Controls::StackPanel();
      stack.Tag(winrt::box_value(pref.Key()));
      stack.Padding({4, 4, 4, 4});
      stack.Orientation(Controls::Orientation::Horizontal);
      auto key = Controls::TextBlock();
      key.Text(promoted ? *description : pref.Key());
      key.Width(350);
      if (!pref.IsDefault()) {
        auto font = winrt::Windows::UI::Text::FontWeights::Bold();
        key.FontWeight(font);
      }
      stack.Children().Append(key);
      ctrl->Width(300);
      ctrl->Margin({4, 0, 40, 0});
      stack.Children().Append(*ctrl);
      auto reset = Controls::Button();
      reset.Content(winrt::box_value(resetStr));
      reset.IsEnabled(!pref.IsDefault());
      reset.Click([=](const auto &, auto const &) {
        auto upref = servoView().ResetPref(pref.Key());
        UpdatePref(upref, *ctrl);
      });
      stack.Children().Append(reset);
      prefList().Children().Append(stack);
    }
  }
}

void BrowserPage::OnPrefererenceSearchboxEdited(
    IInspectable const &, Input::KeyRoutedEventArgs const &) {
  auto input = preferenceSearchbox().Text();
  for (auto element : prefList().Children()) {
    auto ctrl = (Controls::Control &)element;
    if (input.size() == 0) {
      ctrl.Visibility(Visibility::Visible);
    } else {
      auto tag = ctrl.Tag();
      std::wstring key = static_cast<std::wstring>(unbox_value<hstring>(tag));
      bool not_found = key.find(input) == std::wstring::npos;
      ctrl.Visibility(not_found ? Visibility::Collapsed : Visibility::Visible);
    }
  }
}

void BrowserPage::ClearConsole() {
  Dispatcher().RunAsync(CoreDispatcherPriority::High, [=] { mLogs.Clear(); });
}

void BrowserPage::OnDevtoolsMessage(DevtoolsMessageLevel level, hstring source,
                                    hstring body) {
  Dispatcher().RunAsync(CoreDispatcherPriority::High, [=] {
    auto glyphColor = UI::Colors::Transparent();
    auto glyph = L"";
    if (level == servo::DevtoolsMessageLevel::Error) {
      glyphColor = UI::Colors::Red();
      glyph = L"\xEA39"; // ErrorBadge
    } else if (level == servo::DevtoolsMessageLevel::Warn) {
      glyphColor = UI::Colors::Orange();
      glyph = L"\xE7BA"; // Warning
    }
    mLogs.Append(make<ConsoleLog>(glyphColor, glyph, body, source));
  });
}

void BrowserPage::CheckCrashReport() {
  Concurrency::create_task([=] {
    auto pref = servo::Servo::GetPref(L"shell.crash_reporter.enabled");
    bool reporter_enabled = unbox_value<bool>(std::get<1>(pref));
    auto storageFolder = ApplicationData::Current().LocalFolder();
    bool file_exist =
        storageFolder.TryGetItemAsync(L"crash-report.txt").get() != nullptr;
    if (reporter_enabled && file_exist) {
      auto crash_file = storageFolder.GetFileAsync(L"crash-report.txt").get();
      auto content = FileIO::ReadTextAsync(crash_file).get();
      Dispatcher().RunAsync(CoreDispatcherPriority::High, [=] {
        auto resourceLoader = ResourceLoader::GetForCurrentView();
        auto message = resourceLoader.GetString(mPanicking ? L"crash/Happening"
                                                           : L"crash/Happened");
        crashTabMessage().Text(message);
        crashReport().Text(content);
        crashTab().Visibility(Visibility::Visible);
        crashTab().IsSelected(true);
        ShowToolbox();
      });
    } else {
      Dispatcher().RunAsync(CoreDispatcherPriority::High, [=] {
        crashTab().Visibility(Visibility::Collapsed);
        devtoolsTabConsole().IsSelected(true);
      });
    }
  });
}

void BrowserPage::OnDismissCrashReport(IInspectable const &,
                                       RoutedEventArgs const &) {
  Concurrency::create_task([=] {
    auto storageFolder = ApplicationData::Current().LocalFolder();
    auto crash_file = storageFolder.GetFileAsync(L"crash-report.txt").get();
    crash_file.DeleteAsync().get();
  });
  HideToolbox();
}

void BrowserPage::OnSubmitCrashReport(IInspectable const &,
                                      RoutedEventArgs const &) {
  // FIXME
}

void BrowserPage::OnDevtoolsDetached() {}

void BrowserPage::ShowToolbox() {
  if (toolbox().Visibility() == Visibility::Visible) {
    return;
  }
  toolbox().Visibility(Visibility::Visible);
  CheckCrashReport();
  BuildPrefList();
  auto resourceLoader = ResourceLoader::GetForCurrentView();
  if (mDevtoolsStatus == DevtoolsStatus::Running) {
    hstring port = to_hstring(mDevtoolsPort);
    if (mDevtoolsClient == nullptr) {
      DevtoolsDelegate *dd = static_cast<DevtoolsDelegate *>(this);
      mDevtoolsClient = std::make_unique<DevtoolsClient>(L"localhost", port,
                                                         mDevtoolsToken, *dd);
    }
    mDevtoolsClient->Run();
    std::wstring message =
        resourceLoader.GetString(L"devtoolsStatus/Running").c_str();
    hstring formatted{format(message, port.c_str())};
    OnDevtoolsMessage(servo::DevtoolsMessageLevel::None, L"", formatted);
  } else if (mDevtoolsStatus == DevtoolsStatus::Failed) {
    auto body = resourceLoader.GetString(L"devtoolsStatus/Failed");
    OnDevtoolsMessage(servo::DevtoolsMessageLevel::Error, L"", body);
  } else if (mDevtoolsStatus == DevtoolsStatus::Stopped) {
    auto body = resourceLoader.GetString(L"devtoolsStatus/Stopped");
    OnDevtoolsMessage(servo::DevtoolsMessageLevel::None, L"", body);
  }
}

void BrowserPage::HideToolbox() {
  prefList().Children().Clear();
  toolbox().Visibility(Visibility::Collapsed);
  ClearConsole();
  if (mDevtoolsClient != nullptr) {
    mDevtoolsClient->Stop();
  }
}

void BrowserPage::OnDevtoolsButtonClicked(IInspectable const &,
                                          RoutedEventArgs const &) {
  if (toolbox().Visibility() == Visibility::Visible) {
    HideToolbox();
  } else {
    ShowToolbox();
  }
}

void BrowserPage::OnBookmarkDBChanged() {
  Dispatcher().RunAsync(CoreDispatcherPriority::High,
                        [=] { UpdateBookmarkPanel(); });
}

void BrowserPage::UpdateBookmarkPanel() {
  if (mCurrentUrl.has_value()) {
    bookmarkButton().IsEnabled(true);
    if (mBookmarks.Contains(*mCurrentUrl)) {
      bookmarkPanelIcon().Symbol(Controls::Symbol::SolidStar);
      auto name = mBookmarks.GetName(*mCurrentUrl);
      bookmarkPanelTitle().Text(name);
    } else {
      bookmarkPanelIcon().Symbol(Controls::Symbol::OutlineStar);
      auto label = mCurrentTitle.value_or(*mCurrentUrl);
      bookmarkPanelTitle().Text(label);
    }
  } else {
    bookmarkButton().IsEnabled(false);
  }
  if (mBookmarks.TemplateSource().Size() == 0) {
    bookmarkToolbar().Visibility(Visibility::Collapsed);
  } else {
    bookmarkToolbar().Visibility(Visibility::Visible);
  }
}

void BrowserPage::OnBookmarkEdited(IInspectable const &,
                                   Input::KeyRoutedEventArgs const &e) {
  if (e.Key() == Windows::System::VirtualKey::Enter) {
    UpdateBookmark();
  }
}

void BrowserPage::OnBookmarkClicked(IInspectable const &sender,
                                    RoutedEventArgs const &) {
  auto button = sender.as<Controls::Button>();
  auto url = winrt::unbox_value<hstring>(button.Tag());
  servoView().LoadURIOrSearch(url);
}

void BrowserPage::RemoveBookmark() {
  mBookmarks.Delete(*mCurrentUrl);
  bookmarkPanel().Hide();
}

void BrowserPage::UpdateBookmark() {
  mBookmarks.Set(*mCurrentUrl, bookmarkPanelTitle().Text());
  bookmarkPanel().Hide();
}

void BrowserPage::OnJSInputEdited(IInspectable const &,
                                  Input::KeyRoutedEventArgs const &e) {
  if (e.Key() == Windows::System::VirtualKey::Enter) {
    auto input = JSInput().Text();
    JSInput().Text(L"");
    mDevtoolsClient->Evaluate(input);
  }
}

void BrowserPage::OnURLEdited(IInspectable const &,
                              Input::KeyRoutedEventArgs const &e) {
  if (e.Key() == Windows::System::VirtualKey::Enter) {
    servoView().Focus(FocusState::Programmatic);
    auto input = urlTextbox().Text();
    auto uri = servoView().LoadURIOrSearch(input);
    urlTextbox().Text(uri);
  }
}

void BrowserPage::OnMediaControlsPlayClicked(IInspectable const &,
                                             RoutedEventArgs const &) {
  servoView().SendMediaSessionAction(
      static_cast<int32_t>(Servo::MediaSessionActionType::Play));
}
void BrowserPage::OnMediaControlsPauseClicked(IInspectable const &,
                                              RoutedEventArgs const &) {
  servoView().SendMediaSessionAction(
      static_cast<int32_t>(Servo::MediaSessionActionType::Pause));
}

} // namespace winrt::ServoApp::implementation