using System; using System.Collections.Generic; using System.Diagnostics; using System.Threading; using System.Threading.Tasks; using System.Windows; using System.Windows.Input; using NHotkey; using NHotkey.Wpf; using Wox.Core.Plugin; using Wox.Core.Resource; using Wox.Core.UserSettings; using Wox.Helper; using Wox.Infrastructure; using Wox.Infrastructure.Hotkey; using Wox.Infrastructure.Image; using Wox.Infrastructure.Storage; using Wox.Plugin; using Wox.Storage; namespace Wox.ViewModel { public class MainViewModel : BaseModel, ISavable { #region Private Fields private Visibility _contextMenuVisibility; private bool _queryHasReturn; private Query _lastQuery; private bool _ignoreTextChange; private string _queryTextBeforeLoadContextMenu; private string _queryText; private readonly JsonStrorage _queryHistoryStorage; private readonly JsonStrorage _userSelectedRecordStorage; private readonly JsonStrorage _topMostRecordStorage; private readonly Settings _settings; private readonly QueryHistory _queryHistory; private readonly UserSelectedRecord _userSelectedRecord; private readonly TopMostRecord _topMostRecord; private CancellationTokenSource _updateSource; private CancellationToken _updateToken; private bool _saved; #endregion #region Constructor public MainViewModel(Settings settings) { _saved = false; _queryTextBeforeLoadContextMenu = ""; _queryText = ""; _lastQuery = new Query(); _settings = settings; // happlebao todo temp fix for instance code logic HttpProxy.Instance.Settings = _settings; InternationalizationManager.Instance.Settings = _settings; InternationalizationManager.Instance.ChangeLanguage(_settings.Language); ThemeManager.Instance.Settings = _settings; _queryHistoryStorage = new JsonStrorage(); _userSelectedRecordStorage = new JsonStrorage(); _topMostRecordStorage = new JsonStrorage(); _queryHistory = _queryHistoryStorage.Load(); _userSelectedRecord = _userSelectedRecordStorage.Load(); _topMostRecord = _topMostRecordStorage.Load(); InitializeResultListBox(); InitializeContextMenu(); InitializeKeyCommands(); RegisterResultsUpdatedEvent(); SetHotkey(_settings.Hotkey, OnHotkey); SetCustomPluginHotkey(); } private void RegisterResultsUpdatedEvent() { foreach (var pair in PluginManager.GetPluginsForInterface()) { var plugin = (IResultUpdated) pair.Plugin; plugin.ResultsUpdated += (s, e) => { Task.Run(() => { PluginManager.UpdatePluginMetadata(e.Results, pair.Metadata, e.Query); UpdateResultView(e.Results, pair.Metadata, e.Query); }, _updateToken); }; } } private void InitializeKeyCommands() { EscCommand = new RelayCommand(_ => { if (ContextMenuVisibility.IsVisible()) { ContextMenuVisibility = Visibility.Collapsed; } else { MainWindowVisibility = Visibility.Collapsed; } }); SelectNextItemCommand = new RelayCommand(_ => { if (ContextMenuVisibility.IsVisible()) { ContextMenu.SelectNextResult(); } else { Results.SelectNextResult(); } }); SelectPrevItemCommand = new RelayCommand(_ => { if (ContextMenuVisibility.IsVisible()) { ContextMenu.SelectPrevResult(); } else { Results.SelectPrevResult(); } }); DisplayNextQueryCommand = new RelayCommand(_ => { var nextQuery = _queryHistory.Next(); DisplayQueryHistory(nextQuery); }); DisplayPrevQueryCommand = new RelayCommand(_ => { var prev = _queryHistory.Previous(); DisplayQueryHistory(prev); }); SelectNextPageCommand = new RelayCommand(_ => { Results.SelectNextPage(); }); SelectPrevPageCommand = new RelayCommand(_ => { Results.SelectPrevPage(); }); StartHelpCommand = new RelayCommand(_ => { Process.Start("http://doc.getwox.com"); }); OpenResultCommand = new RelayCommand(index => { var results = ContextMenuVisibility.IsVisible() ? ContextMenu : Results; if (index != null) { results.SelectedIndex = int.Parse(index.ToString()); } var result = results.SelectedItem?.RawResult; if (result != null) // SelectedItem returns null if selection is empty. { bool hideWindow = result.Action != null && result.Action(new ActionContext { SpecialKeyState = GlobalHotkey.Instance.CheckModifiers() }); if (hideWindow) { MainWindowVisibility = Visibility.Collapsed; } if (!ContextMenuVisibility.IsVisible()) { _userSelectedRecord.Add(result); _queryHistory.Add(result.OriginQuery.RawQuery); } } }); LoadContextMenuCommand = new RelayCommand(_ => { if (!ContextMenuVisibility.IsVisible()) { var result = Results.SelectedItem?.RawResult; if (result != null) // SelectedItem returns null if selection is empty. { var id = result.PluginID; var menus = PluginManager.GetContextMenusForPlugin(result); menus.Add(ContextMenuTopMost(result)); menus.Add(ContextMenuPluginInfo(id)); ContextMenu.Clear(); Task.Run(() => { ContextMenu.AddResults(menus, id); }, _updateToken); ContextMenuVisibility = Visibility.Visible; } } else { ContextMenuVisibility = Visibility.Collapsed; } }); } private void InitializeResultListBox() { Results = new ResultsViewModel(_settings); ResultListBoxVisibility = Visibility.Collapsed; } private void InitializeContextMenu() { ContextMenu = new ResultsViewModel(_settings); ContextMenuVisibility = Visibility.Collapsed; } private void HandleQueryTextUpdated() { ProgressBarVisibility = Visibility.Hidden; _updateSource?.Cancel(); _updateSource = new CancellationTokenSource(); _updateToken = _updateSource.Token; if (ContextMenuVisibility.IsVisible()) { QueryContextMenu(); } else { string query = QueryText.Trim(); if (!string.IsNullOrEmpty(query)) { Query(query); //reset query history index after user start new query ResetQueryHistoryIndex(); } else { Results.Clear(); ResultListBoxVisibility = Visibility.Collapsed; } } } #endregion #region ViewModel Properties public ResultsViewModel Results { get; private set; } public ResultsViewModel ContextMenu { get; private set; } public string QueryText { get { return _queryText; } set { _queryText = value; if (_ignoreTextChange) { _ignoreTextChange = false; } else { HandleQueryTextUpdated(); } } } public double Left { get; set; } public double Top { get; set; } public Visibility ContextMenuVisibility { get { return _contextMenuVisibility; } set { _contextMenuVisibility = value; _ignoreTextChange = true; if (!value.IsVisible()) { QueryText = _queryTextBeforeLoadContextMenu; ResultListBoxVisibility = Visibility.Visible; OnCursorMovedToEnd(); } else { _queryTextBeforeLoadContextMenu = QueryText; QueryText = ""; ResultListBoxVisibility = Visibility.Collapsed; } } } public Visibility ProgressBarVisibility { get; set; } public Visibility ResultListBoxVisibility { get; set; } public Visibility MainWindowVisibility { get; set; } public ICommand EscCommand { get; set; } public ICommand SelectNextItemCommand { get; set; } public ICommand SelectPrevItemCommand { get; set; } //todo happlebao restore history command public ICommand DisplayNextQueryCommand { get; set; } public ICommand DisplayPrevQueryCommand { get; set; } public ICommand SelectNextPageCommand { get; set; } public ICommand SelectPrevPageCommand { get; set; } public ICommand StartHelpCommand { get; set; } public ICommand LoadContextMenuCommand { get; set; } public ICommand OpenResultCommand { get; set; } #endregion #region Private Methods private void QueryContextMenu() { var contextMenuId = "Context Menu Id"; var query = QueryText.ToLower(); if (!string.IsNullOrEmpty(query)) { List filterResults = new List(); foreach (var contextMenu in ContextMenu.Results) { if (StringMatcher.IsMatch(contextMenu.Title, query) || StringMatcher.IsMatch(contextMenu.SubTitle, query)) { filterResults.Add(contextMenu.RawResult); } } ContextMenu.Clear(); Task.Run(() => { ContextMenu.AddResults(filterResults, contextMenuId); }, _updateToken); } } private void Query(string text) { _queryHasReturn = false; var query = PluginManager.QueryInit(text); if (query != null) { // handle the exclusiveness of plugin using action keyword string lastKeyword = _lastQuery.ActionKeyword; string keyword = query.ActionKeyword; if (string.IsNullOrEmpty(lastKeyword)) { if (!string.IsNullOrEmpty(keyword)) { Results.RemoveResultsExcept(PluginManager.NonGlobalPlugins[keyword].Metadata); } } else { if (string.IsNullOrEmpty(keyword)) { Results.RemoveResultsFor(PluginManager.NonGlobalPlugins[lastKeyword].Metadata); } else if (lastKeyword != keyword) { Results.RemoveResultsExcept(PluginManager.NonGlobalPlugins[keyword].Metadata); } } _lastQuery = query; Task.Delay(200, _updateToken).ContinueWith(_ => { if (query.RawQuery == _lastQuery.RawQuery && !_queryHasReturn) { ProgressBarVisibility = Visibility.Visible; } }, _updateToken); var plugins = PluginManager.ValidPluginsForQuery(query); Task.Run(() => { Parallel.ForEach(plugins, plugin => { var config = _settings.PluginSettings.Plugins[plugin.Metadata.ID]; if (!config.Disabled) { var results = PluginManager.QueryForPlugin(plugin, query); UpdateResultView(results, plugin.Metadata, query); } }); }, _updateToken); } } private void ResetQueryHistoryIndex() { Results.RemoveResultsFor(QueryHistory.MetaData); _queryHistory.Reset(); } private void DisplayQueryHistory(HistoryItem history) { if (history != null) { var historyMetadata = QueryHistory.MetaData; QueryText = history.Query; OnTextBoxSelected(); var executeQueryHistoryTitle = InternationalizationManager.Instance.GetTranslation("executeQuery"); var lastExecuteTime = InternationalizationManager.Instance.GetTranslation("lastExecuteTime"); Results.RemoveResultsExcept(historyMetadata); var result = new Result { Title = string.Format(executeQueryHistoryTitle, history.Query), SubTitle = string.Format(lastExecuteTime, history.ExecutedDateTime), IcoPath = "Images\\history.png", PluginDirectory = Infrastructure.Constant.ProgramDirectory, Action = _ => { QueryText = history.Query; OnTextBoxSelected(); return false; } }; Task.Run(() => { Results.AddResults(new List {result}, historyMetadata.ID); }, _updateToken); } } private Result ContextMenuTopMost(Result result) { Result menu; if (_topMostRecord.IsTopMost(result)) { menu = new Result { Title = InternationalizationManager.Instance.GetTranslation("cancelTopMostInThisQuery"), IcoPath = "Images\\down.png", PluginDirectory = Infrastructure.Constant.ProgramDirectory, Action = _ => { _topMostRecord.Remove(result); App.API.ShowMsg("Succeed"); return false; } }; } else { menu = new Result { Title = InternationalizationManager.Instance.GetTranslation("setAsTopMostInThisQuery"), IcoPath = "Images\\up.png", PluginDirectory = Infrastructure.Constant.ProgramDirectory, Action = _ => { _topMostRecord.AddOrUpdate(result); App.API.ShowMsg("Succeed"); return false; } }; } return menu; } private Result ContextMenuPluginInfo(string id) { var metadata = PluginManager.GetPluginForId(id).Metadata; var translator = InternationalizationManager.Instance; var author = translator.GetTranslation("author"); var website = translator.GetTranslation("website"); var version = translator.GetTranslation("version"); var plugin = translator.GetTranslation("plugin"); var title = $"{plugin}: {metadata.Name}"; var icon = metadata.IcoPath; var subtitle = $"{author}: {metadata.Author}, {website}: {metadata.Website} {version}: {metadata.Version}"; var menu = new Result { Title = title, IcoPath = icon, SubTitle = subtitle, PluginDirectory = metadata.PluginDirectory, Action = _ => false }; return menu; } #endregion #region Hotkey internal void SetHotkey(string hotkeyStr, EventHandler action) { var hotkey = new HotkeyModel(hotkeyStr); SetHotkey(hotkey, action); } public void SetHotkey(HotkeyModel hotkey, EventHandler action) { string hotkeyStr = hotkey.ToString(); try { HotkeyManager.Current.AddOrReplace(hotkeyStr, hotkey.CharKey, hotkey.ModifierKeys, action); } catch (Exception) { string errorMsg = string.Format(InternationalizationManager.Instance.GetTranslation("registerHotkeyFailed"), hotkeyStr); MessageBox.Show(errorMsg); } } public void RemoveHotkey(string hotkeyStr) { if (!string.IsNullOrEmpty(hotkeyStr)) { HotkeyManager.Current.Remove(hotkeyStr); } } /// /// Checks if Wox should ignore any hotkeys /// /// private bool ShouldIgnoreHotkeys() { //double if to omit calling win32 function if (_settings.IgnoreHotkeysOnFullscreen) if (WindowIntelopHelper.IsWindowFullscreen()) return true; return false; } private void SetCustomPluginHotkey() { if (_settings.CustomPluginHotkeys == null) return; foreach (CustomPluginHotkey hotkey in _settings.CustomPluginHotkeys) { CustomPluginHotkey hotkey1 = hotkey; SetHotkey(hotkey.Hotkey, delegate { if (ShouldIgnoreHotkeys()) return; App.API.ShowApp(); App.API.ChangeQuery(hotkey1.ActionKeyword, true); }); } } private void OnHotkey(object sender, HotkeyEventArgs e) { if (ShouldIgnoreHotkeys()) return; ToggleWox(); e.Handled = true; } private void ToggleWox() { if (!MainWindowVisibility.IsVisible()) { MainWindowVisibility = Visibility.Visible; OnTextBoxSelected(); } else { MainWindowVisibility = Visibility.Collapsed; } } #endregion #region Public Methods public void Save() { if (!_saved) { _queryHistoryStorage.Save(); _userSelectedRecordStorage.Save(); _topMostRecordStorage.Save(); PluginManager.Save(); ImageLoader.Save(); _saved = true; } } /// /// To avoid deadlock, this method should not called from main thread /// public void UpdateResultView(List list, PluginMetadata metadata, Query originQuery) { _queryHasReturn = true; ProgressBarVisibility = Visibility.Hidden; foreach (var result in list) { if (_topMostRecord.IsTopMost(result)) { result.Score = int.MaxValue; } else { result.Score += _userSelectedRecord.GetSelectedCount(result)*5; } } if (originQuery.RawQuery == _lastQuery.RawQuery) { Results.AddResults(list, metadata.ID); } if (list.Count > 0 && !ResultListBoxVisibility.IsVisible()) { ResultListBoxVisibility = Visibility.Visible; } } #endregion public event EventHandler CursorMovedToEnd; public void OnCursorMovedToEnd() { CursorMovedToEnd?.Invoke(this, new EventArgs()); } public event EventHandler TextBoxSelected; public void OnTextBoxSelected() { TextBoxSelected?.Invoke(this, new EventArgs()); } } }