feat: support srt

srt.go

@@ -0,0 +1,86 @@
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+)
+
+func parseSRT(filePath string) ([]SRTEntry, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+
+	scanner := bufio.NewScanner(file)
+	var entries []SRTEntry
+	var currentEntry SRTEntry
+
+	for scanner.Scan() {
+		line := strings.TrimSpace(scanner.Text())
+		if line == "" {
+			if currentEntry.Number != 0 {
+				entries = append(entries, currentEntry)
+				currentEntry = SRTEntry{}
+			}
+			continue
+		}
+
+		if currentEntry.Number == 0 {
+			currentEntry.Number, _ = strconv.Atoi(line)
+		} else if currentEntry.StartTime.Hours == 0 {
+			times := strings.Split(line, " --> ")
+			currentEntry.StartTime = parseSRTTimestamp(times[0])
+			currentEntry.EndTime = parseSRTTimestamp(times[1])
+		} else {
+			currentEntry.Content += line + "\n"
+		}
+	}
+
+	if currentEntry.Number != 0 {
+		entries = append(entries, currentEntry)
+	}
+
+	return entries, scanner.Err()
+}
+
+func convertSRT(sourceFile, targetFile, targetFmt string) {
+	switch targetFmt {
+	case "txt":
+		srtToTxt(sourceFile, targetFile)
+	case "lrc":
+		srtToLrc(sourceFile, targetFile)
+	default:
+		fmt.Printf("unsupported target format: %s\n", targetFmt)
+	}
+}
+
+func formatSRTTimestamp(ts Timestamp) string {
+	return fmt.Sprintf("%02d:%02d:%02d,%03d", ts.Hours, ts.Minutes, ts.Seconds, ts.Milliseconds)
+}
+
+func parseSRTTimestamp(timeStr string) Timestamp {
+	t, _ := time.Parse("15:04:05,000", timeStr)
+	return Timestamp{
+		Hours:        t.Hour(),
+		Minutes:      t.Minute(),
+		Seconds:      t.Second(),
+		Milliseconds: t.Nanosecond() / 1e6,
+	}
+}
+
+// basically for the last line of lrc
+func addSeconds(ts Timestamp, seconds int) Timestamp {
+	t := time.Date(0, 1, 1, ts.Hours, ts.Minutes, ts.Seconds, ts.Milliseconds*1e6, time.UTC)
+	t = t.Add(time.Duration(seconds) * time.Second)
+	return Timestamp{
+		Hours:        t.Hour(),
+		Minutes:      t.Minute(),
+		Seconds:      t.Second(),
+		Milliseconds: t.Nanosecond() / 1e6,
+	}
+}