94 lines
3.8 KiB
C
94 lines
3.8 KiB
C
#include "attachment_sender.h"
|
|
#include "instance_lite.h"
|
|
#include "../../../lib/debug_config.h"
|
|
#include "../../../lib/mem.h"
|
|
#include "../../../lib/u_async.h"
|
|
#include "../../../src/chat/chat_core.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#include <libgen.h>
|
|
|
|
int attachment_send(const char* channel_id, const char* src_file_path, const char* db_path,
|
|
const uint8_t reply_to[32], uint64_t request_id) {
|
|
if (!channel_id || !src_file_path || !db_path) {
|
|
DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "attachment_send: invalid args");
|
|
return -1;
|
|
}
|
|
|
|
struct UASYNC* ua = instance_lite_get_uasync();
|
|
if (!uasync_is_running(ua)) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "attachment_send: core stopped"); return -1; }
|
|
|
|
/* get original basename for display, then make unique filename for storage */
|
|
char path_copy[1024];
|
|
snprintf(path_copy, sizeof(path_copy), "%s", src_file_path);
|
|
char* fname = basename(path_copy);
|
|
char original_basename[256];
|
|
snprintf(original_basename, sizeof(original_basename), "%s", fname);
|
|
char* dot = strrchr(fname, '.');
|
|
char ext[32] = "";
|
|
if (dot) { snprintf(ext, sizeof(ext), "%s", dot + 1); *dot = '\0'; }
|
|
|
|
struct stat st; memset(&st, 0, sizeof(st));
|
|
stat(src_file_path, &st);
|
|
|
|
/* generate dt/suffix for unique filename */
|
|
time_t t = time(NULL);
|
|
struct tm tm_buf; localtime_r(&t, &tm_buf);
|
|
char dt[32]; snprintf(dt, sizeof(dt), "%04d%02d%02d-%02d%02d%02d",
|
|
tm_buf.tm_year + 1900, tm_buf.tm_mon + 1, tm_buf.tm_mday,
|
|
tm_buf.tm_hour, tm_buf.tm_min, tm_buf.tm_sec);
|
|
int rnd = rand() & 0xFFFF;
|
|
char suffix[8]; snprintf(suffix, sizeof(suffix), "%04x", rnd);
|
|
|
|
/* ensure media dir exists */
|
|
char media_dir[1024];
|
|
snprintf(media_dir, sizeof(media_dir), "%s/media/%s", db_path, channel_id);
|
|
{
|
|
char tmp[1024]; size_t off = 0;
|
|
for (size_t i = 0; media_dir[i] && off < sizeof(tmp) - 1; i++) {
|
|
tmp[off++] = media_dir[i];
|
|
if (media_dir[i] == '/' && off > 1) { tmp[off] = '\0'; mkdir(tmp, 0755); }
|
|
}
|
|
tmp[off] = '\0';
|
|
mkdir(tmp, 0755);
|
|
}
|
|
|
|
/* single file path — no splitting */
|
|
char media_file[1280];
|
|
snprintf(media_file, sizeof(media_file), "%s/%s_%s_%s.%s",
|
|
media_dir, dt, fname, suffix, ext[0] ? ext : "bin");
|
|
|
|
/* post chat_msg_submit — media_copy=1 so media_index copies src to dest
|
|
data = original_basename — becomes first part of message "d" field: "name.ext|size|..." */
|
|
size_t datalen = strlen(original_basename);
|
|
struct chat_msg_submit* req = u_calloc(1, sizeof(struct chat_msg_submit) + datalen + 1);
|
|
struct posted_task* post = u_calloc(1, sizeof(*post));
|
|
if (!req || !post) {
|
|
DEBUG_ERROR(DEBUG_CATEGORY_MEDIA, "attachment_send: allocation failed"); u_free(req); u_free(post); return -1;
|
|
}
|
|
|
|
req->inst = instance_lite_get_instance();
|
|
memcpy(req->reply_to, reply_to, sizeof(req->reply_to));
|
|
req->request_id = request_id;
|
|
snprintf(req->channel_id, sizeof(req->channel_id), "%s", channel_id);
|
|
snprintf(req->content_type, sizeof(req->content_type), "application/octet-stream");
|
|
snprintf(req->media_src, sizeof(req->media_src), "%s", src_file_path);
|
|
snprintf(req->media_dest, sizeof(req->media_dest), "%s", media_file);
|
|
req->media_copy = 1;
|
|
req->data = (uint8_t*)(req + 1);
|
|
memcpy(req->data, original_basename, datalen);
|
|
req->data_len = (uint32_t)datalen;
|
|
req->timestamp = 0;
|
|
|
|
DEBUG_INFO(DEBUG_CATEGORY_GENERAL, "attachment_send: ch=%s src=%s dst=%s name=%s size=%lld",
|
|
channel_id, src_file_path, media_file, original_basename, (long long)st.st_size);
|
|
|
|
post->callback = chat_core_submit_trampoline; post->arg = req;
|
|
uasync_post_reserved(ua, post);
|
|
return 0;
|
|
}
|