distribution_wrappers 0.2.2

data/lib/distribution_wrappers/hipchat/hipchat.rb

@@ -0,0 +1,80 @@
+module DistributionWrappers
+  class HipChatWrapper < DistributionWrappers::Base
+    
+    def send_message(recipient)
+      super
+      client = get_access(@auth[:key])
+      
+      case recipient['contact_type']
+      when 'channel'
+        client[recipient['identifier']].send('backstitch', @message, :color => "purple", :message_format => "html")
+      when 'user'
+        client.user(recipient['identifier']).send(@message, :color => "purple", :message_format => "html")
+      end
+    end
+    
+    def get_contacts
+      client = get_access(@auth[:key])
+      rooms = client.rooms
+
+      users = client.users
+      
+      return nil if (rooms.nil? || rooms.length == 0) && (users.nil? || users.length == 0)
+      
+      csv_sting = ""
+      
+      # Loop through the channels just grabbing the names of them (that's all we need for the user)
+      rooms.each_with_index do |room, index|          
+        csv_sting += CSV.generate_line [room.name, room.name, '{"url": null}', 'channel', @params[:channel_id]]
+
+        if index % 100 == 0
+          @params[:temp_file].write(csv_sting)
+          csv_sting = ""
+        end
+      end
+      
+      @params[:temp_file].write(csv_sting)
+      
+      csv_sting = ""
+      
+      users.each_with_index do |user, index| 
+        icon_url = user.photo_url ? user.photo_url : "null"
+        csv_sting += CSV.generate_line [user.name, user.user_id, "{\"url\": \"#{icon_url}\"}", 'user', @params[:channel_id]]
+
+        if index % 100 == 0
+          @params[:temp_file].write(csv_sting)
+          csv_sting = ""
+        end
+      end
+      
+      @params[:temp_file].write(csv_sting)
+      
+      return true
+    end
+    
+    private
+    
+    def prepare(identifier=nil)
+      base_url = ""
+      if ENV['ENVIRONMENT'] == "development"
+        base_url = "http://localhost:3000"
+      else
+        base_url = "https://studio.backstit.ch"
+      end
+      
+      params = "contact_id=#{@version_contact.id}&version_id=#{@version_contact.studio_post_version_id}&organization_id=#{@params[:organization_id]}"
+      
+      body_text = "<a href='#{base_url}/posts/#{@params[:post].id}?#{params}'><strong>#{@params[:post].title}</strong></a>"
+      body_text += "<br><br>#{TextHelper.clean_text(@params[:post].draft_content)[0..280]}..."
+      body_text += "<br><br><a href='#{base_url}/posts/#{@params[:post].id}?#{params}'><img src='#{@params[:post].screenshot_url}'/></a>"
+      body_text += "<br><a href='#{base_url}/posts/#{@params[:post].id}?#{params}'><strong>Read More...</strong></a>"
+      
+      return body_text
+    end
+    
+    def get_access(key)
+      client = HipChat::Client.new(key, :api_version => 'v2')
+      client
+    end
+  end
+end

data/lib/distribution_wrappers/office365/office365.rb

@@ -0,0 +1,51 @@
+module DistributionWrappers
+  class Office365 < DistributionWrappers::Email
+    
+    def send_message(email)
+      super
+      key = @auth[:key]
+      recipients = []
+      
+      # Office 365's API takes addresses in a specific way, so we have to build up an array 
+      # of recipients before we hit their API 
+      recipients << {'emailAddress' => {"address" => email['identifier']}}
+      
+      message = {
+        'subject' => @params[:subject],
+        'body' => {
+          'contentType' => 'HTML',
+          'content' => @message
+        },
+        "toRecipients" => recipients,
+        "sender" => {
+          "emailAddress" => {
+            "address" => @params[:custom_opts]['send_from']
+          }
+        }
+      }
+      
+      send_response = Curl.post('https://graph.microsoft.com/v1.0/me/sendMail', {"Message" => message}.to_json) do |http|
+        http.headers['Authorization'] = "Bearer #{key}"
+        http.headers['Content-Type'] = 'application/json'
+      end
+    end
+    
+    def get_contacts
+      key = @auth[:key]
+      results = []
+      
+      response = RestClient.get('https://graph.microsoft.com/v1.0/me/contacts', {"Authorization" => "Bearer #{key}"})
+      contacts = JSON.parse(response.body)
+      
+      csv_string = ""
+      
+      contacts['value'].each do |contact|
+        csv_string += CSV.generate_line [contact['displayName'], contact['emailAddresses'].first['address'], '{"url": null}', 'email', @params[:channel_id]]
+      end  
+      
+      @params[:temp_file].write(csv_string)
+      
+      return true
+    end
+  end
+end

data/lib/distribution_wrappers/sendgrid/sendgrid.rb

@@ -0,0 +1,35 @@
+module DistributionWrappers
+  class Sendgrid < DistributionWrappers::Email
+    include SendGrid
+    
+    def send_message(recipient)
+      super
+      email = recipient['identifier']
+      metadata = {}
+      metadata["user_id"] = @params[:user].id if @params[:user]
+      
+      ### Send Sendgrid Email ###
+      
+      # Build request
+      data = {"metadata" => metadata}
+              
+      # Post to SendGrid
+      api_key = Keys.sendgrid.studio.api_key
+      client = SendGrid::API.new(:api_key => api_key)
+      
+      send_from = {
+          :email => "#{@params[:custom_opts]['send_from']}@#{Keys.sendgrid.studio.send_from_domain}",
+          :name => @params[:user].display_name
+        }
+        
+      from = SendGrid::Email.new(send_from)
+      subject = @params[:subject]
+      to = SendGrid::Email.new(email: email)
+      content = SendGrid::Content.new(type: 'text/html', value: @message)
+      mail = Mail.new(from, subject, to, content)
+      
+      
+      response = client.client.mail._('send').post(request_body: mail.to_json)
+    end
+  end
+end

data/lib/distribution_wrappers/slack/slack.rb

@@ -0,0 +1,87 @@
+module DistributionWrappers
+  class SlackWrapper < DistributionWrappers::Base
+
+    def send_message(recipient)
+      super
+      client = get_access(@auth[:key])
+
+      client.chat_postMessage(channel: recipient['identifier'], attachments: [@message], as_user: true)
+    end
+
+    def get_contacts
+      url = "https://slack.com/api/channels.list?token=#{@auth[:key]}"
+      channels = JSON.parse(RestClient.get(url))['channels']
+
+      url = "https://slack.com/api/users.list?token=#{@auth[:key]}"
+      members = JSON.parse(RestClient.get(url))["members"]
+
+      return nil if (channels.nil? || channels.length == 0) && (members.nil? || members.length == 0)
+
+      csv_sting = ""
+
+      # Loop through the channels just grabbing the names of them (that's all we need for the user)
+      channels.each_with_index do |channel, index|
+        csv_sting += CSV.generate_line [channel["name"], channel["name"], '{"url": null}', 'channel', @params[:channel_id]]
+
+        if index % 100 == 0
+          @params[:temp_file].write(csv_sting)
+          csv_sting = ""
+        end
+      end
+
+      @params[:temp_file].write(csv_sting)
+
+      csv_sting = ""
+
+      members.each_with_index do |member, index|
+        icon_url = member['profile']['image_192'] ? member['profile']['image_192'] : "null"
+        csv_sting += CSV.generate_line [member["real_name"], member["id"], "{\"url\": #{icon_url}}", 'user', @params[:channel_id]]
+
+        if index % 100 == 0
+          @params[:temp_file].write(csv_sting)
+          csv_sting = ""
+        end
+      end
+
+      @params[:temp_file].write(csv_sting)
+
+      return true
+    end
+
+    private
+
+    def prepare(identifier=nil)
+      base_url = ""
+      if ENV['ENVIRONMENT'] == "development"
+        base_url = "http://localhost:3000"
+      else
+        base_url = "https://studio.backstit.ch"
+      end
+
+      params = "contact_id=#{@version_contact.id}&version_id=#{@version_contact.studio_post_version_id}&organization_id=#{@params[:organization].id}"
+
+      return {
+        title: @params[:post].title,
+        title_link: "#{base_url}/posts/#{@params[:post].id}?#{params}",
+        author_name: @params[:user].display_name,
+        color: @params[:organization].highlight_color,
+        fallback: @params[:post].title,
+        image_url: @params[:post].screenshot_url,
+        fields: [
+          {
+            value: "#{TextHelper.clean_text(@params[:post].draft_content)[0..280]}...",
+            short: false
+          }
+        ]
+      }
+    end
+
+    def get_access(key)
+      Slack.configure do |config|
+        config.token = key
+      end
+
+      return Slack::Web::Client.new
+    end
+  end
+end

data/lib/helpers/keys.rb

@@ -0,0 +1,10 @@
+require 'settingslogic'
+module DistributionWrappers
+  class Keys < Settingslogic
+    if ENV['RAILS_ENV'] == 'production'
+      source File.join(File.dirname(__FILE__), '..', 'config/key_chain.yml')
+    else
+      source File.join(File.dirname(__FILE__), '..', 'config/key_chain_dev.yml')
+    end
+  end
+end

data/lib/helpers/text_helper.rb

@@ -0,0 +1,9 @@
+class TextHelper
+  def self.clean_text(text)
+    clean_text = Sanitize.clean(text)
+    clean_text.gsub!("\n", " ")
+    clean_text.gsub!("\t", " ")
+    clean_text.gsub!(/((http|https|ftp|ftps):\/\/)?([a-zA-Z0-9\-]*\.)+[a-zA-Z0-9]{2,4}(\/[a-zA-Z0-9=.?&-]*)?/i, "")
+    return clean_text.strip.gsub("&", "&")
+  end
+end

data/lib/models/device_session_distro.rb

@@ -0,0 +1,5 @@
+class DeviceSessionDistro < ActiveRecord::Base
+  self.table_name = "device_sessions"
+  
+  belongs_to :user, :class_name => "UserDistro", :foreign_key => :user_id
+end

data/lib/models/distribution_channel_distro.rb

@@ -0,0 +1,4 @@
+class DistributionChannelDistro < ActiveRecord::Base
+  self.table_name = 'distribution_channels'
+  belongs_to :owner, polymorphic: true
+end

data/lib/models/feed_distro.rb

@@ -0,0 +1,11 @@
+class FeedDistro < ActiveRecord::Base
+  self.table_name = "feeds"
+  serialize :params, JSON
+  
+  belongs_to :owner, polymorphic: true
+  
+  has_many :team_feeds, :class_name => "TeamFeedDistro", :foreign_key => :feed_id 
+  has_many :teams, :through => :team_feeds
+  
+  has_many :topic_feeds, :class_name => "TopicFeedDistro", :foreign_key => :feed_id 
+end

data/lib/models/organization_admin_distro.rb

@@ -0,0 +1,5 @@
+class OrganizationAdminDistro < ActiveRecord::Base
+  self.table_name = 'organization_admins'
+  belongs_to :organization, :class_name => "OrganizationDistro", :foreign_key => :organization_id
+  belongs_to :user, :class_name => "UserDistro", :foreign_key => :user_id
+end

data/lib/models/organization_distro.rb

@@ -0,0 +1,10 @@
+class OrganizationDistro < ActiveRecord::Base
+  self.table_name = 'organizations'
+  
+  has_many :feeds, :as => :owner, :class_name => "FeedDistro"
+  has_many :topics, :as => :owner, :class_name => "TopicDistro"
+  
+  has_many :teams, -> { order("is_primary DESC, name") }, :class_name => "TeamDistro", :foreign_key => :organization_id
+  has_many :team_members, :through => :teams
+  has_many :members, :through => :team_members
+end

data/lib/models/studio_post_version_contact_distro.rb

@@ -0,0 +1,5 @@
+class PostVersionContactDistro < ActiveRecord::Base
+  self.table_name = "studio_post_version_contacts"
+  
+  belongs_to :version, :foreign_key => "studio_post_version_id", :class_name => "PostVersionDistro"
+end

data/lib/models/studio_post_version_distro.rb

@@ -0,0 +1,5 @@
+class PostVersionDistro < ActiveRecord::Base
+  self.table_name = "studio_post_versions"
+  
+  has_many :contacts, :foreign_key => 'studio_post_version_id', :class_name => "PostVersionContact"
+end

data/lib/models/team_distro.rb

@@ -0,0 +1,13 @@
+class TeamDistro < ActiveRecord::Base
+  self.table_name = "teams"
+  belongs_to :organization, :class_name => "OrganizationDistro", :foreign_key => :organization_id
+  
+  has_many :team_members, :class_name => "TeamMemberDistro", :foreign_key => :team_id
+  has_many :members, :through => :team_members
+  
+  has_many :team_feeds, :class_name => "TeamFeedDistro", :foreign_key => :team_id
+  has_many :feeds, :through => :team_feeds
+  
+  has_many :owned_feeds, :class_name => 'FeedDistro', :as => :owner
+  
+end

data/lib/models/team_feed_distro.rb

@@ -0,0 +1,5 @@
+class TeamFeedDistro < ActiveRecord::Base
+  self.table_name = 'team_feeds'
+  belongs_to :team, :class_name => "TeamDistro", :foreign_key => :team_id
+  belongs_to :feed, :class_name => "FeedDistro", :foreign_key => :feed_id
+end

data/lib/models/team_member_distro.rb

@@ -0,0 +1,6 @@
+class TeamMemberDistro < ActiveRecord::Base
+  self.table_name = "team_members"
+  belongs_to :team, :class_name => "TeamDistro", :foreign_key => :team_id
+  delegate :organization, :to => :team
+  belongs_to :member, :class_name => "UserDistro", :foreign_key => :user_id
+end

data/lib/models/topic_distro.rb

@@ -0,0 +1,11 @@
+class TopicDistro < ActiveRecord::Base
+  self.table_name = "topics"
+  serialize :icon, JSON
+  serialize :banner, JSON
+  serialize :serialized_alias, JSON
+  serialize :serialized_promoted_alias, JSON
+  
+  belongs_to :owner, polymorphic: true
+  has_many :topic_feeds, :class_name => "TopicFeedDistro", :foreign_key => :topic_id
+  has_many :feeds, :through => :topic_feeds
+end

data/lib/models/topic_feed_distro.rb

@@ -0,0 +1,5 @@
+class TopicFeedDistro < ActiveRecord::Base
+  self.table_name = "topic_feeds"
+  belongs_to :topic, :class_name => "TopicDistro", :foreign_key => :topic_id
+  belongs_to :feed, :class_name => "FeedDistro", :foreign_key => :feed_id
+end

data/lib/models/topic_subscription_distro.rb

@@ -0,0 +1,9 @@
+class TopicSubscriptionDistro < ActiveRecord::Base
+  self.table_name = "topic_subscriptions"
+  
+  serialize :serialized_alias, JSON
+  serialize :serialized_promoted_alias, JSON
+  
+  belongs_to :user, :foreign_key => "user_id", :class_name => "UserDistro"
+  belongs_to :topic, :foreign_key => "topic_id", :class_name => "TopicDistro"
+end

data/lib/models/user_distro.rb

@@ -0,0 +1,34 @@
+class UserDistro < ActiveRecord::Base  
+  self.table_name = "users"
+  
+  has_many :team_memberships, :foreign_key => "user_id", :class_name => "TeamMemberDistro"
+  has_many :teams, :through => :team_memberships
+  
+  has_many :organizations, :through => :teams
+
+  has_many :organization_admins, :foreign_key => "user_id", :class_name => "OrganizationAdminDistro"
+  has_many :managed_organizations, :through => :organization_admins, :source => :organization
+  
+  has_many :topic_subscriptions, :class_name => 'TopicSubscriptionDistro', :foreign_key => "user_id"
+  has_many :topics, :through => :topic_subscriptions
+  has_many :owned_topics, :class_name => 'TopicDistro', :as => :owner
+  
+  def managed_feeds
+    self.managed_organizations
+        .select("team_feeds.id, feeds.id AS feed_id, teams.id AS team_id, organizations.id AS organization_id, feeds.name, organizations.highlight_color, team_feeds.feed_type")
+        .joins({:teams => {:team_feeds => :feed}}).where("team_feeds.feed_type = 'manual'").uniq
+  end
+  
+  def avatar_url 
+    avatar = ""
+    if self.avatar != nil
+      avatar = self.avatar
+    else
+      hash = Digest::MD5.hexdigest(self.email)
+      default = "https://images-backstitch.s3.amazonaws.com/next/logos/backstitch_purple_icon.png"
+      avatar = "https://www.gravatar.com/avatar/#{hash}?d=#{default}"
+      # "http://unicornify.appspot.com/avatar/#{hash}?s=128"
+    end
+    avatar
+  end
+end

data/test/helper.rb

@@ -0,0 +1,34 @@
+require 'simplecov'
+
+module SimpleCov::Configuration
+  def clean_filters
+    @filters = []
+  end
+end
+
+SimpleCov.configure do
+  clean_filters
+  load_adapter 'test_frameworks'
+end
+
+ENV["COVERAGE"] && SimpleCov.start do
+  add_filter "/.rvm/"
+end
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'distribution_wrappers'
+
+class Test::Unit::TestCase
+end