pokedex-terminal 0.1.0 → 0.1.1

data/lib/pokedex/classes/Delete.rb

@@ -0,0 +1,30 @@
+require 'tty-prompt'
+require_relative './Main_menu'
+
+class Delete
+  
+  def self.delete(data)
+    delete_prompt = TTY::Prompt.new(active_color: :red)
+      poke_array = []
+      data.each do |hash|
+        hash.each do |k,v|
+          if k == :name
+            poke_array << v
+          end
+        end
+      end
+      user_input = delete_prompt.select('Please enter the name of the Pokemon you\'re looking to delete', poke_array, filter: true)
+      data.each_with_index do |hash, index|
+        if hash[:name] == user_input
+          user_input_2 = delete_prompt.yes?("Are you sure you want to permanently delete #{hash[:name]}?")
+          if user_input_2 == true
+            data.delete_at(index)
+            return data
+          else
+            Main_menu.run
+          end
+        end
+      end
+  end
+
+end

data/lib/pokedex/classes/List.rb

@@ -0,0 +1,73 @@
+require 'tty-prompt'
+
+class List
+  
+  def self.list_menu(data)
+    while true
+      list_prompt = TTY::Prompt.new(active_color: :red)
+      user_input = list_prompt.select("How would you like to list the Pokemon?\nPlease enter one of the following options.") do |menu|
+      menu.choice "List all", 1
+      menu.choice "List by Type", 2
+      menu.choice "List by Secondary Type",3
+      menu.choice "List by Generation", 4
+      menu.choice "List Legendary Pokemon", 5
+      end
+      case user_input
+        when 1
+          puts "-" * 40
+          puts "Listing all Pokemon"
+          puts "-" * 40
+          list_all(data)
+        when 2
+          puts "Which Type would you like to list by?"
+          user_input = New.add_type
+          puts "-" * 40
+          puts "Listing by #{user_input} Type:"
+          puts "-" * 40
+          list_by(data,:type_1,user_input)
+        when 3
+          puts "Which Secondary Type would you like to list by?"
+          user_input = New.add_type
+          puts "-" * 40
+          puts "Listing by #{user_input} Secondary Type:"
+          puts "-" * 40
+          list_by(data,:type_2,user_input)
+        when 4
+          while true
+            user_input = list_prompt.ask("Which generation would you like to list by?\nInput a number between 1-8.") { |q| q.in('1-8') }
+            puts "-" * 40
+            puts "Listing by Generation #{user_input}"
+            puts "-" * 40
+            list_by(data,:generation, user_input.to_i)
+          end
+        when 5
+          puts "-" * 40
+          puts "Listing Legendary Pokemon"
+          puts "-" * 40
+          list_by(data,:legendary,'True')
+        else
+          puts "Invalid selection, please enter a number between 1-5."
+      end
+    end
+  end
+
+  def self.list_all(data)
+    data.each do |hash|
+      Print.print_pokemon_condensed(hash)
+    end
+  puts "-" * 40
+  Main_menu.return?
+  end
+
+  def self.list_by(data,attribute,value)
+    data.each do |hash|
+      if hash[attribute] == value
+      Print.print_pokemon_condensed(hash)
+      end
+    end
+  puts "-" * 40
+  Main_menu.return?
+  end
+
+end
+

data/lib/pokedex/classes/Main_menu.rb

@@ -0,0 +1,67 @@
+require_relative './Search'
+require_relative './Update'
+require_relative './New'
+require_relative './List'
+require_relative './Delete'
+require 'tty-prompt'
+require 'smarter_csv'
+
+class Main_menu
+@@pokemon_data = SmarterCSV.process('../../data/pokemon.csv')
+
+  def self.run
+    while true
+      main_menu_prompt = TTY::Prompt.new(active_color: :red)
+      puts "-" * 40
+      # main_menu_prompt.warn("        Welcome to your Pokedex!")
+      puts "        Welcome to your Pokedex!        ".blue.on_red.blink
+      puts "-" * 40
+      user_input = main_menu_prompt.select("Please select from the following:") do |menu|
+      menu.choice 'List Pokemon', 1 
+      menu.choice 'Search for a Pokemon', 2
+      menu.choice 'Update an existing Pokemon', 3
+      menu.choice 'Add a new Pokemon', 4
+      menu.choice 'Delete a Pokemon', 5 
+      menu.choice 'Exit', 6
+      menu.choice 'Print hashes', 7
+      end
+      case user_input
+        when 1
+          List.list_menu(@@pokemon_data)
+        when 2
+          name = Search.by_name(@@pokemon_data)
+          hash = Search.return_hash(@@pokemon_data, name)
+          Print.print_pokemon_expanded(hash)
+        when 3
+          Update.update_pokemon_menu(@@pokemon_data)
+          ## Incomplete
+        when 4
+          new_hash = New.add_pokemon
+          @@pokemon_data << new_hash
+          Print.print_pokemon_expanded(new_hash)
+        when 5
+          @@pokemon_data = Delete.delete(@@pokemon_data)
+        when 6
+          CSV.open('../../data/pokemon.csv', "wb") do |csv|
+            keys = @@pokemon_data.first.keys
+            csv << keys
+            @@pokemon_data.each do |hash|
+              csv << hash.values_at(*keys)
+            end
+          end
+          exit
+        when 7
+          pp @@pokemon_data
+        else
+          "Invalid selection, please select from the following options."
+      end
+    end
+  end
+
+  def self.return?
+    puts "Press enter to return to the Main Menu."
+    user_input = gets.chomp
+    run()
+  end
+
+end

data/lib/pokedex/classes/New.rb

@@ -0,0 +1,98 @@
+class New
+
+  def self.add_pokemon
+    puts "So you want to create a new Pokemon, what will you call it?"
+    name = add_name()
+    puts "What type will your Pokemon belong to?"
+    type_1 = add_type()
+    puts "Now for the secondary type."
+    type_2 = add_type()
+    hp = add_points('HP')
+    attack = add_points('Attack')
+    defense = add_points('Defense')
+    sp_atk = add_points('Special Attack')
+    sp_def = add_points('Special Defense')
+    speed = add_points('Speed')
+    generation = add_generation()
+    legendary = add_legendary()
+    total = hp + attack + sp_atk + sp_def + speed
+    hash = {}
+    hash[:name] = name
+    hash[:type_1] = type_1
+    hash[:type_2] = type_2
+    hash[:total] = total
+    hash[:hp] = hp
+    hash[:attack] = attack
+    hash[:defense] = defense
+    hash[:"sp._atk"] = sp_atk
+    hash[:"sp._def"] = sp_def
+    hash[:speed] = speed
+    hash[:generation] = generation
+    hash[:legendary] = legendary
+    pp hash
+    return hash
+  end
+
+
+  def self.add_name
+    print "> "
+    return gets.chomp
+  end
+
+  def self.add_type
+    new_prompt = TTY::Prompt.new(active_color: :red)
+    user_input = new_prompt.select("Please select a Type.") do |menu|
+      menu.choice "Normal"
+      menu.choice "Fire"
+      menu.choice "Water" 	
+      menu.choice "Grass"
+      menu.choice "Electric" 	
+      menu.choice "Ice"
+      menu.choice "Fighting" 	
+      menu.choice "Poison"
+      menu.choice "Ground"
+      menu.choice "Flying"
+      menu.choice "Psychic"
+      menu.choice "Bug"
+      menu.choice "Rock"
+      menu.choice "Ghost"
+      menu.choice "Dark" 	
+      menu.choice "Dragon"
+      menu.choice "Steel" 	
+      menu.choice "Fairy"
+    end
+    return user_input
+  end
+
+  def self.add_points(attribute)
+    while true
+      puts "How many points would you like this Pokemon to have for #{attribute}?"
+      print "> "
+      user_input = gets.chomp.to_i
+      if user_input > 0
+        return user_input
+      else
+        puts "Invalid selection. Please input a number greater than 0."
+      end
+    end
+  end
+
+  def self.add_generation
+    new_prompt = TTY::Prompt.new(active_color: :red)
+    user_input = new_prompt.ask("Which generation does this Pokemon belong to?\nInput a number between 1-8.") do |q|
+     q.in('1-8')
+      q.messages[:valid?] = 'Invalid number, select a number from 1-8'
+    end
+    return user_input
+  end
+
+  def self.add_legendary
+    new_prompt = TTY::Prompt.new(active_color: :red)
+    user_input = new_prompt.yes?('Is this Pokemon legendary?') do |q|
+      q.messages[:valid?] = "Invalid response, select 'Y' or 'n'"
+    end
+    p user_input
+    return user_input = true ? 'True' : 'False'
+  end
+
+end

data/lib/pokedex/classes/Pokemon.rb

@@ -0,0 +1,16 @@
+class Pokemon
+  def initialize(hash)
+    @name = hash[:name]
+    @type_1 = hash[:type_1]
+    @type_2 = hash[:type_2]
+    @total = hash[:total]
+    @hp = hash[:hp]
+    @attack = hash[:attack]
+    @defense = hash[:defense]
+    @sp_atk = hash[:"sp._atk"]
+    @sp_def = hash[:"sp._def"]
+    @speed = hash[:speed]
+    @generation = hash[:generation]
+    @legendary = hash[:legendary]
+  end
+end

data/lib/pokedex/classes/Print.rb

@@ -0,0 +1,62 @@
+require_relative './Main_menu'
+require 'colorize'
+
+class Print
+
+  def self.print_pokemon_expanded(hash)
+    puts "-" * 40
+    puts "#{hash[:name]}"
+    puts "-" * 40
+    print "#{hash[:name]} belongs to the #{hash[:type_1]} type"
+    if hash[:type_2] != nil
+      print ", with a secondary type of #{hash[:type_2]} \n"
+    else
+      print ".\n"
+    end
+    puts "Total: #{hash[:total]}"
+    puts "HP: #{hash[:hp]}"
+    print "Attack: #{hash[:attack]}             Defense :#{hash[:defense]}\n"
+    print "Special Attack: #{hash[:"sp._atk"]}     Special Defense: #{hash[:"sp._def"]}\n"
+    puts "Speed: #{hash[:speed]}"
+    puts "They are of Generation #{hash[:generation]}."
+    if hash[:legendary] == 'True'
+      puts "Most importantly, #{hash[:name]} is a legendary Pokemon!"
+    end
+    puts "-" * 40
+    Main_menu.return?
+  end
+
+  def self.print_pokemon_condensed(hash)
+    name = "#{hash[:name]}"
+    case hash[:type_1]
+      when "Normal"
+        puts name
+      when "Fire"
+        puts name.colorize(:red)
+      when "Water" 	
+        puts name.colorize(:blue)
+      when "Grass"
+        puts name.colorize(:green)
+      when "Electric"
+        puts name.colorize(:light_blue)
+      when "Ice"
+        puts name.colorize(:grey)
+      when "Fighting" 
+        puts name.colorize(:yellow)	
+      when "Poison"
+        puts name.colorize(:purple)
+      when "Ground"
+        puts name.colorize(:brown)
+      when "Flying"
+      when "Psychic"
+      when "Bug"
+      when "Rock"
+      when "Ghost"
+      when "Dark" 	
+      when "Dragon"
+      when "Steel" 	
+      when "Fairy"
+    end
+  end
+
+end

data/lib/pokedex/classes/Search.rb

@@ -0,0 +1,27 @@
+require_relative './Print'
+require_relative './Main_menu'
+
+class Search
+
+  def self.by_name(data)
+    search_prompt = TTY::Prompt.new(active_color: :red)
+    poke_array = []
+    data.each do |hash|
+      hash.each do |k,v|
+        if k == :name
+          poke_array << v
+        end
+      end
+    end
+    return search_prompt.select('Please enter the name of the Pokemon you\'re looking for.', poke_array, filter: true)
+  end
+
+  def self.return_hash(data, name)
+    data.each do |hash|
+      if hash[:name] == name
+        return hash
+      end
+    end
+  end
+
+end

data/lib/pokedex/classes/Update.rb

@@ -0,0 +1,69 @@
+require_relative './New'
+require_relative './Search'
+
+class Update
+
+  def self.update_pokemon_menu(data)
+    update_prompt = TTY::Prompt.new(active_color: :red)
+    name = Search.by_name(data)
+    data.each do |hash|
+      if hash[:name] == name
+        user_input = update_prompt.select("Which attribute would you like to update?") do |menu|
+          menu.choice "Name", 1
+          menu.choice "Type", 2
+          menu.choice "Secondary Type", 3
+          menu.choice "HP", 4
+          menu.choice "Attack", 5
+          menu.choice "Defense", 6
+          menu.choice "Special Attack", 7
+          menu.choice "Special Defense", 8
+          menu.choice "Speed", 9
+          menu.choice "Generation", 10
+          menu.choice "Legendary", 11
+          menu.choice "Exit this menu", 12
+        end
+        case user_input
+          when 1
+            puts "What would you like #{hash[:name]}'s new name to be?"
+            name = New.add_name
+            hash[:name] = name
+          when 2
+            puts "What would you like #{hash[:name]}'s new type to be?"
+            type_1 = New.add_type
+            hash[:type_1] = type_1
+          when 3
+            puts "What would you like #{hash[:name]}'s new secondary type to be?"
+            type_2 = New.add_type
+            hash[:type_2] = type_2
+          when 4
+            hp = New.add_points('HP')
+            hash[:hp] = hp
+          when 5
+            attack = New.add_points('Attack')
+            hash[:attack] = attack
+          when 6
+            defense = New.add_points('Defense')
+            hash[:defense] = defense
+          when 7
+            sp_atk = New.add_points('Special Attack')
+            hash[:"sp._atk"] = sp_atk
+          when 8
+            sp_def = New.add_points('Special Defense')
+            hash[:"sp._def"] = sp_def
+          when 9
+            speed = New.add_points('Speed')
+            hash[:speed] = speed
+          when 10
+            generation = New.add_generation
+            hash[:generation] = generation
+          when 11
+            legendary = New.add_legendary
+            hash[:legendary] = legendary
+          when 12
+            Main_menu.run
+        end
+      end
+    end
+  end
+
+end