AI Companion Head

// PC Mirror Companion — ESP32 body firmware
// GENERATE: the PC's AI sends state/text/audio commands below.
// MODIFY: edit face and animation drawing in companion_body.cpp.
// SETTINGS: edit Wi-Fi and audio values in companion_settings.h.
#include <Arduino.h>
#include <WiFi.h>
#include <WebServer.h>
#include "driver/i2s.h"
#include "companion_settings.h"
#include "companion_body.h"


// Forward declarations
void startSpeaker();
void sendJson(const String &json);
void handleStatus();
void handleSet();
void handleSpeak();
void handleRoot();

constexpr int AMP_BCLK = 26;
constexpr int AMP_LRCLK = 25;
constexpr int AMP_DIN = 27;
WebServer server(BODY_HTTP_PORT);

void startSpeaker() {
  i2s_config_t config = {};
  config.mode = (i2s_mode_t)(I2S_MODE_MASTER | I2S_MODE_TX);
  config.sample_rate = PCM_SAMPLE_RATE;
  config.bits_per_sample = I2S_BITS_PER_SAMPLE_16BIT;
  config.channel_format = I2S_CHANNEL_FMT_RIGHT_LEFT;
  config.communication_format = I2S_COMM_FORMAT_I2S;
  config.intr_alloc_flags = ESP_INTR_FLAG_LEVEL1;
  config.dma_buf_count = 4;
  config.dma_buf_len = 128;
  config.tx_desc_auto_clear = true;
  i2s_driver_install(I2S_NUM_1, &config, 0, nullptr);
  i2s_pin_config_t pins = {};
  pins.bck_io_num = AMP_BCLK;
  pins.ws_io_num = AMP_LRCLK;
  pins.data_out_num = AMP_DIN;
  pins.data_in_num = I2S_PIN_NO_CHANGE;
  i2s_set_pin(I2S_NUM_1, &pins);
  i2s_zero_dma_buffer(I2S_NUM_1);
}

void sendJson(const String &json) {
  server.sendHeader("Access-Control-Allow-Origin", "*");
  server.send(200, "application/json", json);
}
void handleStatus() {
  String json = "{\"state\":\"" + bodyStateName() + "\",\"text\":\"" + bodyText() +
                "\",\"ip\":\"" + WiFi.softAPIP().toString() + "\"}";
  sendJson(json);
}
void handleSet() {
  if (!server.hasArg("state") && !server.hasArg("text")) {
    server.send(400, "text/plain", "Use state and/or text query values.");
    return;
  }
  if (server.hasArg("state")) bodySetState(server.arg("state"));
  if (server.hasArg("text")) bodySetText(server.arg("text"));
  if (server.hasArg("contrast")) bodySetContrast((uint8_t)constrain(server.arg("contrast").toInt(), 0, 255));
  handleStatus();
}
void handleSpeak() {
  // PC TTS sends raw signed 16-bit, 16 kHz, mono PCM in the POST body.
  if (server.method() != HTTP_POST) {
    server.send(405, "text/plain", "POST raw 16-bit mono PCM data.");
    return;
  }
  bodySetState("talking");
  const String &raw = server.arg("plain");
  const uint8_t *bytes = (const uint8_t *)raw.c_str();
  for (size_t offset = 0; offset + 1 < raw.length(); offset += 2) {
    int16_t sample = (int16_t)((uint16_t)bytes[offset] | ((uint16_t)bytes[offset + 1] << 8));
    int16_t stereo[2] = {sample, sample};
    size_t written;
    i2s_write(I2S_NUM_1, stereo, sizeof(stereo), &written, portMAX_DELAY);
  }
  bodySetState("idle");
  sendJson("{\"ok\":true}");
}
void handleRoot() {
  server.send(200, "text/html", "<h2>PC Mirror Companion</h2><p>PC control endpoint ready.</p><p>GET /set?state=listening&text=Hello</p><p>GET /status</p>");
}

void setup() {
  Serial.begin(115200);
  bodyBegin();
  startSpeaker();
  WiFi.mode(WIFI_AP);
  WiFi.softAP(BODY_AP_NAME, BODY_AP_PASSWORD);
  bodySetText("Connect PC to Wi-Fi");
  server.on("/", HTTP_GET, handleRoot);
  server.on("/status", HTTP_GET, handleStatus);
  server.on("/set", HTTP_GET, handleSet);
  server.on("/speak", HTTP_POST, handleSpeak);
  server.begin();
}
void loop() {
  server.handleClient();
  bodyTick();
}