public override object ReadJson(JsonReader reader, Type objectType, object existingValue, JsonSerializer serializer)
{
JObject jo = JObject.Load(reader);
var model = new DashboardModel(m_engine);
model.TotalDataBlocked = ByteSize.FromBytes(jo["TotalDataBlocked"]["Bytes"].ToObject<double>());
model.TotalHtmlElementsRemoved = jo["TotalHtmlElementsRemoved"].ToObject<uint>();
model.TotalRequestsBlocked = jo["TotalRequestsBlocked"].ToObject<uint>();
return model;
}
/// <summary>
/// Constructs a new DashboardViewModel instance.
/// </summary>
/// <param name="model">
/// The underlying DashboardModel from which to derive and modify state.
/// </param>
/// <exception cref="ArgumentException">
/// In the event that the model parameter is null, will throw ArgumentException.
/// </exception>
public DashboardViewModel(DashboardModel model)
{
m_model = model;
if (m_model == null)
{
throw new ArgumentException("Expected valid DashboardModel instance.");
}
FilteredApplications = new BindingList<FilteredAppViewModel>();
}
/// <summary>
/// Loads the models, if present, from their serialized JSON files.
/// </summary>
private void LoadProgramState()
{
var stateOutputDir = AppDomain.CurrentDomain.BaseDirectory + @"User\";
JsonSerializerSettings settings = new JsonSerializerSettings();
settings.Converters.Add(new FilteringCategoryConverter(m_filteringEngine));
settings.Converters.Add(new DashboardConverter(m_filteringEngine));
settings.Converters.Add(new FilteredAppConverter());
// This thread != UI thread.
Current.Dispatcher.BeginInvoke(
System.Windows.Threading.DispatcherPriority.Normal,
(Action)delegate ()
{
if (m_filteringCategoriesObservable == null)
{
m_filteringCategoriesObservable = new ObservableCollection<CategorizedFilteredRequestsViewModel>();
}
}
);
// Restore filtering categories.
if (File.Exists(stateOutputDir + "FilterCategories.json"))
{
var filterCatsSerialized = File.ReadAllText(stateOutputDir + "FilterCategories.json");
var filteringCatsList = JsonConvert.DeserializeObject<List<FilteringCategory>>(filterCatsSerialized, settings);
foreach (var filteringList in filteringCatsList)
{
// Ensure lists are up to date.
try
{
filteringList.UpdateAndLoad();
Current.Dispatcher.BeginInvoke(
System.Windows.Threading.DispatcherPriority.Normal,
(Action)delegate ()
{
m_filteringCategoriesObservable.Add(new CategorizedFilteredRequestsViewModel(filteringList));
}
);
}
catch (Exception err)
{
m_logger.Error("Error while updating filtering list: {0}.", err.Message);
}
}
}
// Restore dashboard stats.
if (File.Exists(stateOutputDir + "Dashboard.json"))
{
var dashboardSerialized = File.ReadAllText(stateOutputDir + "Dashboard.json");
m_modelDashboard = JsonConvert.DeserializeObject<DashboardModel>(dashboardSerialized, settings);
m_viewModelDashboard = new DashboardViewModel(m_modelDashboard);
}
// Restore settings.
if (File.Exists(stateOutputDir + "Settings.json"))
{
var settingsSerialized = File.ReadAllText(stateOutputDir + "Settings.json");
m_modelSettings = JsonConvert.DeserializeObject<SettingsModel>(settingsSerialized, settings);
m_viewModelSettings = new SettingsViewModel(m_modelSettings);
}
// Restore filtered apps.
if (File.Exists(stateOutputDir + "FilteredApps.json"))
{
var filteredAppsSerialized = File.ReadAllText(stateOutputDir + "FilteredApps.json");
m_filteredApplicationsTable = JsonConvert.DeserializeObject<ConcurrentDictionary<string, FilteredAppModel>>(filteredAppsSerialized, settings);
foreach (var entry in m_filteredApplicationsTable)
{
m_logger.Info(entry.Key);
Current.Dispatcher.BeginInvoke(
System.Windows.Threading.DispatcherPriority.Normal,
(Action)delegate ()
{
m_viewModelDashboard.FilteredApplications.Add(new FilteredAppViewModel(entry.Value));
}
);
}
}
}
/// <summary>
/// Inits all the various views for the application, which will be pushed and popped on the
/// primary window as requested or required.
/// </summary>
private void InitViews()
{
if (m_filteringEngine == null)
{
throw new Exception("Engine must be initialized prior to initializing views, as views require references to allow user control.");
}
// We null check, because if the save state load worked, these conditions will be
// false and we won't overwrite our restored state.
// This collection is initialized here because it has a direct connection to the UI.
if (m_filteredApplicationsTable == null)
{
m_filteredApplicationsTable = new ConcurrentDictionary<string, FilteredAppModel>();
}
if (m_modelDashboard == null)
{
m_modelDashboard = new DashboardModel(m_filteringEngine);
}
if (m_viewModelDashboard == null)
{
m_viewModelDashboard = new DashboardViewModel(m_modelDashboard);
}
if (m_modelStatistics == null)
{
m_modelStatistics = new StatisticsModel();
}
if (m_viewModelStatistics == null)
{
m_viewModelStatistics = new StatisticsViewModel(m_modelStatistics);
}
if (m_modelSettings == null)
{
m_modelSettings = new SettingsModel();
}
if (m_viewModelSettings == null)
{
m_viewModelSettings = new SettingsViewModel(m_modelSettings);
}
if(m_viewModelWaste == null)
{
m_viewModelWaste = new WasteViewModel(m_viewModelSettings, m_viewModelDashboard);
}
// Necessary because we use a background worker. This thread != UI thread.
Current.Dispatcher.BeginInvoke(
System.Windows.Threading.DispatcherPriority.Normal,
(Action)delegate ()
{
m_modelStatistics.FilterCategories = m_filteringCategoriesObservable;
m_modelSettings.FilterCategories = m_filteringCategoriesObservable;
m_primaryWindow = new MainWindow();
m_viewProgressWait = new ProgressWait();
m_viewStatistics = new Statistics(m_viewModelStatistics);
m_viewDashboard = new Dashboard(m_viewModelDashboard);
m_viewSettings = new Settings(m_viewModelSettings, new AddCategoryControl(m_filteringEngine));
m_viewWaste = new Waste(m_viewModelWaste);
m_primaryWindow.ViewChangeRequest += OnViewChangeRequest;
m_viewDashboard.ViewChangeRequest += OnViewChangeRequest;
m_viewStatistics.ViewChangeRequest += OnViewChangeRequest;
m_viewSettings.ViewChangeRequest += OnViewChangeRequest;
m_viewWaste.ViewChangeRequest += OnViewChangeRequest;
// Listen for the statistics view requests for app-wide stats deletion.
m_viewStatistics.ClearStatisticsRequested += OnClearAllStatsRequest;
MainWindow = m_primaryWindow;
}
);
}