song cleanup (#511)

* refactor song background logic * move song video generation * move subtitle generation * build ASS subtitles * randomize song detail layout * update changelog
2021-11-27 11:15:53 -06:00
parent 919325033d
commit 7edf6f5d13
10 changed files with 463 additions and 197 deletions
@@ -0,0 +1,138 @@
+using System.IO;
+using System.Text;
+using System.Threading.Tasks;
+using ErsatzTV.Core.Interfaces.FFmpeg;
+using LanguageExt;
+using static LanguageExt.Prelude;
+
+namespace ErsatzTV.Core.FFmpeg
+{
+    public class SubtitleBuilder
+    {
+        private readonly ITempFilePool _tempFilePool;
+        private string _content;
+        private Option<IDisplaySize> _resolution = None;
+        private Option<string> _fontName;
+        private Option<int> _fontSize;
+        private Option<string> _primaryColor;
+        private Option<string> _outlineColor;
+        private Option<int> _alignment;
+        private int _marginRight;
+        private int _marginLeft;
+        private int _marginV;
+        private Option<int> _borderStyle;
+        private Option<int> _shadow;
+
+        public SubtitleBuilder(ITempFilePool tempFilePool)
+        {
+            _tempFilePool = tempFilePool;
+        }
+
+        public SubtitleBuilder WithResolution(IDisplaySize resolution)
+        {
+            _resolution = Some(resolution);
+            return this;
+        }
+
+        public SubtitleBuilder WithFontName(string fontName)
+        {
+            _fontName = fontName;
+            return this;
+        }
+
+        public SubtitleBuilder WithFontSize(int fontSize)
+        {
+            _fontSize = fontSize;
+            return this;
+        }
+
+        public SubtitleBuilder WithPrimaryColor(string primaryColor)
+        {
+            _primaryColor = primaryColor;
+            return this;
+        }
+
+        public SubtitleBuilder WithOutlineColor(string outlineColor)
+        {
+            _outlineColor = outlineColor;
+            return this;
+        }
+
+        public SubtitleBuilder WithAlignment(int alignment)
+        {
+            _alignment = alignment;
+            return this;
+        }
+
+        public SubtitleBuilder WithMarginRight(int marginRight)
+        {
+            _marginRight = marginRight;
+            return this;
+        }
+
+        public SubtitleBuilder WithMarginLeft(int marginLeft)
+        {
+            _marginLeft = marginLeft;
+            return this;
+        }
+        
+        public SubtitleBuilder WithMarginV(int marginV)
+        {
+            _marginV = marginV;
+            return this;
+        }
+
+        public SubtitleBuilder WithBorderStyle(int borderStyle)
+        {
+            _borderStyle = borderStyle;
+            return this;
+        }
+
+        public SubtitleBuilder WithShadow(int shadow)
+        {
+            _shadow = shadow;
+            return this;
+        }
+
+        public SubtitleBuilder WithFormattedContent(string content)
+        {
+            _content = content;
+            return this;
+        }
+        
+        public async Task<string> BuildFile()
+        {
+            string fileName = _tempFilePool.GetNextTempFile(TempFileCategory.Subtitle);
+
+            var sb = new StringBuilder();
+            sb.AppendLine("[Script Info]");
+            sb.AppendLine("ScriptType: v4.00+");
+            sb.AppendLine("WrapStyle: 0");
+            sb.AppendLine("ScaledBorderAndShadow: yes");
+            sb.AppendLine("YCbCr Matrix: None");
+
+            foreach (IDisplaySize resolution in _resolution)
+            {
+                sb.AppendLine($"PlayResX: {resolution.Width}");
+                sb.AppendLine($"PlayResY: {resolution.Height}");
+            }
+
+            sb.AppendLine("[V4+ Styles]");
+            sb.AppendLine("Format: Name, Fontname, Fontsize, PrimaryColour, OutlineColour, BorderStyle, Shadow, Alignment, Encoding");
+            sb.AppendLine($"Style: Default,{await _fontName.IfNoneAsync("")},{await _fontSize.IfNoneAsync(32)},{await _primaryColor.IfNoneAsync("")},{await _outlineColor.IfNoneAsync("")},{await _borderStyle.IfNoneAsync(0)},{await _shadow.IfNoneAsync(0)}, {await _alignment.IfNoneAsync(0)},1");
+
+            sb.AppendLine("[Events]");
+            sb.AppendLine("Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text");
+            sb.AppendLine($"Dialogue: 0,0:00:00.00,99:99:99.99,Default,,{_marginLeft},{_marginRight},{_marginV},,{_content}");
+
+            if (!string.IsNullOrWhiteSpace(_content))
+            {
+                sb.AppendLine(_content);
+            }
+
+            await File.WriteAllTextAsync(fileName, sb.ToString());
+
+            return fileName;
+        }
+    }
+}