Closes #2420

ArchiSteamFarm/Core/ASF.cs

@@ -120,8 +120,9 @@ public static class ASF {
 		await InitRateLimiters().ConfigureAwait(false);
 
 		StringComparer botsComparer = await PluginsCore.GetBotsComparer().ConfigureAwait(false);
+		IMachineInfoProvider? customMachineInfoProvider = await PluginsCore.GetCustomMachineInfoProvider().ConfigureAwait(false);
 
-		InitBotsComparer(botsComparer);
+		Bot.Init(botsComparer, customMachineInfoProvider);
 
 		if (!Program.Service && !GlobalConfig.Headless && !Console.IsInputRedirected) {
 			Logging.StartInteractiveConsole();
@@ -364,18 +365,6 @@ public static class ASF {
 		return LastWriteEvents.TryGetValue(filePath, out object? savedWriteEvent) && (currentWriteEvent == savedWriteEvent) && LastWriteEvents.TryRemove(filePath, out _);
 	}
 
-	private static void InitBotsComparer(StringComparer botsComparer) {
-		if (botsComparer == null) {
-			throw new ArgumentNullException(nameof(botsComparer));
-		}
-
-		if (Bot.Bots != null) {
-			return;
-		}
-
-		Bot.Init(botsComparer);
-	}
-
 	private static void InitConfigWatchEvents() {
 		if ((FileSystemWatcher != null) || (LastWriteEvents != null)) {
 			return;

ArchiSteamFarm/Plugins/Interfaces/ICustomMachineInfoProvider.cs

@@ -0,0 +1,35 @@
+//     _                _      _  ____   _                           _____
+//    / \    _ __  ___ | |__  (_)/ ___| | |_  ___   __ _  _ __ ___  |  ___|__ _  _ __  _ __ ___
+//   / _ \  | '__|/ __|| '_ \ | |\___ \ | __|/ _ \ / _` || '_ ` _ \ | |_  / _` || '__|| '_ ` _ \
+//  / ___ \ | |  | (__ | | | || | ___) || |_|  __/| (_| || | | | | ||  _|| (_| || |   | | | | | |
+// /_/   \_\|_|   \___||_| |_||_||____/  \__|\___| \__,_||_| |_| |_||_|   \__,_||_|   |_| |_| |_|
+// |
+// Copyright 2015-2021 Łukasz "JustArchi" Domeradzki
+// Contact: JustArchi@JustArchi.net
+// |
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// |
+// http://www.apache.org/licenses/LICENSE-2.0
+// |
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+using JetBrains.Annotations;
+using SteamKit2;
+
+namespace ArchiSteamFarm.Plugins.Interfaces;
+
+[PublicAPI]
+public interface ICustomMachineInfoProvider : IPlugin {
+	/// <summary>
+	///     ASF will use this property as the <see cref="IMachineInfoProvider" /> for the bots.
+	///     Unless you know what you're doing, you should not implement this property yourself and let ASF decide.
+	/// </summary>
+	/// <returns><see cref="IMachineInfoProvider" /> that will be used for the bots.</returns>
+	IMachineInfoProvider MachineInfoProvider { get; }
+}

ArchiSteamFarm/Plugins/PluginsCore.cs

@@ -117,6 +117,24 @@ internal static class PluginsCore {
 		return responses.FirstOrDefault(static response => response != null) ?? new CrossProcessFileBasedSemaphore(resourceName);
 	}
 
+	internal static async Task<IMachineInfoProvider?> GetCustomMachineInfoProvider() {
+		if (ActivePlugins == null) {
+			return null;
+		}
+
+		IList<IMachineInfoProvider> results;
+
+		try {
+			results = await Utilities.InParallel(ActivePlugins.OfType<ICustomMachineInfoProvider>().Select(static plugin => Task.Run(() => plugin.MachineInfoProvider))).ConfigureAwait(false);
+		} catch (Exception e) {
+			ASF.ArchiLogger.LogGenericException(e);
+
+			return null;
+		}
+
+		return results.FirstOrDefault();
+	}
+
 	internal static bool InitPlugins() {
 		if (ActivePlugins != null) {
 			return false;

ArchiSteamFarm/Steam/Bot.cs

@@ -79,6 +79,8 @@ public sealed class Bot : IAsyncDisposable {
 
 	private static readonly SemaphoreSlim BotsSemaphore = new(1, 1);
 
+	private static IMachineInfoProvider? CustomMachineInfoProvider;
+
 	[JsonIgnore]
 	[PublicAPI]
 	public Actions Actions { get; }
@@ -279,7 +281,18 @@ public sealed class Bot : IAsyncDisposable {
 
 		ArchiWebHandler = new ArchiWebHandler(this);
 
-		SteamConfiguration = SteamConfiguration.Create(builder => builder.WithProtocolTypes(ASF.GlobalConfig.SteamProtocols).WithCellID(ASF.GlobalDatabase.CellID).WithServerListProvider(ASF.GlobalDatabase.ServerListProvider).WithHttpClientFactory(ArchiWebHandler.GenerateDisposableHttpClient));
+		SteamConfiguration = SteamConfiguration.Create(
+			builder => {
+				builder.WithCellID(ASF.GlobalDatabase.CellID);
+				builder.WithHttpClientFactory(ArchiWebHandler.GenerateDisposableHttpClient);
+				builder.WithProtocolTypes(ASF.GlobalConfig.SteamProtocols);
+				builder.WithServerListProvider(ASF.GlobalDatabase.ServerListProvider);
+
+				if (CustomMachineInfoProvider != null) {
+					builder.WithMachineInfoProvider(CustomMachineInfoProvider);
+				}
+			}
+		);
 
 		// Initialize
 		SteamClient = new SteamClient(SteamConfiguration, botName);
@@ -1333,12 +1346,14 @@ public sealed class Bot : IAsyncDisposable {
 		}
 	}
 
-	internal static void Init(StringComparer botsComparer) {
+	internal static void Init(StringComparer botsComparer, IMachineInfoProvider? customMachineInfoProvider = null) {
 		if (Bots != null) {
 			throw new InvalidOperationException(nameof(Bots));
 		}
 
 		BotsComparer = botsComparer ?? throw new ArgumentNullException(nameof(botsComparer));
+
+		CustomMachineInfoProvider = customMachineInfoProvider;
 		Bots = new ConcurrentDictionary<string, Bot>(botsComparer);
 	}