Merge pull request #115 from IRONICBo/feat/add-feishu-integration
This commit is contained in:
13
.gitignore
vendored
13
.gitignore
vendored
@@ -35,6 +35,18 @@ main/mimi_secrets.h
|
||||
*.pyc
|
||||
*.bin
|
||||
|
||||
# Documentation (generated)
|
||||
BUILD_TEST_GUIDE.md
|
||||
COMPILE_TEST_REPORT.md
|
||||
CURRENT_STATUS.md
|
||||
FEISHU_INTEGRATION_SUMMARY.md
|
||||
QUICK_START.md
|
||||
QWEN_INTEGRATION_PLAN.md
|
||||
QWEN_INTEGRATION_COMPLETE.md
|
||||
build_test.sh
|
||||
build_and_integrate_qwen.sh
|
||||
verify_integration.sh
|
||||
|
||||
# Reference repos
|
||||
nanobot/
|
||||
|
||||
@@ -46,3 +58,4 @@ nanobot/
|
||||
.DS_Store
|
||||
Thumbs.db
|
||||
references/
|
||||
.venv/
|
||||
|
||||
@@ -3,7 +3,8 @@ idf_component_register(
|
||||
"mimi.c"
|
||||
"bus/message_bus.c"
|
||||
"wifi/wifi_manager.c"
|
||||
"telegram/telegram_bot.c"
|
||||
"channels/telegram/telegram_bot.c"
|
||||
"channels/feishu/feishu_bot.c"
|
||||
"llm/llm_proxy.c"
|
||||
"agent/agent_loop.c"
|
||||
"agent/context_builder.c"
|
||||
@@ -25,5 +26,5 @@ idf_component_register(
|
||||
REQUIRES
|
||||
nvs_flash esp_wifi esp_netif esp_http_client esp_http_server
|
||||
esp_https_ota esp_event json spiffs console vfs app_update esp-tls
|
||||
esp_timer
|
||||
esp_timer esp_websocket_client
|
||||
)
|
||||
|
||||
@@ -6,6 +6,7 @@
|
||||
|
||||
/* Channel identifiers */
|
||||
#define MIMI_CHAN_TELEGRAM "telegram"
|
||||
#define MIMI_CHAN_FEISHU "feishu"
|
||||
#define MIMI_CHAN_WEBSOCKET "websocket"
|
||||
#define MIMI_CHAN_CLI "cli"
|
||||
#define MIMI_CHAN_SYSTEM "system"
|
||||
@@ -13,7 +14,7 @@
|
||||
/* Message types on the bus */
|
||||
typedef struct {
|
||||
char channel[16]; /* "telegram", "websocket", "cli" */
|
||||
char chat_id[32]; /* Telegram chat_id or WS client id */
|
||||
char chat_id[96]; /* Telegram/Feishu chat_id, open_id, or WS client id */
|
||||
char *content; /* Heap-allocated message text (caller must free) */
|
||||
} mimi_msg_t;
|
||||
|
||||
|
||||
85
main/channels/feishu/README.md
Normal file
85
main/channels/feishu/README.md
Normal file
@@ -0,0 +1,85 @@
|
||||
# Feishu/Lark Bot Integration
|
||||
|
||||
This directory contains the Feishu bot integration for MimiClaw.
|
||||
|
||||
## Features
|
||||
|
||||
- Send text messages to Feishu chats
|
||||
- Receive messages via webhook (HTTP event subscription)
|
||||
- Automatic message chunking (4096 chars per message)
|
||||
- Tenant access token management with auto-refresh
|
||||
- Message deduplication
|
||||
- Reply to specific messages
|
||||
- Support for both DM (p2p) and group chats
|
||||
|
||||
## Configuration
|
||||
|
||||
### Option 1: Build-time Configuration
|
||||
|
||||
1. Copy the secrets template:
|
||||
```bash
|
||||
cp main/mimi_secrets.h.example main/mimi_secrets.h
|
||||
```
|
||||
|
||||
2. Edit `main/mimi_secrets.h`:
|
||||
```c
|
||||
#define MIMI_SECRET_FEISHU_APP_ID "cli_xxxxxxxxxxxxxx"
|
||||
#define MIMI_SECRET_FEISHU_APP_SECRET "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
||||
```
|
||||
|
||||
3. Rebuild:
|
||||
```bash
|
||||
idf.py fullclean && idf.py build
|
||||
```
|
||||
|
||||
### Option 2: Runtime Configuration (CLI)
|
||||
|
||||
```
|
||||
mimi> set_feishu_creds cli_xxxxxxxxxxxxxx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
|
||||
```
|
||||
|
||||
## Feishu App Setup
|
||||
|
||||
1. Go to [Feishu Open Platform](https://open.feishu.cn/)
|
||||
2. Create an app and get **App ID** / **App Secret**
|
||||
3. Enable permissions:
|
||||
- `im:message` - Send and receive messages
|
||||
- `im:message:send_as_bot` - Send messages as bot
|
||||
4. Configure Event Subscription:
|
||||
- Request URL: `http://<ESP32_IP>:18790/feishu/events`
|
||||
- Subscribe to: `im.message.receive_v1`
|
||||
5. The ESP32 will auto-respond to the URL verification challenge
|
||||
|
||||
## Architecture
|
||||
|
||||
```
|
||||
Feishu Server
|
||||
|
|
||||
v (HTTP POST /feishu/events)
|
||||
[ESP32 Webhook Server :18790]
|
||||
|
|
||||
v (message_bus_push_inbound)
|
||||
[Message Bus] --> [Agent Loop] --> [Message Bus]
|
||||
| |
|
||||
v (outbound dispatch) |
|
||||
[feishu_send_message] <-----------------+
|
||||
|
|
||||
v (POST /im/v1/messages)
|
||||
Feishu API
|
||||
```
|
||||
|
||||
## API Reference
|
||||
|
||||
| Function | Description |
|
||||
|----------|-------------|
|
||||
| `feishu_bot_init()` | Load credentials from NVS/build-time |
|
||||
| `feishu_bot_start()` | Start webhook HTTP server |
|
||||
| `feishu_send_message(chat_id, text)` | Send text message |
|
||||
| `feishu_reply_message(message_id, text)` | Reply to a specific message |
|
||||
| `feishu_set_credentials(app_id, secret)` | Save credentials to NVS |
|
||||
|
||||
## References
|
||||
|
||||
- [Feishu Open Platform Docs](https://open.feishu.cn/document/home/index)
|
||||
- [Message API](https://open.feishu.cn/document/server-docs/im-v1/message/create)
|
||||
- [Event Subscription](https://open.feishu.cn/document/server-docs/event-subscription/event-subscription-guide)
|
||||
991
main/channels/feishu/feishu_bot.c
Normal file
991
main/channels/feishu/feishu_bot.c
Normal file
@@ -0,0 +1,991 @@
|
||||
#include "feishu_bot.h"
|
||||
#include "mimi_config.h"
|
||||
#include "bus/message_bus.h"
|
||||
#include "proxy/http_proxy.h"
|
||||
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdbool.h>
|
||||
#include "freertos/FreeRTOS.h"
|
||||
#include "freertos/task.h"
|
||||
#include "esp_log.h"
|
||||
#include "esp_http_client.h"
|
||||
#include "esp_websocket_client.h"
|
||||
#include "esp_crt_bundle.h"
|
||||
#include "esp_timer.h"
|
||||
#include "esp_event.h"
|
||||
#include "nvs.h"
|
||||
#include "cJSON.h"
|
||||
|
||||
static const char *TAG = "feishu";
|
||||
|
||||
/* ── Feishu API endpoints ──────────────────────────────────── */
|
||||
#define FEISHU_API_BASE "https://open.feishu.cn/open-apis"
|
||||
#define FEISHU_AUTH_URL FEISHU_API_BASE "/auth/v3/tenant_access_token/internal"
|
||||
#define FEISHU_SEND_MSG_URL FEISHU_API_BASE "/im/v1/messages"
|
||||
#define FEISHU_REPLY_MSG_URL FEISHU_API_BASE "/im/v1/messages/%s/reply"
|
||||
#define FEISHU_WS_CONFIG_URL "https://open.feishu.cn/callback/ws/endpoint"
|
||||
|
||||
/* ── Credentials & token state ─────────────────────────────── */
|
||||
static char s_app_id[64] = MIMI_SECRET_FEISHU_APP_ID;
|
||||
static char s_app_secret[128] = MIMI_SECRET_FEISHU_APP_SECRET;
|
||||
static char s_tenant_token[512] = {0};
|
||||
static int64_t s_token_expire_time = 0;
|
||||
|
||||
/* ── Feishu WebSocket state ────────────────────────────────── */
|
||||
static esp_websocket_client_handle_t s_ws_client = NULL;
|
||||
static TaskHandle_t s_ws_task = NULL;
|
||||
static char s_ws_url[512] = {0};
|
||||
static int s_ws_ping_interval_ms = 120000;
|
||||
static int s_ws_reconnect_interval_ms = 30000;
|
||||
static int s_ws_reconnect_nonce_ms = 30000;
|
||||
static int s_ws_service_id = 0;
|
||||
static bool s_ws_connected = false;
|
||||
|
||||
static void handle_message_event(cJSON *event);
|
||||
|
||||
/* ── Message deduplication ─────────────────────────────────── */
|
||||
#define FEISHU_DEDUP_CACHE_SIZE 64
|
||||
|
||||
static uint64_t s_seen_msg_keys[FEISHU_DEDUP_CACHE_SIZE] = {0};
|
||||
static size_t s_seen_msg_idx = 0;
|
||||
|
||||
static uint64_t fnv1a64(const char *s)
|
||||
{
|
||||
uint64_t h = 1469598103934665603ULL;
|
||||
if (!s) return h;
|
||||
while (*s) {
|
||||
h ^= (unsigned char)(*s++);
|
||||
h *= 1099511628211ULL;
|
||||
}
|
||||
return h;
|
||||
}
|
||||
|
||||
static bool dedup_check_and_record(const char *message_id)
|
||||
{
|
||||
uint64_t key = fnv1a64(message_id);
|
||||
for (size_t i = 0; i < FEISHU_DEDUP_CACHE_SIZE; i++) {
|
||||
if (s_seen_msg_keys[i] == key) return true;
|
||||
}
|
||||
s_seen_msg_keys[s_seen_msg_idx] = key;
|
||||
s_seen_msg_idx = (s_seen_msg_idx + 1) % FEISHU_DEDUP_CACHE_SIZE;
|
||||
return false;
|
||||
}
|
||||
|
||||
/* ── HTTP response accumulator ─────────────────────────────── */
|
||||
typedef struct {
|
||||
char *buf;
|
||||
size_t len;
|
||||
size_t cap;
|
||||
} http_resp_t;
|
||||
|
||||
static esp_err_t http_event_handler(esp_http_client_event_t *evt)
|
||||
{
|
||||
http_resp_t *resp = (http_resp_t *)evt->user_data;
|
||||
if (evt->event_id == HTTP_EVENT_ON_DATA) {
|
||||
if (resp->len + evt->data_len >= resp->cap) {
|
||||
size_t new_cap = resp->cap * 2;
|
||||
if (new_cap < resp->len + evt->data_len + 1) {
|
||||
new_cap = resp->len + evt->data_len + 1;
|
||||
}
|
||||
char *tmp = realloc(resp->buf, new_cap);
|
||||
if (!tmp) return ESP_ERR_NO_MEM;
|
||||
resp->buf = tmp;
|
||||
resp->cap = new_cap;
|
||||
}
|
||||
memcpy(resp->buf + resp->len, evt->data, evt->data_len);
|
||||
resp->len += evt->data_len;
|
||||
resp->buf[resp->len] = '\0';
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
/* ── Feishu WS frame (protobuf) ────────────────────────────── */
|
||||
typedef struct {
|
||||
char key[32];
|
||||
char value[128];
|
||||
} ws_header_t;
|
||||
|
||||
typedef struct {
|
||||
uint64_t seq_id;
|
||||
uint64_t log_id;
|
||||
int32_t service;
|
||||
int32_t method;
|
||||
ws_header_t headers[16];
|
||||
size_t header_count;
|
||||
const uint8_t *payload;
|
||||
size_t payload_len;
|
||||
} ws_frame_t;
|
||||
|
||||
static bool pb_read_varint(const uint8_t *buf, size_t len, size_t *pos, uint64_t *out)
|
||||
{
|
||||
uint64_t v = 0;
|
||||
int shift = 0;
|
||||
while (*pos < len && shift <= 63) {
|
||||
uint8_t b = buf[(*pos)++];
|
||||
v |= ((uint64_t)(b & 0x7F)) << shift;
|
||||
if ((b & 0x80) == 0) {
|
||||
*out = v;
|
||||
return true;
|
||||
}
|
||||
shift += 7;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool pb_skip_field(const uint8_t *buf, size_t len, size_t *pos, uint8_t wire_type)
|
||||
{
|
||||
uint64_t n = 0;
|
||||
switch (wire_type) {
|
||||
case 0:
|
||||
return pb_read_varint(buf, len, pos, &n);
|
||||
case 1:
|
||||
if (*pos + 8 > len) return false;
|
||||
*pos += 8;
|
||||
return true;
|
||||
case 2:
|
||||
if (!pb_read_varint(buf, len, pos, &n)) return false;
|
||||
if (*pos + (size_t)n > len) return false;
|
||||
*pos += (size_t)n;
|
||||
return true;
|
||||
case 5:
|
||||
if (*pos + 4 > len) return false;
|
||||
*pos += 4;
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
static bool pb_parse_header_msg(const uint8_t *buf, size_t len, ws_header_t *h)
|
||||
{
|
||||
memset(h, 0, sizeof(*h));
|
||||
size_t pos = 0;
|
||||
while (pos < len) {
|
||||
uint64_t tag = 0, slen = 0;
|
||||
if (!pb_read_varint(buf, len, &pos, &tag)) return false;
|
||||
uint32_t field = (uint32_t)(tag >> 3);
|
||||
uint8_t wt = (uint8_t)(tag & 0x07);
|
||||
if (wt != 2) {
|
||||
if (!pb_skip_field(buf, len, &pos, wt)) return false;
|
||||
continue;
|
||||
}
|
||||
if (!pb_read_varint(buf, len, &pos, &slen)) return false;
|
||||
if (pos + (size_t)slen > len) return false;
|
||||
if (field == 1) {
|
||||
size_t n = (slen < sizeof(h->key) - 1) ? (size_t)slen : sizeof(h->key) - 1;
|
||||
memcpy(h->key, buf + pos, n);
|
||||
h->key[n] = '\0';
|
||||
} else if (field == 2) {
|
||||
size_t n = (slen < sizeof(h->value) - 1) ? (size_t)slen : sizeof(h->value) - 1;
|
||||
memcpy(h->value, buf + pos, n);
|
||||
h->value[n] = '\0';
|
||||
}
|
||||
pos += (size_t)slen;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool pb_parse_frame(const uint8_t *buf, size_t len, ws_frame_t *f)
|
||||
{
|
||||
memset(f, 0, sizeof(*f));
|
||||
size_t pos = 0;
|
||||
while (pos < len) {
|
||||
uint64_t tag = 0, v = 0, blen = 0;
|
||||
if (!pb_read_varint(buf, len, &pos, &tag)) return false;
|
||||
uint32_t field = (uint32_t)(tag >> 3);
|
||||
uint8_t wt = (uint8_t)(tag & 0x07);
|
||||
if (field == 1 && wt == 0) {
|
||||
if (!pb_read_varint(buf, len, &pos, &f->seq_id)) return false;
|
||||
} else if (field == 2 && wt == 0) {
|
||||
if (!pb_read_varint(buf, len, &pos, &f->log_id)) return false;
|
||||
} else if (field == 3 && wt == 0) {
|
||||
if (!pb_read_varint(buf, len, &pos, &v)) return false;
|
||||
f->service = (int32_t)v;
|
||||
} else if (field == 4 && wt == 0) {
|
||||
if (!pb_read_varint(buf, len, &pos, &v)) return false;
|
||||
f->method = (int32_t)v;
|
||||
} else if (field == 5 && wt == 2) {
|
||||
if (!pb_read_varint(buf, len, &pos, &blen)) return false;
|
||||
if (pos + (size_t)blen > len) return false;
|
||||
if (f->header_count < 16) {
|
||||
pb_parse_header_msg(buf + pos, (size_t)blen, &f->headers[f->header_count++]);
|
||||
}
|
||||
pos += (size_t)blen;
|
||||
} else if (field == 8 && wt == 2) {
|
||||
if (!pb_read_varint(buf, len, &pos, &blen)) return false;
|
||||
if (pos + (size_t)blen > len) return false;
|
||||
f->payload = buf + pos;
|
||||
f->payload_len = (size_t)blen;
|
||||
pos += (size_t)blen;
|
||||
} else {
|
||||
if (!pb_skip_field(buf, len, &pos, wt)) return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
static const char *frame_header_value(const ws_frame_t *f, const char *key)
|
||||
{
|
||||
for (size_t i = 0; i < f->header_count; i++) {
|
||||
if (strcmp(f->headers[i].key, key) == 0) {
|
||||
return f->headers[i].value;
|
||||
}
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
static bool pb_write_varint(uint8_t *buf, size_t cap, size_t *pos, uint64_t value)
|
||||
{
|
||||
do {
|
||||
if (*pos >= cap) return false;
|
||||
uint8_t byte = (uint8_t)(value & 0x7F);
|
||||
value >>= 7;
|
||||
if (value) byte |= 0x80;
|
||||
buf[(*pos)++] = byte;
|
||||
} while (value);
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool pb_write_tag(uint8_t *buf, size_t cap, size_t *pos, uint32_t field, uint8_t wt)
|
||||
{
|
||||
return pb_write_varint(buf, cap, pos, ((uint64_t)field << 3) | wt);
|
||||
}
|
||||
|
||||
static bool pb_write_bytes(uint8_t *buf, size_t cap, size_t *pos, uint32_t field, const uint8_t *data, size_t len)
|
||||
{
|
||||
if (!pb_write_tag(buf, cap, pos, field, 2)) return false;
|
||||
if (!pb_write_varint(buf, cap, pos, len)) return false;
|
||||
if (*pos + len > cap) return false;
|
||||
memcpy(buf + *pos, data, len);
|
||||
*pos += len;
|
||||
return true;
|
||||
}
|
||||
|
||||
static bool pb_write_string(uint8_t *buf, size_t cap, size_t *pos, uint32_t field, const char *s)
|
||||
{
|
||||
return pb_write_bytes(buf, cap, pos, field, (const uint8_t *)s, strlen(s));
|
||||
}
|
||||
|
||||
static bool ws_encode_header(uint8_t *dst, size_t cap, size_t *out_len, const char *key, const char *value)
|
||||
{
|
||||
size_t pos = 0;
|
||||
if (!pb_write_string(dst, cap, &pos, 1, key)) return false;
|
||||
if (!pb_write_string(dst, cap, &pos, 2, value)) return false;
|
||||
*out_len = pos;
|
||||
return true;
|
||||
}
|
||||
|
||||
static int ws_send_frame(const ws_frame_t *f, const uint8_t *payload, size_t payload_len, int timeout_ms)
|
||||
{
|
||||
uint8_t out[2048];
|
||||
size_t pos = 0;
|
||||
if (!pb_write_tag(out, sizeof(out), &pos, 1, 0) || !pb_write_varint(out, sizeof(out), &pos, f->seq_id)) return -1;
|
||||
if (!pb_write_tag(out, sizeof(out), &pos, 2, 0) || !pb_write_varint(out, sizeof(out), &pos, f->log_id)) return -1;
|
||||
if (!pb_write_tag(out, sizeof(out), &pos, 3, 0) || !pb_write_varint(out, sizeof(out), &pos, (uint32_t)f->service)) return -1;
|
||||
if (!pb_write_tag(out, sizeof(out), &pos, 4, 0) || !pb_write_varint(out, sizeof(out), &pos, (uint32_t)f->method)) return -1;
|
||||
|
||||
for (size_t i = 0; i < f->header_count; i++) {
|
||||
uint8_t hb[256];
|
||||
size_t hlen = 0;
|
||||
if (!ws_encode_header(hb, sizeof(hb), &hlen, f->headers[i].key, f->headers[i].value)) return -1;
|
||||
if (!pb_write_bytes(out, sizeof(out), &pos, 5, hb, hlen)) return -1;
|
||||
}
|
||||
if (payload && payload_len > 0) {
|
||||
if (!pb_write_bytes(out, sizeof(out), &pos, 8, payload, payload_len)) return -1;
|
||||
}
|
||||
return esp_websocket_client_send_bin(s_ws_client, (const char *)out, pos, timeout_ms);
|
||||
}
|
||||
|
||||
/* ── Get / refresh tenant access token ─────────────────────── */
|
||||
static esp_err_t feishu_get_tenant_token(void)
|
||||
{
|
||||
if (s_app_id[0] == '\0' || s_app_secret[0] == '\0') {
|
||||
ESP_LOGW(TAG, "No Feishu credentials configured");
|
||||
return ESP_ERR_INVALID_STATE;
|
||||
}
|
||||
|
||||
int64_t now = esp_timer_get_time() / 1000000LL;
|
||||
if (s_tenant_token[0] != '\0' && s_token_expire_time > now + 300) {
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
cJSON *body = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(body, "app_id", s_app_id);
|
||||
cJSON_AddStringToObject(body, "app_secret", s_app_secret);
|
||||
char *json_str = cJSON_PrintUnformatted(body);
|
||||
cJSON_Delete(body);
|
||||
if (!json_str) return ESP_ERR_NO_MEM;
|
||||
|
||||
http_resp_t resp = { .buf = calloc(1, 2048), .len = 0, .cap = 2048 };
|
||||
if (!resp.buf) { free(json_str); return ESP_ERR_NO_MEM; }
|
||||
|
||||
esp_http_client_config_t config = {
|
||||
.url = FEISHU_AUTH_URL,
|
||||
.event_handler = http_event_handler,
|
||||
.user_data = &resp,
|
||||
.timeout_ms = 10000,
|
||||
.buffer_size = 2048,
|
||||
.buffer_size_tx = 2048,
|
||||
.crt_bundle_attach = esp_crt_bundle_attach,
|
||||
};
|
||||
|
||||
esp_http_client_handle_t client = esp_http_client_init(&config);
|
||||
if (!client) { free(json_str); free(resp.buf); return ESP_FAIL; }
|
||||
|
||||
esp_http_client_set_method(client, HTTP_METHOD_POST);
|
||||
esp_http_client_set_header(client, "Content-Type", "application/json");
|
||||
esp_http_client_set_post_field(client, json_str, strlen(json_str));
|
||||
|
||||
esp_err_t err = esp_http_client_perform(client);
|
||||
esp_http_client_cleanup(client);
|
||||
free(json_str);
|
||||
|
||||
if (err != ESP_OK) {
|
||||
ESP_LOGE(TAG, "Token request HTTP failed: %s", esp_err_to_name(err));
|
||||
free(resp.buf);
|
||||
return err;
|
||||
}
|
||||
|
||||
cJSON *root = cJSON_Parse(resp.buf);
|
||||
free(resp.buf);
|
||||
if (!root) { ESP_LOGE(TAG, "Failed to parse token response"); return ESP_FAIL; }
|
||||
|
||||
cJSON *code = cJSON_GetObjectItem(root, "code");
|
||||
if (!code || code->valueint != 0) {
|
||||
ESP_LOGE(TAG, "Token request failed: code=%d", code ? code->valueint : -1);
|
||||
cJSON_Delete(root);
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
cJSON *token = cJSON_GetObjectItem(root, "tenant_access_token");
|
||||
cJSON *expire = cJSON_GetObjectItem(root, "expire");
|
||||
|
||||
if (token && cJSON_IsString(token)) {
|
||||
strncpy(s_tenant_token, token->valuestring, sizeof(s_tenant_token) - 1);
|
||||
s_token_expire_time = now + (expire ? expire->valueint : 7200) - 300;
|
||||
ESP_LOGI(TAG, "Got tenant access token (expires in %ds)",
|
||||
expire ? expire->valueint : 7200);
|
||||
}
|
||||
|
||||
cJSON_Delete(root);
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
/* ── Feishu API call helper ────────────────────────────────── */
|
||||
static char *feishu_api_call(const char *url, const char *method, const char *post_data)
|
||||
{
|
||||
if (feishu_get_tenant_token() != ESP_OK) return NULL;
|
||||
|
||||
http_resp_t resp = { .buf = calloc(1, 4096), .len = 0, .cap = 4096 };
|
||||
if (!resp.buf) return NULL;
|
||||
|
||||
esp_http_client_config_t config = {
|
||||
.url = url,
|
||||
.event_handler = http_event_handler,
|
||||
.user_data = &resp,
|
||||
.timeout_ms = 15000,
|
||||
.buffer_size = 2048,
|
||||
.buffer_size_tx = 2048,
|
||||
.crt_bundle_attach = esp_crt_bundle_attach,
|
||||
};
|
||||
|
||||
esp_http_client_handle_t client = esp_http_client_init(&config);
|
||||
if (!client) { free(resp.buf); return NULL; }
|
||||
|
||||
char auth_header[600];
|
||||
snprintf(auth_header, sizeof(auth_header), "Bearer %s", s_tenant_token);
|
||||
esp_http_client_set_header(client, "Authorization", auth_header);
|
||||
esp_http_client_set_header(client, "Content-Type", "application/json; charset=utf-8");
|
||||
|
||||
if (strcmp(method, "POST") == 0) {
|
||||
esp_http_client_set_method(client, HTTP_METHOD_POST);
|
||||
if (post_data) {
|
||||
esp_http_client_set_post_field(client, post_data, strlen(post_data));
|
||||
}
|
||||
}
|
||||
|
||||
esp_err_t err = esp_http_client_perform(client);
|
||||
esp_http_client_cleanup(client);
|
||||
|
||||
if (err != ESP_OK) {
|
||||
ESP_LOGE(TAG, "API call failed: %s", esp_err_to_name(err));
|
||||
free(resp.buf);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return resp.buf;
|
||||
}
|
||||
|
||||
/* ── WS long connection ────────────────────────────────────── */
|
||||
static bool parse_query_param(const char *url, const char *key, char *out, size_t out_size)
|
||||
{
|
||||
const char *q = strchr(url, '?');
|
||||
if (!q) return false;
|
||||
q++;
|
||||
size_t key_len = strlen(key);
|
||||
while (*q) {
|
||||
const char *eq = strchr(q, '=');
|
||||
if (!eq) break;
|
||||
const char *amp = strchr(eq + 1, '&');
|
||||
size_t name_len = (size_t)(eq - q);
|
||||
if (name_len == key_len && strncmp(q, key, key_len) == 0) {
|
||||
size_t val_len = amp ? (size_t)(amp - (eq + 1)) : strlen(eq + 1);
|
||||
size_t n = (val_len < out_size - 1) ? val_len : out_size - 1;
|
||||
memcpy(out, eq + 1, n);
|
||||
out[n] = '\0';
|
||||
return true;
|
||||
}
|
||||
if (!amp) break;
|
||||
q = amp + 1;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static esp_err_t feishu_pull_ws_config(void)
|
||||
{
|
||||
cJSON *body = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(body, "AppID", s_app_id);
|
||||
cJSON_AddStringToObject(body, "AppSecret", s_app_secret);
|
||||
char *json_str = cJSON_PrintUnformatted(body);
|
||||
cJSON_Delete(body);
|
||||
if (!json_str) return ESP_ERR_NO_MEM;
|
||||
|
||||
http_resp_t resp = { .buf = calloc(1, 4096), .len = 0, .cap = 4096 };
|
||||
if (!resp.buf) {
|
||||
free(json_str);
|
||||
return ESP_ERR_NO_MEM;
|
||||
}
|
||||
|
||||
esp_http_client_config_t config = {
|
||||
.url = FEISHU_WS_CONFIG_URL,
|
||||
.event_handler = http_event_handler,
|
||||
.user_data = &resp,
|
||||
.timeout_ms = 15000,
|
||||
.buffer_size = 2048,
|
||||
.buffer_size_tx = 1024,
|
||||
.crt_bundle_attach = esp_crt_bundle_attach,
|
||||
};
|
||||
esp_http_client_handle_t client = esp_http_client_init(&config);
|
||||
if (!client) {
|
||||
free(json_str);
|
||||
free(resp.buf);
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
esp_http_client_set_method(client, HTTP_METHOD_POST);
|
||||
esp_http_client_set_header(client, "Content-Type", "application/json");
|
||||
esp_http_client_set_header(client, "locale", "zh");
|
||||
esp_http_client_set_post_field(client, json_str, strlen(json_str));
|
||||
esp_err_t err = esp_http_client_perform(client);
|
||||
int status = esp_http_client_get_status_code(client);
|
||||
esp_http_client_cleanup(client);
|
||||
free(json_str);
|
||||
|
||||
if (err != ESP_OK || status != 200) {
|
||||
ESP_LOGE(TAG, "WS config request failed: err=%s http=%d", esp_err_to_name(err), status);
|
||||
free(resp.buf);
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
cJSON *root = cJSON_Parse(resp.buf);
|
||||
free(resp.buf);
|
||||
if (!root) return ESP_FAIL;
|
||||
|
||||
cJSON *code = cJSON_GetObjectItem(root, "code");
|
||||
cJSON *data = cJSON_GetObjectItem(root, "data");
|
||||
cJSON *url = data ? cJSON_GetObjectItem(data, "URL") : NULL;
|
||||
cJSON *ccfg = data ? cJSON_GetObjectItem(data, "ClientConfig") : NULL;
|
||||
if (!code || code->valueint != 0 || !url || !cJSON_IsString(url)) {
|
||||
ESP_LOGE(TAG, "Invalid WS config response");
|
||||
cJSON_Delete(root);
|
||||
return ESP_FAIL;
|
||||
}
|
||||
|
||||
strncpy(s_ws_url, url->valuestring, sizeof(s_ws_url) - 1);
|
||||
char sid[24] = {0};
|
||||
if (parse_query_param(s_ws_url, "service_id", sid, sizeof(sid))) {
|
||||
s_ws_service_id = atoi(sid);
|
||||
}
|
||||
if (ccfg) {
|
||||
cJSON *pi = cJSON_GetObjectItem(ccfg, "PingInterval");
|
||||
cJSON *ri = cJSON_GetObjectItem(ccfg, "ReconnectInterval");
|
||||
cJSON *rn = cJSON_GetObjectItem(ccfg, "ReconnectNonce");
|
||||
if (pi && cJSON_IsNumber(pi)) s_ws_ping_interval_ms = pi->valueint * 1000;
|
||||
if (ri && cJSON_IsNumber(ri)) s_ws_reconnect_interval_ms = ri->valueint * 1000;
|
||||
if (rn && cJSON_IsNumber(rn)) s_ws_reconnect_nonce_ms = rn->valueint * 1000;
|
||||
}
|
||||
cJSON_Delete(root);
|
||||
ESP_LOGI(TAG, "WS config ready: service_id=%d ping=%dms", s_ws_service_id, s_ws_ping_interval_ms);
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
static void feishu_process_ws_event_json(const char *json, size_t len)
|
||||
{
|
||||
cJSON *root = cJSON_ParseWithLength(json, len);
|
||||
if (!root) return;
|
||||
cJSON *event = cJSON_GetObjectItem(root, "event");
|
||||
cJSON *header = cJSON_GetObjectItem(root, "header");
|
||||
if (event && header) {
|
||||
cJSON *event_type = cJSON_GetObjectItem(header, "event_type");
|
||||
if (event_type && cJSON_IsString(event_type) &&
|
||||
strcmp(event_type->valuestring, "im.message.receive_v1") == 0) {
|
||||
handle_message_event(event);
|
||||
}
|
||||
} else if (event) {
|
||||
handle_message_event(event);
|
||||
}
|
||||
cJSON_Delete(root);
|
||||
}
|
||||
|
||||
static void feishu_handle_ws_frame(const uint8_t *buf, size_t len)
|
||||
{
|
||||
ws_frame_t frame = {0};
|
||||
if (!pb_parse_frame(buf, len, &frame)) {
|
||||
ESP_LOGW(TAG, "WS frame parse failed");
|
||||
return;
|
||||
}
|
||||
|
||||
const char *type = frame_header_value(&frame, "type");
|
||||
if (frame.method == 0) {
|
||||
if (type && strcmp(type, "pong") == 0 && frame.payload && frame.payload_len > 0) {
|
||||
cJSON *cfg = cJSON_ParseWithLength((const char *)frame.payload, frame.payload_len);
|
||||
if (cfg) {
|
||||
cJSON *pi = cJSON_GetObjectItem(cfg, "PingInterval");
|
||||
if (pi && cJSON_IsNumber(pi)) s_ws_ping_interval_ms = pi->valueint * 1000;
|
||||
cJSON_Delete(cfg);
|
||||
}
|
||||
}
|
||||
return;
|
||||
}
|
||||
if (!type || strcmp(type, "event") != 0) return;
|
||||
if (!frame.payload || frame.payload_len == 0) return;
|
||||
|
||||
int code = 200;
|
||||
feishu_process_ws_event_json((const char *)frame.payload, frame.payload_len);
|
||||
|
||||
char ack[32];
|
||||
int ack_len = snprintf(ack, sizeof(ack), "{\"code\":%d}", code);
|
||||
ws_frame_t resp = frame;
|
||||
ws_send_frame(&resp, (const uint8_t *)ack, (size_t)ack_len, 1000);
|
||||
}
|
||||
|
||||
static void feishu_ws_event_handler(void *arg, esp_event_base_t base, int32_t event_id, void *event_data)
|
||||
{
|
||||
(void)arg;
|
||||
(void)base;
|
||||
esp_websocket_event_data_t *e = (esp_websocket_event_data_t *)event_data;
|
||||
static uint8_t *rx_buf = NULL;
|
||||
static size_t rx_cap = 0;
|
||||
if (event_id == WEBSOCKET_EVENT_CONNECTED) {
|
||||
s_ws_connected = true;
|
||||
ESP_LOGI(TAG, "Feishu WS connected");
|
||||
} else if (event_id == WEBSOCKET_EVENT_DISCONNECTED) {
|
||||
s_ws_connected = false;
|
||||
ESP_LOGW(TAG, "Feishu WS disconnected");
|
||||
} else if (event_id == WEBSOCKET_EVENT_DATA) {
|
||||
if (e->op_code != WS_TRANSPORT_OPCODES_BINARY) return;
|
||||
size_t need = e->payload_offset + e->data_len;
|
||||
if (e->payload_offset == 0) {
|
||||
if (rx_buf) free(rx_buf);
|
||||
rx_cap = (e->payload_len > need) ? e->payload_len : need;
|
||||
rx_buf = malloc(rx_cap);
|
||||
if (!rx_buf) return;
|
||||
} else if (!rx_buf || need > rx_cap) {
|
||||
return;
|
||||
}
|
||||
memcpy(rx_buf + e->payload_offset, e->data_ptr, e->data_len);
|
||||
if (need >= e->payload_len) {
|
||||
feishu_handle_ws_frame(rx_buf, e->payload_len);
|
||||
free(rx_buf);
|
||||
rx_buf = NULL;
|
||||
rx_cap = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void feishu_ws_task(void *arg)
|
||||
{
|
||||
(void)arg;
|
||||
while (1) {
|
||||
if (feishu_pull_ws_config() != ESP_OK) {
|
||||
vTaskDelay(pdMS_TO_TICKS(5000));
|
||||
continue;
|
||||
}
|
||||
|
||||
esp_websocket_client_config_t ws_cfg = {
|
||||
.uri = s_ws_url,
|
||||
.buffer_size = 2048,
|
||||
.task_stack = MIMI_FEISHU_POLL_STACK,
|
||||
.reconnect_timeout_ms = s_ws_reconnect_interval_ms,
|
||||
.network_timeout_ms = 10000,
|
||||
.disable_auto_reconnect = false,
|
||||
.crt_bundle_attach = esp_crt_bundle_attach,
|
||||
};
|
||||
|
||||
s_ws_client = esp_websocket_client_init(&ws_cfg);
|
||||
if (!s_ws_client) {
|
||||
vTaskDelay(pdMS_TO_TICKS(5000));
|
||||
continue;
|
||||
}
|
||||
esp_websocket_register_events(s_ws_client, WEBSOCKET_EVENT_ANY, feishu_ws_event_handler, NULL);
|
||||
esp_websocket_client_start(s_ws_client);
|
||||
|
||||
int64_t last_ping = 0;
|
||||
while (s_ws_client) {
|
||||
if (s_ws_connected) {
|
||||
int64_t now = esp_timer_get_time() / 1000;
|
||||
if (now - last_ping >= s_ws_ping_interval_ms) {
|
||||
ws_frame_t ping = {0};
|
||||
ping.seq_id = 0;
|
||||
ping.log_id = 0;
|
||||
ping.service = s_ws_service_id;
|
||||
ping.method = 0;
|
||||
ping.header_count = 1;
|
||||
strncpy(ping.headers[0].key, "type", sizeof(ping.headers[0].key) - 1);
|
||||
strncpy(ping.headers[0].value, "ping", sizeof(ping.headers[0].value) - 1);
|
||||
ws_send_frame(&ping, NULL, 0, 1000);
|
||||
last_ping = now;
|
||||
}
|
||||
}
|
||||
if (!esp_websocket_client_is_connected(s_ws_client) && !s_ws_connected) {
|
||||
break;
|
||||
}
|
||||
vTaskDelay(pdMS_TO_TICKS(200));
|
||||
}
|
||||
|
||||
esp_websocket_client_stop(s_ws_client);
|
||||
esp_websocket_client_destroy(s_ws_client);
|
||||
s_ws_client = NULL;
|
||||
s_ws_connected = false;
|
||||
vTaskDelay(pdMS_TO_TICKS(3000));
|
||||
}
|
||||
}
|
||||
|
||||
/* ── Webhook event handler ─────────────────────────────────── */
|
||||
|
||||
/*
|
||||
* Feishu Event Callback v2 schema:
|
||||
* {
|
||||
* "schema": "2.0",
|
||||
* "header": { "event_type": "im.message.receive_v1", "event_id": "...", ... },
|
||||
* "event": {
|
||||
* "sender": { "sender_id": { "open_id": "ou_xxx" }, "sender_type": "user" },
|
||||
* "message": {
|
||||
* "message_id": "om_xxx",
|
||||
* "chat_id": "oc_xxx",
|
||||
* "chat_type": "p2p" | "group",
|
||||
* "message_type": "text",
|
||||
* "content": "{\"text\":\"hello\"}"
|
||||
* }
|
||||
* }
|
||||
* }
|
||||
*
|
||||
* URL verification challenge:
|
||||
* { "challenge": "xxx", "token": "yyy", "type": "url_verification" }
|
||||
*/
|
||||
|
||||
static void handle_message_event(cJSON *event)
|
||||
{
|
||||
cJSON *message = cJSON_GetObjectItem(event, "message");
|
||||
if (!message) return;
|
||||
|
||||
cJSON *message_id_j = cJSON_GetObjectItem(message, "message_id");
|
||||
cJSON *chat_id_j = cJSON_GetObjectItem(message, "chat_id");
|
||||
cJSON *chat_type_j = cJSON_GetObjectItem(message, "chat_type");
|
||||
cJSON *msg_type_j = cJSON_GetObjectItem(message, "message_type");
|
||||
cJSON *content_j = cJSON_GetObjectItem(message, "content");
|
||||
|
||||
if (!chat_id_j || !cJSON_IsString(chat_id_j)) return;
|
||||
if (!content_j || !cJSON_IsString(content_j)) return;
|
||||
|
||||
const char *message_id = cJSON_IsString(message_id_j) ? message_id_j->valuestring : "";
|
||||
const char *chat_id = chat_id_j->valuestring;
|
||||
const char *chat_type = cJSON_IsString(chat_type_j) ? chat_type_j->valuestring : "p2p";
|
||||
const char *msg_type = cJSON_IsString(msg_type_j) ? msg_type_j->valuestring : "text";
|
||||
|
||||
/* Deduplication */
|
||||
if (message_id[0] && dedup_check_and_record(message_id)) {
|
||||
ESP_LOGD(TAG, "Duplicate message %s, skipping", message_id);
|
||||
return;
|
||||
}
|
||||
|
||||
/* Only handle text messages for now */
|
||||
if (strcmp(msg_type, "text") != 0) {
|
||||
ESP_LOGI(TAG, "Ignoring non-text message type: %s", msg_type);
|
||||
return;
|
||||
}
|
||||
|
||||
/* Parse the content JSON to extract text */
|
||||
cJSON *content_obj = cJSON_Parse(content_j->valuestring);
|
||||
if (!content_obj) {
|
||||
ESP_LOGW(TAG, "Failed to parse message content JSON");
|
||||
return;
|
||||
}
|
||||
|
||||
cJSON *text_j = cJSON_GetObjectItem(content_obj, "text");
|
||||
if (!text_j || !cJSON_IsString(text_j)) {
|
||||
cJSON_Delete(content_obj);
|
||||
return;
|
||||
}
|
||||
|
||||
const char *text = text_j->valuestring;
|
||||
|
||||
/* Strip @bot mention prefix if present (Feishu adds @_user_1 for mentions) */
|
||||
const char *cleaned = text;
|
||||
if (strncmp(cleaned, "@_user_1 ", 9) == 0) {
|
||||
cleaned += 9;
|
||||
}
|
||||
/* Skip leading whitespace */
|
||||
while (*cleaned == ' ' || *cleaned == '\n') cleaned++;
|
||||
|
||||
if (cleaned[0] == '\0') {
|
||||
cJSON_Delete(content_obj);
|
||||
return;
|
||||
}
|
||||
|
||||
/* Get sender info */
|
||||
const char *sender_id = "";
|
||||
cJSON *sender = cJSON_GetObjectItem(event, "sender");
|
||||
if (sender) {
|
||||
cJSON *sender_id_obj = cJSON_GetObjectItem(sender, "sender_id");
|
||||
if (sender_id_obj) {
|
||||
cJSON *open_id = cJSON_GetObjectItem(sender_id_obj, "open_id");
|
||||
if (open_id && cJSON_IsString(open_id)) {
|
||||
sender_id = open_id->valuestring;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ESP_LOGI(TAG, "Message from %s in %s(%s): %.60s%s",
|
||||
sender_id, chat_id, chat_type, cleaned,
|
||||
strlen(cleaned) > 60 ? "..." : "");
|
||||
|
||||
/* For p2p (DM) chats, use sender open_id as chat_id for session routing.
|
||||
* For group chats, use the chat_id (group ID).
|
||||
* This matches the moltbot reference pattern where DMs route by sender. */
|
||||
const char *route_id = chat_id;
|
||||
if (strcmp(chat_type, "p2p") == 0 && sender_id[0]) {
|
||||
route_id = sender_id;
|
||||
}
|
||||
|
||||
/* Push to inbound message bus */
|
||||
mimi_msg_t msg = {0};
|
||||
strncpy(msg.channel, MIMI_CHAN_FEISHU, sizeof(msg.channel) - 1);
|
||||
strncpy(msg.chat_id, route_id, sizeof(msg.chat_id) - 1);
|
||||
msg.content = strdup(cleaned);
|
||||
|
||||
if (msg.content) {
|
||||
if (message_bus_push_inbound(&msg) != ESP_OK) {
|
||||
ESP_LOGW(TAG, "Inbound queue full, dropping feishu message");
|
||||
free(msg.content);
|
||||
}
|
||||
}
|
||||
|
||||
cJSON_Delete(content_obj);
|
||||
}
|
||||
|
||||
/* Webhook mode intentionally disabled: Feishu channel runs in WebSocket mode only. */
|
||||
|
||||
/* ── Public API ────────────────────────────────────────────── */
|
||||
|
||||
esp_err_t feishu_bot_init(void)
|
||||
{
|
||||
nvs_handle_t nvs;
|
||||
if (nvs_open(MIMI_NVS_FEISHU, NVS_READONLY, &nvs) == ESP_OK) {
|
||||
char tmp_id[64] = {0};
|
||||
char tmp_secret[128] = {0};
|
||||
size_t len_id = sizeof(tmp_id);
|
||||
size_t len_secret = sizeof(tmp_secret);
|
||||
|
||||
if (nvs_get_str(nvs, MIMI_NVS_KEY_FEISHU_APP_ID, tmp_id, &len_id) == ESP_OK && tmp_id[0]) {
|
||||
strncpy(s_app_id, tmp_id, sizeof(s_app_id) - 1);
|
||||
}
|
||||
if (nvs_get_str(nvs, MIMI_NVS_KEY_FEISHU_APP_SECRET, tmp_secret, &len_secret) == ESP_OK && tmp_secret[0]) {
|
||||
strncpy(s_app_secret, tmp_secret, sizeof(s_app_secret) - 1);
|
||||
}
|
||||
nvs_close(nvs);
|
||||
}
|
||||
|
||||
if (s_app_id[0] && s_app_secret[0]) {
|
||||
ESP_LOGI(TAG, "Feishu credentials loaded (app_id=%.8s...)", s_app_id);
|
||||
} else {
|
||||
ESP_LOGW(TAG, "No Feishu credentials. Use CLI: set_feishu_creds <APP_ID> <APP_SECRET>");
|
||||
}
|
||||
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
esp_err_t feishu_bot_start(void)
|
||||
{
|
||||
if (s_app_id[0] == '\0' || s_app_secret[0] == '\0') {
|
||||
ESP_LOGW(TAG, "Feishu not configured, skipping WebSocket start");
|
||||
return ESP_OK;
|
||||
}
|
||||
if (s_ws_task) {
|
||||
ESP_LOGW(TAG, "Feishu WebSocket task already running");
|
||||
return ESP_OK;
|
||||
}
|
||||
BaseType_t ok = xTaskCreatePinnedToCore(
|
||||
feishu_ws_task,
|
||||
"feishu_ws",
|
||||
MIMI_FEISHU_POLL_STACK,
|
||||
NULL,
|
||||
MIMI_FEISHU_POLL_PRIO,
|
||||
&s_ws_task,
|
||||
MIMI_FEISHU_POLL_CORE);
|
||||
if (ok != pdPASS) {
|
||||
s_ws_task = NULL;
|
||||
return ESP_FAIL;
|
||||
}
|
||||
ESP_LOGI(TAG, "Feishu WebSocket mode enabled");
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
esp_err_t feishu_send_message(const char *chat_id, const char *text)
|
||||
{
|
||||
if (s_app_id[0] == '\0' || s_app_secret[0] == '\0') {
|
||||
ESP_LOGW(TAG, "Cannot send: no credentials configured");
|
||||
return ESP_ERR_INVALID_STATE;
|
||||
}
|
||||
|
||||
/* Determine receive_id_type based on ID prefix */
|
||||
const char *id_type = "chat_id";
|
||||
if (strncmp(chat_id, "ou_", 3) == 0) {
|
||||
id_type = "open_id";
|
||||
}
|
||||
|
||||
char url[256];
|
||||
snprintf(url, sizeof(url), "%s?receive_id_type=%s", FEISHU_SEND_MSG_URL, id_type);
|
||||
|
||||
size_t text_len = strlen(text);
|
||||
size_t offset = 0;
|
||||
int all_ok = 1;
|
||||
|
||||
while (offset < text_len) {
|
||||
size_t chunk = text_len - offset;
|
||||
if (chunk > MIMI_FEISHU_MAX_MSG_LEN) {
|
||||
chunk = MIMI_FEISHU_MAX_MSG_LEN;
|
||||
}
|
||||
|
||||
char *segment = malloc(chunk + 1);
|
||||
if (!segment) return ESP_ERR_NO_MEM;
|
||||
memcpy(segment, text + offset, chunk);
|
||||
segment[chunk] = '\0';
|
||||
|
||||
/* Build content JSON: {"text":"..."} */
|
||||
cJSON *content = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(content, "text", segment);
|
||||
char *content_str = cJSON_PrintUnformatted(content);
|
||||
cJSON_Delete(content);
|
||||
free(segment);
|
||||
|
||||
if (!content_str) { offset += chunk; all_ok = 0; continue; }
|
||||
|
||||
/* Build message body */
|
||||
cJSON *body = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(body, "receive_id", chat_id);
|
||||
cJSON_AddStringToObject(body, "msg_type", "text");
|
||||
cJSON_AddStringToObject(body, "content", content_str);
|
||||
free(content_str);
|
||||
|
||||
char *json_str = cJSON_PrintUnformatted(body);
|
||||
cJSON_Delete(body);
|
||||
|
||||
if (json_str) {
|
||||
char *resp = feishu_api_call(url, "POST", json_str);
|
||||
free(json_str);
|
||||
|
||||
if (resp) {
|
||||
cJSON *root = cJSON_Parse(resp);
|
||||
if (root) {
|
||||
cJSON *code = cJSON_GetObjectItem(root, "code");
|
||||
if (code && code->valueint != 0) {
|
||||
cJSON *msg = cJSON_GetObjectItem(root, "msg");
|
||||
ESP_LOGW(TAG, "Send failed: code=%d, msg=%s",
|
||||
code->valueint, msg ? msg->valuestring : "unknown");
|
||||
all_ok = 0;
|
||||
} else {
|
||||
ESP_LOGI(TAG, "Sent to %s (%d bytes)", chat_id, (int)chunk);
|
||||
}
|
||||
cJSON_Delete(root);
|
||||
}
|
||||
free(resp);
|
||||
} else {
|
||||
ESP_LOGE(TAG, "Failed to send message chunk");
|
||||
all_ok = 0;
|
||||
}
|
||||
}
|
||||
|
||||
offset += chunk;
|
||||
}
|
||||
|
||||
return all_ok ? ESP_OK : ESP_FAIL;
|
||||
}
|
||||
|
||||
esp_err_t feishu_reply_message(const char *message_id, const char *text)
|
||||
{
|
||||
if (s_app_id[0] == '\0' || s_app_secret[0] == '\0') {
|
||||
return ESP_ERR_INVALID_STATE;
|
||||
}
|
||||
|
||||
char url[256];
|
||||
snprintf(url, sizeof(url), FEISHU_REPLY_MSG_URL, message_id);
|
||||
|
||||
cJSON *content = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(content, "text", text);
|
||||
char *content_str = cJSON_PrintUnformatted(content);
|
||||
cJSON_Delete(content);
|
||||
if (!content_str) return ESP_ERR_NO_MEM;
|
||||
|
||||
cJSON *body = cJSON_CreateObject();
|
||||
cJSON_AddStringToObject(body, "msg_type", "text");
|
||||
cJSON_AddStringToObject(body, "content", content_str);
|
||||
free(content_str);
|
||||
|
||||
char *json_str = cJSON_PrintUnformatted(body);
|
||||
cJSON_Delete(body);
|
||||
if (!json_str) return ESP_ERR_NO_MEM;
|
||||
|
||||
char *resp = feishu_api_call(url, "POST", json_str);
|
||||
free(json_str);
|
||||
|
||||
esp_err_t ret = ESP_FAIL;
|
||||
if (resp) {
|
||||
cJSON *root = cJSON_Parse(resp);
|
||||
if (root) {
|
||||
cJSON *code = cJSON_GetObjectItem(root, "code");
|
||||
if (code && code->valueint == 0) {
|
||||
ret = ESP_OK;
|
||||
} else {
|
||||
cJSON *msg = cJSON_GetObjectItem(root, "msg");
|
||||
ESP_LOGW(TAG, "Reply failed: code=%d, msg=%s",
|
||||
code ? code->valueint : -1, msg ? msg->valuestring : "unknown");
|
||||
}
|
||||
cJSON_Delete(root);
|
||||
}
|
||||
free(resp);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
esp_err_t feishu_set_credentials(const char *app_id, const char *app_secret)
|
||||
{
|
||||
nvs_handle_t nvs;
|
||||
ESP_ERROR_CHECK(nvs_open(MIMI_NVS_FEISHU, NVS_READWRITE, &nvs));
|
||||
ESP_ERROR_CHECK(nvs_set_str(nvs, MIMI_NVS_KEY_FEISHU_APP_ID, app_id));
|
||||
ESP_ERROR_CHECK(nvs_set_str(nvs, MIMI_NVS_KEY_FEISHU_APP_SECRET, app_secret));
|
||||
ESP_ERROR_CHECK(nvs_commit(nvs));
|
||||
nvs_close(nvs);
|
||||
|
||||
strncpy(s_app_id, app_id, sizeof(s_app_id) - 1);
|
||||
strncpy(s_app_secret, app_secret, sizeof(s_app_secret) - 1);
|
||||
|
||||
/* Clear cached token to force re-auth */
|
||||
s_tenant_token[0] = '\0';
|
||||
s_token_expire_time = 0;
|
||||
|
||||
ESP_LOGI(TAG, "Feishu credentials saved");
|
||||
return ESP_OK;
|
||||
}
|
||||
36
main/channels/feishu/feishu_bot.h
Normal file
36
main/channels/feishu/feishu_bot.h
Normal file
@@ -0,0 +1,36 @@
|
||||
#pragma once
|
||||
|
||||
#include "esp_err.h"
|
||||
|
||||
/**
|
||||
* Initialize the Feishu bot (load credentials from NVS / build-time).
|
||||
*/
|
||||
esp_err_t feishu_bot_init(void);
|
||||
|
||||
/**
|
||||
* Start the Feishu webhook HTTP server for receiving events.
|
||||
* Listens on MIMI_FEISHU_WEBHOOK_PORT.
|
||||
*/
|
||||
esp_err_t feishu_bot_start(void);
|
||||
|
||||
/**
|
||||
* Send a text message to a Feishu chat.
|
||||
* Automatically splits messages longer than MIMI_FEISHU_MAX_MSG_LEN chars.
|
||||
* @param chat_id Feishu chat ID (open_id or chat_id)
|
||||
* @param text Message text
|
||||
*/
|
||||
esp_err_t feishu_send_message(const char *chat_id, const char *text);
|
||||
|
||||
/**
|
||||
* Reply to a specific message in a Feishu chat.
|
||||
* @param message_id The message_id to reply to
|
||||
* @param text Reply text
|
||||
*/
|
||||
esp_err_t feishu_reply_message(const char *message_id, const char *text);
|
||||
|
||||
/**
|
||||
* Save Feishu app credentials to NVS.
|
||||
* @param app_id Feishu App ID
|
||||
* @param app_secret Feishu App Secret
|
||||
*/
|
||||
esp_err_t feishu_set_credentials(const char *app_id, const char *app_secret);
|
||||
@@ -1,7 +1,8 @@
|
||||
#include "serial_cli.h"
|
||||
#include "mimi_config.h"
|
||||
#include "wifi/wifi_manager.h"
|
||||
#include "telegram/telegram_bot.h"
|
||||
#include "channels/telegram/telegram_bot.h"
|
||||
#include "channels/feishu/feishu_bot.h"
|
||||
#include "llm/llm_proxy.h"
|
||||
#include "memory/memory_store.h"
|
||||
#include "memory/session_mgr.h"
|
||||
@@ -72,6 +73,47 @@ static int cmd_set_tg_token(int argc, char **argv)
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* --- set_feishu_creds command --- */
|
||||
static struct {
|
||||
struct arg_str *app_id;
|
||||
struct arg_str *app_secret;
|
||||
struct arg_end *end;
|
||||
} feishu_creds_args;
|
||||
|
||||
/* --- feishu_send command --- */
|
||||
static struct {
|
||||
struct arg_str *receive_id;
|
||||
struct arg_str *text;
|
||||
struct arg_end *end;
|
||||
} feishu_send_args;
|
||||
|
||||
static int cmd_set_feishu_creds(int argc, char **argv)
|
||||
{
|
||||
int nerrors = arg_parse(argc, argv, (void **)&feishu_creds_args);
|
||||
if (nerrors != 0) {
|
||||
arg_print_errors(stderr, feishu_creds_args.end, argv[0]);
|
||||
return 1;
|
||||
}
|
||||
feishu_set_credentials(feishu_creds_args.app_id->sval[0],
|
||||
feishu_creds_args.app_secret->sval[0]);
|
||||
printf("Feishu credentials saved.\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int cmd_feishu_send(int argc, char **argv)
|
||||
{
|
||||
int nerrors = arg_parse(argc, argv, (void **)&feishu_send_args);
|
||||
if (nerrors != 0) {
|
||||
arg_print_errors(stderr, feishu_send_args.end, argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
esp_err_t err = feishu_send_message(feishu_send_args.receive_id->sval[0],
|
||||
feishu_send_args.text->sval[0]);
|
||||
printf("feishu_send status: %s\n", esp_err_to_name(err));
|
||||
return (err == ESP_OK) ? 0 : 1;
|
||||
}
|
||||
|
||||
/* --- set_api_key command --- */
|
||||
static struct {
|
||||
struct arg_str *key;
|
||||
@@ -619,6 +661,30 @@ esp_err_t serial_cli_init(void)
|
||||
};
|
||||
esp_console_cmd_register(&tg_token_cmd);
|
||||
|
||||
/* set_feishu_creds */
|
||||
feishu_creds_args.app_id = arg_str1(NULL, NULL, "<app_id>", "Feishu App ID");
|
||||
feishu_creds_args.app_secret = arg_str1(NULL, NULL, "<app_secret>", "Feishu App Secret");
|
||||
feishu_creds_args.end = arg_end(2);
|
||||
esp_console_cmd_t feishu_creds_cmd = {
|
||||
.command = "set_feishu_creds",
|
||||
.help = "Set Feishu app credentials (app_id app_secret)",
|
||||
.func = &cmd_set_feishu_creds,
|
||||
.argtable = &feishu_creds_args,
|
||||
};
|
||||
esp_console_cmd_register(&feishu_creds_cmd);
|
||||
|
||||
/* feishu_send */
|
||||
feishu_send_args.receive_id = arg_str1(NULL, NULL, "<receive_id>", "Feishu open_id/chat_id");
|
||||
feishu_send_args.text = arg_str1(NULL, NULL, "<text>", "Text message (quote if contains spaces)");
|
||||
feishu_send_args.end = arg_end(2);
|
||||
esp_console_cmd_t feishu_send_cmd = {
|
||||
.command = "feishu_send",
|
||||
.help = "Send Feishu text: feishu_send <open_id|chat_id> \"hello\"",
|
||||
.func = &cmd_feishu_send,
|
||||
.argtable = &feishu_send_args,
|
||||
};
|
||||
esp_console_cmd_register(&feishu_send_cmd);
|
||||
|
||||
/* set_api_key */
|
||||
api_key_args.key = arg_str1(NULL, NULL, "<key>", "LLM API key");
|
||||
api_key_args.end = arg_end(1);
|
||||
|
||||
@@ -20,7 +20,7 @@ typedef struct {
|
||||
int64_t at_epoch; /* For AT: unix timestamp */
|
||||
char message[256]; /* Message to inject into inbound queue */
|
||||
char channel[16]; /* Reply channel (default "system") */
|
||||
char chat_id[32]; /* Reply chat_id (default "cron") */
|
||||
char chat_id[96]; /* Reply chat_id/open_id (default "cron") */
|
||||
int64_t last_run; /* Last run epoch */
|
||||
int64_t next_run; /* Next run epoch */
|
||||
bool delete_after_run; /* Remove job after firing (for AT jobs) */
|
||||
|
||||
@@ -14,3 +14,4 @@ dependencies:
|
||||
# # `public` flag doesn't have an effect dependencies of the `main` component.
|
||||
# # All dependencies of `main` are public by default.
|
||||
# public: true
|
||||
espressif/esp_websocket_client: ^1.4.0
|
||||
|
||||
12
main/mimi.c
12
main/mimi.c
@@ -12,7 +12,8 @@
|
||||
#include "mimi_config.h"
|
||||
#include "bus/message_bus.h"
|
||||
#include "wifi/wifi_manager.h"
|
||||
#include "telegram/telegram_bot.h"
|
||||
#include "channels/telegram/telegram_bot.h"
|
||||
#include "channels/feishu/feishu_bot.h"
|
||||
#include "llm/llm_proxy.h"
|
||||
#include "agent/agent_loop.h"
|
||||
#include "memory/memory_store.h"
|
||||
@@ -78,6 +79,13 @@ static void outbound_dispatch_task(void *arg)
|
||||
} else {
|
||||
ESP_LOGI(TAG, "Telegram send success for %s (%d bytes)", msg.chat_id, (int)strlen(msg.content));
|
||||
}
|
||||
} else if (strcmp(msg.channel, MIMI_CHAN_FEISHU) == 0) {
|
||||
esp_err_t send_err = feishu_send_message(msg.chat_id, msg.content);
|
||||
if (send_err != ESP_OK) {
|
||||
ESP_LOGE(TAG, "Feishu send failed for %s: %s", msg.chat_id, esp_err_to_name(send_err));
|
||||
} else {
|
||||
ESP_LOGI(TAG, "Feishu send success for %s (%d bytes)", msg.chat_id, (int)strlen(msg.content));
|
||||
}
|
||||
} else if (strcmp(msg.channel, MIMI_CHAN_WEBSOCKET) == 0) {
|
||||
esp_err_t ws_err = ws_server_send(msg.chat_id, msg.content);
|
||||
if (ws_err != ESP_OK) {
|
||||
@@ -121,6 +129,7 @@ void app_main(void)
|
||||
ESP_ERROR_CHECK(wifi_manager_init());
|
||||
ESP_ERROR_CHECK(http_proxy_init());
|
||||
ESP_ERROR_CHECK(telegram_bot_init());
|
||||
ESP_ERROR_CHECK(feishu_bot_init());
|
||||
ESP_ERROR_CHECK(llm_proxy_init());
|
||||
ESP_ERROR_CHECK(tool_registry_init());
|
||||
ESP_ERROR_CHECK(cron_service_init());
|
||||
@@ -149,6 +158,7 @@ void app_main(void)
|
||||
/* Start network-dependent services */
|
||||
ESP_ERROR_CHECK(agent_loop_start());
|
||||
ESP_ERROR_CHECK(telegram_bot_start());
|
||||
ESP_ERROR_CHECK(feishu_bot_start());
|
||||
cron_service_start();
|
||||
heartbeat_start();
|
||||
ESP_ERROR_CHECK(ws_server_start());
|
||||
|
||||
@@ -37,6 +37,12 @@
|
||||
#ifndef MIMI_SECRET_SEARCH_KEY
|
||||
#define MIMI_SECRET_SEARCH_KEY ""
|
||||
#endif
|
||||
#ifndef MIMI_SECRET_FEISHU_APP_ID
|
||||
#define MIMI_SECRET_FEISHU_APP_ID ""
|
||||
#endif
|
||||
#ifndef MIMI_SECRET_FEISHU_APP_SECRET
|
||||
#define MIMI_SECRET_FEISHU_APP_SECRET ""
|
||||
#endif
|
||||
|
||||
/* WiFi */
|
||||
#define MIMI_WIFI_MAX_RETRY 10
|
||||
@@ -52,6 +58,15 @@
|
||||
#define MIMI_TG_CARD_SHOW_MS 3000
|
||||
#define MIMI_TG_CARD_BODY_SCALE 3
|
||||
|
||||
/* Feishu Bot */
|
||||
#define MIMI_FEISHU_MAX_MSG_LEN 4096
|
||||
#define MIMI_FEISHU_POLL_STACK (12 * 1024)
|
||||
#define MIMI_FEISHU_POLL_PRIO 5
|
||||
#define MIMI_FEISHU_POLL_CORE 0
|
||||
#define MIMI_FEISHU_WEBHOOK_PORT 18790
|
||||
#define MIMI_FEISHU_WEBHOOK_PATH "/feishu/events"
|
||||
#define MIMI_FEISHU_WEBHOOK_MAX_BODY (16 * 1024)
|
||||
|
||||
/* Agent Loop */
|
||||
#define MIMI_AGENT_STACK (24 * 1024)
|
||||
#define MIMI_AGENT_PRIO 6
|
||||
@@ -114,6 +129,7 @@
|
||||
/* NVS Namespaces */
|
||||
#define MIMI_NVS_WIFI "wifi_config"
|
||||
#define MIMI_NVS_TG "tg_config"
|
||||
#define MIMI_NVS_FEISHU "feishu_config"
|
||||
#define MIMI_NVS_LLM "llm_config"
|
||||
#define MIMI_NVS_PROXY "proxy_config"
|
||||
#define MIMI_NVS_SEARCH "search_config"
|
||||
@@ -122,6 +138,8 @@
|
||||
#define MIMI_NVS_KEY_SSID "ssid"
|
||||
#define MIMI_NVS_KEY_PASS "password"
|
||||
#define MIMI_NVS_KEY_TG_TOKEN "bot_token"
|
||||
#define MIMI_NVS_KEY_FEISHU_APP_ID "app_id"
|
||||
#define MIMI_NVS_KEY_FEISHU_APP_SECRET "app_secret"
|
||||
#define MIMI_NVS_KEY_API_KEY "api_key"
|
||||
#define MIMI_NVS_KEY_MODEL "model"
|
||||
#define MIMI_NVS_KEY_PROVIDER "provider"
|
||||
|
||||
@@ -17,6 +17,10 @@
|
||||
/* Telegram Bot */
|
||||
#define MIMI_SECRET_TG_TOKEN ""
|
||||
|
||||
/* Feishu Bot */
|
||||
#define MIMI_SECRET_FEISHU_APP_ID ""
|
||||
#define MIMI_SECRET_FEISHU_APP_SECRET ""
|
||||
|
||||
/* Anthropic API */
|
||||
#define MIMI_SECRET_API_KEY ""
|
||||
#define MIMI_SECRET_MODEL ""
|
||||
|
||||
Reference in New Issue
Block a user