gamerom 0.1.0 → 0.4.0

data/lib/gamerom/config.rb

@@ -1,4 +1,4 @@
-# 'frozen_string_literal' => true
+# frozen_string_literal: true
 
 require 'fileutils'
 require 'logger'
@@ -6,7 +6,7 @@ require 'rest-client'
 require 'mechanize'
 
 module Gamerom
-  ROM_ROOT = ENV['ROM_ROOT'] || File.expand_path("~/.gamerom")
+  ROM_ROOT = ENV['ROM_ROOT'] || File.expand_path('~/.gamerom')
   CACHE_DIR = ENV['CACHE_DIR'] || "#{ROM_ROOT}/cache"
   GAME_DIR = ENV['GAME_DIR'] || "#{ROM_ROOT}/games"
   LOG_DIR = ENV['LOG_DIR'] || "#{ROM_ROOT}/logs"

data/lib/gamerom/game.rb

@@ -1,47 +1,56 @@
-# 'frozen_string_literal' => true
+# frozen_string_literal: true
 
 require 'fileutils'
 require 'ostruct'
+require 'yaml'
 
 module Gamerom
+  # Game - Represents a game ROM
   class Game < OpenStruct
-    def filename
-      "#{self.filepath}/#{File.read(self.state_filename)}"
+    def filenames
+      YAML.load_file(state_filename).map do |filename|
+        "#{filepath}/#{filename}"
+      end
     end
 
     def filepath
-      "#{Gamerom::GAME_DIR}/#{self.repo.name}/#{self.platform}/#{self.region}"
+      "#{Gamerom::GAME_DIR}/#{repo.name}/#{platform}/#{region}"
     end
 
     def install
-      self.repo.install self do |filename|
-        self.update_state filename
+      repo.install self do |filenames|
+        update_state filenames
       end
     end
 
     def installed?
-      File.exists? self.state_filename
+      File.exist? state_filename
     end
 
     def state_filename
-      "#{Gamerom::STATE_DIR}/#{self.repo.name}/#{self.platform}/#{self.region}/#{self.id}"
+      "#{Gamerom::STATE_DIR}/#{repo.name}/#{platform}/#{region}/#{id}"
     end
 
     def to_s
-      "#{self.id} - #{self.name} - #{self.region}#{self.installed? ? " (#{shell.set_color 'installed', :green})" : ''}"
+      install_status = ''
+      install_status = " (#{shell.set_color "installed", :green})" if installed?
+      tags = ''
+      tags = " - tags: #{tags.join(", ")}" if respond_to?(:tags) && !tags.empty?
+      "#{id} - #{name} - #{region}#{install_status}#{tags}"
     end
 
     def uninstall
-      FileUtils.rm_rf self.filename
-      FileUtils.rm_rf self.state_filename
+      FileUtils.rm_rf filenames
+      FileUtils.rm_rf state_filename
     end
 
-    def update_state filename
-      FileUtils.mkdir_p("#{Gamerom::STATE_DIR}/#{self.repo.name}/#{self.platform}/#{self.region}")
-      File.write(self.state_filename, filename)
+    def update_state(filenames)
+      FileUtils.mkdir_p("#{Gamerom::STATE_DIR}/#{repo.name}/#{platform}/#{region}")
+      File.write(state_filename, filenames.to_yaml)
     end
 
     private
+
     def shell
       @shell ||= Thor::Shell::Color.new
     end

data/lib/gamerom/game_info.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+module Gamerom
+  # GameInfo - Extracts region and tags from game name
+  class GameInfo
+    REGIONS = {
+      '1' => 'Japan & Korea',
+      '4' => 'USA & Brazil - NTSC',
+      '5' => 'NTSC',
+      '8' => 'PAL',
+      'A' => 'Australia',
+      'As' => 'Asia',
+      'B' => 'Brazil',
+      'C' => 'Canada',
+      'Ch' => 'China',
+      'D' => 'Netherlands (Dutch)',
+      'E' => 'Europe',
+      'F' => 'France',
+      'FC' => 'French Canadian',
+      'FN' => 'Finland',
+      'G' => 'Germany',
+      'GR' => 'Greece',
+      'H' => 'Holland',
+      'HK' => 'Hong Kong',
+      'I' => 'Italy',
+      'J' => 'Japan',
+      'JUE' => 'Japan & USA & Europe',
+      'K' => 'Korea',
+      'Nl' => 'Netherlands',
+      'NL' => 'Netherlands',
+      'No' => 'Norway',
+      'PD' => 'Public Domain',
+      'R' => 'Russia',
+      'S' => 'Spain',
+      'Sw' => 'Sweden',
+      'SW' => 'Sweden',
+      'U' => 'USA',
+      'UK' => 'England',
+      'Unk' => 'Unknown Country',
+      'Unl' => 'Unlicensed',
+      'PAL' => 'PAL regions (Australia, Europe)',
+      'NTSC' => 'NTSC regions (Japan, USA, Latin America)',
+    }.freeze
+
+    TAGS = {
+      '!' => :good,
+      '!p' => :pending,
+      'a' => :alternate,
+      'b' => :bad,
+      'BF' => :bung,
+      'c' => :checksum,
+      'C' => :color,
+      'f' => :fixed,
+      'h' => :hack,
+      'J' => :japanese_translation,
+      'o' => :overdump,
+      'p' => :pirate,
+      'PC10' => :pc10,
+      'S' => :super,
+      'T-' => :old_translation,
+      't' => :trained,
+      'T+' => :newer_translation,
+      'VS' => :vs,
+      'x' => :bad_checksum,
+    }.freeze
+
+    attr_reader :name
+
+    def initialize(name)
+      @name = name
+    end
+
+    def region
+      identifiers = @name.scan(/\((?<region>[A-Za-z0-9]+)\)/).flatten
+      region_id = identifiers.find { |i| REGIONS.include? i }
+      if region_id
+        REGIONS[region_id]
+      else
+        'USA'
+      end
+    end
+
+    def tags
+      tags = []
+      codes = @name.scan(/\[(?<code>[^\]]+)\]/).flatten
+      codes.each do |code|
+        code = Regexp.last_match(1) if code.match /^(?<code>[abcfhop])[0-9]*/
+        tags << TAGS[code] if TAGS.include?(code)
+      end
+      tags
+    end
+  end
+end

data/lib/gamerom/repo.rb

@@ -1,4 +1,4 @@
-# 'frozen_string_literal' => true
+# frozen_string_literal: true
 
 require 'yaml'
 
@@ -6,30 +6,34 @@ REPOSITORIES = Dir["#{File.dirname(__FILE__)}/repo_adapters/*"].map do |file|
   File.basename(file, '.rb')
 end
 
+require_relative 'repo_adapter'
 REPOSITORIES.each do |repo|
   require_relative "repo_adapters/#{repo}"
 end
 
 module Gamerom
+  # Repo - Represents a game ROM repository
   class Repo
+    attr_reader :name
+
     def self.list
       REPOSITORIES.map do |repo|
-        self.new repo
+        new repo
       end
     end
 
-    def initialize name
+    def initialize(name)
       @name = name
       @repo = Gamerom::RepoAdapters.const_get(name.capitalize)
     end
 
-    def install game, &block
+    def install(game, &block)
       @repo.install game, &block
     end
 
-    def find platform, game_identifier
+    def find(platform, game_identifier)
       games(platform).find do |game|
-        if Float(game_identifier, exception: false)
+        if Integer(game_identifier, exception: false)
           game.id == game_identifier.to_i
         else
           game.name.downcase == game_identifier.downcase
@@ -37,47 +41,42 @@ module Gamerom
       end
     end
 
-    def games platform, options={}
+    def games(platform, options = {})
       platform_database = "#{Gamerom::CACHE_DIR}/#{@name}/#{platform}.yml"
-      update_database platform unless File.exists? platform_database
-      games = YAML.load_file(platform_database).map { |game|
+      update_database platform unless File.exist? platform_database
+      games = YAML.load_file(platform_database).map do |game|
         Game.new(game.merge(platform: platform, repo: self))
-      }
+      end
 
-      if !options[:region].nil?
-        games = games.select { |game|
-          options[:region].nil? || game.region == options[:region]
-        }
+      unless options[:region].nil?
+        games.select! do |game|
+          game.region == options[:region]
+        end
       end
 
-      if !options[:keyword].nil?
-        games = games.select { |game|
+      unless options[:keyword].nil?
+        games.select! do |game|
           game.name =~ /#{options[:keyword]}/i
-        }
+        end
       end
 
       games
     end
 
-    def name
-      @name
-    end
-
     def platforms
       @repo.platforms
     end
 
-    def regions platform
-      games(platform).map { |game| game.region }.sort.uniq
+    def regions(platform)
+      games(platform).map(&:region).sort.uniq
     end
 
     def to_s
       @name
     end
 
-    def update_database platform
+    def update_database(platform)
       games = @repo.games platform
-      puts
       FileUtils.mkdir_p("#{Gamerom::CACHE_DIR}/#{@name}")
       File.write("#{Gamerom::CACHE_DIR}/#{@name}/#{platform}.yml", games.to_yaml)
     end

data/lib/gamerom/repo_adapter.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'nokogiri'
+require 'rest-client'
+
+module Gamerom
+  # RepoAdapter - Common adapter methods
+  module RepoAdapter
+    def nokogiri_get(url)
+      Nokogiri::HTML(RestClient.get(url))
+    end
+
+    def games(platform)
+      games = []
+      progress_bar = ProgressBar.new(platform, sections.count)
+
+      extract_games(platform) do |section_games, section_index|
+        games.append(*section_games)
+        progress_bar.set(section_index + 1)
+      end
+
+      progress_bar.finish
+      games
+    end
+  end
+end

data/lib/gamerom/repo_adapters/coolrom.rb

@@ -1,11 +1,15 @@
 # frozen_string_literal: true
 
-require 'nokogiri'
-require 'rest-client'
+require 'mechanize'
+require 'mechanize/progressbar'
+require 'mechanizeprogress'
 
 module Gamerom
   module RepoAdapters
+    # Coolrom - An adapter for the CoolROM repository website
     class Coolrom
+      extend Gamerom::RepoAdapter
+
       PLATFORM = {
         'atari2600' => 'Atari 2600',
         'atari5200' => 'Atari 5200',
@@ -30,48 +34,51 @@ module Gamerom
         'psx' => 'Sony Playstation',
         'ps2' => 'Sony Playstation 2',
         'psp' => 'Sony Playstation Portable',
-      }
+      }.freeze
 
       def self.platforms
         PLATFORM
       end
 
-      def self.games(platform)
-        games = []
-        sections = ('a'..'z').to_a.unshift("0")
+      def self.sections
+        ('a'..'z').to_a.unshift('0')
+      end
 
-        sections.each do |section|
-          print "#{section} "
-          page = Nokogiri::HTML(RestClient.get("https://coolrom.com.au/roms/#{platform}/#{section}/"))
-          regions = page.css('input.region').map { |i| i["name"] }
+      def self.extract_games(platform)
+        sections.each_with_index do |section, index|
+          page = nokogiri_get("https://coolrom.com.au/roms/#{platform}/#{section}/")
+          regions = page.css('input.region').map { |i| i['name'] }
           regions.each do |region|
-            games.append *page.css("div.#{region} a").map { |game|
-              {
-                id: game['href'].split('/')[3].to_i,
-                name: game.text,
-                region: region,
-              }
-            }
+            game_links = page.css("div.#{region} a")
+            yield game_links.map { |game_link| game(game_link, region) }, index
           end
         end
-        games
+      end
+
+      def self.game(game_link, region)
+        {
+          id: game_link['href'].split('/')[3].to_i,
+          name: game_link.text,
+          region: region,
+        }
       end
 
       def self.install(game)
-        response = RestClient::Request.execute(
-          method: :get,
-          url: "https://coolrom.com.au/downloader.php?id=#{game.id}",
-          headers: {
-            'User-Agent': "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.93 Safari/537.36",
-          },
-          raw_response: true,
-        )
-        if response.code == 200
-          filename = response.headers[:content_disposition].split('; ')[1].split('"')[1]
-          FileUtils.mkdir_p(game.filepath)
-          FileUtils.cp(response.file.path, "#{game.filepath}/#{filename}")
-          yield filename
+        agent = Mechanize.new
+        agent.pluggable_parser.default = Mechanize::Download
+        agent.user_agent = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.93 Safari/537.36'
+
+        response = nil
+        agent.progressbar do
+          response = agent.get("https://coolrom.com.au/downloader.php?id=#{game.id}")
         end
+
+        return unless response.code.to_i == 200
+
+        filename = response.filename
+        FileUtils.mkdir_p(game.filepath)
+        response.save!("#{game.filepath}/#{filename}")
+        yield [filename]
       end
     end
   end

data/lib/gamerom/repo_adapters/romnation.rb

@@ -0,0 +1,110 @@
+# frozen_string_literal: true
+
+require 'cgi'
+require 'mechanize'
+require 'mechanize/progressbar'
+require 'mechanizeprogress'
+
+module Gamerom
+  module RepoAdapters
+    # Romnation - An adapter for the ROMNation repository website
+    class Romnation
+      extend Gamerom::RepoAdapter
+
+      PLATFORM = {
+        'amstrad' => 'Amstrad',
+        'atari2600' => 'Atari 2600',
+        'atari5200' => 'Atari 5200',
+        'atari7800' => 'Atari 7800',
+        'atarijaguar' => 'Atari Jaguar',
+        'atarilynx' => 'Atari Lynx',
+        'colecovision' => 'ColecoVision',
+        'commodore64' => 'Commodore 64',
+        'gamegear' => 'Game Gear',
+        'gb' => 'Game Boy',
+        'gbc' => 'Game Boy Color',
+        'gcdvectrex' => 'Vectrex',
+        'genesis' => 'Genesis',
+        'intellivision' => 'Intellivision',
+        'mame' => 'MAME',
+        'msx1' => 'MSX',
+        'msx2' => 'MSX2',
+        'mtx' => 'MTX',
+        'n64' => 'N64',
+        'neogeocd' => 'Neo Geo CD',
+        'neogeopocket' => 'Neo Geo Pocket',
+        'nes' => 'NES',
+        'oric' => 'Oric',
+        'pce' => 'PC Engine',
+        'radioshackcolorcomputer' => 'TRS-80',
+        'samcoupe' => 'SAM Coupé',
+        'segacd' => 'Sega CD',
+        'segamastersystem' => 'Master System',
+        'snes' => 'SNES',
+        'thompsonmo5' => 'Thomson MO5',
+        'virtualboy' => 'Virtual Boy',
+        'watara' => 'Watara Supervision',
+        'wonderswan' => 'WonderSwan',
+      }.freeze
+
+      def self.platforms
+        PLATFORM
+      end
+
+      def self.sections
+        ('a'..'z').to_a.unshift('0')
+      end
+
+      def self.extract_games(platform)
+        sections.each_with_index do |section, index|
+          pages = extract_pages(platform, section)
+          yield extract_games_from_section_pages(platform, section, pages), index
+        end
+      end
+
+      def self.extract_pages(platform, section)
+        page = nokogiri_get("https://www.romnation.net/srv/roms/#{platform}/#{section}/sort-title.html")
+        pages = ['1']
+        pages = page.css('.pagination').first.css('a').map(&:text).map(&:strip).reject(&:empty?) unless page.css('.pagination').empty?
+        pages
+      end
+
+      def self.extract_games_from_section_pages(platform, section, pages)
+        pages.reduce([]) do |section_games, p|
+          page = nokogiri_get("https://www.romnation.net/srv/roms/#{platform}/#{section}/page-#{p}_sort-title.html")
+          games_links = page.css('table.listings td.title a')
+          section_games.append(*games_links.map { |game_link| game(game_link) })
+        end
+      end
+
+      def self.game(game_link)
+        game_info = GameInfo.new(game_link.text)
+        {
+          id: game_link['href'].split('/')[3].to_i,
+          name: game_link.text,
+          region: game_info.region,
+          tags: game_info.tags,
+        }
+      end
+
+      def self.install(game)
+        agent = Mechanize.new
+        agent.pluggable_parser.default = Mechanize::Download
+        agent.user_agent = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.93 Safari/537.36'
+        page = agent.get("https://www.romnation.net/download/rom/#{game.id}")
+
+        response = nil
+        agent.progressbar do
+          response = page.link_with(text: 'Download This Rom').click
+        end
+
+        return unless response.code.to_i == 200
+
+        filename = CGI.unescape(response.filename.split('_host=').first)
+        FileUtils.mkdir_p(game.filepath)
+        response.save!("#{game.filepath}/#{filename}")
+        yield [filename]
+      end
+    end
+  end
+end