Module: Chit

Extended by:
Chit
Included in:
Chit
Defined in:
lib/chit.rb

Constant Summary collapse

VERSION = 
'0.0.4'
CHITRC = 
File.join("#{ENV['HOME']}",".chitrc")
CONFIG = 
defaults.merge(YAML.load_file(CHITRC))

Instance Method Summary collapse

Instance Method Details

#add(sheet_file) ⇒ Object 



155
156
157
158
159
160
161
162
163
164
165
 
# File 'lib/chit.rb', line 155

def add(sheet_file)
  unless File.exist?(sheet_file)
    breaker = sheet_file.rindex(File::Separator)+1
    path = sheet_file[0,breaker]
    title = @sheet.split(File::Separator).join('::')
    FileUtils.mkdir_p(path)
    yml = {"#{title}" => ''}.to_yaml
    open(sheet_file, 'w') {|f| f << yml}
  end
  edit(sheet_file)
end

#edit(sheet_file) ⇒ Object 



167
168
169
170
171
172
173
174
175
176
177
178
179
 
# File 'lib/chit.rb', line 167

def edit(sheet_file)
  sheet = YAML.load(IO.read(sheet_file)).to_a.first
  sheet[-1] = sheet.last.gsub("\r", '')
  body, title = write_to_tempfile(*sheet), sheet.first
  if body.strip.empty?
    rm(sheet_file)
  else
    open(sheet_file,'w') {|f| f << {title => body}.to_yaml}
    @git.add
    @git.commit_all("-")
  end
  true
end

#init_chit ⇒ Object 



91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/chit.rb', line 91

def init_chit
  FileUtils.mkdir_p(CONFIG['root'])
  if CONFIG['main']['clone-from']
    if File.exist?(main_path)
      puts "Main chit has already been initialized."
    else
      puts "Initialize main chit from #{CONFIG['main']['clone-from']} to #{CONFIG['root']}/main"
      Git.clone(CONFIG['main']['clone-from'], 'main', :path => CONFIG['root'])
      puts "Main chit initialized."        
    end
  else
    puts "ERROR: configuration for main chit repository is missing!"
    return
  end
  
  unless File.exist?(private_path)
    if CONFIG['private'] && CONFIG['private']['clone-from']
      puts "Initialize private chit from #{CONFIG['private']['clone-from']} to #{CONFIG['root']}/private"
      Git.clone(CONFIG['private']['clone-from'], 'private', :path => CONFIG['root'])
      puts "Private chit initialized."
    else
      puts "Initialize private chit from scratch to #{CONFIG['root']}/private"
      Git.init(private_path)
      puts "Private chit initialized."
    end
  else
    puts "Private chit has already been initialized."
  end
  puts "Chit init done."
  true
end

#list_all ⇒ Object 



67
68
69
 
# File 'lib/chit.rb', line 67

def list_all
  puts all_sheets.sort.join("\n")
end

#main_path ⇒ Object 



133
134
135
 
# File 'lib/chit.rb', line 133

def main_path
  File.join(CONFIG['root'], 'main')
end

#parse_args(args) ⇒ Object 



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

def parse_args(args)
  init_chit and return if args.delete('--init')
  update and return if args.delete('--update')
  
  @sheet = args.shift || 'chit'
  is_private = (@sheet =~ /^@(.*)/)
  @sheet = is_private ? $1 : @sheet

  working_dir = is_private ? private_path : main_path
  @git = Git.open(working_dir)

  @fullpath = File.join(working_dir, "#{@sheet}.yml")
  
  add(sheet_file) and return if (args.delete('--add')||args.delete('-a'))
  edit(sheet_file) and return if (args.delete('--edit')||args.delete('-e'))
  search_title and return if (args.delete('--find')||args.delete('-f'))
  search_content and return if (args.delete('--search')||args.delete('-s'))
  true
end

#private_path ⇒ Object 



137
138
139
 
# File 'lib/chit.rb', line 137

def private_path
  File.join(CONFIG['root'], 'private')
end

#rm(sheet_file) ⇒ Object 



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

def rm(sheet_file)
  @git.remove(sheet_file)
  @git.commit_all("-")
rescue Git::GitExecuteError
  FileUtils.rm_rf(sheet_file)
end

#run(args) ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/chit.rb', line 18

def run(args)
  unless File.exist?(main_path) && File.exist?(private_path)
    return unless init_chit
  end
  args = args.dup
  
  return unless parse_args(args)

  if %w[sheets all].include? @sheet
    return list_all()
  end
  
  unless File.exist?(sheet_file)
    update
  end
  
  unless File.exist?(sheet_file)
    if args.delete('--no-add').nil? && CONFIG['add_if_not_exist']
      add(sheet_file)
    else
      puts "Error!:\n  #{@sheet} not found"
      puts "Possible sheets:"
      search_title
    end
  else
    show(sheet_file)
  end
end

#search_content ⇒ Object 



71
72
73
74
75
76
77
78
 
# File 'lib/chit.rb', line 71

def search_content
  @git.grep(@sheet).each {|file, lines|
    title = title_of_file(file.split(':')[1])
    lines.each {|l|
      puts "#{title}:#{l[0]}:  #{l[1]}"
    }
  }
end

#search_title ⇒ Object 



80
81
82
83
84
85
 
# File 'lib/chit.rb', line 80

def search_title
  reg = Regexp.compile("^#{@sheet}")
  files = all_sheets.select {|sheet| sheet =~ reg }
  puts files.sort.join("\n")
  true
end

#sheet_file ⇒ Object 



87
88
89
 
# File 'lib/chit.rb', line 87

def sheet_file
  @fullpath
end

#show(sheet_file) ⇒ Object 



141
142
143
144
145
146
 
# File 'lib/chit.rb', line 141

def show(sheet_file)
  sheet = YAML.load(IO.read(sheet_file)).to_a.first
  sheet[-1] = sheet.last.join("\n") if sheet[-1].is_a?(Array)
  puts sheet.first + ':'
  puts '  ' + sheet.last.gsub("\r",'').gsub("\n", "\n  ").wrap
end

#update ⇒ Object 



123
124
125
126
127
128
129
130
131
 
# File 'lib/chit.rb', line 123

def update
  if CONFIG['main']['clone-from']
    g = Git.open(main_path)
    g.pull
  end
rescue
  puts "ERROR: can not update main chit."
  puts $!
end