84 lines
2.0 KiB
Rust
84 lines
2.0 KiB
Rust
#![feature(proc_macro_hygiene, decl_macro)]
|
|
|
|
#[macro_use] extern crate rocket;
|
|
#[macro_use] extern crate rocket_contrib;
|
|
#[macro_use] extern crate serde_derive;
|
|
|
|
use rocket_contrib::json::Json;
|
|
use pathfinding::grid::Grid;
|
|
|
|
mod battlesnake;
|
|
|
|
#[get("/")]
|
|
fn index() -> &'static str {
|
|
"Hello, world!"
|
|
}
|
|
|
|
#[get("/ping")]
|
|
fn ping() -> &'static str {
|
|
"OK - Pneuma online"
|
|
}
|
|
|
|
#[post("/begin", format = "json", data = "<msg>")]
|
|
fn begin(msg: Json<battlesnake::SnakeRequest>) -> Json<battlesnake::StartResponse> {
|
|
Json(battlesnake::StartResponse{
|
|
color: "#5ce8c3".to_string(),
|
|
head_type: "beluga".to_string(),
|
|
tail_type: "skinny".to_string(),
|
|
})
|
|
}
|
|
|
|
#[post("/move", format = "json", data = "<msg>")]
|
|
fn make_move(msg: Json<battlesnake::SnakeRequest>) -> Json<battlesnake::MoveResponse> {
|
|
let target = find_target(&msg);
|
|
println!("target: {:?}", target);
|
|
|
|
let head = &msg.you.body[0];
|
|
for ne in &msg.board.safe_neighbors(&head) {
|
|
println!("{:?} {}", ne.0, battlesnake::Line{
|
|
start: &head,
|
|
end: &ne.0,
|
|
}.direction());
|
|
}
|
|
|
|
pathfinding::directed::astar::astar(
|
|
&head,
|
|
|n| msg.board.safe_neighbors(n).iter(),
|
|
|n| (battlesnake::manhattan(n, &target) as usize),
|
|
|n| n == &target,
|
|
);
|
|
|
|
Json(battlesnake::MoveResponse {
|
|
move_field: "up".to_string(),
|
|
})
|
|
}
|
|
|
|
fn find_target<'a>(gs: &'a battlesnake::SnakeRequest) -> &'a battlesnake::Coord {
|
|
let head = &gs.you.body[0];
|
|
if gs.you.health > 30 {
|
|
let mut lowest_score: u32 = 99999;
|
|
let mut coord: &battlesnake::Coord = &gs.you.body.last().unwrap();
|
|
|
|
for food in &gs.board.food {
|
|
let score = battlesnake::manhattan(&head, &food);
|
|
if score < lowest_score {
|
|
lowest_score = score;
|
|
coord = food;
|
|
}
|
|
}
|
|
|
|
return coord;
|
|
}
|
|
|
|
return gs.you.body.last().unwrap();
|
|
}
|
|
|
|
fn main() {
|
|
rocket::ignite().mount("/", routes![
|
|
index,
|
|
begin,
|
|
ping,
|
|
make_move,
|
|
]).launch();
|
|
}
|