apiotics-test 0.1.49

data/lib/apiotics/railtie.rb

@@ -0,0 +1,14 @@
+require 'apiotics'
+require 'rails'
+
+module Apiotics
+  class Railtie < Rails::Railtie
+    generators do
+      require 'lib/generators/apiotics/model'
+      require 'lib/generators/apiotics/create_table'
+      require 'lib/generators/apiotics/create_model'
+      require 'lib/generators/apiotics/initializer'
+      require 'lib/generators/apiotics/migration'
+    end
+  end
+end

data/lib/apiotics/server.rb

@@ -0,0 +1,238 @@
+
+# This class creates a local server listening on port 8001, and connects to a remote server on port 8000.  Messages sent to the local server are transmitted
+# to the remote server.  Messages received from the remote server are forwarded to the appropriate logic.
+
+require "socket"
+require 'openssl'
+require 'json'
+
+module Apiotics
+  class Server
+    def initialize
+        @error_msg = nil
+        server_details = Server.lookup
+        rgs = ApioticsSetting.find_by(key: "server")
+        rgs_port = ApioticsSetting.find_by(key: "port")
+        if rgs == nil
+          if server_details["status"] == "ok"
+            rgs = server_details["ip"]
+            c = ApioticsSetting.new
+            c.key = "server"
+            c.value = rgs
+            c.save
+            rgs_port = server_details["port"]
+            c = ApioticsSetting.new
+            c.key = "port"
+            c.value = rgs_port
+            c.save
+          else
+            @error_msg = server_details["status_msg"]
+          end
+        else
+          if server_details["status"] == "ok"
+            rgs.value = server_details["ip"]
+            rgs.save
+            rgs = rgs.value
+            rgs_port.value = server_details["port"]
+            rgs_port.save
+            rgs_port = rgs_port.value
+          else
+            @error_msg = server_details["status_msg"]
+          end
+        end
+      if Apiotics.configuration.tls == true
+        socket = TCPSocket.new(rgs, rgs_port)
+        context = OpenSSL::SSL::SSLContext.new
+        context.key = Server.key
+        context.cert = Server.cert
+        if Apiotics.configuration.verify_peer == true
+          ca_tempfile = Tempfile.new
+          ca_tempfile.write Server.ca_cert.to_pem
+          ca_tempfile.rewind
+          context.ca_file = ca_tempfile.path
+          context.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        end
+        server = OpenSSL::SSL::SSLSocket.new socket, context
+        server.sync_close = true
+        server.connect
+        if Apiotics.configuration.verify_peer == true
+          ca_tempfile.close(true)
+        end
+      else
+        server = TCPSocket.open(rgs, rgs_port)
+      end
+      @server = server
+      @localport = Apiotics.configuration.local_port
+      listen_remote
+      listen_local
+    end
+
+    def send(msg)
+      puts "Message sent: #{msg}"
+      @server.puts( msg )
+    end
+    
+    def close
+      @server.close
+    end
+    
+    def listen_remote
+      begin
+        Thread.new do
+          loop do
+        	  msg = @server.gets
+        	  puts msg
+            msg_hash = Apiotics::Parse.message(msg)
+            r = Apiotics::Insert.new(msg_hash)
+            puts "Message received": msg_hash
+            if r.valid == true
+              if r.action == "set-request-ack" || r.action == "set-complete" || r.action == "get-ack"
+                r.save
+                unless Apiotics.configuration.local_logging == false
+                  r.save_log
+                end
+              end
+            end
+          end
+        end
+      rescue => e
+        puts e
+        listen_remote
+      end
+      
+    end
+    
+    def listen_local
+      begin
+        server = TCPServer.open(@localport)
+        if Apiotics.configuration.handshake == true
+          self.send('{"action":"connect"}')
+        end
+        loop do
+        	Thread.fork(server.accept) do |client| 
+        		s = client.gets
+        		if @error_msg != nil
+        		  string = '{"error":"' + error_msg + '"}'
+        		  client.puts(string)
+        		end
+        		#puts s
+        		self.send(s)
+        	end
+        end
+      rescue => e
+        puts e
+        listen_local
+      end
+    end
+    
+    def self.lookup
+      if Apiotics.configuration.tls == true
+        socket = TCPSocket.new(Apiotics.configuration.server, Apiotics.configuration.server_port)
+        context = OpenSSL::SSL::SSLContext.new
+        context.key = Server.key
+        context.cert = Server.cert
+        if Apiotics.configuration.verify_peer == true
+          ca_tempfile = Tempfile.new
+          ca_tempfile.write Server.ca_cert.to_pem
+          ca_tempfile.rewind
+          context.ca_file = ca_tempfile.path
+          context.verify_mode = OpenSSL::SSL::VERIFY_PEER
+        end
+        server = OpenSSL::SSL::SSLSocket.new socket, context
+        server.sync_close = true
+        server.connect
+      else
+        server = TCPSocket.open(Apiotics.configuration.server, Apiotics.configuration.server_port)
+      end
+      server.puts('{"action":"lookup"}')
+      msg = server.gets
+      hash = JSON.parse(msg)
+      server.close
+      if Apiotics.configuration.verify_peer == true
+        ca_tempfile.close(true)
+      end
+      return hash
+    end
+    
+    def self.key
+      key = ApioticsSetting.find_by(key: "key")
+      if key == nil
+        key = Server.generate_key
+      else
+        key = key.value
+      end
+      pass_phrase = ApioticsSetting.find_by(key: "key_pass_phrase").value
+      key = OpenSSL::PKey::RSA.new key, pass_phrase
+      return key
+    end
+    
+    def self.cert
+      cert = ApioticsSetting.find_by(key: "cert")
+      if cert == nil
+        key = Server.key
+        public_key = key.public_key
+        cert = Server.generate_cert(key, public_key)
+        c = ApioticsSetting.new
+        c.key = "cert"
+        c.value = cert
+        c.save
+      else
+        cert = cert.value
+      end
+      cert = OpenSSL::X509::Certificate.new cert
+      return cert
+    end
+    
+    def self.generate_key
+      key = OpenSSL::PKey::RSA.new 2048
+      pass_phrase = 'simbiotes'
+      cipher = OpenSSL::Cipher.new 'AES-128-CBC'
+      s = ApioticsSetting.new
+      s.key = "key_cipher"
+      s.value = "OpenSSL::Cipher.new 'AES-128-CBC'"
+      s.save
+      s = ApioticsSetting.new
+      s.key = "key_pass_phrase"
+      s.value = 'simbiotes'
+      s.save
+      s = ApioticsSetting.new
+      s.key = "public_key"
+      s.value = key.public_key.to_pem
+      s.save
+      s = ApioticsSetting.new
+      s.key = "key"
+      s.value = key.export(cipher, pass_phrase)
+      s.save
+      return s.value
+    end
+    
+    def self.generate_cert(key, public_key)
+      csr = OpenSSL::X509::Request.new
+      csr.version = 0
+      csr.subject = OpenSSL::X509::Name.parse "CN=simbiotes.com/O=#{Apiotics.configuration.public_key}/OU=#{Apiotics.configuration.private_key}"
+      csr.public_key = key.public_key
+      csr.sign key, OpenSSL::Digest::SHA1.new
+      cert = Apiotics::Portal.generate_certificate(csr)
+      return cert
+    end
+    
+    def self.ca_cert
+      ca_cert = ApioticsSetting.find_by(key: "ca_cert")
+      if ca_cert == nil
+        ca_cert = Apiotics::Portal.ca_certificate
+        c = ApioticsSetting.new
+        c.key = "ca_cert"
+        c.value = ca_cert
+        c.save
+      else
+        ca_cert = ca_cert.value
+      end
+      ca_cert = OpenSSL::X509::Certificate.new ca_cert
+      return ca_cert
+    end
+    
+    def do_at_exit
+    end
+    
+  end
+end

data/lib/apiotics/version.rb

@@ -0,0 +1,3 @@
+module Apiotics
+  VERSION = '0.1.49'
+end

data/lib/generators/apiotics/channel/USAGE

@@ -0,0 +1,10 @@
+Description:
+    This generator creates channel files with the right code for Apiotics interactions.
+
+Example:
+    rails generate apiotics:channel Worker Driver
+		rails generate apiotics:channel Worker Script
+
+    This will create:
+        app/channel/worker/driver_channel.rb
+				app/assets/javascripts/channels/worker_driver.coffee

data/lib/generators/apiotics/channel/channel_generator.rb

@@ -0,0 +1,35 @@
+module Apiotics
+  class ChannelGenerator < Rails::Generators::Base
+    source_root File.expand_path('../templates', __FILE__)
+    argument :parent, :type => :string
+  
+    def copy_channel_files
+      template "apiotics_channel.rb.erb", "app/channels/#{module_file_name}_channel.rb"
+      template "apiotics_channel_client.coffee.erb", "app/assets/javascripts/channels/#{module_file_name}.coffee"
+      template "apiotics_channel_initializer.rb.erb", "config/initializers/apiotics_channel.rb"
+      route "mount ActionCable.server => '/cable'"
+    end
+    
+    private
+    
+    def file_name
+      name.underscore
+    end
+    
+    def module_file_name
+      parent.underscore
+    end
+    
+    def class_name
+      name.classify
+    end
+    
+    def module_name
+      parent.classify
+    end
+    
+    def table_prefix
+      parent.underscore + "_"
+    end
+  end
+end

data/lib/generators/apiotics/channel/templates/apiotics_channel.rb.erb

@@ -0,0 +1,9 @@
+class <%= module_name + "Channel" %> < ApplicationCable::Channel
+  def subscribed
+    stream_from "<%= "#{module_file_name}_channel" %>"
+  end
+
+  def unsubscribed
+    # Any cleanup needed when channel is unsubscribed
+  end
+end

data/lib/generators/apiotics/channel/templates/apiotics_channel_client.coffee.erb

@@ -0,0 +1,16 @@
+$(document).ready ->
+	App.alert = App.cable.subscriptions.create('<%= "#{module_name}" + "Channel" %>',
+		connected: ->
+			# Called when the subscription is ready for use on the server
+			return
+		disconnected: ->
+			# Called when the subscription has been terminated by the server
+			return
+		received: (data) ->
+			console.log data
+			tr = $("#" + data["apiotics_instance"])
+			td = tr.children("." + data["worker_name"] + "-" + data["model_name"] + "-" + data["interface"])
+			td.text(data["value"])
+			# Called when there's incoming data on the websocket for this channel
+			return
+	)

data/lib/generators/apiotics/channel/templates/apiotics_channel_initializer.rb.erb

@@ -0,0 +1,3 @@
+Apiotics.configure do |config|
+  config.push = true
+end

data/lib/generators/apiotics/controller/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Creates a model and associated database table with the attributes specified.  If using data from the portal (recommended) Worker should be the name of a worker defined in your Hive, and Model should be the name of a Driver or Script in that Worker.
+
+Example:
+    rails generate apiotics:scaffold Worker
+
+    This will create:
+				app/controllers/worker.rb

data/lib/generators/apiotics/controller/controller_generator.rb

@@ -0,0 +1,25 @@
+module Apiotics
+  class ControllerGenerator < Rails::Generators::Base
+    source_root File.expand_path('../templates', __FILE__)
+    argument :parent, :type => :string
+  
+    def copy_scaffold_files
+      @models = Apiotics.configuration.targets[module_name.to_s]
+      template "apiotics_scaffold.rb.erb", "app/controllers/#{module_file_name}/#{module_file_name.pluralize}_controller.rb"
+      generate "apiotics:view", "#{parent}"
+      route "scope module: :#{module_file_name} do \n\t\tresources :#{module_file_name}s\n\tend"
+      generate "simple_form:install"
+    end
+    
+    private
+    
+    def module_file_name
+      parent.underscore
+    end
+    
+    def module_name
+      parent.classify
+    end
+    
+  end
+end

data/lib/generators/apiotics/controller/templates/apiotics_scaffold.rb.erb

@@ -0,0 +1,60 @@
+module <%= module_name %>
+  class <%= module_name.pluralize + "Controller"%> < ApplicationController
+    before_action <%= ":set_" + module_file_name %>, only: [:show, :edit, :update, :destroy]
+
+    # GET /alert/leds
+    def index
+      @<%= module_file_name + "s" %> = <%= "::" + module_name + "::" + module_name + ".all" %>
+    end
+
+    # GET /alert/leds/1
+    def show
+    end
+
+    # GET /alert/leds/new
+    def new
+      redirect_to <%= module_file_name %>s_url, notice: 'You are not authorized to do that.'
+    end
+
+    # GET /alert/leds/1/edit
+    def edit
+    end
+
+    # POST /alert/leds
+    def create
+      @<%= module_file_name %> = <%= "::" + module_name + "::" + module_name + ".new(#{module_file_name}_params)" %>
+
+      if @<%= module_file_name %>.save
+        redirect_to <%= module_file_name %>_path(@<%= module_file_name %>), notice: '<%= module_name %> was successfully created.'
+      else
+        render :new
+      end
+    end
+
+    # PATCH/PUT /alert/leds/1
+    def update
+      if @<%= module_file_name %>.update(<%= module_file_name %>_params)
+        redirect_to <%= module_file_name %>_path(@<%= module_file_name %>), notice: '<%= module_name %> was successfully updated.'
+      else
+        render :edit
+      end
+    end
+
+    # DELETE /alert/leds/1
+    def destroy
+      @<%= module_file_name %>.destroy
+      redirect_to <%= module_file_name %>s_url, notice: '<%= module_file_name %> was successfully destroyed.'
+    end
+
+    private
+      # Use callbacks to share common setup or constraints between actions.
+      def set_<%= module_file_name %>
+        @<%= module_file_name %> = <%= "::" + module_name + "::" + module_name + ".find(params[:id])" %>
+      end
+
+      # Only allow a trusted parameter "white list" through.
+      def <%= module_file_name %>_params
+        params.fetch(:<%= module_file_name %>_<%= module_file_name %>, {}).permit(:id, :apiotics_instance<% Apiotics.configuration.targets[module_name].keys.each do |key| %>, :<%= key.underscore.downcase %>_attributes => <%= Apiotics.configuration.targets[module_name][key] + ["id"] %><% end %>)
+      end
+  end
+end

data/lib/generators/apiotics/create_model/USAGE

@@ -0,0 +1,10 @@
+Description:
+    This generator creates model files with the right code for Apiotics interactions.
+
+Example:
+    rails generate apiotics:model Worker Driver
+		rails generate apiotics:model Worker Script
+
+    This will create:
+				app/models/worker.rb
+        app/models/worker/driver.rb