|
1 | 1 | using System.Linq; |
2 | | -using System.Text.RegularExpressions; |
3 | 2 | using Content.Client.Players.PlayTimeTracking; |
4 | 3 | using Content.Client.UserInterface.Controls; |
| 4 | +using Content.Shared.Roles; |
| 5 | +using Content.Shared.Players.PlayTimeTracking; |
5 | 6 | using Robust.Client.AutoGenerated; |
6 | 7 | using Robust.Client.Graphics; |
| 8 | +using Robust.Client.UserInterface; |
| 9 | +using Robust.Client.UserInterface.Controls; |
7 | 10 | using Robust.Client.UserInterface.XAML; |
8 | | -//starlight |
9 | | -using Robust.Shared.Prototypes; |
10 | 11 |
|
11 | 12 | namespace Content.Client.Info.PlaytimeStats; |
12 | 13 |
|
| 14 | +//starlight start |
13 | 15 | [GenerateTypedNameReferences] |
14 | 16 | public sealed partial class PlaytimeStatsWindow : FancyWindow |
15 | 17 | { |
16 | 18 | [Dependency] private readonly JobRequirementsManager _jobRequirementsManager = default!; |
17 | | - [Dependency] private readonly IPrototypeManager _prototypes = default!; |
18 | | - private ISawmill _sawmill = Logger.GetSawmill("PlaytimeStatsWindow"); |
| 19 | + [Dependency] private readonly ILocalizationManager _loc = default!; |
19 | 20 | private readonly Color _altColor = Color.FromHex("#292B38"); |
20 | 21 | private readonly Color _defaultColor = Color.FromHex("#2F2F3B"); |
21 | 22 | private readonly Color _antagColor = Color.FromHex("#fe7676"); |
22 | 23 | private readonly Color _ghostColor = Color.FromHex("#c996e0"); |
23 | | - private bool _useAltColor; |
24 | 24 |
|
25 | 25 | public PlaytimeStatsWindow() |
26 | 26 | { |
27 | 27 | IoCManager.InjectDependencies(this); |
28 | 28 | RobustXamlLoader.Load(this); |
29 | 29 |
|
30 | | - PopulatePlaytimeHeader(); |
31 | | - PopulatePlaytimeData(); |
| 30 | + BuildTabs(); |
32 | 31 | } |
33 | 32 |
|
34 | | - private void PopulatePlaytimeHeader() |
| 33 | + private void BuildTabs() |
35 | 34 | { |
36 | | - var header = new PlaytimeStatsHeader(); |
37 | | - header.OnHeaderClicked += HeaderClicked; |
38 | | - header.BackgroundColorPlaytimePanel.PanelOverride = new StyleBoxFlat(_altColor); |
39 | | - RolesPlaytimeList.AddChild(header); |
40 | | - } |
| 35 | + ServerTabs.RemoveAllChildren(); |
| 36 | + |
| 37 | + var currentServer = _jobRequirementsManager.GetCurrentServerName(); |
| 38 | + |
| 39 | + // Tab 1: merged playtime |
| 40 | + var mergedRolePlaytimes = _jobRequirementsManager.FetchPlaytimeByRoles().ToList(); |
| 41 | + var mergedAntagPlaytimes = _jobRequirementsManager.FetchPlaytimeByAntags().ToList(); |
| 42 | + var mergedTab = BuildPlaytimeTab( |
| 43 | + _jobRequirementsManager.FetchOverallPlaytime(), |
| 44 | + mergedRolePlaytimes, |
| 45 | + _jobRequirementsManager.FetchPlaytimeByDepartments(), |
| 46 | + mergedAntagPlaytimes, |
| 47 | + _jobRequirementsManager.FetchPlaytimeMiscellaneous(mergedRolePlaytimes, mergedAntagPlaytimes)); |
| 48 | + ServerTabs.AddChild(mergedTab); |
| 49 | + TabContainer.SetTabTitle(mergedTab, _loc.GetString("ui-playtime-tab-merged")); |
| 50 | + |
| 51 | + // Tab 2: current server (_originalRoles) |
| 52 | + var originalRoles = _jobRequirementsManager.GetOriginalRoles(); |
| 53 | + if (originalRoles.Count > 0) |
| 54 | + { |
| 55 | + var currentTab = BuildServerTab(originalRoles); |
| 56 | + ServerTabs.AddChild(currentTab); |
| 57 | + TabContainer.SetTabTitle(currentTab, _loc.GetString("ui-playtime-tab-current") + " ★"); |
| 58 | + } |
41 | 59 |
|
42 | | - private void HeaderClicked(PlaytimeStatsHeader.Header header, PlaytimeStatsHeader.SortDirection direction) |
43 | | - { |
44 | | - switch (header) |
| 60 | + // Remaining tabs: NullLink servers grouped by project prefix |
| 61 | + var rolesPerServer = _jobRequirementsManager.GetRolesPerServer(); |
| 62 | + var groups = rolesPerServer |
| 63 | + .GroupBy(kv => |
| 64 | + { |
| 65 | + var dotIdx = kv.Key.IndexOf('.'); |
| 66 | + return dotIdx >= 0 ? kv.Key[..dotIdx] : kv.Key; |
| 67 | + }) |
| 68 | + .OrderBy(g => g.Key); |
| 69 | + |
| 70 | + foreach (var group in groups) |
45 | 71 | { |
46 | | - case PlaytimeStatsHeader.Header.Role: |
47 | | - SortByRole(direction); |
48 | | - break; |
49 | | - case PlaytimeStatsHeader.Header.Playtime: |
50 | | - SortByPlaytime(direction); |
51 | | - break; |
| 72 | + var servers = group.ToList(); |
| 73 | + |
| 74 | + if (servers.Count == 1) |
| 75 | + { |
| 76 | + var (serverName, serverRoles) = servers[0]; |
| 77 | + var tab = BuildServerTab(serverRoles); |
| 78 | + ServerTabs.AddChild(tab); |
| 79 | + TabContainer.SetTabTitle(tab, serverName); |
| 80 | + } |
| 81 | + else |
| 82 | + { |
| 83 | + var innerTabs = new TabContainer |
| 84 | + { |
| 85 | + HorizontalExpand = true, |
| 86 | + VerticalExpand = true, |
| 87 | + }; |
| 88 | + |
| 89 | + foreach (var (serverName, serverRoles) in servers) |
| 90 | + { |
| 91 | + var tab = BuildServerTab(serverRoles); |
| 92 | + innerTabs.AddChild(tab); |
| 93 | + |
| 94 | + var dotIdx = serverName.IndexOf('.'); |
| 95 | + var subTitle = dotIdx >= 0 ? serverName[(dotIdx + 1)..] : serverName; |
| 96 | + TabContainer.SetTabTitle(tab, subTitle); |
| 97 | + } |
| 98 | + |
| 99 | + ServerTabs.AddChild(innerTabs); |
| 100 | + TabContainer.SetTabTitle(innerTabs, group.Key); |
| 101 | + } |
52 | 102 | } |
53 | 103 | } |
54 | 104 |
|
55 | | - private void SortByRole(PlaytimeStatsHeader.SortDirection direction) |
| 105 | + private BoxContainer BuildServerTab(Dictionary<string, TimeSpan> serverRoles) |
56 | 106 | { |
57 | | - var header = RolesPlaytimeList.GetChild(0) as PlaytimeStatsHeader; |
| 107 | + var rolePlaytimes = _jobRequirementsManager.FetchPlaytimeByRoles(serverRoles).ToList(); |
| 108 | + var departmentPlaytimes = _jobRequirementsManager.FetchPlaytimeByDepartments(serverRoles); |
| 109 | + var antagPlaytimes = _jobRequirementsManager.FetchPlaytimeByAntags(serverRoles).ToList(); |
| 110 | + var overallPlaytime = _jobRequirementsManager.FetchOverallPlaytime(serverRoles); |
| 111 | + var miscPlaytimes = _jobRequirementsManager.FetchPlaytimeMiscellaneous(serverRoles, rolePlaytimes, antagPlaytimes); |
58 | 112 |
|
59 | | - var entries = RolesPlaytimeList.Children.OfType<PlaytimeStatsEntry>().ToList(); |
60 | | - |
61 | | - RolesPlaytimeList.RemoveAllChildren(); |
62 | | - |
63 | | - if (header != null) |
64 | | - RolesPlaytimeList.AddChild(header); |
| 113 | + return BuildPlaytimeTab(overallPlaytime, rolePlaytimes, departmentPlaytimes, antagPlaytimes, miscPlaytimes); |
| 114 | + } |
65 | 115 |
|
66 | | - var sortedEntries = (direction == PlaytimeStatsHeader.SortDirection.Ascending) |
67 | | - ? entries.OrderBy(entry => entry.RoleText).ToList() |
68 | | - : entries.OrderByDescending(entry => entry.RoleText).ToList(); |
| 116 | + private BoxContainer BuildPlaytimeTab( |
| 117 | + TimeSpan overallPlaytime, |
| 118 | + List<KeyValuePair<JobPrototype, TimeSpan>> rolePlaytimes, |
| 119 | + IEnumerable<KeyValuePair<DepartmentPrototype, TimeSpan>> departmentPlaytimes, |
| 120 | + IEnumerable<KeyValuePair<AntagPrototype, TimeSpan>> antagPlaytimes, |
| 121 | + IEnumerable<KeyValuePair<PlayTimeTrackerPrototype, TimeSpan>> miscPlaytimes) |
| 122 | + { |
| 123 | + var container = new BoxContainer |
| 124 | + { |
| 125 | + Orientation = BoxContainer.LayoutOrientation.Vertical, |
| 126 | + HorizontalExpand = true, |
| 127 | + VerticalExpand = true, |
| 128 | + }; |
69 | 129 |
|
70 | | - _useAltColor = false; |
| 130 | + // Overall label |
| 131 | + var overallLabel = new Label |
| 132 | + { |
| 133 | + HorizontalExpand = true, |
| 134 | + Text = _loc.GetString("ui-playtime-overall", ("time", overallPlaytime)), |
| 135 | + }; |
| 136 | + container.AddChild(overallLabel); |
| 137 | + container.AddChild(new Control { MinSize = new System.Numerics.Vector2(0, 5) }); |
| 138 | + |
| 139 | + // Scrollable list |
| 140 | + var scrollContainer = new ScrollContainer |
| 141 | + { |
| 142 | + HorizontalExpand = true, |
| 143 | + VerticalExpand = true, |
| 144 | + }; |
71 | 145 |
|
72 | | - foreach (var entry in sortedEntries) |
| 146 | + var rolesListBox = new BoxContainer |
73 | 147 | { |
74 | | - var styleBox = new StyleBoxFlat { BackgroundColor = _useAltColor ? _altColor : _defaultColor }; |
75 | | - entry.UpdateShading(styleBox); |
76 | | - RolesPlaytimeList.AddChild(entry); |
77 | | - _useAltColor ^= true; |
78 | | - } |
79 | | - } |
| 148 | + Orientation = BoxContainer.LayoutOrientation.Vertical, |
| 149 | + HorizontalExpand = true, |
| 150 | + }; |
80 | 151 |
|
81 | | - private void SortByPlaytime(PlaytimeStatsHeader.SortDirection direction) |
82 | | - { |
83 | | - var header = RolesPlaytimeList.GetChild(0) as PlaytimeStatsHeader; |
| 152 | + // Header |
| 153 | + var header = new PlaytimeStatsHeader(); |
| 154 | + header.OnHeaderClicked += (h, dir) => SortList(rolesListBox, h, dir); |
| 155 | + header.BackgroundColorPlaytimePanel.PanelOverride = new StyleBoxFlat(_altColor); |
| 156 | + rolesListBox.AddChild(header); |
84 | 157 |
|
85 | | - var entries = RolesPlaytimeList.Children.OfType<PlaytimeStatsEntry>().ToList(); |
| 158 | + var useAltColor = false; |
86 | 159 |
|
87 | | - RolesPlaytimeList.RemoveAllChildren(); |
| 160 | + // Roles |
| 161 | + foreach (var rolePlaytime in rolePlaytimes) |
| 162 | + AddEntry(rolesListBox, _loc.GetString(rolePlaytime.Key.Name), rolePlaytime.Value, null, ref useAltColor); |
88 | 163 |
|
89 | | - if (header != null) |
90 | | - RolesPlaytimeList.AddChild(header); |
| 164 | + // Departments |
| 165 | + foreach (var departmentPlaytime in departmentPlaytimes) |
| 166 | + AddEntry(rolesListBox, _loc.GetString(departmentPlaytime.Key.Name), departmentPlaytime.Value, departmentPlaytime.Key.Color, ref useAltColor); |
91 | 167 |
|
92 | | - var sortedEntries = (direction == PlaytimeStatsHeader.SortDirection.Ascending) |
93 | | - ? entries.OrderBy(entry => entry.Playtime).ToList() |
94 | | - : entries.OrderByDescending(entry => entry.Playtime).ToList(); |
| 168 | + // Antags |
| 169 | + foreach (var antagPlaytime in antagPlaytimes) |
| 170 | + AddEntry(rolesListBox, _loc.GetString(antagPlaytime.Key.Name), antagPlaytime.Value, _antagColor, ref useAltColor); |
95 | 171 |
|
96 | | - _useAltColor = false; |
| 172 | + // Misc |
| 173 | + foreach (var miscPlaytime in miscPlaytimes) |
| 174 | + AddEntry(rolesListBox, _loc.GetString(miscPlaytime.Key.Name), miscPlaytime.Value, _ghostColor, ref useAltColor); |
97 | 175 |
|
98 | | - foreach (var entry in sortedEntries) |
99 | | - { |
100 | | - var styleBox = new StyleBoxFlat { BackgroundColor = _useAltColor ? _altColor : _defaultColor }; |
101 | | - entry.UpdateShading(styleBox); |
102 | | - RolesPlaytimeList.AddChild(entry); |
103 | | - _useAltColor ^= true; |
104 | | - } |
| 176 | + scrollContainer.AddChild(rolesListBox); |
| 177 | + container.AddChild(scrollContainer); |
| 178 | + return container; |
105 | 179 | } |
106 | 180 |
|
107 | | - |
108 | | - private void PopulatePlaytimeData() |
| 181 | + private void AddEntry(BoxContainer list, string role, TimeSpan playtime, Color? textColor, ref bool useAltColor) |
109 | 182 | { |
110 | | - var overallPlaytime = _jobRequirementsManager.FetchOverallPlaytime(); |
111 | | - |
112 | | - OverallPlaytimeLabel.Text = Loc.GetString("ui-playtime-overall", ("time", overallPlaytime)); |
113 | | - |
114 | | - // Starlight BEGIN |
115 | | - var rolePlaytimes = _jobRequirementsManager.FetchPlaytimeByRoles().ToList(); |
116 | | - var departmentPlaytimes = _jobRequirementsManager.FetchPlaytimeByDepartments(); |
117 | | - var antagPlaytimes = _jobRequirementsManager.FetchPlaytimeByAntags(); |
118 | | - var miscellaneousPlaytimes = _jobRequirementsManager.FetchPlaytimeMiscellaneous(rolePlaytimes, antagPlaytimes); |
119 | | - // Starlight END |
| 183 | + var entry = new PlaytimeStatsEntry(role, playtime, textColor, |
| 184 | + new StyleBoxFlat(useAltColor ? _altColor : _defaultColor)); |
| 185 | + list.AddChild(entry); |
| 186 | + useAltColor ^= true; |
| 187 | + } |
120 | 188 |
|
121 | | - RolesPlaytimeList.RemoveAllChildren(); |
122 | | - PopulatePlaytimeHeader(); |
| 189 | + private void SortList(BoxContainer list, PlaytimeStatsHeader.Header header, PlaytimeStatsHeader.SortDirection direction) |
| 190 | + { |
| 191 | + var headerControl = list.Children.OfType<PlaytimeStatsHeader>().FirstOrDefault(); |
| 192 | + var entries = list.Children.OfType<PlaytimeStatsEntry>().ToList(); |
123 | 193 |
|
124 | | - foreach (var rolePlaytime in rolePlaytimes) |
125 | | - { |
126 | | - var role = rolePlaytime.Key; |
127 | | - var playtime = rolePlaytime.Value; |
128 | | - AddRolePlaytimeEntryToTable(Loc.GetString(role.Name), playtime.ToString()); //starlight edit |
129 | | - } |
| 194 | + list.RemoveAllChildren(); |
130 | 195 |
|
131 | | - //starlight |
132 | | - foreach (var departmentPlaytime in departmentPlaytimes) |
133 | | - { |
134 | | - var department = departmentPlaytime.Key; |
135 | | - var playtime = departmentPlaytime.Value; |
136 | | - AddRolePlaytimeEntryToTable(Loc.GetString(department.Name), playtime.ToString(), textColor: department.Color); //starlight edit |
137 | | - } |
138 | | - foreach (var antagPlaytime in antagPlaytimes) |
139 | | - { |
140 | | - AddRolePlaytimeEntryToTable(Loc.GetString(antagPlaytime.Key.Name), antagPlaytime.Value.ToString(), textColor: _antagColor); |
141 | | - } |
142 | | - foreach (var miscellaneousPlaytime in miscellaneousPlaytimes) |
143 | | - { |
144 | | - var role = miscellaneousPlaytime.Key; |
145 | | - var playtime = miscellaneousPlaytime.Value; |
146 | | - AddRolePlaytimeEntryToTable(Loc.GetString(role.Name), playtime.ToString(), textColor: _ghostColor); |
147 | | - } |
148 | | - //starlight end |
149 | | - } |
| 196 | + if (headerControl != null) |
| 197 | + list.AddChild(headerControl); |
150 | 198 |
|
151 | | - private void AddRolePlaytimeEntryToTable(string role, string playtimeString, Color? textColor = null) //starlight edit |
152 | | - { |
153 | | - if (TimeSpan.TryParse(playtimeString, out var playtime)) |
| 199 | + var sortedEntries = header switch |
154 | 200 | { |
155 | | - var entry = new PlaytimeStatsEntry(role, playtime, textColor, //starlight edit |
156 | | - new StyleBoxFlat(_useAltColor ? _altColor : _defaultColor)); |
157 | | - RolesPlaytimeList.AddChild(entry); |
158 | | - _useAltColor ^= true; |
159 | | - } |
160 | | - else |
| 201 | + PlaytimeStatsHeader.Header.Role => direction == PlaytimeStatsHeader.SortDirection.Ascending |
| 202 | + ? [.. entries.OrderBy(e => e.RoleText)] |
| 203 | + : [.. entries.OrderByDescending(e => e.RoleText)], |
| 204 | + PlaytimeStatsHeader.Header.Playtime => direction == PlaytimeStatsHeader.SortDirection.Ascending |
| 205 | + ? [.. entries.OrderBy(e => e.Playtime)] |
| 206 | + : [.. entries.OrderByDescending(e => e.Playtime)], |
| 207 | + _ => entries, |
| 208 | + }; |
| 209 | + |
| 210 | + var useAlt = false; |
| 211 | + foreach (var entry in sortedEntries) |
161 | 212 | { |
162 | | - _sawmill.Error($"The provided playtime string '{playtimeString}' is not in the correct format."); |
| 213 | + entry.UpdateShading(new StyleBoxFlat(useAlt ? _altColor : _defaultColor)); |
| 214 | + list.AddChild(entry); |
| 215 | + useAlt ^= true; |
163 | 216 | } |
164 | 217 | } |
165 | 218 | } |
| 219 | +//starlight end |
0 commit comments