voidf1sh/hestia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

New MQTT Config

Browse Source
This commit is contained in:
Skylar Grant 2024-11-30 11:30:23 -05:00
parent c3a33b6dac
commit 7872054122
3 changed files with 57 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/custom_modules/HestiaClasses.js
View File

@ -3,8 +3,9 @@ const mqtt = require('mqtt');
module.exports = { module.exports = {
// State class // State class
State: class State { State: class State extends EventEmitter {
constructor(config) { constructor(config) {
super();
this.publisher = 'backend'; this.publisher = 'backend';
this.igniter = { this.igniter = {
@ -75,7 +76,7 @@ module.exports = {
this.shutdown = { this.shutdown = {
topic: config.mqtt.topics.shutdown topic: config.mqtt.topics.shutdown
}; };
console.log(`State initialized.`) this.emit('announcement', `State initialized.`)
return this; return this;
}; };
}, },
@ -84,30 +85,31 @@ module.exports = {
constructor(state) { constructor(state) {
super(); super();
this.publisher = state.publisher; this.publisher = state.publisher;
return this;
} }
init(state, config) { init(state, config) {
// Connect to the MQTT Broker // Connect to the MQTT Broker
console.log(`Attempting MQTT connection to broker: ${config.mqtt.address}`); this.emit('announcement', `Attempting MQTT connection to broker: ${config.mqtt.address}`);
this.client = mqtt.connect(config.mqtt.address, { this.client = mqtt.connect(config.mqtt.address, {
port: config.mqtt.port, port: config.mqtt.port,
}); });
const { client } = this; const { client } = this;
client.on('connect', () => { client.on('connect', () => {
console.log('Connected to MQTT broker'); this.emit('announcement', 'Connected to MQTT broker');
// Subscribe to status topics // Subscribe to status topics
config.states.elements.forEach(element => { config.states.elements.forEach(element => {
client.subscribe(state[element].topic, (err) => { client.subscribe(state[element].topic, (err) => {
if (!err) { if (!err) {
console.log(`Subscribed to ${state[element].topic}`); this.emit('announcement', `Subscribed to ${state[element].topic}`);
} }
}); });
}); });
}); });
client.on('disconnect', () => { client.on('disconnect', () => {
console.log('Disconnected from MQTT broker'); this.emit('announcement', 'Disconnected from MQTT broker');
}); });
// Handle when the Broker sends us a message // Handle when the Broker sends us a message
@ -120,14 +122,14 @@ module.exports = {
// Since the message is a JSON object, we can parse it // Since the message is a JSON object, we can parse it
const msgJson = JSON.parse(msgStr); const msgJson = JSON.parse(msgStr);
// Log the message // Log the message
// console.log(`Message received on topic ${topic}:`); // this.emit('announcement', `Message received on topic ${topic}:`);
// console.log(msgJson); // this.emit('announcement', msgJson);
// Check if the message is from the backend // Check if the message is from the backend
if (msgJson.publisher === this.publisher) { if (msgJson.publisher === this.publisher) {
// console.log('Message is from the backend, ignoring'); // this.emit('announcement', 'Message is from the backend, ignoring');
return; return;
} }
// console.log('Message is from the frontend, updating state'); // this.emit('announcement', 'Message is from the frontend, updating state');
// Update the state // Update the state
state[msgJson.name].on = msgJson.on; state[msgJson.name].on = msgJson.on;
// Emit the state change // Emit the state change
@ -139,7 +141,7 @@ module.exports = {
// Empty block for 'hestia/command' topics // Empty block for 'hestia/command' topics
this.emit('shutdown'); this.emit('shutdown');
} else { } else {
console.log(`Unknown topic: ${topic}`); this.emit('announcement', `Unknown topic: ${topic}`);
} }
}); });
} }
@ -151,7 +153,7 @@ module.exports = {
if (err) { if (err) {
console.error('Failed to publish message:', err); console.error('Failed to publish message:', err);
} else { } else {
console.log('Message published and retained on topic:', topic); this.emit('announcement', 'Message published and retained on topic:', topic);
} }
}); });
} }

3
src/custom_modules/functions.js
View File

@ -21,6 +21,9 @@ module.exports = {
console.log(`DEBUG: ${message}`); console.log(`DEBUG: ${message}`);
} }
break; break;
case 'ERROR':
console.error(`ERROR: ${message}`);
break;
default: default:
break; break;
} }

65
src/main.js
View File

@ -1,37 +1,41 @@
// Variables /***************************************************************************************/
process.debug = true;
// Import modules // Import modules
/***************************************************************************************/
const gpio = require('./custom_modules/VoidGPIO.js'); const gpio = require('./custom_modules/VoidGPIO.js');
const config = require('./custom_modules/config.json'); const config = require('./custom_modules/config.json');
const fn = require('./custom_modules/functions.js'); const fn = require('./custom_modules/functions.js');
const { State, Communicator } = require('./custom_modules/HestiaClasses.js'); const { State, Communicator } = require('./custom_modules/HestiaClasses.js');
/***************************************************************************************/
// Variables // Variables
/***************************************************************************************/
process.debug = true;
process.pinMap = new Map(); process.pinMap = new Map();
for (const pin of config.pins) { for (const pin of config.pins) {
process.pinMap.set(pin.key, pin); process.pinMap.set(pin.key, pin);
} }
// Initialize state and comlink /***************************************************************************************/
// Initialization
/***************************************************************************************/
process.psState = new State(config); process.psState = new State(config);
const comms = new Communicator(process.psState); process.comlink = new Communicator(process.psState);
comms.init(process.psState, config); process.comlink.init(process.psState, config);
fn.gpio.init(process.comlink, process.psState);
// Initialize GPIO /***************************************************************************************/
fn.gpio.init(comms, process.psState); // Loops
/***************************************************************************************/
// Sensor detection loop // Sensor Detection
setInterval(() => { setInterval(() => {
// Iterate through pins // Iterate through pins
for (const pin of config.pins) { for (const pin of config.pins) {
// If pin is an input, read it // If pin is an input, read it
if (pin.mode === 'IN') { if (pin.mode === 'IN') {
fn.log(`I: Sensor Detection Loop: Reading pin ${pin.board}`); fn.log(`Sensor Detection Loop: Reading pin ${pin.board}`, 'DEBUG');
// Read pin // Read pin
gpio.readPin(pin.board).then(state => { gpio.readPin(pin.board).then(state => {
fn.log(`I: Sensor Detection Loop: Pin ${pin.board} is ${state}`); fn.log(`Sensor Detection Loop: Pin ${pin.board} is ${state}`, 'DEBUG');
// Convert the state from string to boolean // Convert the state from string to boolean
const boolState = state === '1' ? true : false; const boolState = state === '1' ? true : false;
// Compare the state to the last known state // Compare the state to the last known state
@ -39,28 +43,39 @@ setInterval(() => {
// Update the state // Update the state
process.psState[pin.key].on = boolState; process.psState[pin.key].on = boolState;
// Send the state to the MQTT broker // Send the state to the MQTT broker
comms.send(config.mqtt.topics[pin.key], JSON.stringify(process.psState[pin.key])); process.comlink.send(config.mqtt.topics[pin.key], JSON.stringify(process.psState[pin.key]));
fn.log(`I: Sensor Detection Loop: ${pin.key}: ${state}`); fn.log(`Sensor Detection Loop: ${pin.key}: ${state}`, 'DEBUG');
} }
}).catch(e => console.error(`E: Sensor Detection Loop: ${e}`)); }).catch(e => fn.log(`Sensor Detection Loop: ${e}`, 'ERROR'));
} }
} }
}, 1000); }, 1000);
// Auger feed loop // Auger Feed
setInterval(fn.routines.cycleAuger, config.augerTotalCycleTime); setInterval(fn.routines.cycleAuger, config.augerTotalCycleTime);
comms.on('stateChange', (oldState, state) => { /***************************************************************************************/
fn.log(`Event: State change detected.`); // Event listeners
/***************************************************************************************/
process.comlink.on('stateChange', (oldState, state) => {
fn.log(`ComLink: State change detected.`, 'INFO');
fn.handlers.stateChange(oldState, state); fn.handlers.stateChange(oldState, state);
}); });
comms.on('startup', () => { process.comlink.on('startup', () => {
fn.log(`I: Event: Startup detected.`); fn.log(`ComLink: Startup detected.`, 'INFO');
fn.power.start.init(comms, process.psState).catch(e => console.error(`E: Power Start Init: ${e}`)); fn.power.start.init(process.comlink, process.psState).catch(e => console.error(`E: Power Start Init: ${e}`));
}); });
comms.on('shutdown', () => { process.comlink.on('shutdown', () => {
fn.log(`I: Event: Shutdown detected.`); fn.log(`ComLink: Shutdown detected.`, 'INFO');
fn.power.stop.init(comms, process.psState).catch(e => console.error(`E: Power Stop Init: ${e}`)); fn.power.stop.init(process.comlink, process.psState).catch(e => console.error(`E: Power Stop Init: ${e}`));
});
process.comlink.on('announcement', msg => {
fn.log(`ComLink: ${msg}`, 'INFO');
});
process.psState.on('announcement', msg => {
fn.log(`State: ${msg}`, 'INFO');
}); });