Contents

#!/usr/bin/env ruby

require 'twitter_ebooks'

module Ebooks
  APP_PATH = Dir.pwd # XXX do some recursive thing instead

  def self.new(target)
    usage = "Usage: ebooks new <reponame>"

    if target.nil?
      log usage
      exit
    end

    target = "./#{reponame}"

    if File.exists?(target)
      log "#{target} already exists. Please remove if you want to recreate."
      exit
    end

    FileUtils.cp_r(SKELETON_PATH, target) 

    File.open(File.join(target, 'bots.rb'), 'w') do |f|
      template = File.read(File.join(SKELETON_PATH, 'bots.rb'))
      f.write(template.gsub("{{BOT_NAME}}", reponame))
    end

    log "New twitter_ebooks app created at #{target}"
  end

  def self.consume(pathes)
    pathes.each do |path|
      filename = File.basename(path)
      shortname = filename.split('.')[0..-2].join('.')
      hash = Digest::MD5.hexdigest(File.read(path))

      log "Consuming text corpus: #{filename}"
      outpath = File.join(APP_PATH, 'model', "#{shortname}.model")
      Model.consume(path).save(outpath)
      log "Corpus consumed"
    end
  end

  def self.gen(model_path, input)
    require 'benchmark'
    model = nil;
    puts Benchmark.measure {
      model = Model.load(model_path)
    }
    if input && !input.empty?
      puts "@cmd " + model.markov_response(input, 135)
    else
      puts model.markov_statement
    end
  end

  def self.archive(username, outpath)
    Archiver.new(username, outpath).fetch_tweets
  end

  def self.command(args)
    usage = """Usage: 
     ebooks new <reponame>
     ebooks consume <corpus_path> [...]
     ebooks gen <model> [input]
     ebooks archive <@user> <outpath>
"""

    if args.length == 0
      log usage
      exit
    end

    case args[0]
    when "new" then new(args[1])
    when "consume" then consume(args[1..-1])
    when "gen" then gen(args[1], args[2..-1].join(' '))
    when "archive" then archive(args[1], args[2])
    end
  end
end

Ebooks.command(ARGV)

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
twitter_ebooks-2.0.3	bin/ebooks