-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathStartup.cs
100 lines (82 loc) · 3.44 KB
/
Startup.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.Generic;
using HomeworkTrackerServer.Storage;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
namespace HomeworkTrackerServer;
public class Startup {
public Startup(IConfiguration config) {
Configuration = config;
// Set storage method
switch (Program.Config["StorageMethod"]) {
default:
const string error = "Invalid StorageMethod value in config, must be MySQL or RAM or File";
Logger.Error(error);
throw new ArgumentException(error);
case "MySQL":
Program.Storage = new MySqlStorage();
break;
case "RAM":
Logger.Info("Warning: StorageMethod is set to RAM, data will be deleted upon restart");
Program.Storage = new RamStorage();
break;
case "File":
Program.Storage = new FileStorage();
break;
}
Logger.Info("Initialising Storage");
try {
Program.Storage.Init();
}
catch (Exception e) {
Logger.Error("Failed to initialize storage: " + e.Message);
Logger.Debug(e.ToString()); // Debug whole error
throw;
}
Program.StorageInitialized = true;
Logger.Info("Initialised Storage");
// Load custom headers
Logger.Info("Loading custom headers");
List<(string, string)> headers = new ();
string headersString = Program.Config["CustomHeaders"];
string[] headersArray = headersString.Split(';');
foreach (string header in headersArray) {
string[] headerSplit = header.Split(':');
if (headerSplit.Length != 2) {
Logger.Error("Invalid custom header: " + header);
continue;
}
headers.Add((headerSplit[0], headerSplit[1]));
Logger.Debug("Added custom header (" + headerSplit[0] + ": " + headerSplit[1] + ")");
}
Program.CustomHeaders = headers.ToArray();
Logger.Info("Loaded " + headers.Count + " custom headers");
Logger.Debug("Finished startup");
}
private IConfiguration Configuration { get; }
// This method gets called by the runtime. Use this method to add services to the container.
public void ConfigureServices(IServiceCollection services) {
Logger.Info("Configuring services");
services.AddControllers();
services.AddControllers().AddNewtonsoftJson();
}
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IWebHostEnvironment env) {
Logger.Info("Configuring app");
if (env.IsDevelopment()) {
app.UseDeveloperExceptionPage();
}
// Only do it if our master says so
if (Program.Config["HttpsRedirection"] == "true") {
app.UseHttpsRedirection();
}
app.UseRouting();
app.UseAuthorization();
app.UseEndpoints(endpoints => {
endpoints.MapControllers();
});
}
}