ahoy_matey 1.6.1 → 2.0.0

data/lib/ahoy/stores/kafka_store.rb

@@ -1,42 +0,0 @@
-module Ahoy
-  module Stores
-    class KafkaStore < LogStore
-      def log_visit(data)
-        post(visits_topic, data)
-      end
-
-      def log_event(data)
-        post(events_topic, data)
-      end
-
-      def client
-        @client ||= begin
-          Kafka.new(
-            seed_brokers: ENV["KAFKA_URL"] || "localhost:9092",
-            logger: Rails.logger
-          )
-        end
-      end
-
-      def producer
-        @producer ||= begin
-          producer = client.async_producer(delivery_interval: 3)
-          at_exit { producer.shutdown }
-          producer
-        end
-      end
-
-      def post(topic, data)
-        producer.produce(data.to_json, topic: topic)
-      end
-
-      def visits_topic
-        "ahoy_visits"
-      end
-
-      def events_topic
-        "ahoy_events"
-      end
-    end
-  end
-end

data/lib/ahoy/stores/kinesis_firehose_store.rb

@@ -1,42 +0,0 @@
-module Ahoy
-  module Stores
-    class KinesisFirehoseStore < LogStore
-      def log_visit(data)
-        post(visits_stream, data)
-      end
-
-      def log_event(data)
-        post(events_stream, data)
-      end
-
-      def client
-        @client ||= Aws::Firehose::Client.new(credentials)
-      end
-
-      def post(stream, data)
-        client.put_record(
-          delivery_stream_name: stream,
-          record: {
-            data: "#{data.to_json}\n"
-          }
-        )
-      end
-
-      def credentials
-        {
-          access_key_id: ENV["AWS_ACCESS_KEY_ID"],
-          secret_access_key: ENV["AWS_SECRET_ACCESS_KEY"],
-          region: "us-east-1"
-        }
-      end
-
-      def visits_stream
-        "ahoy_visits"
-      end
-
-      def events_stream
-        "ahoy_events"
-      end
-    end
-  end
-end

data/lib/ahoy/stores/log_store.rb

@@ -1,53 +0,0 @@
-module Ahoy
-  module Stores
-    class LogStore < BaseStore
-      def track_visit(options, &block)
-        data = {
-          id: ahoy.visit_id,
-          visitor_id: ahoy.visitor_id
-        }.merge(visit_properties.to_hash)
-        data[:user_id] = user.id if user
-        data[:started_at] = options[:started_at]
-
-        yield(data) if block_given?
-
-        log_visit(data)
-      end
-
-      def track_event(name, properties, options, &block)
-        data = {
-          id: options[:id],
-          name: name,
-          properties: properties,
-          visit_id: ahoy.visit_id,
-          visitor_id: ahoy.visitor_id
-        }
-        data[:user_id] = user.id if user
-        data[:time] = options[:time]
-
-        yield(data) if block_given?
-
-        log_event(data)
-      end
-
-      protected
-
-      def log_visit(data)
-        visit_logger.info data.to_json
-      end
-
-      def log_event(data)
-        event_logger.info data.to_json
-      end
-
-      # TODO disable header
-      def visit_logger
-        @visit_logger ||= ActiveSupport::Logger.new(Rails.root.join("log/visits.log"))
-      end
-
-      def event_logger
-        @event_logger ||= ActiveSupport::Logger.new(Rails.root.join("log/events.log"))
-      end
-    end
-  end
-end

data/lib/ahoy/stores/mongoid_store.rb

@@ -1,63 +0,0 @@
-module Ahoy
-  module Stores
-    class MongoidStore < BaseStore
-      def track_visit(options, &block)
-        @visit =
-          visit_model.new do |v|
-            v.id = binary(ahoy.visit_id)
-            v.visitor_id = binary(ahoy.visitor_id)
-            v.user = user if v.respond_to?(:user=) && user
-            v.started_at = options[:started_at]
-          end
-
-        set_visit_properties(visit)
-
-        yield(visit) if block_given?
-
-        visit.upsert
-        geocode(visit)
-      end
-
-      def track_event(name, properties, options, &block)
-        event =
-          event_model.new do |e|
-            e.id = binary(options[:id])
-            e.visit_id = binary(ahoy.visit_id)
-            e.user = user if e.respond_to?(:user)
-            e.name = name
-            e.properties = properties
-            e.time = options[:time]
-          end
-
-        yield(event) if block_given?
-
-        event.upsert
-      end
-
-      def visit
-        @visit ||= visit_model.where(_id: binary(ahoy.visit_id)).first if ahoy.visit_id
-      end
-
-      protected
-
-      def visit_model
-        ::Visit
-      end
-
-      def event_model
-        ::Ahoy::Event
-      end
-
-      def binary(token)
-        token = token.delete("-")
-        if defined?(::BSON)
-          ::BSON::Binary.new(token, :uuid)
-        elsif defined?(::Moped::BSON)
-          ::Moped::BSON::Binary.new(:uuid, token)
-        else
-          token
-        end
-      end
-    end
-  end
-end

data/lib/ahoy/stores/nats_store.rb

@@ -1,34 +0,0 @@
-module Ahoy
-  module Stores
-    class NatsStore < LogStore
-      def log_visit(data)
-        publish(visits_subject, data)
-      end
-
-      def log_event(data)
-        publish(events_subject, data)
-      end
-
-      def publish(subject, data)
-        client.publish(subject, data.to_json)
-      end
-
-      def client
-        @client ||= begin
-          require "nats/io/client"
-          client = NATS::IO::Client.new
-          client.connect(servers: (ENV["NATS_URL"] || "nats://127.0.0.1:4222").split(","))
-          client
-        end
-      end
-
-      def visits_subject
-        "ahoy_visits"
-      end
-
-      def events_subject
-        "ahoy_events"
-      end
-    end
-  end
-end

data/lib/ahoy/stores/nsq_store.rb

@@ -1,36 +0,0 @@
-module Ahoy
-  module Stores
-    class NsqStore < LogStore
-      def log_visit(data)
-        post(visits_topic, data)
-      end
-
-      def log_event(data)
-        post(events_topic, data)
-      end
-
-      def client
-        @client ||= begin
-          require "nsq"
-          client = Nsq::Producer.new(
-            nsqd: ENV["NSQ_URL"] || "127.0.0.1:4150"
-          )
-          at_exit { client.terminate }
-          client
-        end
-      end
-
-      def post(topic, data)
-        client.write_to_topic(topic, data.to_json)
-      end
-
-      def visits_topic
-        "ahoy_visits"
-      end
-
-      def events_topic
-        "ahoy_events"
-      end
-    end
-  end
-end

data/lib/ahoy/subscribers/active_record.rb

@@ -1,19 +0,0 @@
-module Ahoy
-  module Subscribers
-    class ActiveRecord
-      def initialize(options = {})
-        @model = options[:model] || Ahoy::Event
-      end
-
-      def track(name, properties, options = {})
-        @model.create! do |e|
-          e.visit = options[:visit]
-          e.user = options[:user]
-          e.name = name
-          e.properties = properties
-          e.time = options[:time]
-        end
-      end
-    end
-  end
-end

data/lib/ahoy/throttle.rb

@@ -1,17 +0,0 @@
-require "rack/attack"
-
-module Ahoy
-  class Throttle < Rack::Attack
-    throttle("ahoy/ip", limit: Ahoy.throttle_limit, period: Ahoy.throttle_period) do |req|
-      if req.path.start_with?("/ahoy/")
-        req.ip
-      end
-    end
-
-    def_delegators self, :whitelisted?, :blacklisted?, :throttled?, :tracked?, :blocklisted?, :safelisted?
-
-    def self.throttled_response
-      Rack::Attack.throttled_response
-    end
-  end
-end

data/lib/generators/ahoy/stores/active_record_events_generator.rb

@@ -1,59 +0,0 @@
-# taken from https://github.com/collectiveidea/audited/blob/master/lib/generators/audited/install_generator.rb
-require "rails/generators"
-require "rails/generators/migration"
-require "active_record"
-require "rails/generators/active_record"
-
-module Ahoy
-  module Stores
-    module Generators
-      class ActiveRecordEventsGenerator < Rails::Generators::Base
-        include Rails::Generators::Migration
-        source_root File.expand_path("../templates", __FILE__)
-
-        class_option :database, type: :string, aliases: "-d"
-
-        # Implement the required interface for Rails::Generators::Migration.
-        def self.next_migration_number(dirname) #:nodoc:
-          next_migration_number = current_migration_number(dirname) + 1
-          if ::ActiveRecord::Base.timestamped_migrations
-            [Time.now.utc.strftime("%Y%m%d%H%M%S"), "%.14d" % next_migration_number].max
-          else
-            "%.3d" % next_migration_number
-          end
-        end
-
-        def copy_migration
-          @database = options["database"] || detect_database
-          unless @database.in?([nil, "postgresql", "postgresql-jsonb", "mysql", "sqlite"])
-            raise Thor::Error, "Unknown database option"
-          end
-          migration_template "active_record_events_migration.rb", "db/migrate/create_ahoy_events.rb", migration_version: migration_version
-        end
-
-        def generate_model
-          template "active_record_event_model.rb", "app/models/ahoy/event.rb"
-        end
-
-        def create_initializer
-          template "active_record_initializer.rb", "config/initializers/ahoy.rb"
-        end
-
-        def detect_database
-          postgresql_version = ActiveRecord::Base.connection.send(:postgresql_version) rescue 0
-          if postgresql_version >= 90400
-            "postgresql-jsonb"
-          elsif postgresql_version >= 90200
-            "postgresql"
-          end
-        end
-
-        def migration_version
-          if ActiveRecord::VERSION::MAJOR >= 5
-            "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
-          end
-        end
-      end
-    end
-  end
-end

data/lib/generators/ahoy/stores/active_record_generator.rb

@@ -1,16 +0,0 @@
-require "rails/generators"
-
-module Ahoy
-  module Stores
-    module Generators
-      class ActiveRecordGenerator < Rails::Generators::Base
-        class_option :database, type: :string, aliases: "-d"
-
-        def boom
-          invoke "ahoy:stores:active_record_visits", nil, options
-          invoke "ahoy:stores:active_record_events", nil, options
-        end
-      end
-    end
-  end
-end

data/lib/generators/ahoy/stores/active_record_visits_generator.rb

@@ -1,49 +0,0 @@
-# taken from https://github.com/collectiveidea/audited/blob/master/lib/generators/audited/install_generator.rb
-require "rails/generators"
-require "rails/generators/migration"
-require "active_record"
-require "rails/generators/active_record"
-
-module Ahoy
-  module Stores
-    module Generators
-      class ActiveRecordVisitsGenerator < Rails::Generators::Base
-        include Rails::Generators::Migration
-        source_root File.expand_path("../templates", __FILE__)
-
-        class_option :database, type: :string, aliases: "-d"
-
-        # Implement the required interface for Rails::Generators::Migration.
-        def self.next_migration_number(dirname) #:nodoc:
-          next_migration_number = current_migration_number(dirname) + 1
-          if ::ActiveRecord::Base.timestamped_migrations
-            [Time.now.utc.strftime("%Y%m%d%H%M%S"), "%.14d" % next_migration_number].max
-          else
-            "%.3d" % next_migration_number
-          end
-        end
-
-        def copy_migration
-          unless options["database"].in?([nil, "postgresql", "postgresql-jsonb"])
-            raise Thor::Error, "Unknown database option"
-          end
-          migration_template "active_record_visits_migration.rb", "db/migrate/create_visits.rb", migration_version: migration_version
-        end
-
-        def generate_model
-          template "active_record_visit_model.rb", "app/models/visit.rb"
-        end
-
-        def create_initializer
-          template "active_record_initializer.rb", "config/initializers/ahoy.rb"
-        end
-
-        def migration_version
-          if ActiveRecord::VERSION::MAJOR >= 5
-            "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
-          end
-        end
-      end
-    end
-  end
-end