util.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792
  1. package my_util
  2. import (
  3. "bytes"
  4. "crypto/md5"
  5. "crypto/sha1"
  6. "crypto/sha256"
  7. "encoding/binary"
  8. "encoding/hex"
  9. "fmt"
  10. "github.com/allanpk716/ChineseSubFinder/internal/pkg/decode"
  11. "github.com/allanpk716/ChineseSubFinder/internal/pkg/regex_things"
  12. "github.com/allanpk716/ChineseSubFinder/internal/pkg/settings"
  13. "github.com/allanpk716/ChineseSubFinder/internal/types/common"
  14. browser "github.com/allanpk716/fake-useragent"
  15. "github.com/go-resty/resty/v2"
  16. "github.com/google/uuid"
  17. "github.com/sirupsen/logrus"
  18. "io"
  19. "math"
  20. "net/http"
  21. "net/url"
  22. "os"
  23. "os/exec"
  24. "path"
  25. "path/filepath"
  26. "regexp"
  27. "runtime"
  28. "strconv"
  29. "strings"
  30. "time"
  31. )
  32. // NewHttpClient 新建一个 resty 的对象
  33. func NewHttpClient(_proxySettings ...*settings.ProxySettings) *resty.Client {
  34. //const defUserAgent = "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50"
  35. //const defUserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.77 Safari/537.36 Edg/91.0.864.41"
  36. var proxySettings *settings.ProxySettings
  37. var HttpProxy, UserAgent, Referer string
  38. if len(_proxySettings) > 0 {
  39. proxySettings = _proxySettings[0]
  40. }
  41. if proxySettings.UseHttpProxy == true && len(proxySettings.HttpProxyAddress) > 0 {
  42. HttpProxy = proxySettings.HttpProxyAddress
  43. }
  44. // 随机的 Browser
  45. UserAgent = browser.Random()
  46. httpClient := resty.New()
  47. httpClient.SetTimeout(common.HTMLTimeOut)
  48. httpClient.SetRetryCount(2)
  49. if HttpProxy != "" {
  50. httpClient.SetProxy(HttpProxy)
  51. } else {
  52. httpClient.RemoveProxy()
  53. }
  54. if len(proxySettings.Referer) > 0 {
  55. Referer = proxySettings.Referer
  56. }
  57. httpClient.SetHeaders(map[string]string{
  58. "Content-Type": "application/json",
  59. "User-Agent": UserAgent,
  60. })
  61. if len(Referer) > 0 {
  62. httpClient.SetHeader("Referer", Referer)
  63. }
  64. return httpClient
  65. }
  66. func GetPublicIP(queue *settings.TaskQueue, _proxySettings ...*settings.ProxySettings) string {
  67. var client *resty.Client
  68. if len(_proxySettings) > 0 {
  69. client = NewHttpClient(_proxySettings[0])
  70. } else {
  71. client = NewHttpClient()
  72. }
  73. targetSite := "http://myexternalip.com/raw"
  74. if queue.CheckPublicIPTargetSite != "" {
  75. targetSite = queue.CheckPublicIPTargetSite
  76. }
  77. response, err := client.R().Get(targetSite)
  78. if err != nil {
  79. return ""
  80. }
  81. return response.String()
  82. }
  83. // DownFile 从指定的 url 下载文件
  84. func DownFile(l *logrus.Logger, urlStr string, _proxySettings ...*settings.ProxySettings) ([]byte, string, error) {
  85. var proxySettings *settings.ProxySettings
  86. if len(_proxySettings) > 0 {
  87. proxySettings = _proxySettings[0]
  88. }
  89. httpClient := NewHttpClient(proxySettings)
  90. resp, err := httpClient.R().Get(urlStr)
  91. if err != nil {
  92. return nil, "", err
  93. }
  94. filename := GetFileName(l, resp.RawResponse)
  95. if filename == "" {
  96. l.Warningln("DownFile.GetFileName is string.empty", urlStr)
  97. }
  98. return resp.Body(), filename, nil
  99. }
  100. // GetFileName 获取下载文件的文件名
  101. func GetFileName(l *logrus.Logger, resp *http.Response) string {
  102. contentDisposition := resp.Header.Get("Content-Disposition")
  103. if len(contentDisposition) == 0 {
  104. m := regexp.MustCompile(`^(.*/)?(?:$|(.+?)(?:(\.[^.]*$)|$))`).FindStringSubmatch(resp.Request.URL.String())
  105. if m == nil || len(m) < 4 {
  106. l.Warningln("GetFileName.regexp.MustCompile.FindStringSubmatch", resp.Request.URL.String())
  107. return ""
  108. }
  109. return m[2] + m[3]
  110. }
  111. re := regexp.MustCompile(`filename=["]*([^"]+)["]*`)
  112. matched := re.FindStringSubmatch(contentDisposition)
  113. if matched == nil || len(matched) == 0 || len(matched[0]) == 0 {
  114. l.Errorln("GetFileName.Content-Disposition", contentDisposition)
  115. return ""
  116. }
  117. return matched[1]
  118. }
  119. // AddBaseUrl 判断传入的 url 是否需要拼接 baseUrl
  120. func AddBaseUrl(baseUrl, url string) string {
  121. if strings.Contains(url, "://") {
  122. return url
  123. }
  124. return fmt.Sprintf("%s%s", baseUrl, url)
  125. }
  126. // IsDir 存在且是文件夹
  127. func IsDir(path string) bool {
  128. s, err := os.Stat(path)
  129. if err != nil {
  130. return false
  131. }
  132. return s.IsDir()
  133. }
  134. // IsFile 存在且是文件
  135. func IsFile(filePath string) bool {
  136. s, err := os.Stat(filePath)
  137. if err != nil {
  138. return false
  139. }
  140. return !s.IsDir()
  141. }
  142. // VideoNameSearchKeywordMaker 拼接视频搜索的 title 和 年份
  143. func VideoNameSearchKeywordMaker(l *logrus.Logger, title string, year string) string {
  144. iYear, err := strconv.Atoi(year)
  145. if err != nil {
  146. // 允许的错误
  147. l.Errorln("VideoNameSearchKeywordMaker", "year to int", err)
  148. iYear = 0
  149. }
  150. searchKeyword := title
  151. if iYear >= 2020 {
  152. searchKeyword = searchKeyword + " " + year
  153. }
  154. return searchKeyword
  155. }
  156. // SearchMatchedVideoFileFromDirs 搜索符合后缀名的视频文件
  157. func SearchMatchedVideoFileFromDirs(l *logrus.Logger, dirs []string) ([]string, error) {
  158. defer func() {
  159. l.Debugln("SearchMatchedVideoFileFromDirs End ----------------")
  160. }()
  161. l.Debugln("SearchMatchedVideoFileFromDirs Start ----------------")
  162. var fileFullPathList = make([]string, 0)
  163. for _, dir := range dirs {
  164. matchedVideoFile, err := SearchMatchedVideoFile(l, dir)
  165. if err != nil {
  166. return nil, err
  167. }
  168. fileFullPathList = append(fileFullPathList, matchedVideoFile...)
  169. }
  170. for _, s := range fileFullPathList {
  171. l.Debugln(s)
  172. }
  173. return fileFullPathList, nil
  174. }
  175. // SearchMatchedVideoFile 搜索符合后缀名的视频文件,现在也会把 BDMV 的文件搜索出来,但是这个并不是一个视频文件,需要在后续特殊处理
  176. func SearchMatchedVideoFile(l *logrus.Logger, dir string) ([]string, error) {
  177. var fileFullPathList = make([]string, 0)
  178. pathSep := string(os.PathSeparator)
  179. files, err := os.ReadDir(dir)
  180. if err != nil {
  181. return nil, err
  182. }
  183. for _, curFile := range files {
  184. fullPath := dir + pathSep + curFile.Name()
  185. if curFile.IsDir() {
  186. // 内层的错误就无视了
  187. oneList, _ := SearchMatchedVideoFile(l, fullPath)
  188. if oneList != nil {
  189. fileFullPathList = append(fileFullPathList, oneList...)
  190. }
  191. } else {
  192. // 这里就是文件了
  193. bok, fakeBDMVVideoFile := FileNameIsBDMV(fullPath)
  194. if bok == true {
  195. // 这类文件后续的扫描字幕操作需要额外的处理
  196. fileFullPathList = append(fileFullPathList, fakeBDMVVideoFile)
  197. continue
  198. }
  199. if IsWantedVideoExtDef(curFile.Name()) == false {
  200. // 不是期望的视频后缀名则跳过
  201. continue
  202. } else {
  203. // 这里还有一种情况,就是蓝光, BDMV 下面会有一个 STREAM 文件夹,里面很多 m2ts 的视频组成
  204. if filepath.Base(filepath.Dir(fullPath)) == "STREAM" {
  205. l.Debugln("SearchMatchedVideoFile, Skip BDMV.STREAM:", fullPath)
  206. continue
  207. }
  208. // 跳过不符合的文件,比如 MAC OS 下可能有缓存文件,见 #138
  209. fi, err := curFile.Info()
  210. if err != nil {
  211. l.Debugln("SearchMatchedVideoFile, file.Info:", fullPath, err)
  212. continue
  213. }
  214. if fi.Size() == 4096 && strings.HasPrefix(curFile.Name(), "._") == true {
  215. l.Debugln("SearchMatchedVideoFile file.Size() == 4096 && Prefix Name == ._*", fullPath)
  216. continue
  217. }
  218. fileFullPathList = append(fileFullPathList, fullPath)
  219. }
  220. }
  221. }
  222. return fileFullPathList, nil
  223. }
  224. // FileNameIsBDMV 是否是 BDMV 蓝光目录,符合返回 true,以及 fakseVideoFPath
  225. func FileNameIsBDMV(id_bdmv_fileFPath string) (bool, string) {
  226. /*
  227. 这类蓝光视频比较特殊,它没有具体的一个后缀名的视频文件而是由两个文件夹来存储视频数据
  228. * BDMV
  229. * CERTIFICATE
  230. 但是不管如何,都需要使用一个文件作为锚点,就选定 CERTIFICATE 中的 id.bdmv 文件
  231. 后续的下载逻辑也需要单独为这个文件进行处理,比如,从这个文件向上一层获取 nfo 文件,
  232. 以及再上一层得到视频文件夹名称等
  233. */
  234. if strings.ToLower(filepath.Base(id_bdmv_fileFPath)) == common.FileBDMV {
  235. // 这个文件是确认了,那么就需要查看这个文件父级目录是不是 CERTIFICATE 文件夹
  236. // 且 CERTIFICATE 需要和 BDMV 文件夹都存在
  237. CERDir := filepath.Dir(id_bdmv_fileFPath)
  238. BDMVDir := filepath.Join(filepath.Dir(CERDir), "BDMV")
  239. if IsDir(CERDir) == true && IsDir(BDMVDir) == true {
  240. return true, filepath.Join(filepath.Dir(CERDir), filepath.Base(filepath.Dir(CERDir))+common.VideoExtMp4)
  241. }
  242. }
  243. return false, ""
  244. }
  245. func SearchTVNfo(l *logrus.Logger, dir string) ([]string, error) {
  246. var fileFullPathList = make([]string, 0)
  247. pathSep := string(os.PathSeparator)
  248. files, err := os.ReadDir(dir)
  249. if err != nil {
  250. return nil, err
  251. }
  252. for _, curFile := range files {
  253. fullPath := dir + pathSep + curFile.Name()
  254. if curFile.IsDir() {
  255. // 内层的错误就无视了
  256. oneList, _ := SearchTVNfo(l, fullPath)
  257. if oneList != nil {
  258. fileFullPathList = append(fileFullPathList, oneList...)
  259. }
  260. } else {
  261. // 这里就是文件了
  262. if strings.ToLower(curFile.Name()) != decode.MetadateTVNfo {
  263. continue
  264. } else {
  265. // 跳过不符合的文件,比如 MAC OS 下可能有缓存文件,见 #138
  266. fi, err := curFile.Info()
  267. if err != nil {
  268. l.Debugln("SearchTVNfo, file.Info:", fullPath, err)
  269. continue
  270. }
  271. if fi.Size() == 4096 && strings.HasPrefix(curFile.Name(), "._") == true {
  272. l.Debugln("SearchTVNfo file.Size() == 4096 && Prefix Name == ._*", fullPath)
  273. continue
  274. }
  275. fileFullPathList = append(fileFullPathList, fullPath)
  276. }
  277. }
  278. }
  279. return fileFullPathList, nil
  280. }
  281. // IsWantedVideoExtDef 后缀名是否符合规则
  282. func IsWantedVideoExtDef(fileName string) bool {
  283. if len(_wantedExtMap) < 1 {
  284. _defExtMap[common.VideoExtMp4] = common.VideoExtMp4
  285. _defExtMap[common.VideoExtMkv] = common.VideoExtMkv
  286. _defExtMap[common.VideoExtRmvb] = common.VideoExtRmvb
  287. _defExtMap[common.VideoExtIso] = common.VideoExtIso
  288. _defExtMap[common.VideoExtM2ts] = common.VideoExtM2ts
  289. _wantedExtMap[common.VideoExtMp4] = common.VideoExtMp4
  290. _wantedExtMap[common.VideoExtMkv] = common.VideoExtMkv
  291. _wantedExtMap[common.VideoExtRmvb] = common.VideoExtRmvb
  292. _wantedExtMap[common.VideoExtIso] = common.VideoExtIso
  293. _wantedExtMap[common.VideoExtM2ts] = common.VideoExtM2ts
  294. for _, videoExt := range _customVideoExts {
  295. _wantedExtMap[videoExt] = videoExt
  296. }
  297. }
  298. fileExt := strings.ToLower(filepath.Ext(fileName))
  299. _, bFound := _wantedExtMap[fileExt]
  300. return bFound
  301. }
  302. func GetEpisodeKeyName(season, eps int) string {
  303. return "S" + strconv.Itoa(season) + "E" + strconv.Itoa(eps)
  304. }
  305. // CopyFile copies a single file from src to dst
  306. func CopyFile(src, dst string) error {
  307. var err error
  308. var srcFd *os.File
  309. var dstFd *os.File
  310. var srcInfo os.FileInfo
  311. if srcFd, err = os.Open(src); err != nil {
  312. return err
  313. }
  314. defer func() {
  315. _ = srcFd.Close()
  316. }()
  317. if dstFd, err = os.Create(dst); err != nil {
  318. return err
  319. }
  320. defer func() {
  321. _ = dstFd.Close()
  322. }()
  323. if _, err = io.Copy(dstFd, srcFd); err != nil {
  324. return err
  325. }
  326. if srcInfo, err = os.Stat(src); err != nil {
  327. return err
  328. }
  329. return os.Chmod(dst, srcInfo.Mode())
  330. }
  331. // CopyDir copies a whole directory recursively
  332. func CopyDir(src string, dst string) error {
  333. var err error
  334. var fds []os.DirEntry
  335. var srcInfo os.FileInfo
  336. if srcInfo, err = os.Stat(src); err != nil {
  337. return err
  338. }
  339. if err = os.MkdirAll(dst, srcInfo.Mode()); err != nil {
  340. return err
  341. }
  342. if fds, err = os.ReadDir(src); err != nil {
  343. return err
  344. }
  345. for _, fd := range fds {
  346. srcfp := filepath.Join(src, fd.Name())
  347. dstfp := filepath.Join(dst, fd.Name())
  348. if fd.IsDir() {
  349. if err = CopyDir(srcfp, dstfp); err != nil {
  350. fmt.Println(err)
  351. }
  352. } else {
  353. if err = CopyFile(srcfp, dstfp); err != nil {
  354. fmt.Println(err)
  355. }
  356. }
  357. }
  358. return nil
  359. }
  360. // CloseChrome 强行结束没有关闭的 Chrome 进程
  361. func CloseChrome(l *logrus.Logger) {
  362. cmdString := ""
  363. var command *exec.Cmd
  364. sysType := runtime.GOOS
  365. if sysType == "linux" {
  366. // LINUX系统
  367. cmdString = "pkill chrome"
  368. command = exec.Command("/bin/sh", "-c", cmdString)
  369. }
  370. if sysType == "windows" {
  371. // windows系统
  372. cmdString = "taskkill /F /im chrome.exe"
  373. command = exec.Command("cmd.exe", "/c", cmdString)
  374. }
  375. if sysType == "darwin" {
  376. // macOS
  377. // https://stackoverflow.com/questions/57079120/using-exec-command-in-golang-how-do-i-open-a-new-terminal-and-execute-a-command
  378. cmdString = `tell application "/Applications/Google Chrome.app" to quit`
  379. command = exec.Command("osascript", "-s", "h", "-e", cmdString)
  380. }
  381. if cmdString == "" || command == nil {
  382. l.Errorln("CloseChrome OS:", sysType)
  383. return
  384. }
  385. err := command.Run()
  386. if err != nil {
  387. l.Warningln("CloseChrome", err)
  388. }
  389. }
  390. // OSCheck 强制的系统支持检查
  391. func OSCheck() bool {
  392. sysType := runtime.GOOS
  393. if sysType == "linux" {
  394. return true
  395. }
  396. if sysType == "windows" {
  397. return true
  398. }
  399. if sysType == "darwin" {
  400. return true
  401. }
  402. return false
  403. }
  404. // FixWindowPathBackSlash 修复 Windows 反斜杠的梗
  405. func FixWindowPathBackSlash(path string) string {
  406. return strings.Replace(path, string(filepath.Separator), "/", -1)
  407. }
  408. func WriteStrings2File(desfilePath string, strings []string) error {
  409. dstFile, err := os.Create(desfilePath)
  410. if err != nil {
  411. return err
  412. }
  413. defer func() {
  414. _ = dstFile.Close()
  415. }()
  416. allString := ""
  417. for _, s := range strings {
  418. allString += s + "\r\n"
  419. }
  420. _, err = dstFile.WriteString(allString)
  421. if err != nil {
  422. return err
  423. }
  424. return nil
  425. }
  426. func TimeNumber2Time(inputTimeNumber float64) time.Time {
  427. newTime := time.Time{}.Add(time.Duration(inputTimeNumber * math.Pow10(9)))
  428. return newTime
  429. }
  430. func Time2SecondNumber(inTime time.Time) float64 {
  431. outSecond := 0.0
  432. outSecond += float64(inTime.Hour() * 60 * 60)
  433. outSecond += float64(inTime.Minute() * 60)
  434. outSecond += float64(inTime.Second())
  435. outSecond += float64(inTime.Nanosecond()) / 1000 / 1000 / 1000
  436. return outSecond
  437. }
  438. func Time2Duration(inTime time.Time) time.Duration {
  439. return time.Duration(Time2SecondNumber(inTime) * math.Pow10(9))
  440. }
  441. func Second2Time(sec int64) time.Time {
  442. return time.Unix(sec, 0)
  443. }
  444. // ReplaceSpecString 替换特殊的字符
  445. func ReplaceSpecString(inString string, rep string) string {
  446. return regex_things.RegMatchSpString.ReplaceAllString(inString, rep)
  447. }
  448. func Bool2Int(inBool bool) int {
  449. if inBool == true {
  450. return 1
  451. } else {
  452. return 0
  453. }
  454. }
  455. // Round 取整
  456. func Round(x float64) int64 {
  457. if x-float64(int64(x)) > 0 {
  458. return int64(x) + 1
  459. } else {
  460. return int64(x)
  461. }
  462. //return int64(math.Floor(x + 0.5))
  463. }
  464. // MakePowerOfTwo 2的整次幂数 buffer length is not a power of two
  465. func MakePowerOfTwo(x int64) int64 {
  466. power := math.Log2(float64(x))
  467. tmpRound := Round(power)
  468. return int64(math.Pow(2, float64(tmpRound)))
  469. }
  470. // MakeCeil10msMultipleFromFloat 将传入的秒,规整到 10ms 的倍数,返回依然是 秒,向上取整
  471. func MakeCeil10msMultipleFromFloat(input float64) float64 {
  472. const bb = 100
  473. // 先转到 10 ms 单位,比如传入是 1.912 - > 191.2
  474. t10ms := input * bb
  475. // 191.2 - > 192.0
  476. newT10ms := math.Ceil(t10ms)
  477. // 转换回来
  478. return newT10ms / bb
  479. }
  480. // MakeFloor10msMultipleFromFloat 将传入的秒,规整到 10ms 的倍数,返回依然是 秒,向下取整
  481. func MakeFloor10msMultipleFromFloat(input float64) float64 {
  482. const bb = 100
  483. // 先转到 10 ms 单位,比如传入是 1.912 - > 191.2
  484. t10ms := input * bb
  485. // 191.2 - > 191.0
  486. newT10ms := math.Floor(t10ms)
  487. // 转换回来
  488. return newT10ms / bb
  489. }
  490. // MakeCeil10msMultipleFromTime 向上取整,规整到 10ms 的倍数
  491. func MakeCeil10msMultipleFromTime(input time.Time) time.Time {
  492. nowTime := MakeCeil10msMultipleFromFloat(Time2SecondNumber(input))
  493. newTime := time.Time{}.Add(time.Duration(nowTime * math.Pow10(9)))
  494. return newTime
  495. }
  496. // MakeFloor10msMultipleFromTime 向下取整,规整到 10ms 的倍数
  497. func MakeFloor10msMultipleFromTime(input time.Time) time.Time {
  498. nowTime := MakeFloor10msMultipleFromFloat(Time2SecondNumber(input))
  499. newTime := time.Time{}.Add(time.Duration(nowTime * math.Pow10(9)))
  500. return newTime
  501. }
  502. // Time2SubTimeString 时间转字幕格式的时间字符串
  503. func Time2SubTimeString(inTime time.Time, timeFormat string) string {
  504. /*
  505. 这里进行时间转字符串的时候有一点比较特殊
  506. 正常来说输出的格式是类似 15:04:05.00
  507. 那么有个问题,字幕的时间格式是 0:00:12.00, 小时,是个数,除非有跨度到 20 小时的视频,不然小时就应该是个数
  508. 这就需要一个额外的函数去处理这些情况
  509. */
  510. outTimeString := inTime.Format(timeFormat)
  511. if inTime.Hour() > 9 {
  512. // 小时,两位数
  513. return outTimeString
  514. } else {
  515. // 小时,一位数
  516. items := strings.SplitN(outTimeString, ":", -1)
  517. if len(items) == 3 {
  518. outTimeString = strings.Replace(outTimeString, items[0], fmt.Sprintf("%d", inTime.Hour()), 1)
  519. return outTimeString
  520. }
  521. return outTimeString
  522. }
  523. }
  524. // IsEqual 比较 float64
  525. func IsEqual(f1, f2 float64) bool {
  526. const MIN = 0.000001
  527. if f1 > f2 {
  528. return math.Dim(f1, f2) < MIN
  529. } else {
  530. return math.Dim(f2, f1) < MIN
  531. }
  532. }
  533. // ParseTime 解析字幕时间字符串,这里可能小数点后面有 2-4 位
  534. func ParseTime(inTime string) (time.Time, error) {
  535. parseTime, err := time.Parse(common.TimeFormatPoint2, inTime)
  536. if err != nil {
  537. parseTime, err = time.Parse(common.TimeFormatPoint3, inTime)
  538. if err != nil {
  539. parseTime, err = time.Parse(common.TimeFormatPoint4, inTime)
  540. }
  541. }
  542. return parseTime, err
  543. }
  544. // GetFileSHA1 获取文件的 SHA1 值
  545. func GetFileSHA1(srcFileFPath string) (string, error) {
  546. infile, err := os.Open(srcFileFPath)
  547. if err != nil {
  548. return "", err
  549. }
  550. defer func() {
  551. _ = infile.Close()
  552. }()
  553. h := sha1.New()
  554. _, err = io.Copy(h, infile)
  555. if err != nil {
  556. return "", err
  557. }
  558. return hex.EncodeToString(h.Sum(nil)), nil
  559. }
  560. // WriteFile 写文件
  561. func WriteFile(desFileFPath string, bytes []byte) error {
  562. var err error
  563. nowDesPath := desFileFPath
  564. if filepath.IsAbs(nowDesPath) == false {
  565. nowDesPath, err = filepath.Abs(nowDesPath)
  566. if err != nil {
  567. return err
  568. }
  569. }
  570. // 创建对应的目录
  571. nowDirPath := filepath.Dir(nowDesPath)
  572. err = os.MkdirAll(nowDirPath, os.ModePerm)
  573. if err != nil {
  574. return err
  575. }
  576. file, err := os.Create(nowDesPath)
  577. if err != nil {
  578. return err
  579. }
  580. defer func() {
  581. _ = file.Close()
  582. }()
  583. _, err = file.Write(bytes)
  584. if err != nil {
  585. return err
  586. }
  587. return nil
  588. }
  589. // GetNowTimeString 获取当前的时间,没有秒
  590. func GetNowTimeString() (string, int, int, int) {
  591. nowTime := time.Now()
  592. addString := fmt.Sprintf("%d-%d-%d", nowTime.Hour(), nowTime.Minute(), nowTime.Nanosecond())
  593. return addString, nowTime.Hour(), nowTime.Minute(), nowTime.Nanosecond()
  594. }
  595. // GenerateAccessToken 生成随机的 AccessToken
  596. func GenerateAccessToken() string {
  597. u4 := uuid.New()
  598. return u4.String()
  599. }
  600. func Get2UUID() string {
  601. u4 := uuid.New()
  602. u5 := uuid.New()
  603. return u4.String() + u5.String()
  604. }
  605. func UrlJoin(hostUrl, subUrl string) (string, error) {
  606. u, err := url.Parse(hostUrl)
  607. if err != nil {
  608. return "", err
  609. }
  610. u.Path = path.Join(u.Path, subUrl)
  611. return u.String(), nil
  612. }
  613. // GetFileSHA1String 获取文件的 SHA1 字符串
  614. func GetFileSHA1String(fileFPath string) (string, error) {
  615. h := sha1.New()
  616. fp, err := os.Open(fileFPath)
  617. if err != nil {
  618. return "", err
  619. }
  620. defer func() {
  621. _ = fp.Close()
  622. }()
  623. partAll, err := io.ReadAll(fp)
  624. if err != nil {
  625. return "", err
  626. }
  627. h.Write(partAll)
  628. hashBytes := h.Sum(nil)
  629. return fmt.Sprintf("%x", md5.Sum(hashBytes)), nil
  630. }
  631. // GetFileSHA256String 获取文件的 SHA256 字符串
  632. func GetFileSHA256String(fileFPath string) (string, error) {
  633. fp, err := os.Open(fileFPath)
  634. if err != nil {
  635. return "", err
  636. }
  637. defer func() {
  638. _ = fp.Close()
  639. }()
  640. partAll, err := io.ReadAll(fp)
  641. if err != nil {
  642. return "", err
  643. }
  644. return fmt.Sprintf("%x", sha256.Sum256(partAll)), nil
  645. }
  646. func GetRestOfDaySec() time.Duration {
  647. nowTime := time.Now()
  648. todayLast := nowTime.Format("2006-01-02") + " 23:59:59"
  649. todayLastTime, _ := time.ParseInLocation("2006-01-02 15:04:05", todayLast, time.Local)
  650. // 今天剩余的时间(s)
  651. restOfDaySec := time.Duration(todayLastTime.Unix()-time.Now().Local().Unix()) * time.Second
  652. return restOfDaySec
  653. }
  654. // IntToBytes 整形转换成字节
  655. func IntToBytes(n int) ([]byte, error) {
  656. x := int32(n)
  657. bytesBuffer := bytes.NewBuffer([]byte{})
  658. err := binary.Write(bytesBuffer, binary.BigEndian, x)
  659. if err != nil {
  660. return nil, err
  661. }
  662. return bytesBuffer.Bytes(), nil
  663. }
  664. // BytesToInt 字节转换成整形
  665. func BytesToInt(b []byte) (int, error) {
  666. bytesBuffer := bytes.NewBuffer(b)
  667. var x int32
  668. err := binary.Read(bytesBuffer, binary.BigEndian, &x)
  669. if err != nil {
  670. return 0, err
  671. }
  672. return int(x), nil
  673. }
  674. var (
  675. _wantedExtMap = make(map[string]string) // 人工确认的需要监控的视频后缀名
  676. _defExtMap = make(map[string]string) // 内置支持的视频后缀名列表
  677. _customVideoExts = make([]string, 0) // 用户额外自定义的视频后缀名列表
  678. )