From 1c43920fb85387aaf7267b4b79dc829896bd610d Mon Sep 17 00:00:00 2001
From: wnoun <wnoun@outlook.com>
Date: Sun, 7 Apr 2019 21:10:52 +0800
Subject: demux_cue: auto-detect CUE sheet charset

---
 demux/demux_cue.c | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'demux')

diff --git a/demux/demux_cue.c b/demux/demux_cue.c
index 708b742e8f..59628c74b4 100644
--- a/demux/demux_cue.c
+++ b/demux/demux_cue.c
@@ -28,8 +28,11 @@
 #include "mpv_talloc.h"
 
 #include "misc/bstr.h"
+#include "misc/charset_conv.h"
 #include "common/msg.h"
 #include "demux/demux.h"
+#include "options/m_config.h"
+#include "options/m_option.h"
 #include "options/path.h"
 #include "common/common.h"
 #include "stream/stream.h"
@@ -39,8 +42,25 @@
 
 #define PROBE_SIZE 512
 
+#define OPT_BASE_STRUCT struct demux_cue_opts
+struct demux_cue_opts {
+    char *cue_cp;
+};
+
+const struct m_sub_options demux_cue_conf = {
+        .opts = (const m_option_t[]) {
+            OPT_STRING("codepage", cue_cp, 0),
+            {0}
+        },
+        .size = sizeof(struct demux_cue_opts),
+        .defaults = &(const struct demux_cue_opts) {
+            .cue_cp = "auto"
+        }
+};
+
 struct priv {
     struct cue_file *f;
+    struct demux_cue_opts *opts;
 };
 
 static void add_source(struct timeline *tl, struct demuxer *d)
@@ -252,10 +272,21 @@ static int try_open_file(struct demuxer *demuxer, enum demux_check check)
     struct priv *p = talloc_zero(demuxer, struct priv);
     demuxer->priv = p;
     demuxer->fully_read = true;
+    p->opts = mp_get_config_group(p, demuxer->global, &demux_cue_conf);
+    struct demux_cue_opts *cue_opts = p->opts;
 
     bstr data = stream_read_complete(s, p, 1000000);
     if (data.start == NULL)
         return -1;
+    const char *charset = mp_charset_guess(p, demuxer->log, data, cue_opts->cue_cp, 0);
+    if (charset && !mp_charset_is_utf8(charset)) {
+        MP_INFO(demuxer, "Using CUE charset: %s\n", charset);
+        bstr utf8 = mp_iconv_to_utf8(demuxer->log, data, charset, MP_ICONV_VERBOSE);
+        if (utf8.start && utf8.start != data.start) {
+            ta_steal(data.start, utf8.start);
+            data = utf8;
+        }
+    }
     p->f = mp_parse_cue(data);
     talloc_steal(p, p->f);
     if (!p->f) {
-- 
cgit v1.2.3