Feat/mod dub v2 (#22)

* support tts elevenlabs to mod_dub Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * wip Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * wip * wip * fix aws race condition when 2 start transcribes are sent at the same instant * wip * wip * wip * allow queue play on track * wip Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * revert change for aws transcribe Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * fix type * wip * wip * rename parameters * rename paramters * wip * wip * wip * wip * wip * bug: there exists scenarios where callback is not defined * wip * wip * revert unintended changes to mod_google_transcribe * fix bugs w/ streaming tts simplified arg parsing to use freeswitch conventions --------- Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> Co-authored-by: Quan HL <quan.luuhoang8@gmail.com>
2026-01-25 02:08:27 +00:00 · 2024-03-23 15:37:27 -04:00
1 parent 4e57f73c7e
commit 9cdc5fdfca
@@ -0,0 +1,134 @@
+#ifndef __AP_HTTP_H__
+#define __AP_HTTP_H__
+
+#include "ap.h"
+#include <curl/curl.h>
+#include <mpg123.h>
+#include <boost/asio.hpp>
+
+
+typedef struct
+{
+  CURLM *multi;
+  int still_running;
+} GlobalInfo_t;
+
+class AudioProducerHttp : public AudioProducer {
+public:
+
+  typedef enum
+  {
+    STATUS_NONE = 0,
+    STATUS_FAILED,
+    STATUS_DOWNLOAD_IN_PROGRESS,
+    STATUS_DOWNLOAD_PAUSED,
+    STATUS_DOWNLOAD_COMPLETE,
+    STATUS_AWAITING_RESTART,
+    STATUS_STOPPING,
+    STATUS_STOPPED
+  } Status_t;
+
+  typedef enum {
+    HTTP_METHOD_GET = 0,
+    HTTP_METHOD_POST
+  } HttpMethod_t;
+
+  static const char* status2String(Status_t status) {
+    static const char* statusStrings[] = {
+      "STATUS_NONE",
+      "STATUS_FAILED",
+      "STATUS_DOWNLOAD_IN_PROGRESS",
+      "STATUS_DOWNLOAD_PAUSED",
+      "STATUS_DOWNLOAD_COMPLETE",
+      "STATUS_AWAITING_RESTART",
+      "STATUS_STOPPED"
+    };
+
+    if (status >= 0 && status < sizeof(statusStrings) / sizeof(statusStrings[0]))
+    {
+      return statusStrings[status];
+    }
+    else
+    {
+      return "UNKNOWN_STATUS";
+    }
+  }
+  AudioProducerHttp(
+    std::mutex& mutex,
+    CircularBuffer_t& circularBuffer,
+    int sampleRate
+  );
+  virtual ~AudioProducerHttp();
+
+  virtual void start(std::function<void(bool, const std::string&)> callback);
+  void addCurlHandle(const boost::system::error_code& error);
+  virtual void stop();
+  void cleanup(Status_t status, int response_code);
+  void reset();
+
+  void queueHttpGetAudio(const std::string& url, int gain = 0, bool loop = false);
+  void queueHttpPostAudio(const std::string& url, int gain = 0, bool loop = false);
+  void queueHttpPostAudio(const std::string& url, const std::string& body, std::vector<std::string>& headers, int gain = 0, bool loop = false);
+
+  Status_t getStatus() const { return _status; }
+  void setStatus(Status_t status) { _status = status; }
+
+  boost::asio::deadline_timer& getTimer() { return _timer; }
+  
+  static bool initialized;
+  static std::thread worker_thread;
+  static boost::asio::io_service io_service;
+  static void threadFunc();
+  static std::map<curl_socket_t, boost::asio::ip::tcp::socket *> socket_map;
+  static boost::asio::deadline_timer multi_timer;
+
+
+  static GlobalInfo_t global;
+
+  static int sock_cb(CURL *e, curl_socket_t s, int what, void *cbp, void *sockp);
+  static int multi_timer_cb(CURLM *multi, long timeout_ms, GlobalInfo_t *g);
+  static void timer_cb(const boost::system::error_code & error, GlobalInfo_t *g);
+  static void addsock(curl_socket_t s, CURL *easy, int action, GlobalInfo_t *g);
+  static void setsock(int *fdp, curl_socket_t s, CURL *e, int act, int oldact, GlobalInfo_t *g);
+  static int mcode_test(const char *where, CURLMcode code);
+  static void check_multi_info(GlobalInfo_t *g);
+  static void event_cb(GlobalInfo_t *g, curl_socket_t s, int action, const boost::system::error_code & error, int *fdp);
+  static void remsock(int *f, GlobalInfo_t *g);
+  
+  static size_t static_write_cb(char* ptr, size_t size, size_t nmemb, void* userdata);
+  size_t write_cb(void *ptr, size_t size, size_t nmemb);
+
+  static size_t static_header_callback(char *buffer, size_t size, size_t nitems, void* userdata);
+  size_t header_callback(char *buffer, size_t size, size_t nitems);
+
+  void throttling_cb(const boost::system::error_code& error);
+
+  static void static_restart_cb(const boost::system::error_code& error, void* userdata);
+  void restart_cb(const boost::system::error_code& error);
+
+  static int close_socket(void *clientp, curl_socket_t item);
+  static curl_socket_t open_socket(void *clientp, curlsocktype purpose, struct curl_sockaddr *address);
+
+private:
+
+  static void _init();
+  static void _deinit();
+
+  CURL* createEasyHandle();
+  bool parseHeader(const std::string& str, std::string& header, std::string& value);
+  int extract_response_code(const std::string& input);
+
+  HttpMethod_t _method;
+  std::string _url;
+  std::string _body;
+  std::vector<std::string> _headers;
+  Status_t _status;
+  mpg123_handle *_mh;
+  CURL *_easy;
+  char _error[CURL_ERROR_SIZE]; // curl error buffer
+  std::string _err_msg;
+  int _response_code;
+  boost::asio::deadline_timer _timer;
+};
+
+#endif