helios_aim 0.2.2

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler"
+Bundler.setup
+
+gemspec = eval(File.read("helios.gemspec"))
+
+task :build => "#{gemspec.full_name}.gem"
+
+file "#{gemspec.full_name}.gem" => gemspec.files + ["helios.gemspec"] do
+  system "gem build helios.gemspec"
+end

data/bin/helios

@@ -0,0 +1,20 @@
+#!/usr/bin/env ruby
+
+require 'commander/import'
+$:.push File.expand_path("../../lib", __FILE__)
+
+require 'helios/version'
+
+HighLine.track_eof = false # Fix for built-in Ruby
+Signal.trap("INT") {} # Suppress backtrace when exiting command
+
+program :version, Helios::VERSION
+program :description, 'A command-line interface for building mobile infrastructures'
+
+program :help, 'Author', 'Mattt Thompson <m@mattt.me>'
+program :help, 'Website', 'https://helios.io'
+program :help_formatter, :compact
+
+default_command :help
+
+require 'helios/commands'

data/helios_aim.gemspec

@@ -0,0 +1,47 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "helios/version"
+
+Gem::Specification.new do |s|
+  s.name        = "helios_aim"
+  s.authors     = ["Mattt Thompson"]
+  s.email       = "mattt@heroku.com"
+  s.license     = "MIT"
+  s.homepage    = "http://helios.io"
+  s.version     = Helios::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.summary     = "An extensible open-source mobile backend framework"
+  s.description = "Helios is an open-source framework that provides essential backend services for iOS apps, from data synchronization and user accounts to push notifications, in-app purchases, and passbook integration. It allows developers to get a client-server app up-and-running in just a few minutes, and seamlessly incorporate functionality as necessary."
+
+  s.add_dependency "commander", "~> 4.1"
+  s.add_dependency "foreman", "~> 0.63"
+  s.add_dependency "rack-contrib", "~> 1.1"
+  s.add_dependency "rack-push-notification", "~> 0.4"
+  s.add_dependency "rack-in-app-purchase", "~> 0.1"
+  s.add_dependency "rack-passbook", "~> 0.1"
+  s.add_dependency "rack-newsstand", "~> 0.1"
+  s.add_dependency "rack-scaffold_aim"
+  s.add_dependency "core_data"
+  s.add_dependency "json", "~> 1.7"
+  s.add_dependency "coffee-script", "~> 2.2"
+  s.add_dependency "sinatra", "~> 1.3"
+  s.add_dependency "sinatra-contrib", "~> 1.3"
+  s.add_dependency "sinatra-assetpack", "~> 0.2.2"
+  s.add_dependency "sinatra-backbone", "~> 0.1.1"
+  s.add_dependency "sinatra-param", "~> 0.1"
+  s.add_dependency "sinatra-support", "~> 1.2"
+  s.add_dependency "haml", ">= 3.1"
+  s.add_dependency "compass", "~> 0.12"
+  s.add_dependency "zurb-foundation", "4.1.2"
+  s.add_dependency "rails-database-url", "~> 1.0"
+  s.add_dependency "fog", "~> 1.10"
+  s.add_dependency "houston", "~> 0.2"
+
+  s.add_development_dependency "rake"
+  s.add_development_dependency "rspec"
+
+  s.files         = Dir["./**/*"].reject{|file| file =~ /\.\/(bin|example|log|pkg|script|spec|test|vendor)/} + Dir.glob("./lib/helios/templates/*", File::FNM_DOTMATCH)
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+end

data/lib/helios.rb

@@ -0,0 +1,27 @@
+require 'rack'
+
+module Helios
+  class Application
+    def initialize(app = nil, options = {}, &block)
+      @app = Rack::Builder.new do
+        map '/admin' do
+          use Rack::Auth::Basic, "Restricted Area" do |username, password|
+            username == (ENV['HELIOS_ADMIN_USERNAME'] || "") and password == (ENV['HELIOS_ADMIN_PASSWORD'] || "")
+          end if ENV['HELIOS_ADMIN_USERNAME'] or ENV['HELIOS_ADMIN_PASSWORD']
+
+          run Helios::Frontend.new
+        end
+
+        run Helios::Backend.new(&block)
+      end
+    end
+
+    def call(env)
+      @app.call(env)
+    end
+  end
+end
+
+require 'helios/backend'
+require 'helios/frontend'
+require 'helios/version'

data/lib/helios/backend.rb

@@ -0,0 +1,64 @@
+require 'rack'
+
+module Helios
+  class Backend < Rack::Builder
+    DEFAULT_PATHS = {
+      data: '/'
+    }
+
+    require 'rails-database-url' if const_defined?(:Rails)
+
+    def initialize(*args, &block)
+      raise ArgumentError, "Missing block" unless block_given?
+      super(&nil)
+
+      @services = {}
+
+      instance_eval(&block)
+    end
+
+    def call(env)
+      return super(env) unless env["REQUEST_METHOD"] == "OPTIONS" and env["REQUEST_PATH"] == "/"
+
+      links = []
+      @services.each do |path, middleware|
+        links << %{<#{path}>; rel="#{middleware}"}
+      end
+
+      [206, {"Link" => links.join("\n")}, []]
+    end
+
+    private
+
+    def service(identifier, options = {}, &block)
+      if identifier.is_a?(Class)
+        middleware = identifier
+      else
+        begin
+          middleware = Helios::Backend.const_get(constantize(identifier))
+        rescue NameError
+          raise LoadError, "Could not find matching service for #{identifier.inspect} (Helios::Backend::#{constantize(identifier)}). You may need to install an additional gem (such as helios-#{identifier})."
+        end
+      end
+
+      path = "/#{(options.delete(:root) || DEFAULT_PATHS[identifier] || identifier)}".squeeze("/")
+
+      map path do
+        instance_eval(&block) if block_given?
+        run middleware.new(self, options)
+      end
+
+      @services[path] = middleware
+    end
+
+    def constantize(identifier)
+      identifier.to_s.split(/([[:alpha:]]*)/).select{|c| /[[:alpha:]]/ === c}.map(&:capitalize).join("")
+    end
+  end
+end
+
+require 'helios/backend/data'
+require 'helios/backend/in-app-purchase'
+require 'helios/backend/passbook'
+require 'helios/backend/push-notification'
+require 'helios/backend/newsstand'

data/lib/helios/backend/data.rb

@@ -0,0 +1,40 @@
+require 'core_data'
+require 'sequel'
+
+require 'rack/scaffold_aim'
+
+require 'sinatra/base'
+require 'sinatra/param'
+
+
+class Helios::Backend::Data < Sinatra::Base
+  helpers Sinatra::Param
+
+  def initialize(app, options = {})
+    super(Rack::Scaffold.new(options))
+
+    @model = CoreData::DataModel.new(options[:model]) rescue nil
+  end
+
+  before do
+    content_type :json
+  end
+
+  options '/resources' do
+
+    links = []
+    @model.entities.each do |entity|
+      links << %{</#{entity.name.downcase.pluralize}>; rel="resource"}
+    end
+
+    response['Link'] = links.join("\n")
+
+    @model.entities.collect{ |entity|
+      {
+        name: entity.name,
+        url: "/#{entity.name.downcase.pluralize}",
+        attributes: Hash[entity.attributes.collect{|attribute| [attribute.name, attribute.type]}]
+      }
+    }.to_json
+  end
+end

data/lib/helios/backend/in-app-purchase.rb

@@ -0,0 +1,43 @@
+require 'rack/in-app-purchase'
+
+require 'sinatra/base'
+require 'sinatra/param'
+
+class Helios::Backend::InAppPurchase < Sinatra::Base
+  helpers Sinatra::Param
+
+  def initialize(app, options = {}, &block)
+    super(Rack::InAppPurchase.new)
+  end
+
+  before do
+    content_type :json
+  end
+
+  helpers Sinatra::Param
+
+  get '/receipts' do
+    param :q, String
+
+    receipts = Rack::InAppPurchase::Receipt.dataset
+    receipts = receipts.filter("tsv @@ to_tsquery('english', ?)", "#{params[:q]}:*") if params[:q] and not params[:q].empty?
+
+    if params[:page] or params[:per_page]
+      param :page, Integer, default: 1, min: 1
+      param :per_page, Integer, default: 100, in: (1..100)
+
+      {
+        receipts: receipts.limit(params[:per_page], (params[:page] - 1) * params[:per_page]).naked.all,
+        page: params[:page],
+        total: receipts.count
+      }.to_json
+    else
+      param :limit, Integer, default: 100, in: (1..100)
+      param :offset, Integer, default: 0, min: 0
+
+      {
+        receipts: receipts.limit(params[:limit], params[:offset]).naked.all
+      }.to_json
+    end
+  end
+end

data/lib/helios/backend/newsstand.rb

@@ -0,0 +1,97 @@
+require 'rack/newsstand'
+
+require 'sinatra/base'
+require 'sinatra/param'
+
+require 'fog'
+
+class Helios::Backend::Newsstand < Sinatra::Base
+  helpers Sinatra::Param
+
+  def initialize(app, options = {}, &block)
+    super(Rack::Newsstand.new)
+
+    @storage = Fog::Storage.new(options[:storage]) if options[:storage]
+  end
+
+  before do
+    content_type :json
+  end
+
+  get '/issues/?' do
+    pass unless request.accept? 'application/json'
+
+    param :q, String
+
+    issues = Rack::Newsstand::Issue.dataset
+    issues = issues.filter("tsv @@ to_tsquery('english', ?)", "#{params[:q]}:*") if params[:q] and not params[:q].empty?
+
+    if params[:page] or params[:per_page]
+      param :page, Integer, default: 1, min: 1
+      param :per_page, Integer, default: 100, in: (1..100)
+
+      {
+        issues: issues.limit(params[:per_page], (params[:page] - 1) * params[:per_page]).naked.all,
+        page: params[:page],
+        total: issues.count
+      }.to_json
+    else
+      param :limit, Integer, default: 100, in: (1..100)
+      param :offset, Integer, default: 0, min: 0
+
+      {
+        issues: issues.limit(params[:limit], params[:offset]).naked.all
+      }.to_json
+    end
+  end
+
+  head '/issues/new?' do
+    status 503 and return unless @storage
+
+    status 204
+  end
+
+  post '/issues/?' do
+    status 503 and return unless @storage
+
+    param :name, String, empty: false
+    param :summary, String
+
+    issue = Rack::Newsstand::Issue.new(params)
+
+    if issue.valid?
+      directory = @storage.directories.create(key: "newsstand-issue-#{issue.name}-#{Time.now.to_i}", public: true)
+
+      covers, assets = {}, []
+      [:covers, :assets].each do |attribute|
+        (params[attribute] || []).each do |f|
+          file = directory.files.create(
+            key: File.basename(f[:filename]),
+            body: File.open(f[:tempfile]),
+            public: true
+          )
+
+          case attribute
+          when :covers
+            covers["SOURCE"] = file.public_url
+          when :assets
+            assets << file.public_url
+          end
+        end
+      end
+
+      issue.set(cover_urls: covers, asset_urls: assets)
+
+      if issue.save
+        status 201
+        issue.to_json
+      else
+        status 400
+        {errors: issue.errors}.to_json
+      end
+    else
+      status 400
+      {errors: issue.errors}.to_json
+    end
+  end
+end

data/lib/helios/backend/passbook.rb

@@ -0,0 +1,41 @@
+require 'rack/passbook'
+
+require 'sinatra/base'
+require 'sinatra/param'
+
+class Helios::Backend::Passbook < Sinatra::Base
+  helpers Sinatra::Param
+
+  def initialize(app, options = {}, &block)
+    super(Rack::Passbook.new)
+  end
+
+  before do
+    content_type :json
+  end
+
+  get '/passes' do
+    param :q, String
+
+    passes = Rack::Passbook::Pass.dataset
+    passes = passes.filter("tsv @@ to_tsquery('english', ?)", "#{params[:q]}:*") if params[:q] and not params[:q].empty?
+
+    if params[:page] or params[:per_page]
+      param :page, Integer, default: 1, min: 1
+      param :per_page, Integer, default: 100, in: (1..100)
+
+      {
+        passes: passes.limit(params[:per_page], (params[:page] - 1) * params[:per_page]).naked.all,
+        page: params[:page],
+        total: passes.count
+      }.to_json
+    else
+      param :limit, Integer, default: 100, in: (1..100)
+      param :offset, Integer, default: 0, min: 0
+
+      {
+        passes: passes.limit(params[:limit], params[:offset]).naked.all
+      }.to_json
+    end
+  end
+end

data/lib/helios/backend/push-notification.rb

@@ -0,0 +1,110 @@
+require 'rack/push-notification'
+
+require 'sinatra/base'
+require 'sinatra/param'
+
+require 'houston'
+
+class Helios::Backend::PushNotification < Sinatra::Base
+  helpers Sinatra::Param
+  attr_reader :apn_certificate, :apn_environment
+
+  def initialize(app, options = {}, &block)
+    super(Rack::PushNotification.new)
+
+    @apn_certificate = options[:apn_certificate] || ENV['APN_CERTIFICATE']
+    @apn_environment = options[:apn_environment] || ENV['APN_ENVIRONMENT']
+  end
+
+  before do
+    content_type :json
+  end
+
+  get '/devices/?' do
+    param :q, String
+
+    devices = ::Rack::PushNotification::Device.dataset
+    devices = devices.filter("tsv @@ to_tsquery('english', ?)", "#{params[:q]}:*") if params[:q] and not params[:q].empty?
+
+    if params[:page] or params[:per_page]
+      param :page, Integer, default: 1, min: 1
+      param :per_page, Integer, default: 100, in: (1..100)
+
+      {
+        devices: devices.limit(params[:per_page], (params[:page] - 1) * params[:per_page]),
+        page: params[:page],
+        total: devices.count
+      }.to_json
+    else
+      param :limit, Integer, default: 100, in: (1..100)
+      param :offset, Integer, default: 0, min: 0
+
+      {
+        devices: devices.limit(params[:limit], params[:offset])
+      }.to_json
+    end
+  end
+
+  get '/devices/:token/?' do
+    record = ::Rack::PushNotification::Device.find(token: params[:token])
+
+    if record
+      {device: record}.to_json
+    else
+      status 404
+    end
+  end
+
+  head '/message' do
+    status 503 and return unless client
+
+    status 204
+  end
+
+  post '/message' do
+    status 503 and return unless client
+
+    param :payload, String, empty: false
+    param :tokens, Array, empty: false
+
+    tokens = params[:tokens] || ::Rack::PushNotification::Device.all.collect(&:token)
+
+    options = JSON.parse(params[:payload])
+    options[:alert] = options["aps"]["alert"]
+    options[:badge] = options["aps"]["badge"]
+    options[:sound] = options["aps"]["sound"]
+    options.delete("aps")
+
+    begin
+      notifications = tokens.collect{|token| Houston::Notification.new(options.update({device: token}))}
+      client.push(*notifications)
+
+      status 204
+    rescue => error
+      status 500
+
+      {error: error}.to_json
+    end
+  end
+
+  private
+
+  def client
+    begin
+      return nil unless apn_certificate and ::File.exist?(apn_certificate)
+
+      client = case apn_environment.to_sym
+               when :development
+                 Houston::Client.development
+               when :production
+                 Houston::Client.production
+               end
+
+      client.certificate = ::File.read(apn_certificate)
+
+      return client
+    rescue
+      return nil
+    end
+  end
+end