Started writing extractor

8 files changed, 126 insertions, 8 deletions

@@ -2,9 +2,10 @@ PATH
   remote: .
   specs:
     pokeviewer (0.1.0)
+      activerecord-diff
       enumerize
       rails (~> 5.1.4)
-      sequential
+      sequenced
 
 GEM
   remote: https://rubygems.org/
@@ -41,6 +42,7 @@ GEM
       activemodel (= 5.1.4)
       activesupport (= 5.1.4)
       arel (~> 8.0)
+    activerecord-diff (2.0.0)
     activesupport (5.1.4)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (~> 0.7)
@@ -63,11 +65,11 @@ GEM
     mime-types (3.1)
       mime-types-data (~> 3.2015)
     mime-types-data (3.2016.0521)
-    mini_portile2 (2.2.0)
+    mini_portile2 (2.3.0)
     minitest (5.10.3)
     nio4r (2.1.0)
-    nokogiri (1.8.0)
-      mini_portile2 (~> 2.2.0)
+    nokogiri (1.8.1)
+      mini_portile2 (~> 2.3.0)
     rack (2.0.3)
     rack-test (0.7.0)
       rack (>= 1.0, < 3)
@@ -95,7 +97,9 @@ GEM
       rake (>= 0.8.7)
       thor (>= 0.18.1, < 2.0)
     rake (12.1.0)
-    sequential (0.2)
+    sequenced (3.1.1)
+      activerecord (>= 3.0)
+      activesupport (>= 3.0)
     sprockets (3.7.1)
       concurrent-ruby (~> 1.0)
       rack (> 1, < 3)
@@ -0,0 +1,13 @@
+require_dependency "pokeviewer/application_controller"
+
+module Pokeviewer
+  class UploaderController < ApplicationController
+    skip_before_action :verify_authenticity_token
+
+    def submit
+      ExtractSaveDataJob.perform_later params[:game].as_json
+
+      render json: { message: "Data submitted for processing." }
+    end
+  end
+end
@@ -0,0 +1,88 @@
+module Pokeviewer
+  class ExtractSaveDataJob < ApplicationJob
+    queue_as :default
+
+    def perform(args)
+      game = Trainer.find_or_create_by!(
+        name: args["playerName"],
+        number: args["playerId"]) do |r|
+          case args["gameId"].to_i
+          when 1
+            r.game = :ruby
+          when 2
+            r.game = :sapphire
+          when 3
+            r.game = :firered
+          when 4
+            r.game = :leafgreen
+          when 5
+            r.game = :emerald
+          else
+            # Invalid, so just leave the field nil
+          end
+        end
+
+      pokemons = args["pokemon"].map do |param|
+        pk = Pokemon.find_or_create_by!(key: param["key"]) do |r|
+          r.species_id = param["species"]
+          r.ot_name = param["otName"]
+          r.ot_number = param["otId"]
+          r.met_type = :normal
+          r.met_level = param["metLevel"]
+          r.shiny = param["shiny"]
+          r.nature = Pokemon.nature.values[param["nature"]]
+          r.gender = param["gender"]
+          r.second_ability = param["secondAbility"]
+
+          # Handle Unown form
+          if r.species_id == 201
+            r.unown_letter = Pokemon.unown_letter.values[param["unownLetter"]]
+          end
+        end
+
+        rev = Revision.new(pokemon: pk)
+        rev.nickname = param["nickname"]
+        rev.experience = param["experience"]
+        rev.level = param["level"]
+        rev.hp = param["hp"]
+        rev.attack = param["attack"]
+        rev.defense = param["defense"]
+        rev.special_attack = param["spAttack"]
+        rev.special_defense = param["spDefense"]
+        rev.speed = param["speed"]
+        rev.coolness = param["coolness"]
+        rev.beauty = param["beauty"]
+        rev.cuteness = param["cuteness"]
+        rev.smartness = param["smartness"]
+        rev.toughness = param["toughness"]
+        rev.sheen = param["sheen"]
+        rev.hold_item = param["heldItem"]
+        rev.move_1_id = param["moves"][0]["id"]
+        rev.move_1_pp_bonuses = param["moves"][0]["ppBonuses"]
+
+        if param["moves"].length >= 2
+          rev.move_2_id = param["moves"][1]["id"]
+          rev.move_2_pp_bonuses = param["moves"][1]["ppBonuses"]
+        end
+
+        if param["moves"].length >= 3
+          rev.move_3_id = param["moves"][2]["id"]
+          rev.move_3_pp_bonuses = param["moves"][2]["ppBonuses"]
+        end
+
+        if param["moves"].length == 4
+          rev.move_4_id = param["moves"][3]["id"]
+          rev.move_4_pp_bonuses = param["moves"][3]["ppBonuses"]
+        end
+
+        if pk.revisions.empty? or rev.diff?(pk.revisions.last)
+          rev.save!
+        end
+
+        pk
+      end
+
+      game.pokemon = pokemons
+    end
+  end
+end
@@ -4,7 +4,7 @@ module Pokeviewer
 
     belongs_to :species
     belongs_to :trainer, optional: true
-    has_many :revisions, dependent: :destroy
+    has_many :revisions, -> { order "sequential_id ASC" }, dependent: :destroy
 
     validate :uuid_is_constant, on: :update
     before_create :set_uuid
@@ -1,5 +1,15 @@
+require 'active_record/diff'
+
 module Pokeviewer
   class Revision < ApplicationRecord
+    include ActiveRecord::Diff
+
+    diff :nickname, :experience, :level, :hp, :attack, :defense,
+      :special_attack, :special_defense, :speed, :coolness, :beauty, :cuteness,
+      :smartness, :toughness, :sheen, :hold_item, :move_1_id, :move_2_id,
+      :move_3_id, :move_4_id, :move_1_pp_bonuses, :move_2_pp_bonuses,
+      :move_3_pp_bonuses, :move_4_pp_bonuses
+
     belongs_to :pokemon
     acts_as_sequenced scope: :pokemon_id
 
@@ -1,2 +1,4 @@
 Pokeviewer::Engine.routes.draw do
+  post '/', to: 'uploader#submit'
+
 end
@@ -1,5 +1,5 @@
 require "enumerize"
-require "sequential"
+require "sequenced"
 
 module Pokeviewer
   class Engine < ::Rails::Engine
@@ -18,7 +18,8 @@ Gem::Specification.new do |s|
 
   s.add_dependency "rails", "~> 5.1.4"
   s.add_dependency "enumerize"
-  s.add_dependency "sequential"
+  s.add_dependency "sequenced"
+  s.add_dependency "activerecord-diff"
 
   s.add_development_dependency "sqlite3"
 end