pokeplot 0.2.0beta

data/README.md

@@ -0,0 +1,99 @@
+# Pokeplot
+
+[![Gem Version](https://badge.fury.io/rb/pokeplot.svg)](https://badge.fury.io/rb/pokeplot) [![Test Coverage](https://codeclimate.com/github/xssc/pokeplot/badges/coverage.svg)](https://codeclimate.com/github/xssc/pokeplot/coverage)  [![Dependency Status](https://gemnasium.com/badges/github.com/xssc/pokeplot.svg)](https://gemnasium.com/github.com/xssc/pokeplot) [![Gem](https://img.shields.io/gem/dt/pokeplot.svg?maxAge=2592000)](https://rubygems.org/gems/pokeplot)
+
+Pokeplot is a gem to collect Pokemon Go spawning data and visualize it. It is currently a pre-release, expect frequent updates.
+
+![Graph Example](http://i.imgur.com/XRp14KV.png)
+
+### Contents
+
+[**Getting Started**](https://github.com/xssc/pokeplot#getting-started)
+
+[**Usage**](https://github.com/xssc/pokeplot#usage)
+
+[**Features**](https://github.com/xssc/pokeplot#features)
+
+[**Notifications**](https://github.com/xssc/pokeplot#notifications)
+
+[**Updating**](https://github.com/xssc/pokeplot#updating)
+
+[**Future and TODO**](https://github.com/xssc/pokeplot#future--todo)
+
+[**Contanct, Requests, and Issues**](https://github.com/xssc/pokeplot#contact-requests-and-issues)
+
+[**Contributing**](https://github.com/xssc/pokeplot#contributing)
+
+## Getting Started
+
+Installing pokeplot is insanley easy. All you need is ruby 2.3.1. Don't have it? [Follow our easy Ruby install guide!](https://github.com/xssc/pokeplot/wiki/Install-Ruby-2.3.1)
+
+We use MongoDB to quickly store large numbers of encounters. Download and install it from [HERE](https://www.mongodb.com/download-center?jmp=nav#community). Don't worry, it's easy!
+
+Once done the only installation command is:
+
+    $ gem install pokeplot --pre
+
+## Usage
+
+Open a terminal/cmd and make a folder for Pokeplot configuration to be stored
+
+    $ mkdir pokeplot
+    $ cd pokeplot
+
+Then to generate the default config.json file, run
+
+    $ pokeplot config
+
+Then go to the pokeplot folder you just made and open config.json in your favorite text editor. There are `_info` lines to help you along the way.
+
+Then simply run
+
+    $ pokeplot
+
+Real time graphs will be available at `http://localhost:5001` while pokeplot is running.
+
+## Features
+* Single or Multithread
+* Divides coordinates in a way that doesn't cause larger teleportations due to more accounts
+* Multiple accounts (Combo lists also supported)
+* Configurable config.json
+* Real time website with graphs
+* Ability to handle real time data yourslef with Ruby (Examples soon)
+* Notifications
+* A developer who has tons of freetime to work on this project!
+
+
+## Notifications
+
+Currently Pokeplt only supports Pushbullet. [Follow the easy setup guide here](https://github.com/xssc/pokeplot/wiki/Pushbullet)
+
+
+## Updating
+
+When a update is available for pokeplot, only one command is needed to install it
+
+    $ gem update pokeplot
+
+Pokeplot will be updated frequently, I have lots of spare time.
+
+## Future & TODO
+ The following is planned (but not guarenteed)
+
+ * Maps
+ * More graphs
+ * Get Cell Ids in Ruby
+ * Add more notification services
+ * Fix any issues that arrise
+ * Lots more
+
+## Contact, Requests, and Issues
+If you have any problems using this, I will do whatever I can to help you get up and running. Your options are
+
+ * Create an issue [HERE](https://github.com/xssc/pokeplot/issues)
+ * Email me @ xssc820@gmail.com
+ * Or PM me on reddit at [/u/xssc](https://www.reddit.com/message/compose?to=xssc)
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/xssc/pokeplot

data/Rakefile

@@ -0,0 +1,9 @@
+require "bundler/gem_tasks"
+
+begin
+  require 'rspec/core/rake_task'
+  RSpec::Core::RakeTask.new(:spec)
+  task :default => :spec
+rescue LoadError
+  # no rspec available
+end

data/bin/pokeplot

@@ -0,0 +1,223 @@
+#!usr/bin/env ruby
+require 'pokeplot'
+require 'json'
+
+
+default_config =  %{ {
+  "_info": "Enter your accounts here, as many as you want",
+  "accounts": [
+    {
+      "username": "account1",
+      "password": "password1",
+      "provider": "ptc"
+    },
+    {
+      "username": "account2@gmail.com",
+      "password": "password2",
+      "provider": "google"
+    }
+  ],
+
+  "_info": "Enter the origin scan coordinates below [lat, lng]",
+  "location": [
+    40.7829,
+    -73.9654
+  ],
+
+  "_info": "settings for API",
+  "api": {
+    "encryption_path": "/path/to/encryption/file",
+    "log": true
+  },
+
+  "_info": "settings for mongoDB (default host is already set)",
+  "mongo": {
+    "host": "127.0.0.1:27017"
+  },
+
+  "_info": "settings for the data miner",
+  "miner": {
+    "enabled": true,
+    "threaded": true,
+    "range": 40,
+    "interval": 900,
+    "pokemon": true,
+    "forts": true,
+    "log": true
+  },
+
+  "_info": "Pushbullet settings - see the wiki for easy set-up guide",
+  "pushbullet": {
+    "enabled": false,
+    "api_key": "apiKey",
+    "pokemon": [
+      "SNORLAX",
+      "DRATINI",
+      "DRAGONITE",
+      "MEWTWO"
+    ]
+  },
+
+  "_info": "settings for socket for real time webpage updates RECOMENDED YOU DONT CHANGE THIS",
+  "socket": {
+    "enabled": true,
+    "host": "0.0.0.0",
+    "port": "9090",
+    "log": true
+  },
+
+  "_info": "Web server settings",
+  "webserver": {
+    "enabled": true,
+    "host": "0.0.0.0",
+    "port": 5001
+  },
+
+  "_info": "optional ptc account combo list",
+  "combos": [
+    "SToCQSyJuVXj:BGyoYefXjQQZ",
+    "hdOMuITMabLF:kscSpsGGzBxu",
+    "vMpPCulIYkcT:GBbUdGlstxeW",
+    "FZmcNpLdbxha:CurAriesqPFZ",
+    "wVkKQBGmYJXK:XmZoKXLfvNvU",
+    "vgsqGbfVeawM:QhzXnffDjgAt",
+    "dJdUmaNsvuMC:HsHytSfNyvnS",
+    "UPmlEpbNFGYk:tZJMwBeDRAMs",
+    "xcxcPjUKRPal:iFCBrsSjHFRF",
+    "dldjzYhsQbLz:oaQaggnrVIjK",
+    "gZEekjgrBQCK:vUWhIaEHAclX",
+    "wLOToXHcLJBZ:RNIoYdwvDFTH",
+    "tBOjNYBHIJIo:CTqmQaCWaRRT",
+    "pbAGNZsvGnMC:TVWLsmdRRJjG",
+    "cneIEKBIGevh:CnFKUkBDJECj",
+    "WworruNrwycr:gYzkXQnHyPxy",
+    "YtbiVCtnYCIQ:jiBNmgTBWTBw",
+    "rwmUhrHCbWGW:gKdPdvUzbXEB",
+    "kdgqEBIsKMhO:rvESvQkNQtGs",
+    "PssVOTYjurWz:gSMXWHoFkvPq",
+    "kHlinSczQEim:rqFePFPryvUe",
+    "AIAVmolzTbQJ:mxvWoSJNrNRC",
+    "iBESkcpbrIdY:KQnQTqClDykQ",
+    "AXwyjTgvYIfn:vINTlbsjdtCx",
+    "yuHRAgIfVWTZ:VTTnyPBZSojM",
+    "nFrgKgQLzrRe:ooplsOvhQNFQ",
+    "iRFsCpCMUlte:koKrBewFLMxo",
+    "pOqWlFdPGYEl:gkhKikkcwpmh",
+    "NByvaUQkNchO:nPmVzbMPtXWY",
+    "RftfDbyWXmSe:wYLELjKLjkIJ",
+    "gLJMMYXGJibG:AnHNRvNmSkuA",
+    "zBCKrZlciUiV:SZJdvkMSkgRT",
+    "TyqcusJxatgm:EqKtdTEybJpz",
+    "fDDaoMvaOMOs:bmcEVoNcKyJC",
+    "dEgUOTTPOQpQ:elCgIExqkWOp",
+    "ZjZZftPIHKVj:gHzweZzDICAA",
+    "XuinqBhqynbE:PDfXbdSBkoJV",
+    "eYPkNTSSRngF:EShbqUEutCqQ",
+    "PvmFqqXdRnwG:IhDzJDSknest",
+    "CvbgGcNdKumE:gzmlCjYgybpZ",
+    "FiHNyzzpKmOs:tbIXrdCecmlB",
+    "dUsdcfxGJAlf:NTWKivTOeABd",
+    "BvVqSybVGuoD:xhCgeXTtsFST",
+    "GamYBkRElCCY:CFRHRROHlsrY",
+    "zeGrfdhyrtfY:AwSFrczizYsZ",
+    "bivTEnLISOyR:DQLoKikEWJON",
+    "FWMGCfVJcJjV:IJFxXGDqhDRK",
+    "yqDulLursWuk:CKieNAUmgcZM",
+    "TeXIZDsXBEJJ:gQSevIHOVxrd",
+    "iYgqvHkNlGjJ:WbSYJNhlJxbZ"
+  ]
+
+
+}
+}
+
+
+if ARGV[0] == 'config'
+  File.open('config.json', 'w+') do |f|
+    f.puts default_config
+    puts "[+] config.json made, you can now edit it"
+    exit
+  end
+end
+
+
+
+
+
+begin
+  $conf = JSON.parse(File.read("config.json"))
+rescue Exception
+  puts "[-] Something went wrong loading your config.json file"
+  exit
+end
+
+#Make ctrl+c exit whole script
+trap "INT" do
+  Thread.list.each do |thread|
+    thread.exit unless thread == Thread.current
+  end
+ exit
+end
+
+#get rid of annoying messages
+ouput_manager = StringIO.new
+$stdout = ouput_manager
+$stderr = ouput_manager
+
+module Kernel
+  def puts(msg)
+    #if msg[0] == "["
+      ::STDOUT.printf(msg + "\n")
+    #end
+  end
+end
+
+
+module Pokeplot
+  include Database
+
+  Database.mongo_host = $conf["mongo"]["host"]
+
+  #Socket and notifications must come before miner to capture its database requests
+  if $conf["socket"]["enabled"]
+    s = Socket.new($conf["socket"]["host"], $conf["socket"]["port"], $conf["socket"]["log"])
+    Database.mongo_monitor(s)
+  end
+
+  if $conf["pushbullet"]["enabled"]
+    pb = Pushbullet.new($conf["pushbullet"]["api_key"], $conf["pushbullet"]["pokemon"])
+    Database.mongo_monitor(pb)
+  end
+
+  if $conf['combos'].is_a?(Array) && $conf['combos'].count > 0
+    $conf['combos'].each do |c|
+      c = c.split(':')
+      $conf['accounts'] << {'username' => c[0], 'password' => c[1], 'provider' => 'ptc'}
+    end
+  end
+
+
+  if $conf["miner"]["enabled"]
+    miner = Miner.new($conf['accounts'], $conf['location'][0], $conf['location'][1], $conf['miner']['range'], $conf['miner']['interval'], $conf['miner']['pokemon'], $conf['miner']['forts'], $conf['miner']['threaded'], $conf['api']['encryption_path'], $conf['miner']['log'], $conf['api']['log'])
+    miner.start
+  end
+
+  if $conf["webserver"]["enabled"]
+    web = Thread.new do
+      Web.config($conf["webserver"]["host"], $conf["webserver"]["port"])
+      Web.run!
+    end
+  end
+
+  # Join thread so it doesnt exit script
+  if miner.is_a?(Miner) && miner.miner.is_a?(Thread)
+    puts "[+] Joining thread"
+    miner.miner.join
+  end
+
+  if web && web.is_a?(Thread)
+    puts "[+] Joining thread"
+    web.join
+  end
+
+end

data/lib/pokeplot/api.rb

@@ -0,0 +1,109 @@
+require 'poke-api'
+require 'pokeplot/helpers/cell_ids'
+
+module Pokeplot
+  class API
+
+    def initialize(lat, lng, encryption, logging = true)
+      Poke::API::Logging.log_level = :FATAL
+      @api = Poke::API::Client.new
+      set_location(lat, lng)
+      @encryption = encryption
+      @logging = logging
+    end
+
+    def login(username, password, provider)
+      puts "[+] Logging in #{username} with #{provider}" if @logging
+      loop do
+        begin
+          @username = username
+          @password = password
+          @provider = provider
+          @api.login(username, password, provider)
+          @api.activate_signature(@encryption)
+          basic_request
+          break
+        rescue Poke::API::Errors::UnknownProtoFault
+
+          puts "[-] Login for #{username} failed, retrying in 5 seconds" if @logging
+          sleep(5)
+        end
+      end
+    end
+
+    def map_heartbeat
+
+      loop do
+        @api.get_player
+        @api.get_hatched_eggs
+        @api.get_inventory(last_timestamp_ms: (Time.now.to_f * 1000).to_i)
+        @api.check_awarded_badges
+        @api.download_settings(hash: "05daf51635c82611d1aac95c0b051d3ec088a930")
+
+        cell_ids = Helpers::CellIds.get(@api.lat, @api.lng)
+        @api.get_map_objects(
+          latitude: lat,
+          longitude: lng,
+          since_timestamp_ms: [0] * cell_ids.count,
+          cell_id: cell_ids
+        )
+
+        begin
+          response = @api.call.response
+        rescue StandardError
+          puts "[-] Error.. Re-logging in" if @logging
+          login(@username, @password, @provider)
+          next
+        end
+
+        case response[:status_code]
+        when 1
+          return response
+        when 2
+          puts "[-] Error... Retrying..." if @logging
+          basic_request
+        when 102
+          puts "[-] Error, re-logging in" if @logging
+          login(@username, @password, @provider)
+        else
+          puts "[-] Heartbeat error: status code #{response[:status_code]}" if @logging
+          puts "[+] Retrying in 2 seconds..." if @logging
+          sleep(2)
+        end
+      end
+    end
+
+    def set_location(latitude, longitude)
+      @api.lat = latitude
+      @api.lng = longitude
+    end
+
+    def lat
+      return @api.lat
+    end
+
+    def lng
+      return @api.lng
+    end
+
+    def lat=(lat)
+      @api.lat = lat
+    end
+
+    def lng=(lng)
+      @api.lng = lng
+    end
+
+
+    private
+
+    def basic_request
+      @api.get_player
+      @api.get_hatched_eggs
+      @api.get_inventory
+      @api.check_awarded_badges
+      @api.call
+    end
+
+  end
+end

data/lib/pokeplot/database.rb

@@ -0,0 +1,22 @@
+require 'mongo'
+
+module Pokeplot
+  module Database
+
+    @@mongo_host = '127.0.0.1:27017'
+
+    def self.mongo
+      Mongo::Logger.level = Logger::FATAL
+      return Mongo::Client.new([ @@mongo_host ], :database => 'pokeplot')
+    end
+
+    def self.mongo_host=(host)
+      @@mongo_host = host
+    end
+
+    def self.mongo_monitor(class_instance)
+      Mongo::Monitoring::Global.subscribe(Mongo::Monitoring::COMMAND, class_instance)
+    end
+
+  end
+end

data/lib/pokeplot/helpers/array.rb

@@ -0,0 +1,5 @@
+class Array; def in_groups(num_groups)
+  return [] if num_groups == 0
+  slice_size = (self.size/Float(num_groups)).ceil
+  self.each_slice(slice_size).to_a
+end; end

data/lib/pokeplot/helpers/cell_ids.rb

@@ -0,0 +1,14 @@
+require 'poke-api/helpers'
+module Pokeplot
+  module Helpers
+    class CellIds
+      class << self
+
+        def get(lat, long)
+          return ::Poke::API::Helpers.get_cells(lat, long)
+        end
+
+      end
+    end
+  end
+end

data/lib/pokeplot/helpers/math.rb

@@ -0,0 +1,20 @@
+module Pokeplot
+  module Helpers
+    class Math
+      class << self
+
+        def deg_to_rad(deg)
+          return deg * (::Math::PI / 180)
+        end
+
+        def get_earth_radius(lat)
+          earth_radius_max = 6378137.0
+          earth_radius_min = 6356752.3
+          latrad = deg_to_rad(lat)
+          return (1.0/(((::Math.cos(latrad))/earth_radius_max)**(2) + ((::Math.sin(latrad))/earth_radius_min)**(2)))**(1.0/2)
+        end
+
+      end
+    end
+  end
+end

data/lib/pokeplot/helpers/time.rb

@@ -0,0 +1,15 @@
+module Pokeplot
+  module Helpers
+    class Time
+      class << self
+
+        def to_s(seconds)
+          m = (seconds / 60).floor
+          s = seconds % 60
+          return "#{m} minutes and #{s} seconds"
+        end
+
+      end
+    end
+  end
+end