From 25a09d120098574711a3f007f35df44771da387c Mon Sep 17 00:00:00 2001
From: ac79b0c6 <105047743+ac79b0c6@users.noreply.github.com>
Date: Fri, 20 May 2022 18:16:59 +0800
Subject: update mkvlib:add overwrite switch

---
 mkvlib/mkv.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'mkvlib/mkv.go')

diff --git a/mkvlib/mkv.go b/mkvlib/mkv.go
index 46da74b..d85e8d9 100644
--- a/mkvlib/mkv.go
+++ b/mkvlib/mkv.go
@@ -48,6 +48,7 @@ type mkvProcessor struct {
 	nrename    bool
 	check      bool
 	strict     bool
+	overwrite  bool
 }
 
 func (self *mkvProcessor) GetMKVInfo(file string) *mkvInfo {
@@ -256,7 +257,7 @@ func (self *mkvProcessor) CreateMKVs(vDir, sDir, fDir, tDir, oDir, slang, stitle
 		} else {
 			fn += ".mkv"
 		}
-		if _a, _ := isExists(fn); _a {
+		if _a, _ := isExists(fn); _a && !self.overwrite {
 			printLog(lcb, `@Warning@ Existing file: "%s",skip.`, fn)
 			_ok++
 			printLog(lcb, "Create (%d/%d) done.", _ok, l)
@@ -315,7 +316,7 @@ func (self *mkvProcessor) MakeMKVs(dir, data, output, slang, stitle string, lcb
 		} else {
 			fn += ".mkv"
 		}
-		if _a, _ := isExists(fn); _a {
+		if _a, _ := isExists(fn); _a && !self.overwrite {
 			printLog(lcb, `@Warning@ Existing file: "%s",skip.`, fn)
 			_ok++
 			printLog(lcb, "Make (%d/%d) done.", _ok, l)
@@ -445,6 +446,10 @@ func (self *mkvProcessor) NRename(nrename bool) {
 	self.nrename = nrename
 }
 
+func (self *mkvProcessor) Overwrite(o bool) {
+	self.overwrite = o
+}
+
 func (self *mkvProcessor) CreateBlankOrBurnVideo(t int64, s, enc, ass, fontdir, output string) bool {
 	if !self.ffmpeg {
 		return false
-- 
cgit v1.2.1