/
MMPatchRunner.cs
100 lines (77 loc) · 3.51 KB
/
MMPatchRunner.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using UnityEngine;
using ModuleManager.Collections;
using ModuleManager.Extensions;
using ModuleManager.Logging;
using ModuleManager.Threading;
using static ModuleManager.FilePathRepository;
namespace ModuleManager
{
public class MMPatchRunner
{
private const float TIME_TO_WAIT_FOR_LOGS = 0.05f;
private readonly IBasicLogger kspLogger;
public string Status { get; private set; } = "";
public string Errors { get; private set; } = "";
public MMPatchRunner(IBasicLogger kspLogger)
{
this.kspLogger = kspLogger ?? throw new ArgumentNullException(nameof(kspLogger));
}
public IEnumerator Run()
{
PostPatchLoader.Instance.databaseConfigs = null;
if (!Directory.Exists(logsDirPath)) Directory.CreateDirectory(logsDirPath);
kspLogger.Info("Patching started on a new thread, all output will be directed to " + logPath);
MessageQueue<ILogMessage> mmLogQueue = new MessageQueue<ILogMessage>();
QueueLogRunner logRunner = new QueueLogRunner(mmLogQueue);
ITaskStatus loggingThreadStatus = BackgroundTask.Start(delegate
{
using (StreamLogger streamLogger = new StreamLogger(new FileStream(logPath, FileMode.Create)))
{
logRunner.Run(streamLogger);
streamLogger.Info("Done!");
}
});
// Wait for game database to be initialized for the 2nd time and wait for any plugins to initialize
yield return null;
yield return null;
IBasicLogger mmLogger = new QueueLogger(mmLogQueue);
IEnumerable<ModListGenerator.ModAddedByAssembly> modsAddedByAssemblies = ModListGenerator.GetAdditionalModsFromStaticMethods(mmLogger);
IEnumerable<IProtoUrlConfig> databaseConfigs = null;
MMPatchLoader patchLoader = new MMPatchLoader(modsAddedByAssemblies, mmLogger);
ITaskStatus patchingThreadStatus = BackgroundTask.Start(delegate
{
databaseConfigs = patchLoader.Run();
});
while(true)
{
yield return null;
if (!patchingThreadStatus.IsRunning)
logRunner.RequestStop();
Status = patchLoader.status;
Errors = patchLoader.errors;
if (!patchingThreadStatus.IsRunning && !loggingThreadStatus.IsRunning) break;
}
if (patchingThreadStatus.IsExitedWithError)
{
kspLogger.Exception("The patching thread threw an exception", patchingThreadStatus.Exception);
FatalErrorHandler.HandleFatalError("The patching thread threw an exception");
}
if (loggingThreadStatus.IsExitedWithError)
{
kspLogger.Exception("The logging thread threw an exception", loggingThreadStatus.Exception);
FatalErrorHandler.HandleFatalError("The logging thread threw an exception");
}
if (databaseConfigs == null)
{
kspLogger.Error("The patcher returned a null collection of configs");
FatalErrorHandler.HandleFatalError("The patcher returned a null collection of configs");
yield break;
}
PostPatchLoader.Instance.databaseConfigs = databaseConfigs;
}
}
}