Merge branch 'dyf-device' into 6170

fys-admin/src/main/java/com/fuyuanshen/app/service/AudioProcessService.java

@@ -8,10 +8,19 @@ import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
+import javax.xml.stream.XMLInputFactory;
+import javax.xml.stream.XMLStreamConstants;
+import javax.xml.stream.XMLStreamException;
+import javax.xml.stream.XMLStreamReader;
+import java.io.*;
+import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.util.Arrays;
 import java.util.List;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipInputStream;
 
 /**
  * 音频处理服务
@@ -101,6 +110,12 @@ public class AudioProcessService {
 //                log.info("测试文件已保存: {}", savedPath);
 //            }
 
+            // 保存WAV文件到本地
+            String savedPath = saveByteArrayToFile(pcmData, "tts_output.wav");
+            if (savedPath != null) {
+                log.info("WAV文件已保存: {}", savedPath);
+            }
+
             // 将byte[]转换为16进制字符串列表
             List<String> hexList = audioProcessUtil.bytesToHexList(pcmData);
 
@@ -113,6 +128,55 @@ public class AudioProcessService {
         }
     }
 
+    public String saveWavFileLocally(String text, String filename) throws IOException {
+        // 参数校验
+        if (text == null || text.trim().isEmpty()) {
+            throw new IllegalArgumentException("文本内容不能为空");
+        }
+
+        if (filename == null || filename.trim().isEmpty()) {
+            filename = "tts_output.wav"; // 默认文件名
+        }
+
+        try {
+            // 生成PCM数据
+            byte[] rawPcmData = alibabaTTSUtil.generateStandardPcmData(text);
+
+            // 转换为标准WAV格式（添加44字节头部）
+            byte[] wavData = audioProcessUtil.rawPcmToStandardWav(rawPcmData);
+
+            // 保存到本地文件
+            String filePath = saveByteArrayToFile(wavData, filename);
+
+            log.info("WAV文件已保存: {}", filePath);
+            return filePath;
+        } catch (Exception e) {
+            log.error("保存WAV文件失败: {}", e.getMessage(), e);
+            throw new IOException("保存WAV文件失败", e);
+        }
+    }
+
+    private String saveByteArrayToFile(byte[] data, String filename) throws IOException {
+        // 确定保存路径（可以是临时目录或指定目录）
+        String directory = System.getProperty("java.io.tmpdir"); // 使用系统临时目录
+        File dir = new File(directory);
+        if (!dir.exists()) {
+            dir.mkdirs();
+        }
+
+        // 创建完整文件路径
+        File file = new File(dir, filename);
+
+        // 写入文件
+        try (FileOutputStream fos = new FileOutputStream(file)) {
+            fos.write(data);
+        }
+
+        return file.getAbsolutePath();
+    }
+
+
+
     /**
      * 验证音频文件
      */
@@ -170,5 +234,75 @@ public class AudioProcessService {
         }
     }
 
+    /**
+     * 提取文本
+     */
+    public String extract(MultipartFile file) throws Exception {
+        String name = file.getOriginalFilename();
+        if (name == null ||
+                (!name.endsWith(".txt") && !name.endsWith(".docx"))) {
+            throw new IllegalArgumentException("仅支持 .txt 或 .docx");
+        }
+        if (file.getSize() > MAX_AUDIO_SIZE) {
+            throw new IllegalArgumentException("文件超过5MB");
+        }
+
+        String text;
+        /* 全程流式，不落地磁盘，不一次性读字节数组 */
+        try (InputStream in = file.getInputStream()) {
+            if (name.endsWith(".txt")) {
+                text = readTxt(in);
+            } else {
+                text = readDocx(in);
+            }
+        }
+        return text;
+    }
+
+    /* ---------- txt：按行读，StringBuilder 复用 ---------- */
+    private String readTxt(InputStream in) throws IOException {
+        BufferedReader br = new BufferedReader(new InputStreamReader(in, StandardCharsets.UTF_8));
+        StringBuilder sb = new StringBuilder(4096);
+        String line;
+        while ((line = br.readLine()) != null) {
+            sb.append(line).append('\n');
+        }
+        return sb.toString();
+    }
+
+    /* ---------- docx：ZipInputStream 只扫 document.xml ---------- */
+    private String readDocx(InputStream in) throws IOException {
+        ZipInputStream zin = new ZipInputStream(in);
+        ZipEntry e;
+        while ((e = zin.getNextEntry()) != null) {
+            if ("word/document.xml".equals(e.getName())) {
+                return staxExtract(zin);   // 流式读 XML
+            }
+        }
+        return "";
+    }
+
+    /* ---------- StAX 流式提取 <w:t> ---------- */
+    private String staxExtract(InputStream xml) throws IOException {
+        XMLStreamReader r = null;
+        StringBuilder sb = new StringBuilder(4096);
+        try {
+            //System.out.println(new String(xml.readAllBytes()));
+            r = XMLInputFactory.newInstance().createXMLStreamReader(xml);
+            while (r.hasNext()) {
+                if (r.next() == XMLStreamConstants.START_ELEMENT &&
+                        "t".equals(r.getLocalName())) {
+                    String elementText = r.getElementText();
+                    sb.append(elementText);
+                }
+            }
+        } catch (XMLStreamException ex) {
+            throw new IOException(ex);
+        } finally {
+            if (r != null) try { r.close(); } catch (XMLStreamException ignore) {}
+        }
+        return sb.toString();
+    }
+
 
 }