Add example

2025-02-10 16:12:57 +00:00
parent cd26c6ca87
commit 3dee6318da
6 changed files with 319 additions and 11 deletions
@@ -0,0 +1,86 @@
+using Core.Models.Eft.Common.Tables;
+using Core.Models.External;
+using Core.Models.Spt.Config;
+using Core.Models.Utils;
+using Core.Routers;
+using Core.Servers;
+using Core.Services;
+using Core.Utils;
+
+namespace _13AddTraderWithAssortJson
+{
+    public class AddTraderWithAssortJson : IPostDBLoadMod
+    {
+        private readonly ISptLogger<AddTraderWithAssortJson> _logger;
+        private readonly JsonUtil _jsonUtil;
+        private readonly FileUtil _fileUtil;
+        private readonly DatabaseService _databaseService;
+        private readonly ImageRouter _imageRouter;
+        private readonly ConfigServer _configServer;
+        private readonly TraderConfig _traderConfig;
+        private readonly RagfairConfig _ragfairConfig;
+
+        public AddTraderWithAssortJson(
+            ISptLogger<AddTraderWithAssortJson> logger,
+            JsonUtil jsonUtil,
+            FileUtil fileUtil,
+            DatabaseService databaseService,
+            ImageRouter imageRouter,
+            ConfigServer configServer)
+        {
+            _logger = logger;
+            _jsonUtil = jsonUtil;
+            _fileUtil = fileUtil;
+            _databaseService = databaseService;
+            _imageRouter = imageRouter;
+            _configServer = configServer;
+
+            _traderConfig = _configServer.GetConfig<TraderConfig>();
+            _ragfairConfig = _configServer.GetConfig<RagfairConfig>();
+        }
+
+        public void PostDBLoad()
+        {
+            var traderImagePath = "./db/cat.jpg";
+            
+            var baseJson = _fileUtil.ReadFile("./db/base.json");
+            var traderBase = _jsonUtil.Deserialize<TraderBase>(baseJson);
+
+            var assortJson = _fileUtil.ReadFile("./db/assort.json");
+            var assort = _jsonUtil.Deserialize<TraderAssort>(assortJson);
+
+            // Create helper class and use it to register our traders image/icon + set its stock refresh time
+            var addTraderHelper = new AddTraderHelper();
+            _imageRouter.AddRoute(traderBase.Avatar.Replace(".jpg", ""), System.IO.Path.GetFullPath(traderImagePath));
+            addTraderHelper.SetTraderUpdateTime(_traderConfig, traderBase, 3600, 4000);
+
+            // Add trader to flea market
+            _ragfairConfig.Traders[traderBase.Id] = true;
+
+            // Add new trader to the trader dictionary in DatabaseServer - this is where the assort json is loaded
+            /*
+             * The assortJSON includes the following:
+             * Milk available for roubles at LL1
+             * Milk available for item barter at LL1
+             * Helmet w/ soft armour available for roubles at LL1
+             * Helmet w/ soft armour available for item barter at LL1
+             *
+             * It is *REQUIRED* to use MongoIDs for IDs in the assort
+             */
+            addTraderHelper.AddTraderToDb(
+                baseJson,
+                _databaseService.GetTables(),
+                _jsonUtil,
+                assortJson);
+
+            addTraderHelper.AddTraderToLocales(
+                baseJson,
+                _databaseService.GetTables(),
+                traderBase.Name,
+                "Cat",
+                traderBase.Nickname,
+                traderBase.Location,
+                "This is the cat shop. Meow.");
+        }
+    }
+}