plextvtool 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 373445d3db12d5ac5fac4b27b4dca542cfc8f8614ccd564fb3f2420b348ad3a1
+  data.tar.gz: c8e3b00c395047e852dc8a8c96dbe11fe09a8aea0261cbeb85372c7017d688ad
+SHA512:
+  metadata.gz: ca1e07becd24b2c504fcef9a3d429bc15c5599b079c831c57af7fe7510a7a18eaa558b30a2897091ae67909feec99303d08cc84f7f322510a568def016e2cbf5
+  data.tar.gz: ceaa85681401f9de9eb9a31dc3122a46b2ba0e61c6ac7c3516db894bb4cc8d38f55e5d42b481827b591f3e1d9d573da89075043baee9437ee3804e6a554a2bf5

data/bin/plextvtool

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'thor'
+require 'plextvtoolservice'
+
+# CLI for plextvtool
+class PlexTvTool < Thor
+  desc 'fix <DIR>', 'rename episodes in <DIR>'
+  options title: :required, season: :required
+  def fix(dir)
+    PlexTvToolService.new(dir, options).rename_files!
+  end
+
+  desc 'config', 'set configurable parameters for fetching data'
+  options key: :required
+  def config
+    loaded_config = PlexTvToolService.load_config
+    loaded_config['plextvt']['omdb']['api_key'] = options[:key]
+    PlexTvToolService.write_config! loaded_config
+  end
+end
+PlexTvTool.start(ARGV)

data/lib/config.yml

@@ -0,0 +1,4 @@
+---
+plextvt:
+  omdb:
+    base_uri: https://www.omdbapi.com/

data/lib/plextvtoolservice.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'net/http'
+require 'thor'
+require 'yaml'
+
+# Service for renaming tv show filenames in a way that is friendly with Plex
+class PlexTvToolService
+  attr_reader :dir, :episodes, :title, :season
+
+  def initialize(dir, options)
+    @dir = dir
+    @title = options[:title]
+    @season = options[:season]
+    @episodes = fetch_episodes
+  end
+
+  def rename_files!
+    sort_and_rename!
+  rescue StandardError => e
+    puts e.message
+    puts e.backtrace.inspect
+  end
+
+  class << self
+    def load_config
+      config_path = File.join File.dirname(__FILE__), 'config.yml'
+      YAML.safe_load File.read(config_path)
+    end
+
+    def write_config!(config)
+      File.open(File.join(File.dirname(__FILE__), 'config.yml'), 'w') do |f|
+        YAML.dump config, f
+      end
+    end
+  end
+
+  private
+
+  def fetch_episodes
+    api_key, base_uri = api_params
+    params = { t: title, Season: season, apikey: api_key }
+    uri = URI base_uri
+    uri.query = URI.encode_www_form params
+    response = Net::HTTP.get_response uri
+    raise 'Bad response from OMDB API' unless response.is_a? Net::HTTPSuccess
+
+    body = JSON.parse response.body
+    check_response body
+    body['Episodes']
+  end
+
+  def check_response(response)
+    return unless response['Response'] == 'False'
+
+    puts "Error response from OMDB API: #{response['Error']}"
+    exit 1
+  end
+
+  def api_params
+    loaded_config = self.class.load_config
+    [loaded_config.dig('plextvt', 'omdb', 'api_key'),
+     loaded_config.dig('plextvt', 'omdb', 'base_uri')]
+  end
+
+  def sort_and_rename!
+    files = Dir.glob("#{dir}/*").sort
+    files.zip(episodes).each do |file, episode|
+      new_name = format_new_name(file, episode)
+      print "Renaming #{File.basename(file)} to #{new_name} ... "
+      File.rename(file, "#{dir}/#{new_name}")
+      puts 'DONE'
+    end
+  end
+
+  def format_new_name(file, episode)
+    new_name = "#{title} s#{season.to_s.rjust(2, '0')}"\
+      "e#{episode['Episode'].rjust(2, '0')} #{episode['Title']}#{File.extname(file)}"
+    new_name.tr('<>:"/\\|?*', '') # Invalid chars for ntfs
+  end
+end

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification
+name: plextvtool
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Ian Wright
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-03-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+description: A tool for renaming TV media files for Plex Media Server
+email: hello@ianwright.dev
+executables:
+- plextvtool
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/plextvtool
+- lib/config.yml
+- lib/plextvtoolservice.rb
+homepage: https://github.com/ns-ian/plextvtool
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.15
+signing_key:
+specification_version: 4
+summary: Plex TV Tool
+test_files: []