333 lines
9.3 KiB
C++
333 lines
9.3 KiB
C++
#include "controller.h"
|
|
#include "spi_master.h"
|
|
#include "config.h"
|
|
#include "playlist.h"
|
|
|
|
Controller::Controller(Player* p, PlaylistManager* pm) {
|
|
_player = p;
|
|
_pm = pm;
|
|
_rfid = new MFRC522(17, MFRC522::UNUSED_PIN);
|
|
|
|
BTN_NEXT_SETUP();
|
|
BTN_PREV_SETUP();
|
|
BTN_VOL_UP_SETUP();
|
|
BTN_VOL_DOWN_SETUP();
|
|
|
|
SPIMaster::select_rc522();
|
|
DEBUG("Initializing RC522...\n");
|
|
_rfid->PCD_Init();
|
|
#ifdef SHOW_DEBUG
|
|
_rfid->PCD_DumpVersionToSerial();
|
|
#endif
|
|
SPIMaster::select_rc522(false);
|
|
INFO("RC522 initialized.\n");
|
|
|
|
for (uint8_t i=0; i<NUM_BUTTONS; i++) _button_last_pressed_at[i]=0;
|
|
}
|
|
|
|
void Controller::set_mqtt_client(MQTTClient* m) {
|
|
_mqtt_client = m;
|
|
}
|
|
|
|
void Controller::loop() {
|
|
unsigned long now = millis();
|
|
if ((_last_rfid_scan_at < now - RFID_SCAN_INTERVAL) || (now < _last_rfid_scan_at)) {
|
|
_check_rfid();
|
|
_last_rfid_scan_at = now;
|
|
}
|
|
_check_serial();
|
|
_check_buttons();
|
|
|
|
if ((_last_mqtt_report_at < now - MQTT_REPORT_INTERVAL) || (now < _last_mqtt_report_at)) {
|
|
_send_mqtt_report();
|
|
_last_mqtt_report_at = now;
|
|
}
|
|
}
|
|
|
|
uint32_t Controller::_get_rfid_card_uid() {
|
|
SPIMaster::select_rc522();
|
|
if (!_rfid->PICC_ReadCardSerial()) {
|
|
if (!_rfid->PICC_IsNewCardPresent()) {
|
|
return 0;
|
|
}
|
|
if (!_rfid->PICC_ReadCardSerial()) {
|
|
return 0;
|
|
}
|
|
}
|
|
SPIMaster::select_rc522(false);
|
|
uint32_t uid = _rfid->uid.uidByte[0]<<24 | _rfid->uid.uidByte[1]<<16 | _rfid->uid.uidByte[2]<<8 | _rfid->uid.uidByte[3];
|
|
return uid;
|
|
}
|
|
|
|
void Controller::_check_rfid() {
|
|
MFRC522::StatusCode status;
|
|
if (_rfid_present) {
|
|
byte buffer[2];
|
|
byte buffer_size = 2;
|
|
SPIMaster::select_rc522();
|
|
status = _rfid->PICC_WakeupA(buffer, &buffer_size);
|
|
if (status == MFRC522::STATUS_OK) {
|
|
// Card is still present.
|
|
_rfid->PICC_HaltA();
|
|
SPIMaster::select_rc522(false);
|
|
return;
|
|
}
|
|
SPIMaster::select_rc522(false);
|
|
// Card is now gone
|
|
_rfid_present = false;
|
|
INFO("No more RFID card.\n");
|
|
if (_state != LOCKED) {
|
|
_player->stop();
|
|
}
|
|
} else {
|
|
uint32_t uid = _get_rfid_card_uid();
|
|
if (uid > 0) {
|
|
_mqtt_client->publish_rfid_uid(uid);
|
|
//_no_rfid_card_count = 0;
|
|
String temp = String(uid, HEX);
|
|
String s_uid = "";
|
|
for (int i=0; i<(8-temp.length()); i++) {
|
|
s_uid.concat("0");
|
|
}
|
|
s_uid.concat(temp);
|
|
INFO("New RFID card uid: %s\n", s_uid.c_str());
|
|
_rfid_present = true;
|
|
|
|
String data = _read_rfid_data();
|
|
|
|
Playlist* pl = _pm->get_playlist_for_id(s_uid);
|
|
if (data.indexOf("[lock]") != -1) {
|
|
if (_state == LOCKED) {
|
|
_state = NORMAL;
|
|
DEBUG("ControllerState is now UNLOCKED\n");
|
|
} else {
|
|
DEBUG("ControllerState is now LOCKING\n");
|
|
_state = LOCKING;
|
|
}
|
|
}
|
|
if (pl==NULL) {
|
|
INFO("Could not find album for id '%s'.", s_uid.c_str());
|
|
return;
|
|
}
|
|
int index;
|
|
if (data.indexOf("[random]") != -1 && pl->is_fresh()) {
|
|
pl->shuffle();
|
|
} else if ((index=data.indexOf("[random:")) != -1 && pl->is_fresh()) {
|
|
String temp = data.substring(index + 8);
|
|
index = temp.indexOf("]");
|
|
TRACE("temp: %s, temp.substring(0, %d): %s\n", temp.c_str(), index, temp.substring(0, index).c_str());
|
|
if (index>0) {
|
|
uint8_t random_offset = temp.substring(0, index).toInt();
|
|
pl->shuffle(random_offset);
|
|
}
|
|
}
|
|
|
|
if (_state == LOCKED) {
|
|
DEBUG("ControllerState is LOCKED, ignoring card.\n");
|
|
return;
|
|
}
|
|
|
|
if (_state == LOCKING) {
|
|
_state = LOCKED;
|
|
DEBUG("ControllerState is now LOCKED.\n");
|
|
}
|
|
|
|
_player->play(pl);
|
|
}
|
|
}
|
|
}
|
|
|
|
String Controller::_read_rfid_data() {
|
|
static MFRC522::MIFARE_Key keys[8] = {
|
|
{{0xd3, 0xf7, 0xd3, 0xf7, 0xd3, 0xf7}}, // D3 F7 D3 F7 D3 F7
|
|
{{0xff, 0xff, 0xff, 0xff, 0xff, 0xff}}, // FF FF FF FF FF FF = factory default
|
|
{{0xa0, 0xa1, 0xa2, 0xa3, 0xa4, 0xa5}}, // A0 A1 A2 A3 A4 A5
|
|
{{0xb0, 0xb1, 0xb2, 0xb3, 0xb4, 0xb5}}, // B0 B1 B2 B3 B4 B5
|
|
{{0x4d, 0x3a, 0x99, 0xc3, 0x51, 0xdd}}, // 4D 3A 99 C3 51 DD
|
|
{{0x1a, 0x98, 0x2c, 0x7e, 0x45, 0x9a}}, // 1A 98 2C 7E 45 9A
|
|
{{0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff}}, // AA BB CC DD EE FF
|
|
{{0x00, 0x00, 0x00, 0x00, 0x00, 0x00}} // 00 00 00 00 00 00
|
|
};
|
|
SPIMaster::select_rc522();
|
|
DEBUG("Trying to read RFID data...\n");
|
|
|
|
String data = "";
|
|
MFRC522::PICC_Type type = _rfid->PICC_GetType(_rfid->uid.sak);
|
|
|
|
uint8_t sectors = 0;
|
|
switch(type) {
|
|
case MFRC522::PICC_TYPE_MIFARE_MINI: sectors = 5; break;
|
|
case MFRC522::PICC_TYPE_MIFARE_1K: sectors = 16; break;
|
|
case MFRC522::PICC_TYPE_MIFARE_4K: sectors = 40; break;
|
|
default: INFO("Unknown PICC type %s\n", String(MFRC522::PICC_GetTypeName(type)).c_str());
|
|
}
|
|
|
|
int good_key_index = -1;
|
|
for (uint8_t sector=1; sector<sectors; sector++) {
|
|
uint8_t blocks = (sector < 32) ? 4 : 16;
|
|
uint8_t block_offset = (sector < 32) ? sector * 4 : 128 + (sector - 32) * 16;
|
|
|
|
MFRC522::StatusCode status;
|
|
|
|
for (int i=0; i<8; i++) {
|
|
MFRC522::MIFARE_Key *k = &keys[i];
|
|
TRACE("Trying MIFARE key %02X %02X %02X %02X %02X %02X...\n", k->keyByte[0], k->keyByte[1], k->keyByte[2], k->keyByte[3], k->keyByte[4], k->keyByte[5]);
|
|
status = _rfid->PCD_Authenticate(MFRC522::PICC_CMD_MF_AUTH_KEY_A, block_offset, k, &_rfid->uid);
|
|
if (status == MFRC522::STATUS_OK) {
|
|
TRACE("Authentication succeeded with key #%d\n", i);
|
|
good_key_index = i;
|
|
break;
|
|
}
|
|
}
|
|
if (good_key_index == -1) {
|
|
TRACE("Could not find a valid MIFARE key.\n");
|
|
} else {
|
|
for (uint8_t block=0; block<blocks-1; block++) {
|
|
byte buffer[18];
|
|
uint8_t byte_count = 18;
|
|
status = _rfid->MIFARE_Read(block_offset + block, buffer, &byte_count);
|
|
if (status != MFRC522::STATUS_OK) {
|
|
DEBUG("MIFARE_Read() failed: %s\n", String(_rfid->GetStatusCodeName(status)).c_str());
|
|
continue;
|
|
}
|
|
for (int i=0; i<16; i++) {
|
|
if (buffer[i]>=0x20 && buffer[i]<0x7F) data.concat((char)buffer[i]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
_rfid->PICC_HaltA();
|
|
_rfid->PCD_StopCrypto1();
|
|
DEBUG("Data from RFID: %s\n", data.c_str());
|
|
SPIMaster::select_rc522(false);
|
|
return data;
|
|
}
|
|
|
|
void Controller::_check_serial() {
|
|
if (Serial.available() > 0) {
|
|
char c = Serial.read();
|
|
Serial.printf("%c", c);
|
|
if (c==10 || c==13) {
|
|
if (_serial_buffer.length()>0) {
|
|
_execute_serial_command(_serial_buffer);
|
|
_serial_buffer = String();
|
|
}
|
|
} else {
|
|
_serial_buffer.concat(c);
|
|
}
|
|
}
|
|
}
|
|
|
|
void Controller::_execute_serial_command(String cmd) {
|
|
DEBUG("Executing command: %s\n", cmd.c_str());
|
|
|
|
if (cmd.startsWith("play ")) {
|
|
Playlist* p = _pm->get_playlist_for_id(cmd.substring(5));
|
|
_player->play(p);
|
|
//} else if (cmd.equals("ls")) {
|
|
// _execute_command_ls("/");
|
|
//} else if (cmd.startsWith("ls ")) {
|
|
// _execute_command_ls(cmd.substring(3));
|
|
//} else if (cmd.equals("play")) {
|
|
// _player->play_random_album();
|
|
} else if (cmd.equals("stop")) {
|
|
_player->stop();
|
|
} else if (cmd.equals("help")) {
|
|
_execute_command_help();
|
|
} else if (cmd.equals("-")) {
|
|
_player->vol_down();
|
|
} else if (cmd.equals("+")) {
|
|
_player->vol_up();
|
|
} else if (cmd.equals("p")) {
|
|
_player->track_prev();
|
|
} else if (cmd.equals("n")) {
|
|
_player->track_next();
|
|
} else if (cmd.equals("ids")) {
|
|
_pm->dump_ids();
|
|
} else {
|
|
ERROR("Unknown command: %s\n", cmd.c_str());
|
|
}
|
|
return;
|
|
}
|
|
|
|
void Controller::_execute_command_ls(String path) {
|
|
INFO("Listing contents of %s:\n", path.c_str());
|
|
// TODO
|
|
//std::list<String> files = _player->ls(path);
|
|
//for(std::list<String>::iterator it=files.begin(); it!=files.end(); ++it) {
|
|
// INFO(" %s\n", (*it).c_str());
|
|
//}
|
|
}
|
|
|
|
void Controller::_execute_command_help() {
|
|
INFO("Valid commands are:");
|
|
INFO(" help - Displays this help\n");
|
|
//INFO(" ls [dir] - Lists the contents of [dir] or, if not given, of /\n");
|
|
INFO(" ids - Lists all known ID-to-folder mappings\n");
|
|
INFO(" play [id] - Plays the album with the given id\n");
|
|
INFO(" stop - Stops playback\n");
|
|
INFO(" - / + - Decrease or increase the volume\n");
|
|
INFO(" p / n - Previous or next track\n");
|
|
}
|
|
|
|
void Controller::_check_buttons() {
|
|
if (BTN_PREV() && _debounce_button(0)) {
|
|
if (_state == NORMAL) {
|
|
_player->track_prev();
|
|
} else {
|
|
DEBUG("Ignoring btn_prev because state is LOCKED.\n");
|
|
}
|
|
} else if (BTN_VOL_UP() && _debounce_button(1)) {
|
|
_player->vol_up();
|
|
} else if (BTN_VOL_DOWN() && _debounce_button(2)) {
|
|
_player->vol_down();
|
|
} else if (BTN_NEXT() && _debounce_button(3)) {
|
|
if (_state == NORMAL) {
|
|
_player->track_next();
|
|
} else {
|
|
DEBUG("Ignoring btn_next because state is LOCKED.\n");
|
|
}
|
|
}
|
|
}
|
|
|
|
bool Controller::_debounce_button(uint8_t index) {
|
|
bool ret = false;
|
|
if (_button_last_pressed_at[index] + DEBOUNCE_MILLIS < millis()) {
|
|
DEBUG("Button %d pressed.\n", index);
|
|
ret = true;
|
|
}
|
|
_button_last_pressed_at[index] = millis();
|
|
return ret;
|
|
}
|
|
String Controller::get_status_json() {
|
|
String response = String("{");
|
|
response.concat("\"state\": \"");
|
|
response.concat(_player->is_playing() ? "playing" : "idle");
|
|
response.concat("\", ");
|
|
if (_player->is_playing()) {
|
|
// TODO
|
|
//response.concat("\"album\": \"");
|
|
//response.concat(_player->album());
|
|
//response.concat("\", \"track\": ");
|
|
//response.concat(_player->track());
|
|
//response.concat(", \"position\": ");
|
|
//response.concat(_player->position());
|
|
//response.concat(", ");
|
|
}
|
|
response.concat("\"volume\": ");
|
|
response.concat(_player->volume());
|
|
response.concat(", \"volume_max\": ");
|
|
response.concat(VOLUME_MAX);
|
|
response.concat(", \"volume_min\": ");
|
|
response.concat(VOLUME_MIN);
|
|
response.concat(", \"rfid_uid\": ");
|
|
response.concat(String(_last_rfid_card_uid, HEX));
|
|
response.concat("}\n");
|
|
return response;
|
|
}
|
|
|
|
void Controller::_send_mqtt_report() {
|
|
|
|
}
|