apiotics 0.2.6 → 0.2.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6919be7f896c869fdd2e20947c61afdf41dd12bc
-  data.tar.gz: 152f2bc9fc9d0eff1b58100f2e48e1baeef650d2
+  metadata.gz: 1e25c98184861bff100688037a23340b6be5385e
+  data.tar.gz: f97ad7382054e1e67090a64f83aff5f12b7ac340
 SHA512:
-  metadata.gz: cb533fdf78271769d7a762d7fed4dd9c1359410cf5978498710d54b2f1c1c501f7e4bf54269741fafe339414fae4087a091c68641ed4e5404b5aa211a9556ee1
-  data.tar.gz: 7d973e60f7d1dfd218ccd72cb2211203d452d2969bccce44ea01884c413b01813676adfe89d9106218735826fe884b1c0f59161095a5415240305327e3a088e8
+  metadata.gz: 568dfad8b2116d196a085861cd3dfea958c57945482c6bb84b8f68024e3bc692df39d71fc7ba45166bd8aeebe2156fc17ce104ac48e3c10d0fb8e3adbc8f8a65
+  data.tar.gz: bce6b5fd058002e67bda0273c035b0794af6e712c9ddbb7a7a3d90418fc3627969045ab51456cf9e195be7a7f0b02c460f6594c930d0f78fae8c215cd8b25768

data/lib/apiotics/hardware.rb

@@ -294,6 +294,174 @@ module Apiotics
       end
     end
     
+    def self.check_sd_card_scripts
+      status = nil
+      download_directory = Rails.root.join('lib', 'sd_card')
+      Dir.mkdir download_directory unless Dir.exist?(download_directory)
+      download_directory = Rails.root.join('lib', 'sd_card', 'downloads')
+      Dir.mkdir download_directory unless Dir.exist?(download_directory)
+      Dir.chdir download_directory
+      system = nil
+      if Hardware.mac == true
+        system = "mac"
+      elsif Hardware.linux == true
+        system = "linux"
+      end
+      downloaded = false
+      unless system == nil
+        if system == "mac"
+          if File.exist?("#{download_directory}/brand_image.mac.sh")
+            downloaded = true
+          end
+        elsif system == "linux"
+          if File.exist?("#{download_directory}/brand_image.sh")
+            downloaded = true
+          end
+        end
+        if downloaded == false
+          scripts = Apiotics::Portal.sd_scripts(system)
+          File.write("#{download_directory}/sd_scripts.tgz")
+          `tar xvzf #{download_directory}/sd_scripts.tgz -C #{download_directory}`
+          `rm #{download_directory}/sd_scripts.tgz`
+        end
+        if system == "mac"
+          if File.exist?("#{download_directory}/brand_image.mac.sh")
+            status = download_directory
+          end
+        elsif system == "linux"
+          if File.exist?("#{download_directory}/brand_image.sh")
+            status = download_directory
+          end
+        end
+      end
+      return status
+    end
+    
+    def self.get_sd_image(download_directory, worker_name)
+      Dir.chdir download_directory
+      image_path = "#{download_directory}/#{worker_name}.img.gz"
+      unless File.exist?(image_path)
+        Apiotics::Portal.get_sd_image(download_directory, worker_name)
+        #zip_path = "#{download_directory}/#{worker_name}.img.gz"
+        #`gunzip -c #{zip_path} > #{image_path}`
+        #`rm #{zip_path}`
+      end
+      return image_path
+    end
+    
+    def self.sd_config_file(worker_name, name, download_directory)
+      worker_directory = download_directory + "/#{worker_name.downcase}"
+      Dir.mkdir worker_directory unless Dir.exist?(worker_directory)
+      config_file_path = "#{worker_directory}/#{worker_name.downcase}.json"
+      if File.exist?(config_file_path)
+        config = File.read(config_file_path)
+        config = JSON.parse(config)
+      else
+        config = {}
+      end
+      config["port"] = "8883"
+      config["download"] = true
+      config["public_key"] = Apiotics.configuration.public_key
+      config["private_key"] = Apiotics.configuration.private_key
+      if name == nil || name == ""
+        config["name"] = ""
+      else
+        config["name"] = name.to_s
+      end
+      config["portal"] = Apiotics.configuration.portal
+      config["message_validation"] = false
+      config["local_server"] = "localhost"
+      config["local_port"] = "8001"
+      config["containers"] = false
+      config["driver_logging"] = true
+      File.write(config_file_path, config.to_json)
+      config_file_path
+    end
+    
+    def self.brand_sd_image(download_directory, image_path, wifi_settings, config_file_path)
+      status = false
+      branding_script_path == nil
+      if Hardware.mac == true
+        branding_script_path = "#{download_directory}/brand_image.mac.sh"
+      elsif Hardware.linux == true
+        branding_script_path = "#{download_directory}/brand_image.sh"
+      end
+      if branding_script_path != nil
+        `#{branding_script_path} #{image_path} #{wifi_settings["ssid"]} #{wifi_settings["password"]} #{config_file_path} dhcp`
+        status = true
+      else
+        puts "Unrecognized system architecture, exiting."
+      end
+      status
+    end
+    
+    def self.find_smallest_disk
+      small_disk_size = 99999999999999999999
+      small_disk = "no_disk"
+      disks = {}
+      output = `df -k`
+      lines = output.split("\n")
+      i = 0
+      lines.each do |line|
+        lines[i] = line.split(" ")
+        i += 1
+      end
+      gigabyte_scaling = 1024 * 1024
+      lines.each do |line|
+        length = line.count
+        unless line[length - 1] == "/"
+          disks[line[0]] = line[1].to_i / gigabyte_scaling
+        end
+      end
+      disks.each do |disk, size|
+        if size < small_disk_size && size > 4 && disk[0..3] == "/dev"
+          small_disk_size = size
+          small_disk = disk
+        end
+      end
+        return small_disk
+      end
+    end
+    
+    def self.burn_sd_image(download_directory, image_path, disk)
+      puts "Burning #{image_path} to #{disk}..."
+      if Hardware.mac == true
+        execute_path = download_directory + "/burn_image_to_disk.mac.sh"
+      elsif Hardware.linux == true
+        execute_path = download_directory + "/burn_image_to_disk.sh"
+      else
+        execute_path = nil
+      end
+      if execute_path == nil
+        puts "This computer is not running a recognized OS.  Exiting."
+      else
+        `#{execute_path} #{image_path} #{disk}`
+      end
+    end
+    
+    def self.sd_card(worker_name, name, config)
+      download_directory = self.check_sd_card_scripts
+      if download_directory != nil
+        image_path = Hardware.get_sd_image(download_directory, worker_name)
+        wifi_settings = Hardware.fetch_wifi
+        if wifi_settings != nil
+          config_file_path = Hardware.sd_config_file(worker_name, name, download_directory)
+          if Hardware.brand_sd_image(download_directory, image_path, wifi_settings, config_file_path)
+            disk = Hardware.find_smallest_disk
+            if disk != "no_disk"
+              Hardware.burn_sd_image(download_directory, image_path, disk)
+            else
+              puts "No plausible SD card disk found.  Please insert an SD card into your computer. Exiting."
+            end
+          end
+        else
+          puts "No WiFi settings found.  Please run rake firmware:wifi[ssid,password,security_mode] to set your default wifi settings"
+        end
+      else
+        puts "SD card installation scripts not available. Exiting."
+      end
+    end
+    
     def self.install(worker_name, name)
       config = Apiotics::Portal.openocd_worker_config(worker_name)
       config = JSON.parse(config)
@@ -301,6 +469,8 @@ module Apiotics
         Hardware.openocd(worker_name, name, config)
       elsif config["installer"] == "Uniflash 4.3.0"
         Hardware.uniflash(worker_name, name, config)
+      elsif config["installer"] == "SD Card"
+        Hardware.sd_card(worker_name, name, config)
       else
         puts "Unknown installer for #{worker_name}.  Please check the installer settings for the #{worker_name} device on the Apiotics portal.\n"
       end
@@ -331,14 +501,29 @@ module Apiotics
 
       encrypted_wifi_config = args.to_json.apiotics_encrypt(pwd)
       
-      download_directory = Rails.root.join('lib', 'uniflash')
-      Dir.mkdir download_directory unless Dir.exist?(download_directory)
-      download_directory = Rails.root.join('lib', 'uniflash', 'downloads')
+      download_directory = Rails.root.join('lib', 'wifi')
       Dir.mkdir download_directory unless Dir.exist?(download_directory)
       Dir.chdir download_directory
-      path = Rails.root.join('lib', 'uniflash', 'downloads', 'wifi_settings')
+      path = Rails.root.join('lib', 'wifi', 'wifi_settings')
       File.write(path, encrypted_wifi_config)
     end
     
+    def self.fetch_wifi
+      return_hash = nil
+      download_directory = Rails.root.join('lib', 'wifi')
+      Dir.mkdir download_directory unless Dir.exist?(download_directory)
+      Dir.chdir download_directory
+      path = Rails.root.join('lib', 'wifi', 'wifi_settings')
+      if File.exist?(path)
+        pwd = ApioticsSetting.find_by(key: "wifi_key")
+        if pwd != nil
+          pwd = pwd.value
+          encrypted_settings = File.read(path)
+          return_hash = encrypted_settings.apiotics_decrypt(pwd)
+        end
+      end
+      return_hash
+    end
+    
   end
 end

data/lib/apiotics/portal.rb

@@ -250,6 +250,10 @@ module Apiotics
       response = HTTParty.post("#{Apiotics.configuration.portal}api/openocd", :query => {:public_key => Apiotics.configuration.public_key, :private_key => Apiotics.configuration.private_key, :system => system}).body
     end
     
+    def self.sd_scripts(system)
+      response = HTTParty.post("#{Apiotics.configuration.portal}api/sd_scripts", :query => {:public_key => Apiotics.configuration.public_key, :private_key => Apiotics.configuration.private_key, :system => system}).body
+    end
+    
     def self.openocd_worker_config(worker_name)
       response = HTTParty.post("#{Apiotics.configuration.portal}api/openocd_details", :query => {:public_key => Apiotics.configuration.public_key, :private_key => Apiotics.configuration.private_key, :worker => worker_name}).body
     end
@@ -262,6 +266,18 @@ module Apiotics
       }
     end
     
+    def self.get_sd_image(download_directory, worker_name)
+      filename = "#{download_directory}/#{worker_name}.img.gz"
+      File.open(filename, "w") do |file|
+        puts "Downloading image.."
+        response = HTTParty.post("#{Apiotics.configuration.portal}api/download", :query => {:public_key => Apiotics.configuration.public_key, :private_key => Apiotics.configuration.private_key, :worker => worker_name}, stream_body: true) do |fragment|
+          print "."
+          file.write(fragment)
+        end
+      end
+      File.unlink(filename)
+    end
+    
     def self.register_client
       response = HTTParty.post("#{Apiotics.configuration.portal}api/register_client", :query => {:public_key => Apiotics.configuration.public_key, :private_key => Apiotics.configuration.private_key}).body
       hash = JSON.parse(response)

data/lib/apiotics/version.rb

@@ -1,3 +1,3 @@
 module Apiotics
-  VERSION = '0.2.6'
+  VERSION = '0.2.7'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: apiotics
 version: !ruby/object:Gem::Version
-  version: 0.2.6
+  version: 0.2.7
 platform: ruby
 authors:
 - MicroArx Corporation
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-05-18 00:00:00.000000000 Z
+date: 2018-05-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails