device-tracker 0.1.0

data/lib/device/tracker/db/migrate/20150521071815_create_users.rb

@@ -0,0 +1,13 @@
+class CreateUsers < ActiveRecord::Migration
+  def change
+    create_table :users do |t|
+      t.string :username, index: true
+      t.string :name
+      t.string :password_digest
+      t.string :email
+      t.boolean :is_active, default: false
+      t.boolean :is_admin, default: false
+      t.timestamps null: false
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20150521082155_create_devices.rb

@@ -0,0 +1,23 @@
+class CreateDevices < ActiveRecord::Migration
+  def change
+
+    create_table :devices do |t|
+
+      t.string :unid, index: true
+      t.string :manufacturer
+      t.string :device
+      t.text :description
+      t.datetime :checked_out_since
+      t.boolean :available, default: true
+      t.string :imei, default: nil
+      t.belongs_to :operating_system
+      t.belongs_to :user
+
+      t.boolean :sim_card, default: false
+      t.boolean :debug_device, default: false
+
+      t.timestamps null: false
+    end
+
+  end
+end

data/lib/device/tracker/db/migrate/20150521120335_create_operating_systems.rb

@@ -0,0 +1,8 @@
+class CreateOperatingSystems < ActiveRecord::Migration
+  def change
+    create_table :operating_systems do |t|
+      t.string :name
+      t.integer :api_level
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20150527162242_create_transactions.rb

@@ -0,0 +1,13 @@
+class CreateTransactions < ActiveRecord::Migration
+  def change
+    create_table :transactions do |t|
+      t.string :transaction_type, index: true
+
+      t.belongs_to :user
+      t.belongs_to :device
+
+      t.text :description
+      t.datetime :created_at
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20151027073050_create_heartbeat.rb

@@ -0,0 +1,10 @@
+class CreateHeartbeat < ActiveRecord::Migration
+  def change
+    create_table :heartbeats do |t|
+      t.decimal :longitude, { precision: 10, scale: 6 }
+      t.decimal :latitude, { precision: 10, scale: 6 }
+      t.belongs_to :device
+      t.timestamps null: false
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20151028132946_add_user_verification.rb

@@ -0,0 +1,7 @@
+class AddUserVerification < ActiveRecord::Migration
+  def change
+    change_table :users do |t|
+        t.boolean :is_verified, default: false
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20151028141328_remove_is_active_from_users.rb

@@ -0,0 +1,5 @@
+class RemoveIsActiveFromUsers < ActiveRecord::Migration
+  def change
+    remove_column :users, :is_active, :boolean, default: false
+  end
+end

data/lib/device/tracker/db/migrate/20151029085629_add_password_reset_code_to_users.rb

@@ -0,0 +1,7 @@
+class AddPasswordResetCodeToUsers < ActiveRecord::Migration
+  def change
+    change_table :users do |t|
+        t.string :reset_code, default: nil
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20151030130341_add_missing_column_to_devices.rb

@@ -0,0 +1,7 @@
+class AddMissingColumnToDevices < ActiveRecord::Migration
+  def change
+    change_table :devices do |t|
+      t.boolean :missing, default: false
+    end
+  end
+end

data/lib/device/tracker/db/migrate/20151102141601_add_serial_number_to_devices.rb

@@ -0,0 +1,7 @@
+class AddSerialNumberToDevices < ActiveRecord::Migration
+  def change
+    change_table :devices do |t|
+      t.string :serial_number
+    end
+  end
+end

data/lib/device/tracker/db/schema.rb

@@ -0,0 +1,74 @@
+# encoding: UTF-8
+# This file is auto-generated from the current state of the database. Instead
+# of editing this file, please use the migrations feature of Active Record to
+# incrementally modify your database, and then regenerate this schema definition.
+#
+# Note that this schema.rb definition is the authoritative source for your
+# database schema. If you need to create the application database on another
+# system, you should be using db:schema:load, not running all the migrations
+# from scratch. The latter is a flawed and unsustainable approach (the more migrations
+# you'll amass, the slower it'll run and the greater likelihood for issues).
+#
+# It's strongly recommended that you check this file into your version control system.
+
+ActiveRecord::Schema.define(version: 20151109121036) do
+
+  create_table "devices", force: :cascade do |t|
+    t.string   "unid"
+    t.string   "manufacturer"
+    t.string   "device"
+    t.text     "description"
+    t.datetime "checked_out_since"
+    t.boolean  "available",           default: true
+    t.string   "imei"
+    t.integer  "operating_system_id"
+    t.integer  "user_id"
+    t.boolean  "sim_card",            default: false
+    t.boolean  "debug_device",        default: false
+    t.datetime "created_at",                          null: false
+    t.datetime "updated_at",                          null: false
+    t.boolean  "missing",             default: false
+    t.integer  "checkout_count",      default: 0
+    t.string   "serial_number"
+  end
+
+  add_index "devices", ["unid"], name: "index_devices_on_unid"
+
+  create_table "heartbeats", force: :cascade do |t|
+    t.decimal  "longitude",  precision: 10, scale: 6
+    t.decimal  "latitude",   precision: 10, scale: 6
+    t.integer  "device_id"
+    t.datetime "created_at",                          null: false
+    t.datetime "updated_at",                          null: false
+  end
+
+  create_table "operating_systems", force: :cascade do |t|
+    t.string  "name"
+    t.integer "api_level"
+  end
+
+  create_table "transactions", force: :cascade do |t|
+    t.string   "transaction_type"
+    t.integer  "user_id"
+    t.integer  "device_id"
+    t.text     "description"
+    t.datetime "created_at"
+  end
+
+  add_index "transactions", ["transaction_type"], name: "index_transactions_on_transaction_type"
+
+  create_table "users", force: :cascade do |t|
+    t.string   "username"
+    t.string   "name"
+    t.string   "password_digest"
+    t.string   "email"
+    t.boolean  "is_admin",        default: false
+    t.datetime "created_at",                      null: false
+    t.datetime "updated_at",                      null: false
+    t.boolean  "is_verified",     default: false
+    t.string   "reset_code"
+  end
+
+  add_index "users", ["username"], name: "index_users_on_username"
+
+end

data/lib/device/tracker/db/seeds.rb

@@ -0,0 +1,18 @@
+module Device
+  module Tracker
+    class Seed
+      def self.seed(name: name, password: password, email: email)
+        user = User.find_or_create_by({
+          name: name,
+          username: 'admin',
+          email: email,
+          is_admin: true,
+          is_verified: true
+        })
+        user.password = password
+        user.save
+        puts "Added #{user.username} to the database."
+      end
+    end
+  end
+end

data/lib/device/tracker/dependencies.rb

@@ -0,0 +1,15 @@
+require 'rubygems'
+require 'bundler'
+
+%w(
+  sinatra
+  sinatra/activerecord
+  sinatra/flash
+  sinatra/partial
+  tilt/erb
+  json
+  pony
+).each { |d| require d }
+
+puts File.dirname(__FILE__)
+Dir.glob("#{File.dirname(__FILE__)}/{helpers,controllers,models}/*.rb").each { |file| require file }

data/lib/device/tracker/helpers/application_helper.rb

@@ -0,0 +1,84 @@
+module Device
+  module Tracker
+    module ApplicationHelper
+
+      SECRET = 'mYR4nd0mAr$eS3cr3t/fuGg1wugGl3'
+
+      def is_admin?
+        user = get_logged_in_user
+        return if user.nil?
+        user[:is_admin]
+      end
+
+      def value_for(name, object)
+        if flash[name]
+          flash[name]
+        elsif object
+          object.send name
+        end
+      end
+
+      def get_logged_in_user
+        session[:user] unless session[:user].nil?
+      end
+
+      def generate_activation_code(size = 6)
+        charset = %w{0 1 2 3 4 6 7 8 9 A C D E F G H J K M N P Q R T V W X Y Z}
+        (0...size).map{ charset.to_a[rand(charset.size)] }.join
+      end
+
+      def change_password?(params)
+        if params[:user][:password].empty? or params[:user][:password_confirmation].empty?
+          false
+        else
+          true
+        end
+      end
+
+      def protected!
+        if session[:user].nil?
+          create_flash "info", ["You must be logged in to see this page."]
+          redirect "/login"
+        elsif !session[:user][:is_verified]
+          create_flash "info", ["You account is awaiting verification by an admin."]
+          redirect "/login"
+        end
+      end
+
+      def report_transaction(message, type, device = nil)
+        transaction = Transaction.new
+
+        if session[:user]
+          transaction.user_id = get_logged_in_user[:id]
+        end
+
+        transaction.description = message
+        transaction.transaction_type = type
+        transaction.device_id = device.id unless device.nil?
+
+        transaction.save!
+      end
+
+      def perform_admin_check
+        user = get_logged_in_user
+        if user.nil? or user[:is_admin] == false
+          create_flash "warning", ["#{user[:name]} does not have permissions to access this page."]
+          redirect back
+        end
+      end
+
+      def create_flash(type, message)
+        flash[:message] = {css_class: type, message: message }
+      end
+
+      def valid_heartbeat?(data)
+        if !data["heartbeat"].nil? and !data["heartbeat"]["longitude"].nil? and
+           !data["heartbeat"]["latitude"].nil? and !data["heartbeat"]["device_id"].nil?
+          return true
+        end
+        false
+      end
+
+    end
+  end
+end

data/lib/device/tracker/models/device.rb

@@ -0,0 +1,39 @@
+require 'date'
+
+module Device
+  module Tracker
+    class Device < ActiveRecord::Base
+
+      belongs_to :operating_system
+      belongs_to :user
+
+      has_many :transactions
+      has_many :heartbeats
+
+      validates :unid, :manufacturer, :device, :description, :operating_system, presence: true
+
+      validates :unid, uniqueness: true
+
+      def unid=(unid)
+        self[:unid] = unid.upcase
+      end
+
+      def full_name
+        self[:manufacturer] + " " + self[:device]
+      end
+
+      def passed_use_by_date?
+        (! self[:available] and (Time.now - self[:checked_out_since]).to_i / 1.day >= 3)
+      end
+
+      def checked_out_since_formatted
+        self[:checked_out_since].strftime("%A, %d %B %Y at %I:%M%p")
+      end
+
+      def days_checked_out
+        (Time.now.to_date - self[:checked_out_since].to_date).floor
+      end
+
+    end
+  end
+end

data/lib/device/tracker/models/heartbeat.rb

@@ -0,0 +1,9 @@
+module Device
+  module Tracker
+    class Heartbeat < ActiveRecord::Base
+      validates :longitude, :latitude, :device_id, presence: true
+
+      belongs_to :device
+    end
+  end
+end

data/lib/device/tracker/models/operating_system.rb

@@ -0,0 +1,8 @@
+module Device
+  module Tracker
+    class OperatingSystem < ActiveRecord::Base
+      validates :name, presence: true, uniqueness: true
+      has_many :devices
+    end
+  end
+end

data/lib/device/tracker/models/transaction.rb

@@ -0,0 +1,56 @@
+require "date"
+
+module Device
+  module Tracker
+    class Transaction < ActiveRecord::Base
+
+      belongs_to :user
+      belongs_to :device
+
+      validates :transaction_type, :description, presence: true
+      validates :transaction_type, inclusion: { in: %w(CHECKOUT RETURN REGISTRATION LOGIN ACTIVATE DEACTIVATE DELETION MISSING FOUND),
+        message: "%{value} is not a valid transaction." }
+
+      def created_at
+        self[:created_at].strftime("%a %d %b %Y at %H:%M:%S")
+      end
+
+      def self.missing
+        "MISSING"
+      end
+
+      def self.found
+        "FOUND"
+      end
+
+      def self.activate
+        "ACTIVATE"
+      end
+
+      def self.deactivate
+        "DEACTIVATE"
+      end
+
+      def self.checkout
+        "CHECKOUT"
+      end
+
+      def self.return
+        "RETURN"
+      end
+
+      def self.registration
+        "REGISTRATION"
+      end
+
+      def self.login
+        "LOGIN"
+      end
+
+      def self.deletion
+        "DELETION"
+      end
+
+    end
+  end
+end

data/lib/device/tracker/models/user.rb

@@ -0,0 +1,22 @@
+require 'bcrypt'
+
+module Device
+  module Tracker
+    class User < ActiveRecord::Base
+      has_secure_password
+      has_many :devices
+      has_many :transactions
+      validates :username, :name, :email, presence: true
+      validates :password, :presence => { :if => :password_required? }
+      validates_format_of :email, :with => /\A([^@\s]+)@((?:[-a-z0-9]+\.)+[a-z]{2,})\Z/i, :on => :create
+      validates :username, uniqueness: true
+      validates :email, uniqueness: true
+      validates_confirmation_of :password
+
+    protected
+      def password_required?
+        self.new_record?
+      end
+    end
+  end
+end