@@ -13,7 +13,7 @@ namespace CreamInstaller.Components;
1313
1414internal sealed class CustomTreeView : TreeView
1515{
16- private const string KoaloaderToggleString = "Koaloader " ;
16+ private const string ProxyToggleString = "代理 " ;
1717
1818 private static readonly Color C1 = ColorTranslator . FromHtml ( "#FFFF99" ) ;
1919 private static readonly Color C2 = ColorTranslator . FromHtml ( "#696900" ) ;
@@ -24,10 +24,10 @@ internal sealed class CustomTreeView : TreeView
2424 private static readonly Color C7 = ColorTranslator . FromHtml ( "#006900" ) ;
2525 private static readonly Color C8 = ColorTranslator . FromHtml ( "#69AA69" ) ;
2626
27- private readonly Dictionary < Selection , Rectangle > checkBoxBounds = new ( ) ;
28- private readonly Dictionary < Selection , Rectangle > comboBoxBounds = new ( ) ;
27+ private readonly Dictionary < Selection , Rectangle > checkBoxBounds = [ ] ;
28+ private readonly Dictionary < Selection , Rectangle > comboBoxBounds = [ ] ;
2929
30- private readonly Dictionary < TreeNode , Rectangle > selectionBounds = new ( ) ;
30+ private readonly Dictionary < TreeNode , Rectangle > selectionBounds = [ ] ;
3131 private SolidBrush backBrush ;
3232 private ToolStripDropDown comboBoxDropDown ;
3333 private Font comboBoxFont ;
@@ -36,6 +36,7 @@ internal sealed class CustomTreeView : TreeView
3636 internal CustomTreeView ( )
3737 {
3838 DrawMode = TreeViewDrawMode . OwnerDrawAll ;
39+ Invalidated += OnInvalidated ;
3940 DrawNode += DrawTreeNode ;
4041 Disposed += OnDisposed ;
4142 }
@@ -59,27 +60,38 @@ private void OnDisposed(object sender, EventArgs e)
5960 comboBoxDropDown = null ;
6061 }
6162
63+ private void OnInvalidated ( object sender , EventArgs e )
64+ {
65+ checkBoxBounds . Clear ( ) ;
66+ comboBoxBounds . Clear ( ) ;
67+ selectionBounds . Clear ( ) ;
68+ }
69+
6270 private void DrawTreeNode ( object sender , DrawTreeNodeEventArgs e )
6371 {
6472 e . DrawDefault = true ;
6573 TreeNode node = e . Node ;
6674 if ( node is not { IsVisible : true } )
6775 return ;
76+
6877 bool highlighted = ( e . State & TreeNodeStates . Selected ) == TreeNodeStates . Selected && Focused ;
6978 Graphics graphics = e . Graphics ;
7079 backBrush ??= new ( BackColor ) ;
7180 Font font = node . NodeFont ?? Font ;
7281 Brush brush = highlighted ? SystemBrushes . Highlight : backBrush ;
7382 Rectangle bounds = node . Bounds ;
7483 Rectangle selectionBounds = bounds ;
84+
7585 Form form = FindForm ( ) ;
7686 if ( form is not SelectForm and not SelectDialogForm )
7787 return ;
88+
7889 string id = node . Name ;
7990 Platform platform = ( node . Tag as Platform ? ) . GetValueOrDefault ( Platform . None ) ;
8091 DLCType dlcType = ( node . Tag as DLCType ? ) . GetValueOrDefault ( DLCType . None ) ;
8192 if ( string . IsNullOrWhiteSpace ( id ) || platform is Platform . None && dlcType is DLCType . None )
8293 return ;
94+
8395 Color color = highlighted
8496 ? C1
8597 : Enabled
@@ -91,15 +103,15 @@ private void DrawTreeNode(object sender, DrawTreeNodeEventArgs e)
91103 SelectionDLC dlc = SelectionDLC . FromId ( dlcType , node . Parent ? . Name , id ) ;
92104 text = dlc ? . Selection != null ? dlc . Selection . Platform . ToString ( ) : dlcType . ToString ( ) ;
93105 }
94- else
95- text = platform . ToString ( ) ;
106+ else text = platform . ToString ( ) ;
96107
97108 Size size = TextRenderer . MeasureText ( graphics , text , font ) ;
98109 bounds = bounds with { X = bounds . X + bounds . Width , Width = size . Width } ;
99110 selectionBounds = new ( selectionBounds . Location , selectionBounds . Size + bounds . Size with { Height = 0 } ) ;
100111 graphics . FillRectangle ( brush , bounds ) ;
101112 Point point = new ( bounds . Location . X - 1 , bounds . Location . Y + 1 ) ;
102113 TextRenderer . DrawText ( graphics , text , font , point , color , TextFormatFlags . Default ) ;
114+
103115 if ( platform is not Platform . Paradox )
104116 {
105117 color = highlighted
@@ -130,7 +142,7 @@ private void DrawTreeNode(object sender, DrawTreeNodeEventArgs e)
130142 graphics . FillRectangle ( brush , bounds ) ;
131143 }
132144
133- CheckBoxState checkBoxState = selection . Koaloader
145+ CheckBoxState checkBoxState = selection . UseProxy
134146 ? Enabled ? CheckBoxState . CheckedPressed : CheckBoxState . CheckedDisabled
135147 : Enabled
136148 ? CheckBoxState . UncheckedPressed
@@ -142,7 +154,8 @@ private void DrawTreeNode(object sender, DrawTreeNodeEventArgs e)
142154 graphics . FillRectangle ( backBrush , bounds ) ;
143155 point = new ( bounds . Left , bounds . Top + bounds . Height / 2 - size . Height / 2 - 1 ) ;
144156 CheckBoxRenderer . DrawCheckBox ( graphics , point , checkBoxState ) ;
145- text = KoaloaderToggleString ;
157+
158+ text = ProxyToggleString ;
146159 size = TextRenderer . MeasureText ( graphics , text , font ) ;
147160 int left = 1 ;
148161 bounds = bounds with { X = bounds . X + bounds . Width , Width = size . Width + left } ;
@@ -151,29 +164,48 @@ private void DrawTreeNode(object sender, DrawTreeNodeEventArgs e)
151164 graphics . FillRectangle ( backBrush , bounds ) ;
152165 point = new ( bounds . Location . X - 1 + left , bounds . Location . Y + 1 ) ;
153166 TextRenderer . DrawText ( graphics , text , font , point , Enabled ? C7 : C8 , TextFormatFlags . Default ) ;
167+
154168 this . checkBoxBounds [ selection ] = RectangleToClient ( checkBoxBounds ) ;
155- if ( selection . Koaloader )
169+
170+ if ( selection . UseProxy )
156171 {
157172 comboBoxFont ??= new ( font . FontFamily , 6 , font . Style , font . Unit , font . GdiCharSet ,
158173 font . GdiVerticalFont ) ;
159174 ComboBoxState comboBoxState = Enabled ? ComboBoxState . Normal : ComboBoxState . Disabled ;
160- text = ( selection . KoaloaderProxy ?? Selection . DefaultKoaloaderProxy ) + ".dll" ;
175+ ButtonState buttonState = Enabled ? ButtonState . Normal : ButtonState . Inactive ;
176+
177+ text = ( selection . Proxy ?? Selection . DefaultProxy ) + ".dll" ;
161178 size = TextRenderer . MeasureText ( graphics , text , comboBoxFont ) + new Size ( 6 , 0 ) ;
162179 const int padding = 2 ;
163180 bounds = new ( bounds . X + bounds . Width , bounds . Y + padding / 2 , size . Width , bounds . Height - padding ) ;
164181 selectionBounds = new ( selectionBounds . Location ,
165182 selectionBounds . Size + bounds . Size with { Height = 0 } ) ;
166183 Rectangle comboBoxBounds = bounds ;
167184 graphics . FillRectangle ( backBrush , bounds ) ;
168- ComboBoxRenderer . DrawTextBox ( graphics , bounds , text , comboBoxFont , comboBoxState ) ;
185+ if ( ComboBoxRenderer . IsSupported )
186+ ComboBoxRenderer . DrawTextBox ( graphics , bounds , text , comboBoxFont , comboBoxState ) ;
187+ else
188+ {
189+ graphics . FillRectangle ( SystemBrushes . ControlText , bounds ) ;
190+ ControlPaint . DrawButton ( graphics , bounds , buttonState ) ;
191+ point = new ( bounds . Location . X + 3 + bounds . Width / 2 - size . Width / 2 ,
192+ bounds . Location . Y + bounds . Height / 2 - size . Height / 2 ) ;
193+ TextRenderer . DrawText ( graphics , text , comboBoxFont , point ,
194+ Enabled ? SystemColors . ControlText : SystemColors . GrayText , TextFormatFlags . Default ) ;
195+ }
196+
169197 size = new ( 14 , 0 ) ;
170198 left = - 1 ;
171199 bounds = bounds with { X = bounds . X + bounds . Width + left , Width = size . Width } ;
172200 selectionBounds = new ( selectionBounds . Location ,
173201 selectionBounds . Size + new Size ( bounds . Size . Width + left , 0 ) ) ;
174202 comboBoxBounds = new ( comboBoxBounds . Location ,
175203 comboBoxBounds . Size + new Size ( bounds . Size . Width + left , 0 ) ) ;
176- ComboBoxRenderer . DrawDropDownButton ( graphics , bounds , comboBoxState ) ;
204+ if ( ComboBoxRenderer . IsSupported )
205+ ComboBoxRenderer . DrawDropDownButton ( graphics , bounds , comboBoxState ) ;
206+ else
207+ ControlPaint . DrawComboButton ( graphics , bounds , buttonState ) ;
208+
177209 this . comboBoxBounds [ selection ] = RectangleToClient ( comboBoxBounds ) ;
178210 }
179211 else
@@ -201,20 +233,16 @@ protected override void OnMouseDown(MouseEventArgs e)
201233 break ;
202234 }
203235
204- if ( e . Button is not MouseButtons . Left )
236+ if ( e . Button is not MouseButtons . Left || ! ComboBoxRenderer . IsSupported )
205237 return ;
238+
206239 if ( comboBoxBounds . Count > 0 && selectForm is not null )
207240 foreach ( KeyValuePair < Selection , Rectangle > pair in comboBoxBounds )
208241 if ( ! Selection . All . ContainsKey ( pair . Key ) )
209242 _ = comboBoxBounds . Remove ( pair . Key ) ;
210243 else if ( pair . Value . Contains ( clickPoint ) )
211244 {
212- HashSet < string > proxies = EmbeddedResources
213- . Where ( r => r . StartsWith ( "Koaloader" , StringComparison . Ordinal ) ) . Select ( p =>
214- {
215- p . GetProxyInfoFromIdentifier ( out string proxyName , out _ ) ;
216- return proxyName ;
217- } ) . ToHashSet ( ) ;
245+ IEnumerable < string > proxies = pair . Key . GetAvailableProxies ( ) ;
218246 comboBoxDropDown ??= new ( ) ;
219247 comboBoxDropDown . ShowItemToolTips = false ;
220248 comboBoxDropDown . Items . Clear ( ) ;
@@ -224,7 +252,9 @@ protected override void OnMouseDown(MouseEventArgs e)
224252 foreach ( ( string directory , BinaryType _ ) in pair . Key . ExecutableDirectories )
225253 {
226254 string path = directory + @"\" + proxy + ".dll" ;
227- if ( ! path . FileExists ( ) || path . IsResourceFile ( ResourceIdentifier . Koaloader ) )
255+ if ( ! path . FileExists ( ) || path . IsResourceFile ( ResourceIdentifier . Koaloader ) ||
256+ path . IsResourceFile ( ResourceIdentifier . Steamworks32 ) ||
257+ path . IsResourceFile ( ResourceIdentifier . Steamworks64 ) )
228258 continue ;
229259 canUse = false ;
230260 break ;
@@ -233,8 +263,8 @@ protected override void OnMouseDown(MouseEventArgs e)
233263 if ( canUse )
234264 _ = comboBoxDropDown . Items . Add ( new ToolStripButton ( proxy + ".dll" , null , ( _ , _ ) =>
235265 {
236- pair . Key . KoaloaderProxy = proxy == Selection . DefaultKoaloaderProxy ? null : proxy ;
237- selectForm . OnKoaloaderChanged ( ) ;
266+ pair . Key . Proxy = proxy == Selection . DefaultProxy ? null : proxy ;
267+ selectForm . OnProxyChanged ( ) ;
238268 } ) { Font = comboBoxFont } ) ;
239269 }
240270
@@ -247,8 +277,8 @@ protected override void OnMouseDown(MouseEventArgs e)
247277 _ = checkBoxBounds . Remove ( pair . Key ) ;
248278 else if ( pair . Value . Contains ( clickPoint ) )
249279 {
250- pair . Key . Koaloader = ! pair . Key . Koaloader ;
251- selectForm ? . OnKoaloaderChanged ( ) ;
280+ pair . Key . UseProxy = ! pair . Key . UseProxy ;
281+ selectForm ? . OnProxyChanged ( ) ;
252282 break ;
253283 }
254284 }
0 commit comments