vtt2ass 0.2.14 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 59af90e625fca8d57a1427d6b520854830536816a2d8c9d4e3bad15a8f5c229f
-  data.tar.gz: 32211cb9f63bb2de48114d071cf04e625e62892750a17143a505f8a2e62a8346
+  metadata.gz: 725876fe775e13997b8bfea4f1f8d1bd87fe18420c4adfabebc7aa6eced12464
+  data.tar.gz: 13ebdd5f0542f75a9a886e028afb3f93621e68c6b61304d8976320d804d447f8
 SHA512:
-  metadata.gz: c3a3c9897a10163f7a9decfd84eef040a12700d031c2ecccfcebecabea794e5e80fb885e6e747e78b8bce69cf35f4c34248d3e90fc79d6b3d4d961b905d27447
-  data.tar.gz: bb112d38d09029baf95cc48baace9a58bfff70f00c4a391bf5bd5dbf4cfae12f0a3b232f308ed70fa3bf15b4e965486616643ca620f205013cfa5a874d813abc
+  metadata.gz: 1656502fd7ff8f0b180a8826d593d763e855641e780b70f3e9f320a05c385c5ebfca97b5f037f87bacadabbbe86c35f9718fb6e9a5dd828bd3d9c7b2ede3ac37
+  data.tar.gz: a6099e227e52f45783a9462b3041b272bbe01ff76c87f10c44508df3711a61559865dd180f06ea38c02b2db245a6a9ed48732626e8ef4dd95b664d1f3177ac8f

data/.gitignore

@@ -9,4 +9,8 @@ Gemfile.lock
 
 # Input and output folders
 input/
-output/
+output/
+
+# Subtitle files
+/*.vtt
+/*.ass

data/README.md

@@ -42,6 +42,7 @@ Options:
                                    # Default: 52
   -f, [--font-family=FONT_FAMILY]  # Specify a font family for the subtitles
                                    # Default: Open Sans Semibold
+  -c, [--css=CSS]                  # Specify a CSS file path for Hidive subs
   -q, [--quiet], [--no-quiet]      # Don't output to the console
 
 Run the VTT to ASS conversion for the specified file(s)

data/lib/vtt2ass/ASSFile.rb

@@ -1,6 +1,8 @@
 # Relative imports
 require_relative 'ASSLine'
 require_relative 'ASSStyle'
+require_relative 'CSSFile'
+require_relative 'CSSRule'
 
 ##
 # This class defines an ASS subtitle file.
@@ -10,9 +12,12 @@ class ASSFile
 
     ##
     # Creates a new ASSFile instance and assigns the default values of instance variables.
-    def initialize(title, width, height)
+    def initialize(title, width, height, css_file_path = nil)
         @width = width
         @height = height
+        if not css_file_path.nil? then
+            @css_file = CSSFile.new(css_file_path)
+        end
         @header = [
             '[Script Info]',
             "Title: #{title}",
@@ -41,6 +46,8 @@ class ASSFile
     # each VTTLine. All those ASSLine are stored in an array. It also creates an array of ASSStyle that
     # will be used in the ASS style list.
     def convertVTTtoASS(vtt_file, font_family, font_size)
+        fs = font_size
+        font_color = '&H00FFFFFF'
         vtt_file.lines.each do |line|
             @ass_lines.push(ASSLine.new(line.style, line.time_start, line.time_end, line.text))
             style_exists = false
@@ -51,7 +58,27 @@ class ASSFile
                 end
             end
             if not style_exists then
-                @ass_styles.push(ASSStyle.new(line.style, line.params, font_family, font_size, @width, @height))
+                if defined?(@css_file) then
+                    css_rule = @css_file.find_rule(line.style)
+                    if not css_rule.nil? then
+                        css_rule.properties.each do |property|
+                            case property[:key]
+                            when 'font-family'
+                                font_family = property[:value].gsub('"', '').split(' ,').last
+                            when 'font-size'
+                                em_size = 1
+                                #em_size = property[:value][0].eql? '.' ? "0#{property[:value]}" : property[:value]
+                                if property[:value][0].eql? '.' then
+                                    em_size = "0#{property[:value]}".gsub('em', '').to_f
+                                end
+                                font_size = (fs * em_size).to_i
+                            when 'color'
+                                font_color = ASSStyle.convert_color(property[:value])
+                            end
+                        end
+                    end
+                end
+                @ass_styles.push(ASSStyle.new(line.style, line.params, font_family, font_size, font_color, @width, @height))
             end
         end
     end

data/lib/vtt2ass/ASSStyle.rb

@@ -1,5 +1,7 @@
 # Relative imports
 require_relative 'ASSStyleParams'
+require_relative 'Validator'
+require 'redgreenblue'
 
 ##
 # This class defines an ASS style that can be applied on a subtitle line.
@@ -13,11 +15,12 @@ class ASSStyle
     # * Requires +params+, a string of VTT styling as input.
     # * Requires a video +width+ as input.
     # * Requires a video +height+ as input. 
-    def initialize(style_name, params, font_family, font_size, width, height)
+    def initialize(style_name, params, font_family, font_size, font_color, width, height)
         @width = width
         @height = height
         @font_family = font_family
         @font_size = font_size
+        @font_color = font_color
         @style_name = style_name
         @s_params = ASSStyleParams.new(params, width, height)
         if style_name.eql? 'MainTop' then
@@ -28,6 +31,12 @@ class ASSStyle
     ##
     # This method assigns the object values to an ASS style line and outputs it.
     def to_s
-        return "Style: #{@style_name},#{@font_family},#{@font_size},&H00FFFFFF,&H000000FF,&H00020713,&H00000000,-1,0,0,0,100,100,0,0,1,2.0,2.0,#{@s_params.alignment},#{@s_params.horizontal_margin},0,#{@s_params.vertical_margin},1"
+        return "Style: #{@style_name},#{@font_family},#{@font_size},#{@font_color},&H000000FF,&H00020713,&H00000000,-1,0,0,0,100,100,0,0,1,2.0,2.0,#{@s_params.alignment},#{@s_params.horizontal_margin},0,#{@s_params.vertical_margin},1"
+    end
+
+    def self.convert_color(color_value)
+        color_value.gsub!('#', '')
+        color = Validator.hex?(color_value) ? RGB.hex(color_value) : RGB.css(color_value)
+        return ("&H00%02x%02x%02x" % [color.b, color.g, color.r]).upcase
     end
 end

data/lib/vtt2ass/Application.rb

@@ -20,7 +20,9 @@ class Application
             @title = options[:title]
         end
         @quiet = options[:quiet]
-        @noout = options[:noout]
+        if options[:css] then
+            @css = options[:css].gsub('\\', '/')
+        end
     end
 
     ##
@@ -55,7 +57,8 @@ class Application
         ass_file = ASSFile.new(
             (defined?(@title) ? @title : File.basename(file_path).gsub('.vtt', '')),
             @width,
-            @height
+            @height,
+            defined?(@css) ? @css : nil
         )
         ass_file.convertVTTtoASS(vtt_file, @font_family, @font_size)
         return ass_file

data/lib/vtt2ass/CSSFile.rb

@@ -0,0 +1,36 @@
+require 'css_parser'
+require_relative 'CSSRule'
+
+class CSSFile
+    attr_reader :rules
+    include CssParser
+
+    def initialize(file_path)
+        @file_path = file_path
+        parser = CssParser::Parser.new
+        parser.load_file!(file_path)
+        @rules = []
+        parser.each_selector do |selector, declarations, specificity|
+            css_obj = CSSRule.new(selector, declarations)
+            if not css_obj.name.empty? then
+                @rules.push(css_obj)
+            end
+        end
+    end
+
+    def find_rule(value)
+        return_rule = nil
+        @rules.each do |rule|
+            if rule.name == value then
+                return_rule = rule
+                break
+            end
+        end
+        return return_rule
+    end
+
+    def to_s
+        return @file_path
+    end
+
+end

data/lib/vtt2ass/CSSRule.rb

@@ -0,0 +1,22 @@
+class CSSRule
+    attr_reader :name, :properties
+
+    def initialize(selector, declarations)
+        @name = reduce_selector(selector)
+        declarations.split(/;\s?/).each do |dec|
+            @properties = []
+            temp = dec.split(/:\s?/)
+            @properties.push(
+                { key: temp.first, value: temp.last}
+            )
+        end
+    end
+
+    def to_s
+        return "#{@name} #{@properties}"
+    end
+
+    def reduce_selector(selector)
+        return selector.to_s.gsub(/\.rmp-container>\.rmp-content>\.rmp-cc-area>\.rmp-cc-container>\.rmp-cc-display>\.rmp-cc-cue\s?\.?/, '')
+    end
+end

data/lib/vtt2ass/Validator.rb

@@ -0,0 +1,10 @@
+class Validator
+    def self.hex?(value)
+        hex = true
+        value.gsub!('#', '')
+        value.chars.each do |digit|
+          hex = false unless digit.match(/\h/)
+        end
+        return hex
+    end
+end

data/lib/vtt2ass/version.rb

@@ -4,5 +4,5 @@ module Vtt2ass
   ##
   # This is the version of the application.
   # This needs to be changed for each gem release.
-  VERSION = "0.2.14"
+  VERSION = "0.3.0"
 end

data/lib/vtt2ass.rb

@@ -15,6 +15,7 @@ class MainCommand < Thor
     method_option :title,       :aliases => '-t', :desc => 'Specify a title for you file. If the input is a directory, all files will share the same title.', :type => :string
     method_option :font_size,   :aliases => '-s', :desc => 'Specify a font size for the subtitles', :default => 52, :type => :numeric
     method_option :font_family, :aliases => '-f', :desc => 'Specify a font family for the subtitles', :default => 'Open Sans Semibold', :type => :string
+    method_option :css,         :aliases => '-c', :desc => 'Specify a CSS file path for Hidive subs', :type => :string
     method_option :quiet,       :aliases => '-q', :desc => 'Don\'t output to the console', :type => :boolean
     def convert(input)
         app = Application.new(input, options)

data/vtt2ass.gemspec

@@ -27,6 +27,8 @@ Gem::Specification.new do |spec|
 
   spec.add_dependency 'htmlentities', '~> 4.3'
   spec.add_dependency 'thor', '~> 1.1'
+  spec.add_dependency 'css_parser', '~> 1.10'
+  spec.add_dependency 'redgreenblue', '~> 0.15'
 
   spec.add_development_dependency 'rake', '~> 12.0'
   spec.add_development_dependency 'minitest', '~> 5.0'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vtt2ass
 version: !ruby/object:Gem::Version
-  version: 0.2.14
+  version: 0.3.0
 platform: ruby
 authors:
 - Louis-Philippe Fortin
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2021-11-23 00:00:00.000000000 Z
+date: 2022-04-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: htmlentities
@@ -38,6 +38,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: css_parser
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: redgreenblue
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.15'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.15'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -130,8 +158,11 @@ files:
 - lib/vtt2ass/ASSStyle.rb
 - lib/vtt2ass/ASSStyleParams.rb
 - lib/vtt2ass/Application.rb
+- lib/vtt2ass/CSSFile.rb
+- lib/vtt2ass/CSSRule.rb
 - lib/vtt2ass/VTTFile.rb
 - lib/vtt2ass/VTTLine.rb
+- lib/vtt2ass/Validator.rb
 - lib/vtt2ass/version.rb
 - vtt2ass.gemspec
 homepage: https://gitlab.com/dkb-weeblets/vtt2ass