Jiri-Kremser/simple-websocket-vcr

require 'simple_websocket_vcr/errors'
require 'fileutils'

module WebSocketVCR
  include Errors

  class Cassette
    attr_reader :name, :options, :recording, :sessions

    alias_method :recording?, :recording

    def initialize(name, options)
      @name = name
      @options = options
      @using_json = WebSocketVCR.configuration.json_cassettes
      @name += @using_json ? '.json' : '.yml'

      if File.exist?(filename) && @options[:record] != :all
        @recording = false
        @sessions = initialize_sessions filename
      else
        fail "No cassette '#{name}' found and recording has been turned off" if @options[:record] == :none
        @recording = true
        @sessions = []
      end
    end

    def next_session
      if recording?
        erb_variables = @options[:reverse_substitution] ? @options[:erb] : nil
        session = if @using_json
                    RecordedJsonSession.new([], erb_variables)
                  else
                    RecordedYamlSession.new([], erb_variables)
                  end
        @sessions.push(session)
        @sessions.last
      else
        fail NoMoreSessionsError if @sessions.empty?
        @sessions.shift
      end
    end

    def save
      return unless recording?
      dirname = File.dirname(filename)
      # make sure the directory structure is there
      FileUtils.mkdir_p(dirname) unless File.directory?(dirname)
      if @using_json
        text = JSON.pretty_generate(@sessions.map(&:record_entries))
      else
        text = { 'websocket_interactions' => @sessions.map(&:record_entries) }.to_yaml(Indent: 8)
      end
      File.open(filename, 'w') { |f| f.write(text) }
    end

    protected

    def filename
      "#{WebSocketVCR.configuration.cassette_library_dir}/#{name}"
    end

    private

    def initialize_sessions(filename)
      file_content = File.open(filename, &:read)

      # parse JSON/YAML
      if @using_json
        parsed_content = JSON.parse(file_content)
        sessions = RecordedJsonSession.load(parsed_content, @options[:erb])
      else
        parsed_content = YAML.load(file_content)
        sessions = RecordedYamlSession.load(parsed_content, @options[:erb])
      end
      sessions
    end
  end

  class RecordedSession
    attr_reader :record_entries

    def initialize(entries, erb_variables = nil)
      @record_entries = entries
      @erb_variables = erb_variables
    end

    def store(entry)
      hash = entry.is_a?(RecordEntry) ? entry.attributes.map(&:to_s) : Hash[entry.map { |k, v| [k.to_s, v.to_s] }]
      if !hash['data'].nil? && !@erb_variables.nil? && hash['type'] != 'binary'
        @erb_variables.each do |k, v|
          hash['data'].gsub! v.to_s, "<%= #{k} %>"
        end
      end
      @record_entries << hash
    end

    def next
      RecordEntry.parse(@record_entries.shift, @erb_variables)
    end

    def head
      @record_entries.empty? ? nil : RecordEntry.parse(@record_entries.first, @erb_variables)
    end

    def method_missing(method_name, *args, &block)
      @record_entries.__send__(method_name, *args, &block)
    end
  end

  class RecordedJsonSession < RecordedSession
    def self.load(json, erb_variables = nil)
      json.map { |session| RecordedJsonSession.new(session, erb_variables) }
    end
  end

  class RecordedYamlSession < RecordedSession
    def self.load(yaml, erb_variables = nil)
      yaml['websocket_interactions'].map { |session| RecordedYamlSession.new(session, erb_variables) }
    end
  end

  class RecordEntry
    attr_accessor :operation, :event, :type, :data

    def self.parse(obj, erb_variables = nil)
      record_entry = RecordEntry.new
      record_entry.operation = obj['operation']
      record_entry.event = obj['event'] if obj['event']
      record_entry.type = obj['type'] if obj['type']
      record_entry.data = obj['data'] if obj['data']

      # do the ERB substitution
      if erb_variables && record_entry.type != 'binary'
        require 'ostruct'
        namespace = OpenStruct.new(erb_variables)
        record_entry.data = ERB.new(record_entry.data).result(namespace.instance_eval { binding })
      end

      record_entry
    end
  end
end