naresh
/
growpi
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

wip

main
Nareshkumar Rao 7 months ago
parent
de6b9cf153
commit
cccf737c1a
14 changed files with 494 additions and 29 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      .vscode/tasks.json
  2. 331
      Cargo.lock
  3. 2
      Cargo.toml
  4. 2
      growpi.service
  5. 6
      growpi.toml
  6. 20
      html/growpi/src/App.vue
  7. 6
      src/config.rs
  8. 18
      src/control/data_logging.rs
  9. 46
      src/control/imaging.rs
  10. 6
      src/control/mod.rs
  11. 19
      src/control/temperature.rs
  12. 41
      src/io.rs
  13. 2
      src/main.rs
  14. 19
      src/server.rs

3
.vscode/tasks.json
View File

@ -7,7 +7,8 @@
"args": [ "args": [
"build", "build",
"--target", "--target",
"arm-unknown-linux-gnueabihf"
"arm-unknown-linux-gnueabihf",
"--release"
], ],
"problemMatcher": [ "problemMatcher": [
"$rustc" "$rustc"

331
Cargo.lock
View File

@ -42,6 +42,93 @@ dependencies = [
"libc", "libc",
] ]
[[package]]
name = "async-channel"
version = "2.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "136d4d23bcc79e27423727b36823d86233aad06dfea531837b038394d11e9928"
dependencies = [
"concurrent-queue",
"event-listener 5.3.0",
"event-listener-strategy 0.5.2",
"futures-core",
"pin-project-lite",
]
[[package]]
name = "async-io"
version = "2.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dcccb0f599cfa2f8ace422d3555572f47424da5648a4382a9dd0310ff8210884"
dependencies = [
"async-lock",
"cfg-if",
"concurrent-queue",
"futures-io",
"futures-lite",
"parking",
"polling",
"rustix",
"slab",
"tracing",
"windows-sys 0.52.0",
]
[[package]]
name = "async-lock"
version = "3.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d034b430882f8381900d3fe6f0aaa3ad94f2cb4ac519b429692a1bc2dda4ae7b"
dependencies = [
"event-listener 4.0.3",
"event-listener-strategy 0.4.0",
"pin-project-lite",
]
[[package]]
name = "async-process"
version = "2.2.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a53fc6301894e04a92cb2584fedde80cb25ba8e02d9dc39d4a87d036e22f397d"
dependencies = [
"async-channel",
"async-io",
"async-lock",
"async-signal",
"async-task",
"blocking",
"cfg-if",
"event-listener 5.3.0",
"futures-lite",
"rustix",
"tracing",
"windows-sys 0.52.0",
]
[[package]]
name = "async-signal"
version = "0.2.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "afe66191c335039c7bb78f99dc7520b0cbb166b3a1cb33a03f53d8a1c6f2afda"
dependencies = [
"async-io",
"async-lock",
"atomic-waker",
"cfg-if",
"futures-core",
"futures-io",
"rustix",
"signal-hook-registry",
"slab",
"windows-sys 0.52.0",
]
[[package]]
name = "async-task"
version = "4.7.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8b75356056920673b02621b35afd0f7dda9306d03c79a30f5c56c44cf256e3de"
[[package]] [[package]]
name = "async-trait" name = "async-trait"
version = "0.1.80" version = "0.1.80"
@ -53,6 +140,12 @@ dependencies = [
"syn", "syn",
] ]
[[package]]
name = "atomic-waker"
version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1505bd5d3d116872e7271a6d4e16d81d0c8570876c8de68093a09ac269d8aac0"
[[package]] [[package]]
name = "autocfg" name = "autocfg"
version = "1.2.0" version = "1.2.0"
@ -157,6 +250,20 @@ dependencies = [
"generic-array", "generic-array",
] ]
[[package]]
name = "blocking"
version = "1.6.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "495f7104e962b7356f0aeb34247aca1fe7d2e783b346582db7f2904cb5717e88"
dependencies = [
"async-channel",
"async-lock",
"async-task",
"futures-io",
"futures-lite",
"piper",
]
[[package]] [[package]]
name = "bumpalo" name = "bumpalo"
version = "3.16.0" version = "3.16.0"
@ -210,6 +317,15 @@ dependencies = [
"error-code", "error-code",
] ]
[[package]]
name = "concurrent-queue"
version = "2.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4ca0197aee26d1ae37445ee532fefce43251d24cc7c166799f4d46817f1d3973"
dependencies = [
"crossbeam-utils",
]
[[package]] [[package]]
name = "core-foundation-sys" name = "core-foundation-sys"
version = "0.8.6" version = "0.8.6"
@ -225,6 +341,12 @@ dependencies = [
"libc", "libc",
] ]
[[package]]
name = "crossbeam-utils"
version = "0.8.19"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "248e3bacc7dc6baa3b21e405ee045c3047101a49145e7e9eca583ab4c2ca5345"
[[package]] [[package]]
name = "crypto-common" name = "crypto-common"
version = "0.1.6" version = "0.1.6"
@ -320,6 +442,54 @@ version = "3.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a0474425d51df81997e2f90a21591180b38eccf27292d755f3e30750225c175b" checksum = "a0474425d51df81997e2f90a21591180b38eccf27292d755f3e30750225c175b"
[[package]]
name = "event-listener"
version = "4.0.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "67b215c49b2b248c855fb73579eb1f4f26c38ffdc12973e20e07b91d78d5646e"
dependencies = [
"concurrent-queue",
"parking",
"pin-project-lite",
]
[[package]]
name = "event-listener"
version = "5.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6d9944b8ca13534cdfb2800775f8dd4902ff3fc75a50101466decadfdf322a24"
dependencies = [
"concurrent-queue",
"parking",
"pin-project-lite",
]
[[package]]
name = "event-listener-strategy"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "958e4d70b6d5e81971bebec42271ec641e7ff4e170a6fa605f2b8a8b65cb97d3"
dependencies = [
"event-listener 4.0.3",
"pin-project-lite",
]
[[package]]
name = "event-listener-strategy"
version = "0.5.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0f214dc438f977e6d4e3500aaa277f5ad94ca83fbbd9b1a15713ce2344ccc5a1"
dependencies = [
"event-listener 5.3.0",
"pin-project-lite",
]
[[package]]
name = "fastrand"
version = "2.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9fc0510504f03c51ada170672ac806f1f105a88aa97a5281117e1ddc3368e51a"
[[package]] [[package]]
name = "fd-lock" name = "fd-lock"
version = "4.0.2" version = "4.0.2"
@ -346,6 +516,12 @@ dependencies = [
"percent-encoding", "percent-encoding",
] ]
[[package]]
name = "fuchsia-cprng"
version = "0.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a06f77d526c1a601b7c4cdd98f54b5eaabffc14d5f2f0296febdc7f357c6d3ba"
[[package]] [[package]]
name = "futures-channel" name = "futures-channel"
version = "0.3.30" version = "0.3.30"
@ -361,6 +537,25 @@ version = "0.3.30"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dfc6580bb841c5a68e9ef15c77ccc837b40a7504914d52e47b8b0e9bbda25a1d" checksum = "dfc6580bb841c5a68e9ef15c77ccc837b40a7504914d52e47b8b0e9bbda25a1d"
[[package]]
name = "futures-io"
version = "0.3.30"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a44623e20b9681a318efdd71c299b6b222ed6f231972bfe2f224ebad6311f0c1"
[[package]]
name = "futures-lite"
version = "2.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "52527eb5074e35e9339c6b4e8d12600c7128b68fb25dcb9fa9dec18f7c25f3a5"
dependencies = [
"fastrand",
"futures-core",
"futures-io",
"parking",
"pin-project-lite",
]
[[package]] [[package]]
name = "futures-task" name = "futures-task"
version = "0.3.30" version = "0.3.30"
@ -400,6 +595,7 @@ name = "growpi"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"ads1x1x", "ads1x1x",
"async-process",
"axum", "axum",
"chrono", "chrono",
"csv", "csv",
@ -410,6 +606,7 @@ dependencies = [
"rust-embed", "rust-embed",
"rustyline", "rustyline",
"serde", "serde",
"tempdir",
"tokio", "tokio",
"toml", "toml",
"tower-http", "tower-http",
@ -427,6 +624,12 @@ version = "0.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "95505c38b4572b2d910cecb0281560f54b440a19336cbbcb27bf6ce6adc6f5a8" checksum = "95505c38b4572b2d910cecb0281560f54b440a19336cbbcb27bf6ce6adc6f5a8"
[[package]]
name = "hermit-abi"
version = "0.3.9"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d231dfb89cfffdbc30e7fc41579ed6066ad03abda9e567ccafae602b97ec5024"
[[package]] [[package]]
name = "home" name = "home"
version = "0.5.9" version = "0.5.9"
@ -691,6 +894,12 @@ version = "1.19.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92" checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92"
[[package]]
name = "parking"
version = "2.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bb813b8af86854136c6922af0598d719255ecb2179515e6e7730d468f05c9cae"
[[package]] [[package]]
name = "percent-encoding" name = "percent-encoding"
version = "2.3.1" version = "2.3.1"
@ -729,6 +938,32 @@ version = "0.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8b870d8c151b6f2fb93e84a13146138f05d02ed11c7e7c54f8826aaaf7c9f184" checksum = "8b870d8c151b6f2fb93e84a13146138f05d02ed11c7e7c54f8826aaaf7c9f184"
[[package]]
name = "piper"
version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "668d31b1c4eba19242f2088b2bf3316b82ca31082a8335764db4e083db7485d4"
dependencies = [
"atomic-waker",
"fastrand",
"futures-io",
]
[[package]]
name = "polling"
version = "3.7.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "645493cf344456ef24219d02a768cf1fb92ddf8c92161679ae3d91b91a637be3"
dependencies = [
"cfg-if",
"concurrent-queue",
"hermit-abi",
"pin-project-lite",
"rustix",
"tracing",
"windows-sys 0.52.0",
]
[[package]] [[package]]
name = "proc-macro2" name = "proc-macro2"
version = "1.0.81" version = "1.0.81"
@ -757,6 +992,52 @@ dependencies = [
"nibble_vec", "nibble_vec",
] ]
[[package]]
name = "rand"
version = "0.4.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "552840b97013b1a26992c11eac34bdd778e464601a4c2054b5f0bff7c6761293"
dependencies = [
"fuchsia-cprng",
"libc",
"rand_core 0.3.1",
"rdrand",
"winapi",
]
[[package]]
name = "rand_core"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7a6fdeb83b075e8266dcc8762c22776f6877a63111121f5f8c7411e5be7eed4b"
dependencies = [
"rand_core 0.4.2",
]
[[package]]
name = "rand_core"
version = "0.4.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9c33a3c44ca05fa6f1807d8e6743f3824e8509beca625669633be0acbdf509dc"
[[package]]
name = "rdrand"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "678054eb77286b51581ba43620cc911abf02758c91f93f479767aed0f90458b2"
dependencies = [
"rand_core 0.3.1",
]
[[package]]
name = "remove_dir_all"
version = "0.5.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3acd125665422973a33ac9d3dd2df85edad0f4ae9b00dafb1a05e43a9f5ef8e7"
dependencies = [
"winapi",
]
[[package]] [[package]]
name = "rppal" name = "rppal"
version = "0.17.1" version = "0.17.1"
@ -941,6 +1222,24 @@ dependencies = [
"digest", "digest",
] ]
[[package]]
name = "signal-hook-registry"
version = "1.4.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a9e9e0b4211b72e7b8b6e85c807d36c212bdb33ea8587f7569562a84df5465b1"
dependencies = [
"libc",
]
[[package]]
name = "slab"
version = "0.4.9"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8f92a496fb766b417c996b9c5e57daf2f7ad3b0bebe1ccfca4856390e3d3bb67"
dependencies = [
"autocfg",
]
[[package]] [[package]]
name = "smallvec" name = "smallvec"
version = "1.13.2" version = "1.13.2"
@ -989,6 +1288,16 @@ version = "1.0.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a7065abeca94b6a8a577f9bd45aa0867a2238b74e8eb67cf10d492bc39351394" checksum = "a7065abeca94b6a8a577f9bd45aa0867a2238b74e8eb67cf10d492bc39351394"
[[package]]
name = "tempdir"
version = "0.3.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "15f2b5fb00ccdf689e0149d1b1b3c03fead81c2b37735d812fa8bddbbf41b6d8"
dependencies = [
"rand",
"remove_dir_all",
]
[[package]] [[package]]
name = "tokio" name = "tokio"
version = "1.37.0" version = "1.37.0"
@ -1234,6 +1543,22 @@ version = "0.2.92"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "af190c94f2773fdb3729c55b007a722abb5384da03bc0986df4c289bf5567e96" checksum = "af190c94f2773fdb3729c55b007a722abb5384da03bc0986df4c289bf5567e96"
[[package]]
name = "winapi"
version = "0.3.9"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
dependencies = [
"winapi-i686-pc-windows-gnu",
"winapi-x86_64-pc-windows-gnu",
]
[[package]]
name = "winapi-i686-pc-windows-gnu"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
[[package]] [[package]]
name = "winapi-util" name = "winapi-util"
version = "0.1.8" version = "0.1.8"
@ -1243,6 +1568,12 @@ dependencies = [
"windows-sys 0.52.0", "windows-sys 0.52.0",
] ]
[[package]]
name = "winapi-x86_64-pc-windows-gnu"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
[[package]] [[package]]
name = "windows-core" name = "windows-core"
version = "0.52.0" version = "0.52.0"

2
Cargo.toml
View File

@ -18,3 +18,5 @@ tower-http = { "version" = "0.5", features = ["cors"] }
csv = "1.3.0" csv = "1.3.0"
rust-embed = { "version" = "8.3.0", features = ["debug-embed"] } rust-embed = { "version" = "8.3.0", features = ["debug-embed"] }
mime_guess = "=2.0.4" mime_guess = "=2.0.4"
tempdir = "0.3.7"
async-process = "2.2.2"

2
growpi.service
View File

@ -10,4 +10,4 @@ ExecStart=/opt/growpi/growpi
WorkingDirectory=/opt/growpi WorkingDirectory=/opt/growpi
[Install] [Install]
WantedBy=mutli-user.target
WantedBy=multi-user.target

6
growpi.toml
View File

@ -34,10 +34,14 @@ temperature_set_point_upper = 35.0
temperature_set_point_lower = 28.0 temperature_set_point_lower = 28.0
temperature_loop_mins = 60 temperature_loop_mins = 60
sunlight_hours = 24 sunlight_hours = 24
soil_loop_hours = 12
watering_frequency_hours = 30 watering_frequency_hours = 30
watering_amount_grams = 200 watering_amount_grams = 200
[data_logging_settings] [data_logging_settings]
enabled = true enabled = true
frequency_mins = 60 frequency_mins = 60
imaging_frequency_minutes = 60
imaging_resolution = "R480p"
[server_settings]
port = 2205

20
html/growpi/src/App.vue
View File

@ -1,5 +1,6 @@
<script> <script>
import Button from 'primevue/button'; import Button from 'primevue/button';
import Image from 'primevue/image';
import InputNumber from 'primevue/inputnumber'; import InputNumber from 'primevue/inputnumber';
import ToggleButton from 'primevue/togglebutton'; import ToggleButton from 'primevue/togglebutton';
export default { export default {
@ -12,9 +13,10 @@ export default {
water_primed: false, water_primed: false,
fan_state: false, fan_state: false,
pumpAmount: 150, pumpAmount: 150,
image_src: "/image"
}; };
}, },
components: { ToggleButton, Button, InputNumber },
components: { ToggleButton, Button, InputNumber, Image },
methods: { methods: {
async updateInfo() { async updateInfo() {
let info = await receiveInfo(); let info = await receiveInfo();
@ -24,6 +26,9 @@ export default {
this.temperature = info.temperature; this.temperature = info.temperature;
this.soil_moisture = info.soil_moisture; this.soil_moisture = info.soil_moisture;
}, },
async updateImage() {
this.image_src = "/image#" + new Date().getTime();
},
async sendSwitch(name, state) { async sendSwitch(name, state) {
await fetch("/api/switch/" + name + "/" + to_state(state)); await fetch("/api/switch/" + name + "/" + to_state(state));
await this.updateInfo(); await this.updateInfo();
@ -32,11 +37,17 @@ export default {
console.log("Pumping: " + this.pumpAmount); console.log("Pumping: " + this.pumpAmount);
await fetch("/api/pump/" + this.pumpAmount); await fetch("/api/pump/" + this.pumpAmount);
await this.updateInfo(); await this.updateInfo();
},
async refreshImage() {
await fetch("/api/refresh_image");
this.updateImage();
} }
}, },
created() { created() {
this.updateInfo(); this.updateInfo();
this.updateImage();
setInterval(this.updateInfo, 1000); setInterval(this.updateInfo, 1000);
setInterval(this.updateImage, 10000);
} }
} }
@ -107,6 +118,13 @@ function to_bool(state) {
:min="100" :max="300" /> :min="100" :max="300" />
</td> </td>
</tr> </tr>
<tr>
<td colspan="2">
<Image :src="image_src" />
<br /><br />
<Button @click="refreshImage();" label="Refresh Image" />
</td>
</tr>
</table> </table>
</template> </template>
<style> <style>

6
src/config.rs
View File

@ -1,6 +1,6 @@
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use crate::error::GenericResult;
use crate::{error::GenericResult, io::ImageResolution};
#[derive(Serialize, Deserialize)] #[derive(Serialize, Deserialize)]
pub struct RelaySettings { pub struct RelaySettings {
@ -58,6 +58,8 @@ pub struct ControllerSettings {
pub struct DataLoggingSettings { pub struct DataLoggingSettings {
pub enabled: bool, pub enabled: bool,
pub frequency_mins: u64, pub frequency_mins: u64,
pub imaging_frequency_minutes: u64,
pub imaging_resolution: ImageResolution,
} }
#[derive(Serialize, Deserialize, Clone)] #[derive(Serialize, Deserialize, Clone)]
@ -128,6 +130,8 @@ impl Default for Configuration {
data_logging_settings: DataLoggingSettings { data_logging_settings: DataLoggingSettings {
enabled: true, enabled: true,
frequency_mins: 60, frequency_mins: 60,
imaging_frequency_minutes: 60,
imaging_resolution: ImageResolution::R480p,
}, },
server_settings: ServerSettings { port: 2205 }, server_settings: ServerSettings { port: 2205 },
} }

18
src/control/data_logging.rs
View File

@ -4,7 +4,6 @@ use chrono::Utc;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use crate::{ use crate::{
config::DataLoggingSettings,
error::GenericResult, error::GenericResult,
sensors, sensors,
state::{lock_state, ProgramStateShared}, state::{lock_state, ProgramStateShared},
@ -43,15 +42,14 @@ impl DataRecords {
pub async fn data_logging_loop(program_state: ProgramStateShared) { pub async fn data_logging_loop(program_state: ProgramStateShared) {
loop { loop {
let DataLoggingSettings {
enabled,
frequency_mins,
} = lock_state(&program_state)
.map(|state| state.config.data_logging_settings.clone())
.unwrap_or(DataLoggingSettings {
enabled: true,
frequency_mins: 60,
});
let (enabled, frequency_mins) = lock_state(&program_state)
.map(|state| {
(
state.config.data_logging_settings.enabled,
state.config.data_logging_settings.frequency_mins,
)
})
.unwrap_or((true, 60));
if enabled { if enabled {
let _ = DataRecords::push(program_state.clone()); let _ = DataRecords::push(program_state.clone());
} }

46
src/control/imaging.rs
View File

@ -0,0 +1,46 @@
use std::{path::Path, time::Duration};
use crate::{
error::GenericResult,
io,
state::{lock_state, ProgramStateShared},
};
pub const IMAGE_PATH: &str = "./growpi.image.jpeg";
pub async fn save_latest_image(program_state: ProgramStateShared) -> GenericResult<()> {
let resolution = lock_state(&program_state)
.map(|state| {
state
.config
.data_logging_settings
.imaging_resolution
.clone()
})
.unwrap_or(io::ImageResolution::R360p);
io::capture_image(&resolution, get_image_path()).await?;
Ok(())
}
pub fn get_image_path() -> &'static Path {
Path::new(IMAGE_PATH)
}
pub async fn imaging_loop(program_state: ProgramStateShared) {
loop {
let imaging_frequency = lock_state(&program_state)
.map(|state| state.config.data_logging_settings.imaging_frequency_minutes)
.map(|f| match f {
0 => None,
n => Some(n),
})
.unwrap_or(None);
match imaging_frequency {
Some(f) => {
let _ = save_latest_image(program_state.clone()).await;
tokio::time::sleep(Duration::from_mins(f)).await;
}
None => tokio::time::sleep(Duration::from_hours(24)).await,
};
}
}

6
src/control/mod.rs
View File

@ -2,13 +2,14 @@ use tokio::join;
use crate::{ use crate::{
control::{ control::{
data_logging::data_logging_loop, light::light_control_loop,
data_logging::data_logging_loop, imaging::imaging_loop, light::light_control_loop,
soil::soil_moisture_control_loop, temperature::temperature_control_loop, soil::soil_moisture_control_loop, temperature::temperature_control_loop,
}, },
state::ProgramStateShared, state::ProgramStateShared,
}; };
mod data_logging; mod data_logging;
pub mod imaging;
mod light; mod light;
mod soil; mod soil;
mod temperature; mod temperature;
@ -18,6 +19,7 @@ pub async fn control_thread(program_state: ProgramStateShared) {
light_control_loop(program_state.clone()), light_control_loop(program_state.clone()),
temperature_control_loop(program_state.clone()), temperature_control_loop(program_state.clone()),
soil_moisture_control_loop(program_state.clone()), soil_moisture_control_loop(program_state.clone()),
data_logging_loop(program_state.clone())
data_logging_loop(program_state.clone()),
imaging_loop(program_state.clone())
); );
} }

19
src/control/temperature.rs
View File

@ -21,17 +21,16 @@ fn temperature_control(program_state: ProgramStateShared) -> GenericResult<()> {
} }
pub async fn temperature_control_loop(program_state: ProgramStateShared) { pub async fn temperature_control_loop(program_state: ProgramStateShared) {
let loop_duration = program_state
.lock()
.map(|program_state| {
program_state
.config
.controller_settings
.temperature_loop_mins
})
.unwrap_or(1);
loop { loop {
let loop_duration = program_state
.lock()
.map(|program_state| {
program_state
.config
.controller_settings
.temperature_loop_mins
})
.unwrap_or(1);
let _ = temperature_control(program_state.clone()); let _ = temperature_control(program_state.clone());
tokio::time::sleep(Duration::from_mins(loop_duration)).await; tokio::time::sleep(Duration::from_mins(loop_duration)).await;
} }

41
src/io.rs
View File

@ -1,6 +1,8 @@
use ads1x1x::{Ads1x1x, ChannelSelection, DynamicOneShot}; use ads1x1x::{Ads1x1x, ChannelSelection, DynamicOneShot};
use async_process::Command;
use nb::block; use nb::block;
use rppal::gpio::{Gpio, OutputPin}; use rppal::gpio::{Gpio, OutputPin};
use serde::{Deserialize, Serialize};
use crate::{config::*, error::GenericResult}; use crate::{config::*, error::GenericResult};
@ -93,3 +95,42 @@ impl Relay {
.ok_or("Pin not configured.")?) .ok_or("Pin not configured.")?)
} }
} }
#[allow(dead_code)]
#[derive(Clone, Serialize, Deserialize)]
pub enum ImageResolution {
R1080p,
R720p,
R480p,
R360p,
}
impl ImageResolution {
fn get_width_height(&self) -> (u64, u64) {
match self {
ImageResolution::R1080p => (1920, 1080),
ImageResolution::R720p => (1280, 720),
ImageResolution::R480p => (640, 480),
ImageResolution::R360p => (480, 360),
}
}
}
pub async fn capture_image(
resolution: &ImageResolution,
path: &std::path::Path,
) -> GenericResult<()> {
let path = std::path::absolute(path)?;
let (width, height) = resolution.get_width_height();
Command::new("/usr/bin/libcamera-jpeg")
.arg("-o")
.arg(path.clone())
.arg("-t")
.arg("1")
.arg("--width")
.arg(width.to_string())
.arg("--height")
.arg(height.to_string())
.status()
.await?
.exit_ok()?;
Ok(())
}

2
src/main.rs
View File

@ -1,4 +1,6 @@
#![feature(duration_constructors)] #![feature(duration_constructors)]
#![feature(exit_status_error)]
#![feature(absolute_path)]
use cli_mode::run_cli; use cli_mode::run_cli;
use config::Configuration; use config::Configuration;

19
src/server.rs
View File

@ -9,7 +9,7 @@ use serde::{Deserialize, Serialize};
use tower_http::cors::{Any, CorsLayer}; use tower_http::cors::{Any, CorsLayer};
use crate::{ use crate::{
actuators,
actuators, control,
error::GenericResult, error::GenericResult,
io::RelaySwitchState, io::RelaySwitchState,
sensors, sensors,
@ -36,6 +36,8 @@ fn setup_router(program_state: ProgramStateShared) -> Router {
.route("/api/info", get(info_handler)) .route("/api/info", get(info_handler))
.route("/api/switch/:device/:state", get(switch_handler)) .route("/api/switch/:device/:state", get(switch_handler))
.route("/api/pump/:quantity", get(pump_handler)) .route("/api/pump/:quantity", get(pump_handler))
.route("/api/refresh_image", get(image_refresh_handler))
.route("/image", get(image_handler))
.route("/*path", get(site_handler)) .route("/*path", get(site_handler))
.route("/", get(root_handler)) .route("/", get(root_handler))
.with_state(program_state) .with_state(program_state)
@ -139,3 +141,18 @@ fn serve_site(path: Option<String>) -> Response {
} }
response response
} }
async fn image_handler() -> Response {
let bytes = std::fs::read(control::imaging::get_image_path());
let response = bytes.map(|bytes| {
let mut r = bytes.into_response();
r.headers_mut()
.append(header::CONTENT_TYPE, HeaderValue::from_static("image/jpeg"));
r
});
response.unwrap_or_else(|e| format!("Error: {}", e).into_response())
}
async fn image_refresh_handler(State(program_state): State<ProgramStateShared>) -> Response {
let _ = control::imaging::save_latest_image(program_state).await;
StatusCode::OK.into_response()
}

Write Preview
Loading…
Cancel
Save