Finish moving and testing functions file, Promises

This commit is contained in:
Skylar Grant 2022-12-03 19:42:47 -05:00
parent 5ba4fbada9
commit a29701e6eb
2 changed files with 115 additions and 65 deletions

View File

@ -2,8 +2,16 @@
const dotenv = require('dotenv').config(); const dotenv = require('dotenv').config();
// Module for working with files // Module for working with files
const fs = require('fs'); const fs = require('fs');
const { resolve } = require('path');
// Setup for use with the Pi's GPIO pins // Setup for use with the Pi's GPIO pins
const gpio = require('rpi-gpio'); if (process.env.ONPI == 'true') {
console.log('Running on a Raspberry Pi.');
const gpio = require('rpi-gpio');
} else if (process.env.ONPI == 'false') {
console.log('Not running on a Raspberry Pi.');
} else {
console.log('Problem with ENV file.');
}
// The functions we'll export to be used in other files // The functions we'll export to be used in other files
const functions = { const functions = {
@ -15,63 +23,90 @@ const functions = {
}, },
// Turns the auger on (Pin 7 high) // Turns the auger on (Pin 7 high)
on() { on() {
return new Promise((resolve) => {
if (process.env.ONPI == 'true') {
gpio.write(7, true, function(err) { gpio.write(7, true, function(err) {
if (err) throw err; if (err) throw err;
if (process.env.DEBUG == "true") console.log('Auger turned on.'); if (process.env.DEBUG == "true") console.log('Auger turned on.');
}); });
} else {
console.log('Auger turned on.');
resolve('Auger turned on.');
}
});
}, },
// Turns the auger off (pin 7 low) // Turns the auger off (pin 7 low)
off() { off() {
return new Promise((resolve) => {
if (process.env.ONPI == 'true') {
gpio.write(7, false, function(err) { gpio.write(7, false, function(err) {
if (err) throw err; if (err) throw err;
if (process.env.DEBUG == "true") console.log('Auger turned off.'); if (process.env.DEBUG == "true") console.log('Auger turned off.');
}); });
} else {
console.log('Auger turned off.');
resolve('Auger turned off.');
}
});
}, },
// Cycles the auger using the two functions above this one (functions.auger.on() and functions.auger.off()) // Cycles the auger using the two functions above this one (functions.auger.on() and functions.auger.off())
// Sleeps in between cycles using functions.sleep() // Sleeps in between cycles using functions.sleep()
cycle() { cycle() {
this.on(); return new Promise((resolve) => {
functions.sleep(process.env.ONTIME); this.on().then(() => {
this.off(); functions.sleep(process.env.ONTIME).then(() => {
functions.sleep(process.env.OFFTIME); this.off().then(() => {
return; functions.sleep(process.env.OFFTIME).then(() => {
resolve("Cycle complete.");
});
});
});
});
});
}, },
}, },
files: { files: {
async check() { check() {
return new Promise((resolve, reject) => {
// TODO this code needs to be finished from migration // TODO this code needs to be finished from migration
// Check for pause file existing, then sleep for preset time, then run the function again. // Check for pause file existing, then sleep for preset time, then run the function again.
if (fs.existsSync('./pause')) { if (fs.existsSync('./pause')) {
return "pause"; resolve("pause");
} }
// Check for reload file existing, then reload environment variables, then delete the file. // Check for reload file existing, then reload environment variables, then delete the file.
if (fs.existsSync('./reload')) { if (fs.existsSync('./reload')) {
return "reload"; resolve("reload");
} }
// Check for quit file existing, then delete it, then quit the program // Check for quit file existing, then delete it, then quit the program
if (fs.existsSync('./quit')) { if (fs.existsSync('./quit')) {
return "quit"; resolve("quit");
} }
return "none"; resolve("none");
});
}, },
}, },
commands: { commands: {
pause() { pause() {
console.log('Paused...'); return new Promise((resolve) => {
this.sleep(process.env.PAUSETIME).then(() => { return; }); functions.sleep(process.env.PAUSETIME).then(() => { resolve(); });
return; });
}, },
reload() { reload(envs) {
dotenv.config({ override: true }) return new Promise((resolve) => {
const dotenv = require('dotenv').config({ override: true });
fs.unlink('./reload', (err) => { fs.unlink('./reload', (err) => {
if (err) throw err; if (err) throw err;
console.log('Deleted reload file.'); console.log('Deleted reload file.');
}); });
console.log('Reloaded environment variables.'); console.log('Reloaded environment variables.');
console.log(`ONTIME=${process.env.ONTIME}\nOFFTIME=${process.env.OFFTIME}\nPAUSETIME=${process.env.PAUSETIME}\nDEBUG=${process.env.DEBUG}`); console.log(`ONTIME=${process.env.ONTIME}\nOFFTIME=${process.env.OFFTIME}\nPAUSETIME=${process.env.PAUSETIME}\nDEBUG=${process.env.DEBUG}\nONPI=${process.env.ONPI}`);
return; resolve();
});
}, },
quit() { quit() {
fs.unlink('./quit', (err) => { fs.unlink('./quit', (err) => {
@ -85,19 +120,26 @@ const functions = {
// Sleeps for any given milliseconds, call with await // Sleeps for any given milliseconds, call with await
sleep(ms) { sleep(ms) {
return new Promise((resolve) => { return new Promise((resolve) => {
setTimeout(resolve, ms); if (process.env.DEBUG == "true") console.log(`Sleeping for ${ms}ms`);
const finish = () => {
if (process.env.DEBUG == "true") console.log(`Slept for ${ms}ms`); if (process.env.DEBUG == "true") console.log(`Slept for ${ms}ms`);
resolve();
}
setTimeout(finish, ms);
}); });
}, },
init() { init() {
// Write the current env vars to console // Write the current env vars to console
console.log('Environment variables:'); console.log('Environment variables:');
console.log(`ONTIME=${process.env.ONTIME}\nOFFTIME=${process.env.OFFTIME}\nPAUSETIME=${process.env.PAUSETIME}\nDEBUG=${process.env.DEBUG}`); console.log(`ONTIME=${process.env.ONTIME}\nOFFTIME=${process.env.OFFTIME}\nPAUSETIME=${process.env.PAUSETIME}\nDEBUG=${process.env.DEBUG}\nONPI=${process.env.ONPI}`);
return true; return true;
}, },
} }
// Set up GPIO 4 (pysical pin 7) as output, then call functions.auger.ready() // Set up GPIO 4 (pysical pin 7) as output, then call functions.auger.ready()
gpio.setup(7, gpio.DIR_OUT, functions.auger.ready()); if (process.env.ONPI == 'true') {
gpio.setup(7, gpio.DIR_OUT, functions.auger.ready());
}
module.exports = { functions } module.exports = { functions };

26
main.js
View File

@ -1,32 +1,40 @@
// Custom functions module to keep main script clean // Custom functions module to keep main script clean
const fn = require('./functions').functions; const fn = require('./functions.js').functions;
// Environment Variables Importing // Environment Variables Importing
const dotenv = require('dotenv').config(); const dotenv = require('dotenv').config();
// TODO Add logic for other sensors // TODO Add logic for other sensors
while (true) { main(fn);
main();
}
// Main function, turns the auger on, sleeps for the time given in environment variables, then turns the auger off, sleeps, repeats. // Main function, turns the auger on, sleeps for the time given in environment variables, then turns the auger off, sleeps, repeats.
async function main() { async function main(fn) {
switch (fn.files.check()) { fn.files.check().then((res,rej) => {
console.log('File Check: ' + res);
switch (res) {
case "pause": case "pause":
fn.commands.pause(); fn.commands.pause().then(() => {
main(fn);
});
break; break;
case "reload": case "reload":
fn.commands.reload(); fn.commands.reload().then(() => {
main(fn);
});
break; break;
case "quit": case "quit":
fn.commands.quit(); fn.commands.quit();
break; break;
case "none": case "none":
fn.auger.cycle(); fn.auger.cycle().then(() => {
main(fn);
});
break; break;
default: default:
main(fn);
break; break;
} }
});
} }