rivendell-import 0.0.1

data/lib/rivendell/import/notifier/mail-body.erb

@@ -0,0 +1,8 @@
+<% tasks.each do |task| -%>
+<%= task.file %>
+<% if task.status.completed? -%>
+  imported in <%= task.destination %>
+<% else -%>
+  failed
+<% end -%>
+<% end %>

data/lib/rivendell/import/notifier/mail-subject.erb

@@ -0,0 +1,11 @@
+[Rivendell Import] 
+<% if many? %>
+  <%= tasks.count %> task(s) completed
+<% else %>
+  <% task = tasks.first %>
+  <% if task.status.completed? %>
+    <%= task.destination %> imported
+  <% else %>
+    <%= task.file.name %> failed
+  <% end %>
+<% end %>

data/lib/rivendell/import/notifier/mail.rb

@@ -0,0 +1,104 @@
+require 'mail'
+require 'erb'
+
+module Rivendell::Import::Notifier
+  class Mail < Rivendell::Import::Notifier::Base
+
+    @@from = nil
+    cattr_accessor :from
+
+    attr_accessor :from, :to, :subject, :body
+
+    def subject
+      @subject ||= File.expand_path("../mail-subject.erb", __FILE__)
+    end
+
+    def body
+      @body ||= File.expand_path("../mail-body.erb", __FILE__)
+    end
+
+    def from
+      @from ||= self.class.from
+    end
+
+    def notify!(tasks)
+      create_message(tasks).deliver!
+    end
+
+    def create_message(tasks)
+      Message.new(tasks).tap do |mail|
+        mail.from = from
+        mail.to = to
+        mail.subject = template(subject)
+        mail.body = template(body)
+      end
+    end
+
+    class Message
+
+      attr_reader :tasks
+      attr_accessor :from, :to, :subject, :body
+      
+      def initialize(tasks)
+        @tasks = tasks
+      end
+
+      def completed
+        tasks.select { |task| task.status.completed? }
+      end
+
+      def failed
+        tasks.select { |task| task.status.failed? }
+      end
+
+      def failed?
+        failed.present?
+      end
+
+      delegate :many?, :to => :tasks
+
+      def mail
+        ::Mail.new.tap do |mail|
+          mail.from = from
+          mail.to = to
+          mail.subject = strip(render(subject))
+          mail.body = render(body)
+        end
+      end
+
+      def deliver!
+        mail.deliver!
+      end
+
+      def render(template)
+        ERB.new(template, nil, '%<>-').result(binding)
+      end
+
+      def strip(text)
+        text.strip.gsub(/\n[ ]*/m,"")
+      end
+
+    end
+
+    # def valid?
+    #   [from, to, subject, body].all?(&:present?)
+    # end
+
+    def parameters
+      %w{from to subject body}.inject({}) do |map, attribute|
+        value = send attribute
+        map[attribute] = value if value
+        map
+      end
+    end
+
+    def template(definition)
+      if File.exists?(definition)
+        File.read definition
+      else
+        definition
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/notifiers.rb

@@ -0,0 +1,24 @@
+module Rivendell::Import
+  class Notifiers
+
+    attr_reader :import, :notifiers
+
+    def initialize
+      @import = import
+    end
+
+    def push(notifier)
+      notifiers << notifier
+    end
+
+    def notify!
+      notified_tasks = waiting_tasks
+      @waiting_tasks = []
+
+      notifiers.each do |notifier|
+        notifier.notify notified_tasks
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/task.rb

@@ -0,0 +1,80 @@
+require 'rivendell/import/tasking/file'
+require 'rivendell/import/tasking/tags'
+require 'rivendell/import/tasking/cart'
+require 'rivendell/import/tasking/status'
+require 'rivendell/import/tasking/destination'
+
+module Rivendell::Import
+  class Task < ActiveRecord::Base
+    include Rivendell::Import::Tasking::File
+    include Rivendell::Import::Tasking::Tags
+    include Rivendell::Import::Tasking::Cart
+    include Rivendell::Import::Tasking::Status
+    include Rivendell::Import::Tasking::Destination
+    
+    def self.pending
+      where :status => "pending"
+    end
+
+    def self.ran
+      where :status => %w{completed failed}
+    end
+
+    def ran?
+      status.completed? or status.failed?
+    end
+
+    def xport
+      @xport ||= Rivendell::API::Xport.new
+    end
+
+    def prepare(&block)
+      begin
+        Context.new(self).run(&block)
+      rescue => e
+        logger.error "Task preparation failed : #{e}"
+      end
+      self
+    end
+
+    def logger
+      Rivendell::Import.logger
+    end
+
+    def to_s
+      "Import '#{file}' in #{destination}"
+    end
+
+    has_many :notifications
+    has_many :notifiers, :through => :notifications
+
+    def notify!
+      notifiers.each do |notifier|
+        notifier.notify
+      end
+    end
+
+    def run
+      logger.debug "Run #{self.inspect}"
+      change_status! :running
+
+      reset_destination!
+
+      cart.create
+      cart.import file
+      cart.update
+
+      save!
+      change_status! :completed
+
+      logger.info "Imported Cart #{cart.number}"
+    rescue Exception => e
+      logger.error "Task failed : #{e}"
+    ensure
+      unless status.completed?
+        change_status! :failed 
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/tasking/cart.rb

@@ -0,0 +1,25 @@
+module Rivendell::Import::Tasking
+  module Cart
+
+    def raw_cart
+      read_attribute :cart
+    end
+
+    def cart
+      @cast ||= Rivendell::Import::Cart.new(self).tap do |cart|
+        cart.from_json raw_cart if raw_cart
+      end
+    end
+
+    def write_cart
+      write_attribute :cart, (cart.present? ? cart.to_json : nil)
+    end
+
+    def self.included(base)
+      base.class_eval do
+        before_save :write_cart
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/tasking/destination.rb

@@ -0,0 +1,28 @@
+module Rivendell::Import::Tasking
+  module Destination
+
+    def destination
+      read_attribute(:destination) or
+        write_attribute(:destination, calculate_destination)
+    end
+
+    def reset_destination!
+      write_attribute :destination, nil
+    end
+
+    def self.included(base)
+      base.class_eval do
+        before_save :destination, :on => :create
+      end
+    end
+
+    def calculate_destination
+      if cart.number
+        "Cart #{cart.number}"
+      elsif cart.group
+        "Cart in group #{cart.group}" 
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/tasking/file.rb

@@ -0,0 +1,20 @@
+module Rivendell::Import::Tasking
+  module File
+    def file=(file)
+      if file
+        @file = file
+
+        self.file_name = file.name
+        self.file_path = file.path
+      else
+        @file = nil
+
+        self.file_name = self.file_path = nil
+      end
+    end
+
+    def file
+      @file ||= Rivendell::Import::File.new(file_name, :path => file_path)
+    end
+  end
+end

data/lib/rivendell/import/tasking/status.rb

@@ -0,0 +1,29 @@
+module Rivendell::Import::Tasking
+  module Status
+
+    def define_default_status
+      self.status ||= "pending"
+    end
+
+    def self.included(base)
+      base.class_eval do
+        after_initialize :define_default_status
+      end
+    end
+
+    def raw_status
+      read_attribute(:status)
+    end
+
+    def status
+      ActiveSupport::StringInquirer.new(raw_status) if raw_status
+    end
+
+    def change_status!(status)
+      logger.debug "Change status to #{status}"
+      update_attribute :status, status.to_s
+      notify! if ran?
+    end
+
+  end
+end

data/lib/rivendell/import/tasking/tags.rb

@@ -0,0 +1,27 @@
+module Rivendell::Import::Tasking
+  module Tags
+
+    def raw_tags
+      read_attribute :tags
+    end
+
+    def tags
+      @tags ||= (raw_tags ? raw_tags.split(",") : [])
+    end
+
+    def tag(tag)
+      self.tags << tag
+    end
+
+    def write_tags
+      write_attribute :tags, tags.join(',') if @tags
+    end
+
+    def self.included(base)
+      base.class_eval do
+        before_save :write_tags        
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/tasks.rb

@@ -0,0 +1,23 @@
+module Rivendell::Import
+  class Tasks
+
+    def pending?
+      not Task.pending.empty?
+    end
+
+    def pop
+      Task.pending.first
+    end
+
+    def run
+      Task.pending.find_each(&:run)
+    end
+
+    def create(file, &block)
+      Rivendell::Import::Task.create({:file => file}, {}, &block).tap do |task|
+        Rivendell::Import.logger.debug "Created task #{task.inspect}"
+      end
+    end
+
+  end
+end

data/lib/rivendell/import/version.rb

@@ -0,0 +1,5 @@
+module Rivendell
+  module Import
+    VERSION = "0.0.1"
+  end
+end

data/lib/rivendell/import/worker.rb

@@ -0,0 +1,34 @@
+module Rivendell::Import
+  class Worker
+
+    attr_reader :import
+
+    def initialize(import)
+      @import = import
+    end
+
+    def start
+      Thread.new do 
+        Rivendell::Import.logger.debug "Start Worker"
+        run 
+      end
+
+      self
+    end
+
+    def run
+      loop do
+        task = import.tasks.pop
+        if task
+          task.run
+        else
+          # Rivendell::Import.logger.debug "No pending task, sleep 10s"
+          sleep 10
+        end
+      end
+    rescue => e
+      Rivendell::Import.logger.error "Worker failed : #{e}"
+    end
+
+  end
+end

data/rivendell-import.gemspec

@@ -0,0 +1,37 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/rivendell/import/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Alban Peignier"]
+  gem.email         = ["alban@tryphon.eu"]
+  gem.description   = %q{Import sound in our Rivendell system}
+  gem.summary       = %q{Import engine for Rivendell}
+  gem.homepage      = "http://wiki.tryphon.eu/rivendell-import/"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.name          = "rivendell-import"
+  gem.require_paths = ["lib"]
+  gem.version       = Rivendell::Import::VERSION
+
+  gem.add_runtime_dependency 'listen'
+  gem.add_runtime_dependency 'httmultiparty'
+  gem.add_runtime_dependency 'rivendell-api', '~> 0.0.2'
+  gem.add_runtime_dependency 'trollop'
+  gem.add_runtime_dependency 'activerecord', '~> 3.2.8'
+  gem.add_runtime_dependency 'activesupport', '~> 3.2.8'
+  gem.add_runtime_dependency 'mail'
+  gem.add_runtime_dependency 'sqlite3'
+
+  gem.add_development_dependency "simplecov"
+  gem.add_development_dependency "rspec"
+  gem.add_development_dependency "guard"
+  gem.add_development_dependency "guard-rspec"
+  gem.add_development_dependency "guard-cucumber"
+  gem.add_development_dependency "rake"
+  gem.add_development_dependency "cucumber"
+  gem.add_development_dependency "database_cleaner"
+  # gem.add_development_dependency "remarkable_activerecord"
+
+end