mirror of
https://github.com/moeru-ai/airi-factorio.git
synced 2025-09-04 17:02:04 +00:00
initial commit
This commit is contained in:
LemonNeko
8
.editorconfig
Normal file
8
.editorconfig
Normal file
@@ -0,0 +1,8 @@
|
||||
[*]
|
||||
indent_style = space
|
||||
indent_size = 2
|
||||
trim_trailing_whitespace = true
|
||||
insert_final_newline = true
|
||||
end_of_line = lf
|
||||
charset = utf-8
|
||||
|
||||
5
.gitignore
vendored
Normal file
5
.gitignore
vendored
Normal file
@@ -0,0 +1,5 @@
|
||||
node_modules
|
||||
.vscode
|
||||
*.lua
|
||||
|
||||
.DS_Store
|
||||
952
control.ts
Normal file
952
control.ts
Normal file
@@ -0,0 +1,952 @@
|
||||
import type { MapPosition, MapPositionStruct } from 'factorio:prototype'
|
||||
import type {
|
||||
BoundingBoxArray,
|
||||
CollisionMask,
|
||||
EquipmentPosition,
|
||||
LuaEntity,
|
||||
LuaPlayer,
|
||||
OnPlayerMinedEntityEvent,
|
||||
OnScriptPathRequestFinishedEvent,
|
||||
PathfinderWaypoint,
|
||||
SurfaceCreateEntity,
|
||||
} from 'factorio:runtime'
|
||||
|
||||
enum TaskStates {
|
||||
IDLE,
|
||||
WALKING_TO_ENTITY,
|
||||
MINING,
|
||||
PLACING,
|
||||
PLACING_IN_CHEST,
|
||||
PICKING_UP,
|
||||
CRAFTING,
|
||||
RESEARCHING,
|
||||
WALKING_DIRECT,
|
||||
AUTO_INSERTING,
|
||||
ATTACKING,
|
||||
}
|
||||
|
||||
interface PlayerParametersWalkToEntity {
|
||||
entity_name: string
|
||||
search_radius: number
|
||||
path: PathfinderWaypoint[] | null
|
||||
path_drawn: boolean
|
||||
path_index: number
|
||||
calculating_path: boolean
|
||||
should_mine: boolean
|
||||
target_position: MapPositionStruct | null
|
||||
}
|
||||
|
||||
interface PlayerParametersWalkingDirect {
|
||||
target_position: MapPositionStruct | null
|
||||
}
|
||||
|
||||
interface PlayerParametersMineEntity {
|
||||
entity_name: string
|
||||
}
|
||||
|
||||
interface PlayerParametersPlaceEntity {
|
||||
entity_name: string
|
||||
position?: MapPositionStruct
|
||||
}
|
||||
|
||||
interface PlayerParametersAutoInsertNearby {
|
||||
item_name: string
|
||||
entity_name: string
|
||||
max_count: number
|
||||
}
|
||||
|
||||
interface PlayerParametersPickupItem {
|
||||
item_name: string
|
||||
count: number
|
||||
container_name: string
|
||||
search_radius: number
|
||||
}
|
||||
|
||||
interface PlayerParametersCraftItem {
|
||||
item_name: string
|
||||
count: number
|
||||
crafted: number
|
||||
}
|
||||
|
||||
interface PlayerParametersAttackNearestEnemy {
|
||||
search_radius: number
|
||||
target: LuaEntity | null
|
||||
}
|
||||
|
||||
interface PlayerParametersResearchTechnology {
|
||||
technology_name: string
|
||||
}
|
||||
|
||||
let setup_complete = false
|
||||
|
||||
interface PlayerState {
|
||||
task_state: TaskStates
|
||||
parameters_walk_to_entity?: PlayerParametersWalkToEntity
|
||||
parameters_walking_direct?: PlayerParametersWalkingDirect
|
||||
parameters_mine_entity?: PlayerParametersMineEntity
|
||||
parameters_place_entity?: PlayerParametersPlaceEntity
|
||||
parameters_auto_insert_nearby?: PlayerParametersAutoInsertNearby
|
||||
parameters_pickup_item?: PlayerParametersPickupItem
|
||||
parameters_craft_item?: PlayerParametersCraftItem
|
||||
parameters_attack_nearest_enemy?: PlayerParametersAttackNearestEnemy
|
||||
parameters_research_technology?: PlayerParametersResearchTechnology
|
||||
}
|
||||
|
||||
const player_state: PlayerState = {
|
||||
task_state: TaskStates.IDLE,
|
||||
}
|
||||
|
||||
function log_player_info(player_id: number) {
|
||||
const player = game.connected_players[player_id]
|
||||
const log_data: {
|
||||
name: string
|
||||
position: MapPosition
|
||||
force: string
|
||||
inventory: { name: string, count: number }[]
|
||||
equipment: { name: string, position: EquipmentPosition }[]
|
||||
nearby_entities: { name: string, position: MapPosition }[]
|
||||
map_info: {
|
||||
surface_name: string
|
||||
daytime: number
|
||||
wind_speed: number
|
||||
wind_orientation: number
|
||||
}
|
||||
research: {
|
||||
current_research: string
|
||||
research_progress: number
|
||||
}
|
||||
technologies: string[]
|
||||
crafting_queue: { name: string, count: number }[]
|
||||
character_stats: {
|
||||
health: number | undefined
|
||||
health_max: number
|
||||
mining_progress: number | undefined
|
||||
mining_target: LuaEntity | undefined
|
||||
vehicle: string
|
||||
}
|
||||
} = {
|
||||
name: player.name,
|
||||
position: player.position,
|
||||
force: player.force.name,
|
||||
inventory: [],
|
||||
equipment: [],
|
||||
nearby_entities: [],
|
||||
map_info: {
|
||||
surface_name: player.surface.name,
|
||||
daytime: player.surface.daytime,
|
||||
wind_speed: player.surface.wind_speed,
|
||||
wind_orientation: player.surface.wind_orientation,
|
||||
},
|
||||
research: {
|
||||
current_research: player.force.current_research?.name ?? 'None',
|
||||
research_progress: player.force.research_progress,
|
||||
},
|
||||
technologies: [],
|
||||
crafting_queue: [],
|
||||
character_stats: {
|
||||
health: undefined,
|
||||
health_max: 0,
|
||||
mining_progress: undefined,
|
||||
mining_target: undefined,
|
||||
vehicle: 'None',
|
||||
},
|
||||
}
|
||||
|
||||
const main_inventory = player.get_main_inventory()
|
||||
if (main_inventory) {
|
||||
main_inventory.get_contents().forEach(({ name, count }) => {
|
||||
log_data.inventory.push({ name, count })
|
||||
})
|
||||
}
|
||||
|
||||
if (player.character?.grid) {
|
||||
player.character.grid.equipment.forEach(({ name, position }) => {
|
||||
log_data.equipment.push({ name, position })
|
||||
})
|
||||
}
|
||||
|
||||
const nearby_entities = player.surface.find_entities_filtered({
|
||||
position: player.position,
|
||||
radius: 20,
|
||||
})
|
||||
nearby_entities.forEach(({ name, position }) => {
|
||||
log_data.nearby_entities.push({ name, position })
|
||||
})
|
||||
|
||||
for (const [name, tech] of pairs(player.force.technologies)) {
|
||||
if (tech.researched) {
|
||||
log_data.technologies.push(name)
|
||||
}
|
||||
}
|
||||
|
||||
for (let i = 1; i < player.crafting_queue_size; i++) {
|
||||
const item = player.crafting_queue?.[i]
|
||||
if (item) {
|
||||
log_data.crafting_queue.push({ name: item.recipe, count: item.count })
|
||||
}
|
||||
}
|
||||
|
||||
if (player.character) {
|
||||
log_data.character_stats = {
|
||||
health: player.character.health,
|
||||
health_max: player.character.max_health,
|
||||
mining_progress: player.character.mining_progress,
|
||||
mining_target: player.character.mining_target,
|
||||
vehicle: player.vehicle?.name ?? 'None',
|
||||
}
|
||||
}
|
||||
|
||||
log(`[AUTORIO] Player ${player.name} info: ${serpent.block(log_data)}`)
|
||||
}
|
||||
|
||||
remote.add_interface('autorio_tasks', {
|
||||
walk_to_entity: (entity_name: string, search_radius: number) => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log(`[AUTORIO] Cannot start walk_to_entity task: Player is not idle`)
|
||||
return false
|
||||
}
|
||||
|
||||
log(`[AUTORIO] New walk_to_entity task: ${entity_name}, radius: ${search_radius}`)
|
||||
|
||||
player_state.task_state = TaskStates.WALKING_TO_ENTITY
|
||||
player_state.parameters_walk_to_entity = {
|
||||
entity_name,
|
||||
search_radius,
|
||||
path: null,
|
||||
path_drawn: false,
|
||||
path_index: 1,
|
||||
calculating_path: false,
|
||||
should_mine: false,
|
||||
target_position: null,
|
||||
}
|
||||
|
||||
return true
|
||||
},
|
||||
|
||||
mine_entity: (entity_name: string) => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start mine_entity task: Player is not idle')
|
||||
return false
|
||||
}
|
||||
|
||||
log(`[AUTORIO] New mine_entity task: ${entity_name}`)
|
||||
player_state.task_state = TaskStates.MINING
|
||||
player_state.parameters_mine_entity = {
|
||||
entity_name,
|
||||
}
|
||||
|
||||
return true
|
||||
},
|
||||
place_entity: (entity_name: string) => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start place_entity task: Player is not idle')
|
||||
return false
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.PLACING
|
||||
player_state.parameters_place_entity = {
|
||||
entity_name,
|
||||
position: undefined,
|
||||
}
|
||||
|
||||
log(`[AUTORIO] New place_entity task: ${entity_name}`)
|
||||
return true
|
||||
},
|
||||
auto_insert_nearby: (item_name: string, entity_name: string, max_count: number): [boolean, string] => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start auto_insert_nearby task: Player is not idle')
|
||||
return [false, 'Task already in progress']
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.AUTO_INSERTING
|
||||
player_state.parameters_auto_insert_nearby = {
|
||||
item_name,
|
||||
entity_name,
|
||||
max_count: max_count || math.huge,
|
||||
}
|
||||
|
||||
log(`[AUTORIO] New auto_insert_nearby task for ${item_name} into ${entity_name}`)
|
||||
return [true, 'Task started']
|
||||
},
|
||||
pick_up_item: (item_name: string, container_name: string, count: number): [boolean, string] => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start pick_up_item task: Player is not idle')
|
||||
return [false, 'Task already in progress']
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.PICKING_UP
|
||||
player_state.parameters_pickup_item = {
|
||||
item_name,
|
||||
count: count || 1,
|
||||
container_name,
|
||||
search_radius: 8,
|
||||
}
|
||||
log(`[AUTORIO] New pick_up_item task: ${item_name} x${count} from ${container_name}`)
|
||||
return [true, 'Task started']
|
||||
},
|
||||
craft_item: (item_name: string, count: number = 1): [boolean, string] => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start craft_item task: Player is not idle')
|
||||
return [false, 'Task already in progress']
|
||||
}
|
||||
const player = game.connected_players[0]
|
||||
if (!player.force.recipes[item_name]) {
|
||||
log('[AUTORIO] Cannot start craft_item task: Recipe not available')
|
||||
return [false, 'Recipe not available']
|
||||
}
|
||||
if (!player.force.recipes[item_name].enabled) {
|
||||
log('[AUTORIO] Cannot start craft_item task: Recipe not unlocked')
|
||||
return [false, 'Recipe not unlocked']
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.CRAFTING
|
||||
player_state.parameters_craft_item = {
|
||||
item_name,
|
||||
count,
|
||||
crafted: 0,
|
||||
}
|
||||
log(`[AUTORIO] New craft_item task: ${item_name} x${count}`)
|
||||
return [true, 'Task started']
|
||||
},
|
||||
attack_nearest_enemy: (search_radius: number = 50): [boolean, string] => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start attack_nearest_enemy task: Player is not idle')
|
||||
return [false, 'Task already in progress']
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.ATTACKING
|
||||
player_state.parameters_attack_nearest_enemy = {
|
||||
search_radius,
|
||||
target: null,
|
||||
}
|
||||
log(`[AUTORIO] New attack nearest enemy task, search radius: ${search_radius}`)
|
||||
return [true, 'Task started']
|
||||
},
|
||||
research_technology: (technology_name: string): [boolean, string] => {
|
||||
if (player_state.task_state !== TaskStates.IDLE) {
|
||||
log('[AUTORIO] Cannot start research_technology task: Player is not idle')
|
||||
return [false, 'Task already in progress']
|
||||
}
|
||||
const player = game.connected_players[0]
|
||||
const force = player.force
|
||||
const tech = force.technologies[technology_name]
|
||||
|
||||
if (!tech) {
|
||||
log('[AUTORIO] Cannot start research_technology task: Technology not found')
|
||||
return [false, 'Technology not found']
|
||||
}
|
||||
|
||||
if (tech.researched) {
|
||||
log('[AUTORIO] Cannot start research_technology task: Technology already researched')
|
||||
return [false, 'Technology already researched']
|
||||
}
|
||||
|
||||
if (!tech.enabled) {
|
||||
log('[AUTORIO] Cannot start research_technology task: Technology not available for research')
|
||||
return [false, 'Technology not available for research']
|
||||
}
|
||||
|
||||
const research_added = force.add_research(tech)
|
||||
if (research_added) {
|
||||
log(`[AUTORIO] New research_technology task: ${technology_name}`)
|
||||
return [true, 'Research started']
|
||||
}
|
||||
log('[AUTORIO] Could not start new research.')
|
||||
return [true, 'Cannot start new research.']
|
||||
},
|
||||
log_player_info: (player_id: number) => {
|
||||
log_player_info(player_id)
|
||||
return true
|
||||
},
|
||||
})
|
||||
|
||||
function get_direction(start_position: MapPositionStruct, end_position: MapPositionStruct) {
|
||||
const angle = math.atan2(end_position.y - start_position.y, start_position.x - end_position.x)
|
||||
const octant = (angle + math.pi) / (2 * math.pi) * 8 + 0.5
|
||||
|
||||
if (octant < 1) {
|
||||
return defines.direction.east
|
||||
}
|
||||
if (octant < 2) {
|
||||
return defines.direction.northeast
|
||||
}
|
||||
if (octant < 3) {
|
||||
return defines.direction.north
|
||||
}
|
||||
if (octant < 4) {
|
||||
return defines.direction.northwest
|
||||
}
|
||||
if (octant < 5) {
|
||||
return defines.direction.west
|
||||
}
|
||||
if (octant < 6) {
|
||||
return defines.direction.southwest
|
||||
}
|
||||
if (octant < 7) {
|
||||
return defines.direction.south
|
||||
}
|
||||
return defines.direction.southeast
|
||||
}
|
||||
|
||||
function start_mining(player: LuaPlayer, entity_position: MapPositionStruct) {
|
||||
player.update_selected_entity(entity_position)
|
||||
player.mining_state = { mining: true, position: entity_position }
|
||||
log(`[AUTORIO] Started mining at position: ${serpent.line(entity_position)}`)
|
||||
}
|
||||
|
||||
script.on_event(defines.events.on_script_path_request_finished, (event: OnScriptPathRequestFinishedEvent) => {
|
||||
if (player_state.task_state !== TaskStates.WALKING_TO_ENTITY) {
|
||||
log('[AUTORIO] Not walking to entity, ignoring path request')
|
||||
return
|
||||
}
|
||||
|
||||
if (!player_state.parameters_walk_to_entity) {
|
||||
log('[AUTORIO] No parameters found when receiving path request')
|
||||
return
|
||||
}
|
||||
|
||||
if (!event.path) {
|
||||
log('[AUTORIO] Path calculation failed, switching to direct walking')
|
||||
player_state.task_state = TaskStates.WALKING_DIRECT
|
||||
player_state.parameters_walk_to_entity.calculating_path = false
|
||||
return
|
||||
}
|
||||
|
||||
player_state.parameters_walk_to_entity.path = event.path
|
||||
player_state.parameters_walk_to_entity.path_drawn = false
|
||||
player_state.parameters_walk_to_entity.path_index = 1
|
||||
player_state.parameters_walk_to_entity.calculating_path = false
|
||||
log(`[AUTORIO] Path calculation completed. Path length: ${event.path}`)
|
||||
})
|
||||
|
||||
script.on_event(defines.events.on_player_mined_entity, (unused_event: OnPlayerMinedEntityEvent) => {
|
||||
log('[AUTORIO] Entity mined, resetting to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walk_to_entity = undefined
|
||||
})
|
||||
|
||||
function setup() {
|
||||
const surface = game.surfaces[0]
|
||||
const enemies = surface.find_entities_filtered({ force: 'enemy' })
|
||||
log(`[AUTORIO] Removing ${enemies.length} enemies`)
|
||||
for (const enemy of enemies) {
|
||||
enemy.destroy()
|
||||
}
|
||||
|
||||
setup_complete = true
|
||||
log('[AUTORIO] Setup complete')
|
||||
}
|
||||
|
||||
function state_walking_to_entity(player: LuaPlayer) {
|
||||
if (!player_state.parameters_walk_to_entity) {
|
||||
log('[AUTORIO] No parameters found when walking to entity')
|
||||
return
|
||||
}
|
||||
|
||||
let nearest_entity: LuaEntity | null = null
|
||||
let min_distance = math.huge
|
||||
|
||||
const entities = player.surface.find_entities_filtered({
|
||||
position: player.position,
|
||||
radius: player_state.parameters_walk_to_entity.search_radius,
|
||||
name: player_state.parameters_walk_to_entity.entity_name,
|
||||
})
|
||||
if (entities.length === 0) {
|
||||
log('[AUTORIO] No entities found, reverting to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walk_to_entity = undefined
|
||||
return
|
||||
}
|
||||
|
||||
for (const entity of entities) {
|
||||
const distance = (entity.position.x - player.position.x) ^ 2 + (entity.position.y - player.position.y) ^ 2
|
||||
if (distance < min_distance) {
|
||||
min_distance = distance
|
||||
nearest_entity = entity
|
||||
}
|
||||
}
|
||||
|
||||
log(`[AUTORIO] Nearest entity position: ${serpent.line(nearest_entity?.position)}`)
|
||||
log(`[AUTORIO] Player position: ${serpent.line(player.position)}`)
|
||||
log(`[AUTORIO] Player bounding box: ${serpent.line(player.character?.bounding_box)}`)
|
||||
|
||||
if (nearest_entity && !player_state.parameters_walk_to_entity.calculating_path && !player_state.parameters_walk_to_entity.path) {
|
||||
const character = player.character
|
||||
if (!character) {
|
||||
log('[AUTORIO] Player character not found, aborting pathfinding')
|
||||
return
|
||||
}
|
||||
|
||||
// TODO: improve path following
|
||||
// currently using larger than character bbox as a workaround for the path following getting stuck on objects
|
||||
// may sometimes still get stuck on trees and will fail to find small passages
|
||||
const bbox: BoundingBoxArray = [[-0.5, -0.5], [0.5, 0.5]]
|
||||
const start = player.surface.find_non_colliding_position(
|
||||
'iron-chest', // TODO: using iron chest bbox so request_path doesn't fail standing near objects using the larger bbox
|
||||
character.position,
|
||||
10,
|
||||
0.5,
|
||||
false,
|
||||
)
|
||||
|
||||
if (!start) {
|
||||
log('[AUTORIO] find_non_colliding_position returned nil! Aborting pathfinding.')
|
||||
return
|
||||
}
|
||||
|
||||
const collision_mask: CollisionMask = {
|
||||
layers: {
|
||||
player: true,
|
||||
train: true,
|
||||
water_tile: true,
|
||||
object: true,
|
||||
// car: true,
|
||||
// cliff: true,
|
||||
},
|
||||
consider_tile_transitions: true,
|
||||
}
|
||||
|
||||
player.surface.request_path({
|
||||
bounding_box: bbox,
|
||||
collision_mask,
|
||||
radius: 2,
|
||||
start,
|
||||
goal: nearest_entity.position,
|
||||
force: player.force,
|
||||
entity_to_ignore: character,
|
||||
pathfind_flags: {
|
||||
cache: false,
|
||||
no_break: true,
|
||||
prefer_straight_paths: false,
|
||||
allow_paths_through_own_entities: false,
|
||||
},
|
||||
})
|
||||
player_state.parameters_walk_to_entity.calculating_path = true
|
||||
player_state.parameters_walk_to_entity.target_position = nearest_entity.position
|
||||
log(`[AUTORIO] Requested path calculation to ${serpent.line(nearest_entity.position)}`)
|
||||
}
|
||||
|
||||
if (player_state.parameters_walk_to_entity.path && nearest_entity) {
|
||||
if (!player_state.parameters_walk_to_entity.path_drawn) {
|
||||
for (let i = 1; i <= player_state.parameters_walk_to_entity.path.length - 1; i++) {
|
||||
rendering.draw_line({
|
||||
color: { r: 0, g: 1, b: 0 },
|
||||
width: 2,
|
||||
from: player_state.parameters_walk_to_entity.path[i].position,
|
||||
to: player_state.parameters_walk_to_entity.path[i + 1].position,
|
||||
surface: player.surface,
|
||||
time_to_live: 600,
|
||||
draw_on_ground: true,
|
||||
})
|
||||
}
|
||||
player_state.parameters_walk_to_entity.path_drawn = true
|
||||
log('[AUTORIO] Path drawn on ground')
|
||||
}
|
||||
|
||||
const path = player_state.parameters_walk_to_entity.path
|
||||
const path_index = player_state.parameters_walk_to_entity.path_index
|
||||
|
||||
if (path_index <= path.length && math.sqrt((nearest_entity.position.x - player.position.x) ^ 2 + (nearest_entity.position.y - player.position.y) ^ 2) > 1) {
|
||||
const next_position = path[path_index].position
|
||||
const direction = get_direction(player.position, next_position)
|
||||
|
||||
player.walking_state = {
|
||||
walking: true,
|
||||
direction,
|
||||
}
|
||||
|
||||
if (((next_position.x - player.position.x) ^ 2 + (next_position.y - player.position.y) ^ 2) < 0.01) {
|
||||
player_state.parameters_walk_to_entity.path_index = path_index + 1
|
||||
log(`[AUTORIO] Moving to next path index: ${player_state.parameters_walk_to_entity.path_index}`)
|
||||
}
|
||||
if (((nearest_entity.position.x - player.position.x) ^ 2 + (nearest_entity.position.y - player.position.y) ^ 2) < 2) {
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walk_to_entity = undefined
|
||||
log('[AUTORIO] Reached target entity, switching to IDLE state')
|
||||
}
|
||||
}
|
||||
else {
|
||||
rendering.clear()
|
||||
player.walking_state = { walking: false, direction: defines.direction.north }
|
||||
|
||||
if (player_state.parameters_walk_to_entity.should_mine) {
|
||||
log('[AUTORIO] Switching to MINING state')
|
||||
player_state.task_state = TaskStates.MINING
|
||||
}
|
||||
else {
|
||||
log('[AUTORIO] Task completed, switching to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walk_to_entity = undefined
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function state_mining(player: LuaPlayer) {
|
||||
if (!player_state.parameters_mine_entity) {
|
||||
log('[AUTORIO] No parameters found when mining')
|
||||
return
|
||||
}
|
||||
|
||||
const nearest_entity = player.surface.find_entities_filtered({
|
||||
position: player.position,
|
||||
radius: 2,
|
||||
name: player_state.parameters_mine_entity.entity_name,
|
||||
limit: 1,
|
||||
})[0]
|
||||
|
||||
log(`[AUTORIO] Mining entity: ${player_state.parameters_mine_entity.entity_name}`)
|
||||
if (nearest_entity !== undefined) {
|
||||
start_mining(player, nearest_entity.position)
|
||||
}
|
||||
else {
|
||||
log('[AUTORIO] No entity found to mine, switching to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_mine_entity = undefined
|
||||
}
|
||||
}
|
||||
|
||||
function state_placing(player: LuaPlayer) {
|
||||
if (!player) {
|
||||
log('[AUTORIO] Invalid player, ending PLACING task')
|
||||
return [false, 'Invalid player']
|
||||
}
|
||||
|
||||
if (!player_state.parameters_place_entity) {
|
||||
log('[AUTORIO] No parameters found when placing')
|
||||
return
|
||||
}
|
||||
|
||||
const surface = player.surface
|
||||
const inventory = player.get_main_inventory()
|
||||
|
||||
if (!inventory) {
|
||||
log('[AUTORIO] Cannot access player inventory, ending PLACING task')
|
||||
return [false, 'Cannot access player inventory']
|
||||
}
|
||||
|
||||
const entity_prototype = prototypes.entity[player_state.parameters_place_entity.entity_name]
|
||||
if (!entity_prototype || !entity_prototype.items_to_place_this) {
|
||||
log('[AUTORIO] Invalid entity name, ending PLACING task')
|
||||
return [false, 'Invalid entity name']
|
||||
}
|
||||
|
||||
const item_name = entity_prototype.items_to_place_this[0]
|
||||
if (!item_name) {
|
||||
log('[AUTORIO] Invalid entity name, ending PLACING task')
|
||||
return [false, 'Invalid entity name']
|
||||
}
|
||||
|
||||
const [item_stack, unused_count] = inventory.find_item_stack(player_state.parameters_place_entity.entity_name)
|
||||
if (!item_stack) {
|
||||
log('[AUTORIO] Entity not found in inventory, ending PLACING task')
|
||||
return [false, 'Entity not found in inventory']
|
||||
}
|
||||
|
||||
if (!player_state.parameters_place_entity.position) {
|
||||
player_state.parameters_place_entity.position = surface.find_non_colliding_position(player_state.parameters_place_entity.entity_name, player.position, 1, 1)
|
||||
if (!player_state.parameters_place_entity.position) {
|
||||
log('[AUTORIO] Could not find a valid position to place the entity, ending PLACING task')
|
||||
return [false, 'Could not find a valid position to place the entity']
|
||||
}
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
const create_entity_args: SurfaceCreateEntity = {
|
||||
name: player_state.parameters_place_entity.entity_name,
|
||||
position: player_state.parameters_place_entity.position,
|
||||
force: player.force,
|
||||
raise_built: true,
|
||||
player,
|
||||
}
|
||||
const entity = surface.create_entity(create_entity_args)
|
||||
|
||||
if (entity) {
|
||||
item_stack.count = item_stack.count - 1
|
||||
log(`[AUTORIO] Entity placed successfully: ${player_state.parameters_place_entity.entity_name}`)
|
||||
return [true, 'Entity placed successfully', entity]
|
||||
}
|
||||
log(`[AUTORIO] Failed to place entity: ${player_state.parameters_place_entity.entity_name}`)
|
||||
return [false, 'Failed to place entity']
|
||||
}
|
||||
|
||||
function state_picking_up(player: LuaPlayer) {
|
||||
if (!player_state.parameters_pickup_item) {
|
||||
log('[AUTORIO] No parameters found when picking up')
|
||||
return
|
||||
}
|
||||
|
||||
const nearby_containers = player.surface.find_entities_filtered({
|
||||
position: player.position,
|
||||
radius: player_state.parameters_pickup_item.search_radius,
|
||||
name: player_state.parameters_pickup_item.container_name,
|
||||
force: player.force,
|
||||
})
|
||||
|
||||
const player_inventory = player.get_main_inventory()
|
||||
if (!player_inventory) {
|
||||
log('[AUTORIO] Cannot access player inventory, ending PICKING_UP task')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_pickup_item = undefined
|
||||
return
|
||||
}
|
||||
|
||||
let picked_up_total = 0
|
||||
|
||||
for (const container of nearby_containers) {
|
||||
const max_index = container.get_max_inventory_index()
|
||||
for (let i = 1; i <= max_index; i++) {
|
||||
const inventory = container.get_inventory(i)
|
||||
if (inventory) { // Check if inventory exists at this index
|
||||
const [item_stack, unused_count] = inventory.find_item_stack(player_state.parameters_pickup_item.item_name)
|
||||
if (item_stack) {
|
||||
const to_pick_up = math.min(item_stack.count, player_state.parameters_pickup_item.count - picked_up_total)
|
||||
const picked_up = player_inventory.insert({ name: player_state.parameters_pickup_item.item_name, count: to_pick_up })
|
||||
|
||||
if (picked_up > 0) {
|
||||
inventory.remove({ name: player_state.parameters_pickup_item.item_name, count: picked_up })
|
||||
picked_up_total = picked_up_total + picked_up
|
||||
log(`[AUTORIO] Picked up ${picked_up} ${player_state.parameters_pickup_item.item_name} from ${container.name} inventory index ${i}`)
|
||||
}
|
||||
|
||||
if (picked_up_total >= player_state.parameters_pickup_item.count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
if (picked_up_total >= player_state.parameters_pickup_item.count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
if (picked_up_total >= player_state.parameters_pickup_item.count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if (picked_up_total === 0) {
|
||||
log('[AUTORIO] No items picked up, ending task')
|
||||
}
|
||||
else {
|
||||
log(`[AUTORIO] Picked up a total of ${picked_up_total} ${player_state.parameters_pickup_item.item_name}`)
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_pickup_item = undefined
|
||||
}
|
||||
|
||||
function state_auto_inserting(player: LuaPlayer) {
|
||||
if (!player_state.parameters_auto_insert_nearby) {
|
||||
log('[AUTORIO] No parameters found when auto inserting')
|
||||
return
|
||||
}
|
||||
|
||||
const nearby_entities = player.surface.find_entities_filtered({
|
||||
position: player.position,
|
||||
radius: 8,
|
||||
name: player_state.parameters_auto_insert_nearby.entity_name,
|
||||
force: player.force,
|
||||
})
|
||||
|
||||
const player_inventory = player.get_main_inventory()
|
||||
if (!player_inventory) {
|
||||
log('[AUTORIO] Cannot access player inventory, ending AUTO_INSERTING task')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_auto_insert_nearby = undefined
|
||||
return
|
||||
}
|
||||
|
||||
const [item_stack, unused_count] = player_inventory.find_item_stack(player_state.parameters_auto_insert_nearby.item_name)
|
||||
let inserted_total = 0
|
||||
|
||||
if (item_stack) {
|
||||
for (const entity of nearby_entities) {
|
||||
const max_index = entity.get_max_inventory_index()
|
||||
if (entity.can_insert({ name: player_state.parameters_auto_insert_nearby.item_name })) {
|
||||
for (let i = 1; i <= max_index; i++) {
|
||||
const inventory = entity.get_inventory(i)
|
||||
if (inventory && inventory.can_insert({ name: player_state.parameters_auto_insert_nearby.item_name })) {
|
||||
const to_insert = math.min(item_stack.count, player_state.parameters_auto_insert_nearby.max_count - inserted_total)
|
||||
const inserted = inventory.insert({ name: player_state.parameters_auto_insert_nearby.item_name, count: to_insert })
|
||||
if (inserted > 0) {
|
||||
player_inventory.remove({ name: player_state.parameters_auto_insert_nearby.item_name, count: inserted })
|
||||
inserted_total = inserted_total + inserted
|
||||
log(`[AUTORIO] Inserted ${inserted} ${player_state.parameters_auto_insert_nearby.item_name} into ${entity.name} inventory index ${i}`)
|
||||
}
|
||||
if (inserted_total >= player_state.parameters_auto_insert_nearby.max_count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
if (inserted_total >= player_state.parameters_auto_insert_nearby.max_count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
if (inserted_total >= player_state.parameters_auto_insert_nearby.max_count) {
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (inserted_total === 0) {
|
||||
log('[AUTORIO] No items inserted, ending task')
|
||||
}
|
||||
else {
|
||||
log(`[AUTORIO] Inserted a total of ${inserted_total} ${player_state.parameters_auto_insert_nearby.item_name}`)
|
||||
}
|
||||
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_auto_insert_nearby = undefined
|
||||
}
|
||||
}
|
||||
|
||||
function state_crafting(player: LuaPlayer) {
|
||||
if (!player_state.parameters_craft_item) {
|
||||
log('[AUTORIO] No parameters found when crafting')
|
||||
return
|
||||
}
|
||||
|
||||
const recipe = player.force.recipes[player_state.parameters_craft_item.item_name]
|
||||
|
||||
if (!recipe) {
|
||||
log('[AUTORIO] Recipe not found, ending task')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_craft_item = undefined
|
||||
return
|
||||
}
|
||||
|
||||
const ingredients = recipe.ingredients
|
||||
const player_inventory = player.get_main_inventory()
|
||||
|
||||
if (!player_inventory) {
|
||||
log('[AUTORIO] Cannot access player inventory, ending CRAFTING task')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_craft_item = undefined
|
||||
return
|
||||
}
|
||||
|
||||
let can_craft = true
|
||||
|
||||
for (const ingredient of ingredients) {
|
||||
if (player_inventory.get_item_count(ingredient.name) < ingredient.amount) {
|
||||
can_craft = false
|
||||
log(`[AUTORIO] Not enough ${ingredient.name} to craft ${player_state.parameters_craft_item.item_name}`)
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if (can_craft) {
|
||||
for (const ingredient of ingredients) {
|
||||
player_inventory.remove({ name: ingredient.name, count: ingredient.amount })
|
||||
}
|
||||
|
||||
recipe.products.forEach((product) => {
|
||||
if (product.type === 'item' && player_state.parameters_craft_item) {
|
||||
player.insert({ name: product.name, count: product.amount })
|
||||
|
||||
player_state.parameters_craft_item.crafted = player_state.parameters_craft_item.crafted + 1
|
||||
|
||||
log(`[AUTORIO] Crafted 1 ${player_state.parameters_craft_item.item_name}`)
|
||||
|
||||
if (player_state.parameters_craft_item.crafted >= player_state.parameters_craft_item.count) {
|
||||
log('[AUTORIO] Crafting task complete')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_craft_item = undefined
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
else {
|
||||
log('[AUTORIO] Not enough ingredients to craft, ending task')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_craft_item = undefined
|
||||
}
|
||||
}
|
||||
|
||||
function state_researching(player: LuaPlayer) {
|
||||
if (!player_state.parameters_research_technology) {
|
||||
log('[AUTORIO] No parameters found when researching')
|
||||
return
|
||||
}
|
||||
|
||||
const force = player.force
|
||||
const tech = force.technologies[player_state.parameters_research_technology.technology_name]
|
||||
|
||||
if (tech.researched) {
|
||||
log(`[AUTORIO] Research completed: ${player_state.parameters_research_technology.technology_name}`)
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_research_technology = undefined
|
||||
}
|
||||
else if (force.current_research !== tech) {
|
||||
log(`[AUTORIO] Research interrupted: ${player_state.parameters_research_technology.technology_name}`)
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_research_technology = undefined
|
||||
}
|
||||
}
|
||||
|
||||
function state_walking_direct(player: LuaPlayer) {
|
||||
if (!player_state.parameters_walking_direct) {
|
||||
log('[AUTORIO] No parameters found when walking directly')
|
||||
return
|
||||
}
|
||||
|
||||
const target = player_state.parameters_walking_direct.target_position
|
||||
|
||||
if (target) {
|
||||
const direction = get_direction(player.position, target)
|
||||
player.walking_state = {
|
||||
walking: true,
|
||||
direction,
|
||||
}
|
||||
|
||||
if (((target.x - player.position.x) ^ 2 + (target.y - player.position.y) ^ 2) < 2) {
|
||||
log('[AUTORIO] Reached target, switching to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walking_direct = undefined
|
||||
}
|
||||
}
|
||||
else {
|
||||
log('[AUTORIO] No target position, switching to IDLE state')
|
||||
player_state.task_state = TaskStates.IDLE
|
||||
player_state.parameters_walking_direct = undefined
|
||||
}
|
||||
}
|
||||
|
||||
script.on_event(defines.events.on_tick, (unused_event) => {
|
||||
if (!setup_complete) {
|
||||
setup()
|
||||
}
|
||||
|
||||
const player = game.connected_players[0]
|
||||
if (!player || !player.character) {
|
||||
log('[AUTORIO] No valid player found')
|
||||
return
|
||||
}
|
||||
|
||||
if (player_state.task_state === TaskStates.IDLE) {
|
||||
return
|
||||
}
|
||||
|
||||
if (player_state.task_state === TaskStates.WALKING_TO_ENTITY) {
|
||||
state_walking_to_entity(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.MINING) {
|
||||
state_mining(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.PLACING) {
|
||||
state_placing(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.AUTO_INSERTING) {
|
||||
state_auto_inserting(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.PICKING_UP) {
|
||||
state_picking_up(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.CRAFTING) {
|
||||
state_crafting(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.RESEARCHING) {
|
||||
state_researching(player)
|
||||
}
|
||||
else if (player_state.task_state === TaskStates.WALKING_DIRECT) {
|
||||
state_walking_direct(player)
|
||||
}
|
||||
})
|
||||
27
eslint.config.mjs
Normal file
27
eslint.config.mjs
Normal file
@@ -0,0 +1,27 @@
|
||||
import antfu from '@antfu/eslint-config'
|
||||
|
||||
export default antfu({
|
||||
rules: {
|
||||
'ts/naming-convention': [
|
||||
'error',
|
||||
{
|
||||
selector: [
|
||||
'property',
|
||||
'parameter',
|
||||
'variable'
|
||||
],
|
||||
format: ['snake_case'],
|
||||
},
|
||||
|
||||
],
|
||||
// rule conflict with ts/naming-convention when using snake_case
|
||||
'unused-imports/no-unused-vars': [
|
||||
'error',
|
||||
{
|
||||
argsIgnorePattern: '^unused_',
|
||||
destructuredArrayIgnorePattern: '^unused_',
|
||||
}
|
||||
]
|
||||
},
|
||||
})
|
||||
|
||||
9
info.json
Normal file
9
info.json
Normal file
@@ -0,0 +1,9 @@
|
||||
{
|
||||
"name": "autorio",
|
||||
"version": "0.1.0",
|
||||
"title": "autorio",
|
||||
"author": ["LemonNekoGH", "naklecha"],
|
||||
"factorio_version": "1.1",
|
||||
"dependencies": ["base >= 1.1"],
|
||||
"description": "an automation mod for factorio."
|
||||
}
|
||||
31
package.json
Normal file
31
package.json
Normal file
@@ -0,0 +1,31 @@
|
||||
{
|
||||
"name": "autorio.ts",
|
||||
"version": "1.0.0",
|
||||
"packageManager": "pnpm@9.15.4",
|
||||
"description": "An automation mod for Factorio written in TypeScript.",
|
||||
"author": "LemonNekoGH",
|
||||
"license": "MIT",
|
||||
"repository": {
|
||||
"type": "git",
|
||||
"url": "https://github.com/LemonNekoGH/autorio.ts"
|
||||
},
|
||||
"keywords": [
|
||||
"factorio",
|
||||
"automation",
|
||||
"mod",
|
||||
"typescript"
|
||||
],
|
||||
"main": "index.js",
|
||||
"scripts": {
|
||||
"build": "tstl",
|
||||
"dev": "tstl --watch"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@antfu/eslint-config": "^3.14.0",
|
||||
"eslint": "^9.18.0",
|
||||
"lua-types": "^2.13.1",
|
||||
"typed-factorio": "^3.11.0",
|
||||
"typescript": "^5.7.3",
|
||||
"typescript-to-lua": "^1.29.1"
|
||||
}
|
||||
}
|
||||
3075
pnpm-lock.yaml
generated
Normal file
3075
pnpm-lock.yaml
generated
Normal file
File diff suppressed because it is too large
Load Diff
16
tsconfig.json
Normal file
16
tsconfig.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"$schema": "https://raw.githubusercontent.com/TypeScriptToLua/TypeScriptToLua/master/tsconfig-schema.json",
|
||||
"compilerOptions": {
|
||||
"target": "ESNext",
|
||||
"lib": ["ESNext"],
|
||||
"moduleResolution": "Node",
|
||||
"types": [
|
||||
"typed-factorio/runtime",
|
||||
"lua-types/jit"
|
||||
],
|
||||
"strict": true
|
||||
},
|
||||
"tstl": {
|
||||
"luaTarget": "JIT"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user