darjeelink 0.11.5

data/app/controllers/darjeelink/application_controller.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ApplicationController < ActionController::Base
+    protect_from_forgery with: :exception
+
+    before_action :check_ip_whitelist
+    before_action :authenticate
+
+    private
+
+    # Check user IP address against whitelist from ENV
+    def check_ip_whitelist
+      return unless Rails.env.production?
+      return unless Rails.application.config.permitted_ips
+      return if Rails.application.config.permitted_ips.split(',').include? request.ip
+
+      render plain: 'Access Denied', status: :forbidden
+    end
+
+    # Authenticate against Google OAuth
+    def authenticate
+      return if session[:email] || request.path =~ /google_oauth2/ || Rails.env.test?
+      # Seperate line to make it easier to disable the bypass if doing some auth
+      # related locally
+      return if Rails.env.development?
+
+      redirect_post '/auth/google_oauth2'
+    end
+  end
+end

data/app/controllers/darjeelink/sessions_controller.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class SessionsController < Darjeelink::ApplicationController
+    def create
+      if auth_domain.present? && auth_domain == Darjeelink.auth_domain
+        update_session
+        redirect_to '/'
+      else
+        render plain: 'Access Denied', status: :forbidden
+      end
+    end
+
+    protected
+
+    def auth_hash
+      request.env['omniauth.auth']
+    end
+
+    def auth_domain
+      auth_hash['info']['email'].split('@')[1] if auth_hash
+    end
+
+    private
+
+    def update_session
+      session[:email] = auth_hash['info']['email']
+      session[:name]  = auth_hash['info']['name']
+    end
+  end
+end

data/app/controllers/darjeelink/short_links_controller.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ShortLinksController < Darjeelink::ApplicationController
+    skip_before_action :check_ip_whitelist, only: :show
+    skip_before_action :authenticate, only: :show
+
+    before_action :check_url_present, only: :create
+    before_action :check_url_valid, only: :create
+
+    class ShortLinkNotFoundError < StandardError
+    end
+
+    def index
+      @short_links = Darjeelink::ShortLink.search(params[:query])
+                                          .order(id: 'desc')
+                                          .paginate(page: params[:page], per_page: 20)
+    end
+
+    def new
+    end
+
+    def create
+      begin
+        Darjeelink::ShortLink.create!(url: params[:url], shortened_path: params[:shortened_path])
+      rescue ActiveRecord::RecordNotUnique
+        flash[:error] = "#{params[:shortened_path]} already used! Choose a different custom path"
+        return redirect_to(darjeelink.new_short_link_path)
+      end
+
+      redirect_to(darjeelink.short_links_path)
+    end
+
+    def edit
+      @short_link = Darjeelink::ShortLink.find(params[:id])
+    end
+
+    def update
+      @short_link = Darjeelink::ShortLink.find(params[:id])
+      @short_link.update!(short_link_params)
+
+      redirect_to(darjeelink.short_links_path)
+    end
+
+    def show
+      shortened_path = CGI.escape(params[:id])
+      short_link = Darjeelink::ShortLink.ifind(shortened_path)
+
+      if short_link.nil?
+        log_missing_shortlink(shortened_path)
+        return redirect_to(Darjeelink.fallback_url)
+      end
+
+      short_link.update!(visits: short_link.visits + 1)
+      redirect_to(build_url(short_link.url))
+    end
+
+    private
+
+    def log_missing_shortlink(shortened_path)
+      Rails.logger.warn("ShortLink not found: #{shortened_path}")
+    end
+
+    def build_url(url)
+      uri = URI(url)
+
+      original_params = Rack::Utils.parse_query(uri.query)
+      # Strong params - throws an error if we just use except.  So we have to
+      # permit everything first
+      new_params = params.permit!.except(:controller, :action, :id)
+
+      short_link_params = merge_params(original_params, new_params)
+
+      # Prevents adding a ? to the end of the url if there are no params
+      uri.query = short_link_params.to_query if short_link_params.present?
+
+      uri.to_s
+    end
+
+    def merge_params(original_params, new_params)
+      # if there are duplicates, new params value will take priority
+      original_params.merge(new_params) do |_key, original_value, new_value|
+        if original_value.present? && new_value.present?
+          new_value
+        elsif original_value.present?
+          original_value
+        else
+          new_value
+        end
+      end
+    end
+
+    def short_link_params
+      params.require(:short_link).permit(:url, :shortened_path)
+    end
+
+    def check_url_present
+      return if params[:url].present?
+
+      flash[:error] = 'URL cannot be blank'
+      redirect_to(darjeelink.new_short_link_path)
+    end
+
+    def check_url_valid
+      return if params[:url] =~ URI::DEFAULT_PARSER.make_regexp
+
+      flash[:error] = 'URL is not valid.  Does it have https:// and are there any spaces?'
+      redirect_to(darjeelink.new_short_link_path)
+    end
+  end
+end

data/app/controllers/darjeelink/tracking_links_controller.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class TrackingLinksController < Darjeelink::ApplicationController
+    def new
+    end
+  end
+end

data/app/helpers/darjeelink/application_helper.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  module ApplicationHelper
+  end
+end

data/app/importers/darjeelink/rebrandly_importer.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class RebrandlyImporter
+    def initialize
+      @duplicates = []
+      @api = Rebrandly::Api.new
+    end
+
+    def import
+      import_rebrandly_links
+
+      Rails.logger.info("Imported #{api.link_count - duplicates.count} links from Rebrandly")
+      duplicate_string = duplicates.join("\n")
+      Rails.logger.warn("Duplicates:\n#{duplicate_string}") unless duplicates.empty?
+    end
+
+    private
+
+    attr_reader :api
+    attr_accessor :duplicates
+
+    def import_rebrandly_links
+      last_id = nil
+      imported_count = 0
+
+      while imported_count < api.link_count
+        api.links(last: last_id).each do |link|
+          begin
+            create_short_link(link)
+            last_id = link.id
+          rescue ActiveRecord::RecordNotUnique
+            duplicates << "#{link.destination}/#{link.slashtag}"
+          end
+          imported_count += 1
+        end
+      end
+    end
+
+    def create_short_link(link)
+      Darjeelink::ShortLink.create!(
+        url: link.destination,
+        shortened_path: link.slashtag,
+        visits: link.clicks
+      )
+    end
+  end
+end

data/app/importers/darjeelink/short_link_importer.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ShortLinkImporter
+    def initialize(path_to_csv)
+      @path_to_csv = path_to_csv
+      @duplicates = []
+    end
+
+    def import
+      CSV.foreach(path_to_csv, headers: true).each do |row|
+        url, auto_generated_key, custom_key = row.fields
+        create_short_link(url, auto_generated_key, custom_key)
+      end
+
+      duplicate_string = duplicates.join("\n")
+      Rails.logger.warn("Duplicates:\n#{duplicate_string}") unless duplicates.empty?
+    end
+
+    private
+
+    attr_reader :path_to_csv
+    attr_accessor :duplicates
+
+    def create_short_link(url, auto_generated_key, custom_key)
+      shortened_path = custom_key.present? ? custom_key : auto_generated_key
+      ::Darjeelink::ShortLink.create!(url: url, shortened_path: shortened_path)
+    rescue ActiveRecord::RecordNotUnique
+      duplicates << "#{url}/#{custom_key || auto_generated_key}"
+    end
+  end
+end

data/app/jobs/darjeelink/application_job.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ApplicationJob < ActiveJob::Base
+  end
+end

data/app/mailers/darjeelink/application_mailer.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ApplicationMailer < ActionMailer::Base
+    default from: 'from@example.com'
+    layout 'mailer'
+  end
+end

data/app/models/darjeelink/application_record.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ApplicationRecord < ActiveRecord::Base
+    self.abstract_class = true
+  end
+end

data/app/models/darjeelink/short_link.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Darjeelink
+  class ShortLink < ApplicationRecord
+    after_save :generate_short_link
+    before_validation :strip_white_space
+
+    validates_presence_of :url
+
+    class << self
+      def search(query)
+        where(
+          'shortened_path ILIKE :query', query: "%#{query}%"
+        ).or(
+          where(
+            'url ILIKE :query', query: "%#{query}%"
+          )
+        )
+      end
+
+      def ifind(shortened_path)
+        where('lower(shortened_path) = ?', shortened_path.downcase).first
+      end
+    end
+
+    private
+
+    def generate_short_link
+      return if shortened_path.present?
+
+      update!(shortened_path: auto_generate_shortened_path)
+    end
+
+    ALPHABET = (('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a + ['-', '_']).freeze
+
+    def auto_generate_shortened_path
+      # from http://refactormycode.com/codes/125-base-62-encoding
+      # with only minor modification
+      i = id
+      return ALPHABET[0] if i.zero?
+
+      generated_path = []
+      base = ALPHABET.length
+
+      while i.positive?
+        generated_path << ALPHABET[i.modulo(base)]
+        i /= base
+      end
+
+      generated_path.join('').reverse
+    end
+
+    def strip_white_space
+      url&.strip!
+      shortened_path&.strip!
+    end
+  end
+end

data/app/views/darjeelink/short_links/_short_link_form.erb

@@ -0,0 +1,24 @@
+<div class="row">
+  <div class="col-md-12">
+    <% if flash[:error] %>
+      <div class="fade in alert alert-danger alert-dismissable">
+        <button type="button" class="close" data-dismiss="alert" aria-hidden="true">&times;</button>
+        <%= flash[:error] %>
+      </div>
+    <% end %>
+
+    <%= form_with url: darjeelink.short_links_path, local: true do |f| %>
+      <div class="form-group">
+        <label for="url">Url</label>
+        <input type="url" class="form-control" id="long_url" name="url">
+      </div>
+      <div class="form-group">
+        <label for="shortened_path">Custom Path</label>
+        <input type="text" class="form-control" name="shortened_path">
+      </div>
+      <div>
+        <button type="submit">Shorten</button>
+      </div>
+    <% end %>
+  </div>
+</div>

data/app/views/darjeelink/short_links/edit.html.erb

@@ -0,0 +1,29 @@
+<div class="row">
+  <div class="col-md-12">
+    <%= link_to "Back", darjeelink.short_links_path, class: "btn btn-default" %>
+  </div>
+</div>
+
+<div class="row">
+  <div class="col-md-12">
+    <h1>Edit Short Link</h1>
+  </div>
+</div>
+
+<div class="row">
+  <div class="col-md-12">
+    <%= form_for @short_link, url: darjeelink.short_link_path do |f| %>
+      <div class="form-group">
+        <%= f.label :url %>
+        <%= f.url_field :url, class: "form-control", value: @short_link.url %>
+      </div>
+      <div class="form-group">
+        <%= f.label :custom_path %>
+        <%= f.text_field :shortened_path, class: "form-control", value: @short_link.shortened_path %>
+      </div>
+      <div>
+        <button type="submit">Update</button>
+      </div>
+    <% end %>
+  </div>
+</div>

data/app/views/darjeelink/short_links/index.html.erb

@@ -0,0 +1,75 @@
+<div class="row">
+  <div class="col-md-12">
+    <div class="page-header">
+      <h2 class="text-center header-text">
+        Short Links
+      </h2>
+    </div>
+  </div>
+</div>
+
+<div class="row">
+  <div class="col-md-12 text-center">
+    <div class="btn-group">
+      <%= link_to "Tracking link generator", darjeelink.new_tracking_link_path, class: "btn btn-success" %>
+      <%= link_to "Create your shortened link", darjeelink.new_short_link_path, class: "btn btn-default" %>
+    </div>
+  </div>
+</div>
+
+<br/>
+<div class="row text-center">
+  <div class="col-sm-4 col-sm-offset-4">
+    <%= form_tag("/short_links", method: "get") do %>
+      <div class="form-group">
+        <%= text_field_tag(:query, nil, class: 'form-control', placeholder: 'Search term') %>
+      </div>
+      <%= submit_tag("Search") %>
+    <% end %>
+  </div>
+</div>
+
+<div class="row">
+  <div class="col-md-12">
+    <table class="table table-striped">
+      <thead>
+        <tr>
+          <th>URL</th>
+          <th>Visits</th>
+          <th>Edit</th>
+        </tr>
+      </thead>
+      <tbody>
+        <% @short_links.each do |short_link| %>
+        <tr>
+          <td>
+            <a href="<%= short_link.url %>" target="_blank" id="copy-text">
+              <%= Darjeelink.domain + '/' + short_link.shortened_path %>
+            </a>
+            <button id="copy-btn" class="btn btn-info btn-sm" data-clipboard-text=<%= Darjeelink.domain + '/' + short_link.shortened_path %>>
+              <span class="glyphicon glyphicon-copy"></span>
+            </button>
+          </td>
+          <td>
+            <p>
+              <%= short_link.visits %>
+            </p>
+          </td>
+          <td>
+            <%= link_to "Edit", darjeelink.edit_short_link_path(short_link) %>
+          </td>
+        </tr>
+        <% end %>
+      </tbody>
+    </table>
+    <div class="row">
+      <div class="col-md-12">
+        <%= will_paginate @short_links %>
+      </div>
+    </div>
+  </div>
+</div>
+
+<script>
+  new ClipboardJS('#copy-btn');
+</script>