using System; using System.Collections.Generic; using System.Diagnostics; using System.Threading; using System.Threading.Tasks; using System.Windows; using System.Windows.Input; using Wox.Core.Plugin; using Wox.Core.Resource; using Wox.Core.Updater; using Wox.Core.UserSettings; using Wox.Helper; using Wox.Infrastructure; using Wox.Infrastructure.Hotkey; using Wox.Infrastructure.Storage; using Wox.Plugin; using Wox.Storage; namespace Wox.ViewModel { public class MainViewModel : BaseViewModel, ISavable { #region Private Fields private bool _isProgressBarTooltipVisible; private double _left; private double _top; private Visibility _contextMenuVisibility; private Visibility _progressBarVisibility; private Visibility _resultListBoxVisibility; private Visibility _mainWindowVisibility; private bool _queryHasReturn; private Query _lastQuery; private bool _ignoreTextChange; private string _queryTextBeforeLoadContextMenu; private string _queryText; private readonly JsonStrorage _settingsStorage; private readonly JsonStrorage _queryHistoryStorage; private readonly JsonStrorage _userSelectedRecordStorage; private readonly JsonStrorage _topMostRecordStorage; // todo happlebao this field should be private in the future public readonly Settings _settings; private readonly QueryHistory _queryHistory; private readonly UserSelectedRecord _userSelectedRecord; private readonly TopMostRecord _topMostRecord; private CancellationTokenSource _updateSource; private CancellationToken _updateToken; #endregion #region Constructor public MainViewModel() { _queryTextBeforeLoadContextMenu = ""; _queryText = ""; _lastQuery = new Query(); _settingsStorage = new JsonStrorage(); _settings = _settingsStorage.Load(); // happlebao todo temp fix for instance code logic HttpProxy.Instance.Settings = _settings; UpdaterManager.Instance.Settings = _settings; InternationalizationManager.Instance.Settings = _settings; 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(); } 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(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; } }); BackCommand = new RelayCommand(_ => { ListeningKeyPressed?.Invoke(this, new ListeningKeyPressedEventArgs(_ as KeyEventArgs)); }); } 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; OnPropertyChanged(); if (_ignoreTextChange) { _ignoreTextChange = false; } else { HandleQueryTextUpdated(); } } } public double Left { get { return _left; } set { _left = value; OnPropertyChanged(); } } public double Top { get { return _top; } set { _top = value; OnPropertyChanged(); } } public Visibility ContextMenuVisibility { get { return _contextMenuVisibility; } set { _contextMenuVisibility = value; OnPropertyChanged(); _ignoreTextChange = true; if (!value.IsVisible()) { QueryText = _queryTextBeforeLoadContextMenu; ResultListBoxVisibility = Visibility.Visible; OnCursorMovedToEnd(); } else { _queryTextBeforeLoadContextMenu = QueryText; QueryText = ""; ResultListBoxVisibility = Visibility.Collapsed; } } } public Visibility ProgressBarVisibility { get { return _progressBarVisibility; } set { _progressBarVisibility = value; OnPropertyChanged(); } } public Visibility ResultListBoxVisibility { get { return _resultListBoxVisibility; } set { _resultListBoxVisibility = value; OnPropertyChanged(); } } public Visibility MainWindowVisibility { get { return _mainWindowVisibility; } set { _mainWindowVisibility = value; OnPropertyChanged(); MainWindowVisibilityChanged?.Invoke(this, new EventArgs()); } } public ICommand EscCommand { get; set; } public ICommand SelectNextItemCommand { get; set; } public ICommand SelectPrevItemCommand { get; set; } 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; } public ICommand BackCommand { 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.Wox.ProgramPath, 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.Wox.ProgramPath, 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.Wox.ProgramPath, 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 Public Methods public void Save() { _settingsStorage.Save(); _queryHistoryStorage.Save(); _userSelectedRecordStorage.Save(); _topMostRecordStorage.Save(); } /// /// 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 ListeningKeyPressed; public event EventHandler MainWindowVisibilityChanged; public event EventHandler CursorMovedToEnd; public void OnCursorMovedToEnd() { CursorMovedToEnd?.Invoke(this, new EventArgs()); } public event EventHandler TextBoxSelected; public void OnTextBoxSelected() { TextBoxSelected?.Invoke(this, new EventArgs()); } } public class ListeningKeyPressedEventArgs : EventArgs { public KeyEventArgs KeyEventArgs { get; private set; } public ListeningKeyPressedEventArgs(KeyEventArgs keyEventArgs) { KeyEventArgs = keyEventArgs; } } }