PlantCtrl/rust/src/main.rs

220 lines
7.1 KiB
Rust
Raw Normal View History

2023-12-07 02:33:17 +01:00
use std::{
ffi::CString,
fs::File,
io::{Read, Write},
str::from_utf8,
};
2023-12-04 00:41:29 +01:00
2023-12-07 02:33:17 +01:00
use chrono::{Datelike, NaiveDateTime, Timelike};
2023-11-23 22:50:17 +01:00
2023-12-07 02:33:17 +01:00
use anyhow::{Context, Result};
2023-11-23 22:50:17 +01:00
use chrono_tz::Europe::Berlin;
use esp_idf_hal::delay::Delay;
2023-11-29 18:27:40 +01:00
use esp_idf_svc::http::server::EspHttpServer;
2023-12-07 02:33:17 +01:00
use plant_hal::{CreatePlantHal, PlantCtrlBoardInteraction, PlantHal, PLANT_COUNT};
2023-11-23 22:50:17 +01:00
use webserver::webserver::httpd;
2023-11-29 18:27:40 +01:00
mod config;
2023-12-07 02:33:17 +01:00
pub mod plant_hal;
2023-11-23 22:50:17 +01:00
mod webserver {
pub mod webserver;
}
2023-11-20 01:46:19 +01:00
2023-12-07 02:33:17 +01:00
fn main() -> Result<()> {
2023-11-20 01:46:19 +01:00
// It is necessary to call this function once. Otherwise some patches to the runtime
// implemented by esp-idf-sys might not link properly. See https://github.com/esp-rs/esp-idf-template/issues/71
esp_idf_svc::sys::link_patches();
// Bind the log crate to the ESP Logging facilities
esp_idf_svc::log::EspLogger::initialize_default();
2023-11-23 22:50:17 +01:00
log::info!("Startup Rust");
2023-11-20 01:46:19 +01:00
2023-11-27 01:07:07 +01:00
let git_hash = env!("VERGEN_GIT_DESCRIBE");
println!("Version useing git has {}", git_hash);
2023-11-23 22:50:17 +01:00
2023-11-21 23:45:15 +01:00
2023-12-07 02:33:17 +01:00
let mut board = PlantHal::create()?;
println!("Board hal init");
let time = board.time();
let mut cur = match time {
Ok(cur) => cur,
Err(err) => {
log::error!("time error {}", err);
NaiveDateTime::from_timestamp_millis(0).unwrap().and_utc()
}
};
//check if we know the time current > 2020
if cur.year() < 2020 {
if board.is_day() {
//assume TZ safe times ;)
cur = *cur.with_hour(15).get_or_insert(cur);
} else {
cur = *cur.with_hour(3).get_or_insert(cur);
}
2023-11-21 23:45:15 +01:00
}
2023-11-20 01:46:19 +01:00
2023-12-07 02:33:17 +01:00
println!("cur is {}", cur);
2023-11-21 01:05:12 +01:00
2023-12-04 00:41:29 +01:00
2023-11-21 01:05:12 +01:00
2023-12-07 02:33:17 +01:00
//continous/interrupt?
//check if boot button is pressed, if longer than 5s delete config and reboot into config mode
2023-11-29 18:27:40 +01:00
2023-12-07 02:33:17 +01:00
let config = board.get_config();
match config {
Ok(conf) => {
2023-11-29 18:27:40 +01:00
},
2023-12-07 02:33:17 +01:00
Err(err) => {
2023-11-23 22:50:17 +01:00
},
}
2023-11-21 23:45:15 +01:00
2023-12-07 02:33:17 +01:00
// let proceed = config.unwrap();
//check if we have a config file
// if not found or parsing error -> error very fast blink general fault
//if this happens after a firmeware upgrade (check image state), mark as invalid
//blink general fault error_reading_config_after_upgrade, reboot after
// open accesspoint with webserver for wlan mqtt setup
//blink general fault error_no_config_after_upgrade
//once config is set store it and reboot
//if proceed.tank_sensor_enabled() {
//}
//is tank sensor enabled in config?
//measure tank level (without wifi due to interference)
//TODO this should be a result// detect invalid measurement value
let tank_value = board.tank_sensor_mv();
match tank_value {
Ok(tank_raw) => {
println!("Tank sensor returned {}", tank_raw);
},
Err(_) => {
//if not possible value, blink general fault error_tank_sensor_fault
board.general_fault(true);
//set general fault persistent
//set tank sensor state to fault
},
}
//measure each plant moisture
let mut initial_measurements_a: [i32;PLANT_COUNT] = [0;PLANT_COUNT];
let mut initial_measurements_b: [i32;PLANT_COUNT] = [0;PLANT_COUNT];
let mut initial_measurements_p: [i32;PLANT_COUNT] = [0;PLANT_COUNT];
for plant in 0..PLANT_COUNT {
initial_measurements_a[plant] = board.measure_moisture_hz(plant, plant_hal::Sensor::A)?;
initial_measurements_b[plant] = board.measure_moisture_hz(plant, plant_hal::Sensor::B)?;
initial_measurements_p[plant] = board.measure_moisture_hz(plant, plant_hal::Sensor::PUMP)?;
}
2023-11-29 18:27:40 +01:00
2023-12-07 02:33:17 +01:00
//match board.wifi("C3MA", Some("chaosimquadrat"), 10000) {
// Ok(_) => println!("online mode"),
// Err(_) => {
// println!("Offline mode");
//},
//}
//try connect wifi and do mqtt roundtrip
// if no wifi, set general fault persistent
//if no mqtt, set general fault persistent
2023-12-07 02:50:50 +01:00
let mut total_size = 0;
let mut used_size = 0;
2023-12-07 02:33:17 +01:00
unsafe {
2023-12-07 02:50:50 +01:00
let base_path = CString::new("/spiffs")?;
let storage = CString::new("storage")?;
2023-12-07 02:33:17 +01:00
let conf = esp_idf_sys::esp_vfs_spiffs_conf_t {
base_path: base_path.as_ptr(),
partition_label: storage.as_ptr(),
max_files: 5,
format_if_mount_failed: true,
};
2023-12-07 02:50:50 +01:00
esp_idf_sys::esp!(esp_idf_sys::esp_vfs_spiffs_register(&conf))?;
2023-12-07 02:33:17 +01:00
2023-12-07 02:50:50 +01:00
esp_idf_sys::esp!(esp_idf_sys::esp_spiffs_info(storage.as_ptr(),&mut total_size,&mut used_size))?;
2023-11-29 18:27:40 +01:00
}
2023-12-07 02:50:50 +01:00
println!("Total spiffs size is {}, used size is {}", total_size, used_size);
2023-12-07 02:33:17 +01:00
println!("writing");
let mut config_file = File::create("/spiffs/config.cfg")?;
config_file.write_all("test stuff".as_bytes())?;
config_file.flush()?;
println!("Reading");
let mut cfg = File::open("/spiffs/config.cfg")?;
let mut data: [u8; 512] = [0; 512];
let read = cfg.read(&mut data)?;
println!("Read file {}", from_utf8(&data[0..read])?);
/*
match board.sntp(1000 * 120) {
2023-11-21 23:45:15 +01:00
Ok(new_time) => cur = new_time,
2023-11-23 22:50:17 +01:00
Err(err) => {
println!("sntp error: {}", err);
}
2023-11-21 23:45:15 +01:00
}
2023-11-23 22:50:17 +01:00
println!("Running logic at utc {}", cur);
let europe_time = cur.with_timezone(&Berlin);
println!("Running logic at europe/berlin {}", europe_time);
2023-11-21 01:05:12 +01:00
//if config battery mode
2023-12-07 02:33:17 +01:00
//read battery level
//if not possible set general fault persistent, but do continue
2023-11-21 01:05:12 +01:00
//else
2023-12-07 02:33:17 +01:00
//assume 12v and max capacity
2023-11-20 02:24:14 +01:00
//if tank sensor is enabled
2023-12-07 02:33:17 +01:00
//if tank sensor fault abort if config require is set
//check if water is > minimum allowed || fault
//if not, set all plants requiring water to persistent fault
2023-11-21 01:05:12 +01:00
//for each plant
2023-12-07 02:33:17 +01:00
//check if moisture is < target
//state += dry
//check if in cooldown
//state += cooldown
//check if consecutive pumps > limit
//state += notworking
//set plant fault persistent
//pump one cycle
// set last pump time to now
//during pump state += active
//after pump check if Pump moisture value is increased by config delta x
// state -= active
// state += cooldown
// if not set plant error persistent fault
// state += notworking
//set consecutive pumps+=1
2023-11-21 01:05:12 +01:00
//check if during light time
2023-12-07 02:33:17 +01:00
//lightstate += out of worktime
2023-11-21 01:05:12 +01:00
//check battery level
2023-12-07 02:33:17 +01:00
//lightstate += battery empty
2023-11-21 01:05:12 +01:00
//check solar level if config requires
2023-12-07 02:33:17 +01:00
//lightstate += stillday
2023-11-21 01:05:12 +01:00
//if no preventing lightstate, enable light
2023-12-07 02:33:17 +01:00
//lightstate = active
//keep webserver in scope
let webserver = httpd(true);
let delay = Delay::new_default();
loop {
//let freertos do shit
delay.delay_ms(1001);
}
*/
return Ok(());
2023-11-20 01:46:19 +01:00
}
2023-11-20 02:24:14 +01:00
//error codes
//error_reading_config_after_upgrade
//error_no_config_after_upgrade
2023-12-07 02:33:17 +01:00
//error_tank_sensor_fault