支持ass转换
This commit is contained in:
Hami Lemon
parent
a29d2c057e
commit
eba676b8de
6 changed files with 406 additions and 105 deletions
|
|
@ -1,135 +1,122 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/Hami-Lemon/ltc"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/Hami-Lemon/ltc"
|
||||
"github.com/jessevdk/go-flags"
|
||||
)
|
||||
|
||||
type Option struct {
|
||||
Id string `short:"i" long:"id" description:"歌曲的id，网易云和QQ音乐均可。"`
|
||||
Input string `short:"I" long:"input" description:"需要转换的LRC文件路径。"`
|
||||
Source string `short:"s" long:"source" description:"当设置id时有效，指定从网易云（163）还是QQ音乐（qq）上获取歌词。" default:"163" choice:"163" choice:"qq" choice:"QQ"`
|
||||
Download bool `short:"d" long:"download" description:"只下载歌词，而不进行解析。"`
|
||||
Mode int `short:"m" long:"mode" default:"1" description:"原文和译文的排列模式,可选值有：[1] [2] [3]" choice:"1" choice:"2" choice:"3"`
|
||||
Version bool `short:"v" long:"version" description:"获取版本信息"`
|
||||
Output string `no-flag:""`
|
||||
}
|
||||
|
||||
const (
|
||||
// VERSION 当前版本
|
||||
VERSION = `"0.2.4" (build 2022.03.29)`
|
||||
)
|
||||
|
||||
var (
|
||||
opt Option
|
||||
VERSION = `"0.3.4" (build 2022.03.30)`
|
||||
VERSION_INFO = "LrcToCaptions(ltc) version: %s\n"
|
||||
)
|
||||
|
||||
func main() {
|
||||
//TODO 支持转ass文件
|
||||
//TODO 酷狗的krc支持逐字，更利于打轴 https://shansing.com/read/392/
|
||||
args, err := flags.Parse(&opt)
|
||||
if err != nil {
|
||||
os.Exit(0)
|
||||
}
|
||||
parseFlag()
|
||||
//TODO 酷狗的krc精准到字，更利于打轴 https://shansing.com/read/392/
|
||||
//显示版本信息
|
||||
if opt.Version {
|
||||
fmt.Printf("LrcToCaptions(ltc) version: %s\n", VERSION)
|
||||
if version.IsSet() {
|
||||
fmt.Printf(VERSION_INFO, VERSION)
|
||||
return
|
||||
}
|
||||
//获取保存的文件名
|
||||
if len(args) != 0 {
|
||||
opt.Output = args[0]
|
||||
//未指定来源
|
||||
if input == "" {
|
||||
fmt.Printf("未指定歌词来源\n")
|
||||
flag.Usage()
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
//获取歌词，lyric为原文歌词，tranLyric为译文歌词
|
||||
var lyric, tranLyric string
|
||||
if opt.Id != "" {
|
||||
if opt.Source != "163" {
|
||||
lyric, tranLyric = ltc.GetQQLyric(opt.Id)
|
||||
} else {
|
||||
lyric, tranLyric = ltc.Get163Lyric(opt.Id)
|
||||
}
|
||||
//下载歌词
|
||||
if opt.Download {
|
||||
//对文件名进行处理
|
||||
o := opt.Output
|
||||
if o == "" {
|
||||
o = opt.Id + ".lrc"
|
||||
} else if !strings.HasSuffix(o, ".lrc") {
|
||||
o += ".lrc"
|
||||
}
|
||||
ltc.WriteFile(o, lyric)
|
||||
if tranLyric != "" {
|
||||
ltc.WriteFile("tran_"+o, tranLyric)
|
||||
}
|
||||
fmt.Println("下载歌词完成！")
|
||||
return
|
||||
}
|
||||
} else if opt.Input != "" {
|
||||
//从文件中获取歌词
|
||||
if !strings.HasSuffix(opt.Input, ".lrc") {
|
||||
//从文件中获取
|
||||
if checkPath(input) {
|
||||
if !strings.HasSuffix(input, ".lrc") {
|
||||
fmt.Println("Error: 不支持的格式，目前只支持lrc歌词文件。")
|
||||
os.Exit(1)
|
||||
panic("")
|
||||
}
|
||||
lyric = ltc.ReadFile(opt.Input)
|
||||
if lyric == "" {
|
||||
fmt.Println("获取歌词失败，文件内容为空。")
|
||||
os.Exit(1)
|
||||
if data, err := ioutil.ReadFile(input); err == nil {
|
||||
if len(data) == 0 {
|
||||
fmt.Println("获取歌词失败，文件内容为空。")
|
||||
panic("")
|
||||
}
|
||||
lyric = string(data)
|
||||
} else {
|
||||
panic("读取文件失败:" + input + err.Error())
|
||||
}
|
||||
} else {
|
||||
fmt.Println("Error: 请指定需要转换的歌词。")
|
||||
os.Exit(1)
|
||||
//从网络上获取
|
||||
if source != "163" {
|
||||
lyric, tranLyric = ltc.GetQQLyric(input)
|
||||
} else {
|
||||
lyric, tranLyric = ltc.Get163Lyric(input)
|
||||
}
|
||||
}
|
||||
|
||||
//下载歌词
|
||||
if download.IsSet() {
|
||||
//对文件名进行处理
|
||||
o := output
|
||||
if o == "" {
|
||||
o = input + ".lrc"
|
||||
} else if !strings.HasSuffix(o, ".lrc") {
|
||||
o += ".lrc"
|
||||
}
|
||||
writeFile(o, lyric)
|
||||
if tranLyric != "" {
|
||||
writeFile("tran_"+o, tranLyric)
|
||||
}
|
||||
fmt.Println("下载歌词完成！")
|
||||
return
|
||||
}
|
||||
|
||||
lrc, lrcT := ltc.ParseLRC(lyric), ltc.ParseLRC(tranLyric)
|
||||
//先转换成srt
|
||||
srt, srtT := ltc.LrcToSrt(lrc), ltc.LrcToSrt(lrcT)
|
||||
if srtT != nil {
|
||||
var mode ltc.SRTMergeMode
|
||||
switch opt.Mode {
|
||||
case 1:
|
||||
mode = ltc.SRT_MERGE_MODE_STACK
|
||||
case 2:
|
||||
mode = ltc.SRT_MERGE_MODE_UP
|
||||
case 3:
|
||||
mode = ltc.SRT_MERGE_MODE_BOTTOM
|
||||
}
|
||||
srt.Merge(srtT, mode)
|
||||
//原文和译文合并
|
||||
srt.Merge(srtT, mode.Mode())
|
||||
}
|
||||
name := opt.Output
|
||||
if name == "" {
|
||||
title := srt.Title
|
||||
if title != "" {
|
||||
//以歌曲名命名
|
||||
name = fmt.Sprintf("%s.srt", title)
|
||||
} else if opt.Id != "" {
|
||||
//以歌曲的id命名
|
||||
name = fmt.Sprintf("%s.srt", opt.Id)
|
||||
} else if opt.Input != "" {
|
||||
//以LRC文件的文件名命名
|
||||
name = fmt.Sprintf("%s.srt", opt.Input[:len(opt.Input)-4])
|
||||
} else {
|
||||
//以当前时间的毫秒值命名
|
||||
name = fmt.Sprintf("%d.srt", time.Now().Unix())
|
||||
switch format.Value() {
|
||||
case FORMAT_SRT:
|
||||
if err := srt.WriteFile(output); err != nil {
|
||||
fmt.Println("出现错误,保存失败")
|
||||
panic(err.Error())
|
||||
}
|
||||
case FORMAT_ASS:
|
||||
ass := ltc.SrtToAss(srt)
|
||||
if err := ass.WriteFile(output); err != nil {
|
||||
fmt.Println("出现错误,保存失败")
|
||||
panic(err.Error())
|
||||
}
|
||||
} else if !strings.HasSuffix(name, ".srt") {
|
||||
name += ".srt"
|
||||
}
|
||||
if err = srt.WriteFile(name); err != nil {
|
||||
fmt.Println("出现错误,保存失败")
|
||||
panic(err.Error())
|
||||
}
|
||||
//如果是相对路径，则获取其对应的绝对路径
|
||||
if !filepath.IsAbs(name) {
|
||||
if !filepath.IsAbs(output) {
|
||||
//如果是相对路径，父目录即是当前运行路径
|
||||
dir, er := os.Getwd()
|
||||
if er == nil {
|
||||
name = dir + string(os.PathSeparator) + name
|
||||
output = dir + string(os.PathSeparator) + output
|
||||
}
|
||||
}
|
||||
fmt.Printf("保存结果为：%s\n", name)
|
||||
fmt.Printf("保存结果为：%s\n", output)
|
||||
}
|
||||
|
||||
func writeFile(file string, content string) {
|
||||
f, err := os.Create(file)
|
||||
if err != nil {
|
||||
fmt.Printf("创建结果文件失败：%v\n", err)
|
||||
panic("")
|
||||
}
|
||||
defer f.Close()
|
||||
writer := bufio.NewWriter(f)
|
||||
_, err = writer.WriteString(content)
|
||||
err = writer.Flush()
|
||||
if err != nil {
|
||||
fmt.Printf("保存文件失败：%v\n", err)
|
||||
panic("")
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Reference in a new issue