diff --git a/cf-lib/data/Item.lua b/cf-lib/data/Item.lua index 1cfffcf..5934a72 100644 --- a/cf-lib/data/Item.lua +++ b/cf-lib/data/Item.lua @@ -57,4 +57,11 @@ function Item:itemsPerRocket(count) return self end +--- Sets the default import location for space platforms +--- @param planetName string The name of the planet +function Item:defaultImportLocation(planetName) + self.prototype.default_import_location = planetName + return self +end + return Item \ No newline at end of file diff --git a/cf-lib/data/Recipe.lua b/cf-lib/data/Recipe.lua index 199005c..213aa14 100644 --- a/cf-lib/data/Recipe.lua +++ b/cf-lib/data/Recipe.lua @@ -187,6 +187,17 @@ function Recipe:removeResult(resultName) return self end +--- Adds an additional crafting category +--- @param categoryName string The name of the crafting category +function Recipe:addCategory(categoryName) + local categories = self.prototype.additional_categories or {} + self.prototype.additional_categories = categories + if not table.contains(categories, categoryName) then + table.insert(categories, categoryName) + end + return self +end + --- Adds the recipe to a technology --- @param technology string|table The name of the technology or the technology table function Recipe:unlockedByTechnology(technology)