Contents

module Twuckoo
  class OneLineFromFile
    # read all lines from the file that contains the strategies
    # select the strategies that are not used yet
    # pick a random strategy

    # store used strategies in a file with (store the md5 sum of the strategies' text)
    attr_reader :lines

    LINES_FILE = 'lines.txt'
    USED_LINES_FILE = 'used_lines.txt'

    def setup
      @lines = []
    end

    #TODO: This could be merged into setup
    def load_tweets
      load_lines
    end

    def get_lines_from_file
      IO::readlines(LINES_FILE).map { |line| line.chomp }
    end

    def get_all_lines
      @fresh_lines ||= get_lines_from_file
    end

    def get_used_lines_from_file
      begin
        IO::readlines(USED_LINES_FILE).map { |line| line.chomp }
      rescue Errno::ENOENT
        []
      end
    end

    def get_used_lines
      @used_lines ||=  get_used_lines_from_file
    end

    def get_unused_lines
      get_all_lines.reject { |line| get_used_lines.include?(line) }
    end

    def load_lines
      add_lines(*get_unused_lines)
    end

    def add_lines(*new_lines)
      @lines.concat(new_lines)
    end

    def pick
      rand(lines.length)
    end

    def store(line)
      open(USED_LINES_FILE, "a") do |file|
        file.write(line + "\n")
      end
    end

    def next
      @lines.delete_at(pick)
    end

    def reset
      File.delete(USED_LINES_FILE) rescue nil
      @used_lines = nil
      load_lines
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
twuckoo-0.5.3	lib/twuckoo/feeders/one_line_from_file.rb