Module: Cheat

Extended by:
Cheat
Includes:
Pager
Included in:
Cheat
Defined in:
lib/cheat.rb,
lib/cheat/diffr.rb,
lib/cheat/version.rb

Defined Under Namespace

Modules: Controllers, Helpers, Models, Views Classes: Diffr

Constant Summary collapse

HOST = 
ARGV.include?('debug') ? 'localhost' : 'cheat.errtheblog.com'
PORT = 
ARGV.include?('debug') ? 3001 : 80
SUFFIX = 
''
VERSION = 
"1.3.1"

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.create ⇒ Object 



592
593
594
 
# File 'lib/cheat/site.rb', line 592

def Cheat.create
  Cheat::Models.create_schema
end

Instance Method Details

#add(title) ⇒ Object 



149
150
151
152
153
 
# File 'lib/cheat.rb', line 149

def add(title)
  body = write_to_tempfile(title)
  res = post_sheet(title, body, true)
  check_errors(res, title, body)
end

#cache_dir ⇒ Object 



189
190
191
 
# File 'lib/cheat.rb', line 189

def cache_dir
  RUBY_PLATFORM =~ /win32/ ? win32_cache_dir : File.join(File.expand_path("~"), ".cheat")
end

#cache_file ⇒ Object 



91
92
93
 
# File 'lib/cheat.rb', line 91

def cache_file
  "#{cache_dir}/#{@sheet}.yml" if cache_dir
end

#cheat_uri ⇒ Object 



99
100
101
 
# File 'lib/cheat.rb', line 99

def cheat_uri
  "#{HOST}:#{PORT}#{SUFFIX}"
end

#check_errors(result, title, text) ⇒ Object 



174
175
176
177
178
179
180
181
182
183
 
# File 'lib/cheat.rb', line 174

def check_errors(result, title, text)
  if result.body =~ /<p class="error">(.+?)<\/p>/m
    puts $1.gsub(/\n/, '').gsub(/<.+?>/, '').squeeze(' ').wrap(80)
    puts
    puts "Here's what you wrote, so it isn't lost in the void:"
    puts text
  else
    puts "Success!  Try it!", "$ cheat #{title}"
  end
end

#clear_cache ⇒ Object 



203
204
205
 
# File 'lib/cheat.rb', line 203

def clear_cache
  FileUtils.rm_rf(cache_dir) if cache_dir
end

#clear_cache_file ⇒ Object 



207
208
209
 
# File 'lib/cheat.rb', line 207

def clear_cache_file
  FileUtils.rm(cache_file) if File.exists?(cache_file)
end

#diff_sheets(sheet, version) ⇒ Object

$ cheat greader –diff 1



74
75
76
77
78
79
80
81
82
 
# File 'lib/cheat.rb', line 74

def diff_sheets(sheet, version)
  return unless version =~ /^(\d+)(:(\d+))?$/
  old_version, new_version = $1, $3

  uri = "http://#{cheat_uri}/d/#{sheet}/#{old_version}"
  uri += "/#{new_version}" if new_version

  fetch_sheet(uri, false)
end

#edit(sheet_yaml) ⇒ Object 



140
141
142
143
144
145
146
147
 
# File 'lib/cheat.rb', line 140

def edit(sheet_yaml)
  sheet = YAML.load(sheet_yaml).to_a.first
  sheet[-1] = sheet.last.gsub("\r", '')
  body, title = write_to_tempfile(*sheet), sheet.first
  return if body.strip == sheet.last.strip
  res = post_sheet(title, body)
  check_errors(res, title, body)
end

#editor ⇒ Object 



185
186
187
 
# File 'lib/cheat.rb', line 185

def editor
  ENV['VISUAL'] || ENV['EDITOR'] || "vim"
end

#execute(sheet_yaml) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/cheat.rb', line 103

def execute(sheet_yaml)
  sheet_body = YAML.load(sheet_yaml).to_a.flatten.last
  puts "\n  " + sheet_body.gsub("\r",'').gsub("\n", "\n  ").wrap
  puts "\nWould you like to execute the above sheet? (Y/N)"
  answer = STDIN.gets
  case answer.chomp
  when "Y" then system YAML.load(sheet_yaml).to_a.flatten.last
  when "N" then puts "Not executing sheet."
  else
    puts "Must be Y or N!"
  end
rescue Errno::EPIPE
  # do nothing
rescue
  puts "That didn't work.  Maybe try `$ cheat cheat' for help?" # Fix Emacs ruby-mode highlighting bug: `"
end

#fetch_sheet(uri, try_to_cache = true) ⇒ Object 



32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/cheat.rb', line 32

def fetch_sheet(uri, try_to_cache = true)
  open(uri, headers) do |body|
    sheet = body.read
    FileUtils.mkdir_p(cache_dir) unless File.exists?(cache_dir)
    File.open(cache_file, 'w') { |f| f.write(sheet) } if try_to_cache && has_content(sheet) && cache_file && !@edit
    @edit ? edit(sheet) : show(sheet)
  end
  exit
rescue OpenURI::HTTPError => e
  puts "Whoa, some kind of Internets error!", "=> #{e} from #{uri}"
end

#has_content(sheet) ⇒ Object 



84
85
86
87
88
89
 
# File 'lib/cheat.rb', line 84

def has_content(sheet)
  if sheet.is_a?(String)
    return (sheet.length > 15) && !sheet[0,14].include?("Error!")
  end
  return true
end

#headers ⇒ Object 



95
96
97
 
# File 'lib/cheat.rb', line 95

def headers
  { 'User-Agent' => 'cheat!', 'Accept' => 'text/yaml' }
end

#parse_args(args) ⇒ Object 



44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/cheat.rb', line 44

def parse_args(args)
  puts "Looking for help?  Try http://cheat.errtheblog.com or `$ cheat cheat'" and return if args.empty?

  if args.delete('--clear-cache') || args.delete('--new')
    clear_cache
    return if args.empty?
  end

  if i = args.index('--diff')
    diff_sheets(args.first, args[i+1])
  end

  show_versions(args.first) if args.delete('--versions')

  add(args.shift) and return if args.delete('--add')
  incoming_file = true if @edit = args.delete('--edit')

  @execute = true if args.delete("--execute") || args.delete("-x")
  @sheet = args.shift

  clear_cache_file if incoming_file
  true
end

#post_sheet(title, body, new = false) ⇒ Object 



155
156
157
158
159
 
# File 'lib/cheat.rb', line 155

def post_sheet(title, body, new = false)
  uri = "http://#{cheat_uri}/w/"
  uri += title unless new
  Net::HTTP.post_form(URI.parse(uri), "sheet_title" => title, "sheet_body" => body.strip, "from_gem" => true)
end

#process(sheet_yaml) ⇒ Object 



120
121
122
123
124
125
126
 
# File 'lib/cheat.rb', line 120

def process(sheet_yaml)
  if @execute
    execute(sheet_yaml)
  else
    show(sheet_yaml)
  end
end

#sheets(args) ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/cheat.rb', line 13

def sheets(args)
  args = args.dup

  return unless parse_args(args)

  FileUtils.mkdir(cache_dir) unless File.exists?(cache_dir) if cache_dir

  uri = "http://#{cheat_uri}/y/"

  if %w[sheets all recent].include? @sheet
    uri = uri.sub('/y/', @sheet == 'recent' ? '/yr/' : '/ya/')
    return open(uri, headers) { |body| process(body.read) }
  end

  return process(File.read(cache_file)) if File.exists?(cache_file) rescue clear_cache if cache_file

  fetch_sheet(uri + @sheet) if @sheet
end

#show(sheet_yaml) ⇒ Object 



128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/cheat.rb', line 128

def show(sheet_yaml)
  sheet = YAML.load(sheet_yaml).to_a.first
  sheet[-1] = sheet.last.join("\n") if sheet[-1].is_a?(Array)
  page
  puts sheet.first + ':'
  puts '  ' + sheet.last.gsub("\r",'').gsub("\n", "\n  ").wrap
rescue Errno::EPIPE
  # do nothing
rescue
  puts "That didn't work.  Maybe try `$ cheat cheat' for help?" # Fix Emacs ruby-mode highlighting bug: `"
end

#show_versions(sheet) ⇒ Object

$ cheat greader –versions



69
70
71
 
# File 'lib/cheat.rb', line 69

def show_versions(sheet)
  fetch_sheet("http://#{cheat_uri}/h/#{sheet}/", false)
end

#win32_cache_dir ⇒ Object 



193
194
195
196
197
198
199
200
201
 
# File 'lib/cheat.rb', line 193

def win32_cache_dir
  unless File.exists?(home = ENV['HOMEDRIVE'] + ENV['HOMEPATH'])
    puts "No HOMEDRIVE or HOMEPATH environment variable.  Set one to save a" +
         "local cache of cheat sheets."
    return false
  else
    return File.join(home, 'Cheat')
  end
end

#write_to_tempfile(title, body = nil) ⇒ Object 



161
162
163
164
165
166
167
168
169
170
171
172
 
# File 'lib/cheat.rb', line 161

def write_to_tempfile(title, body = nil)
  # god dammit i hate tempfile, this is so messy but i think it's
  # the only way.
  tempfile = Tempfile.new(title + '.cheat')
  tempfile.write(body) if body
  tempfile.close
  system "#{editor} #{tempfile.path}"
  tempfile.open
  body = tempfile.read
  tempfile.close
  body
end