Class: Teapot::Generator

Inherits:
Definition show all
Defined in:
lib/teapot/generator.rb

Instance Attribute Summary

Attributes inherited from Definition

#context, #description, #name, #package

Instance Method Summary collapse

Methods inherited from Definition

#freeze, #path, #pretty_print, #to_s

Constructor Details

#initialize(context, package, name) ⇒ Generator

Returns a new instance of Generator.



39
40
41
42
43
 
# File 'lib/teapot/generator.rb', line 39

def initialize(context, package, name)
  super context, package, name

  @generate = nil
end

Instance Method Details

#append(source, destination, substitutions = nil) ⇒ Object 



78
79
80
 
# File 'lib/teapot/generator.rb', line 78

def append(source, destination, substitutions = nil)
  write(source, destination, substitutions, "a")
end

#copy(source, destination, substitutions = nil) ⇒ Object 



120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
 
# File 'lib/teapot/generator.rb', line 120

def copy(source, destination, substitutions = nil)
  source_path = Pathname(path) + source
  destination_path = Pathname(context.root) + destination

  if source_path.directory?
    source_path.children(false).each do |child_path|
      copy(source_path + child_path, destination_path + substitute(child_path.to_s, substitutions), substitutions)
    end
  else
    if is_binary(source_path) or is_binary(destination)
      destination_path = Pathname(context.root) + destination
      copy_binary(source_path, destination_path)
    else
      merge(source_path, destination, substitutions)
    end
  end
end

#copy_binary(source_path, destination_path) ⇒ Object 



115
116
117
118
 
# File 'lib/teapot/generator.rb', line 115

def copy_binary(source_path, destination_path)
  destination_path.dirname.create
  FileUtils.cp source_path, destination_path
end

#generate(&block) ⇒ Object 



45
46
47
 
# File 'lib/teapot/generator.rb', line 45

def generate(&block)
  @generate = Proc.new(&block)
end

#generate!(*args) ⇒ Object 



49
50
51
 
# File 'lib/teapot/generator.rb', line 49

def generate!(*args)
  @generate[*args]
end

#is_binary(path) ⇒ Object 



107
108
109
110
111
112
113
 
# File 'lib/teapot/generator.rb', line 107

def is_binary(path)
  if path.exist?
    return path.read(1024).bytes.find{|byte| byte >= 0 and byte <= 6}
  else
    return false
  end
end

#merge(source, destination, substitutions = nil) ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
 
# File 'lib/teapot/generator.rb', line 82

def merge(source, destination, substitutions = nil)
  source_path = Pathname(path) + source
  destination_path = Pathname(context.root) + destination

  if destination_path.exist?
    temporary_file = Tempfile.new(destination_path.basename.to_s)

    # This functionality works, but needs improvements.
    begin
      # Need to ask user what to do?
      write(source_path, temporary_file.path, substitutions, "w")

      result = Merge::combine(destination.readlines, temporary_file.readlines)

      destination.open("w") do |file|
        file.write result.join
      end
    ensure
      temporary_file.unlink
    end
  else
    write(source_path, destination_path, substitutions, "w")
  end
end

#substitute(text, substitutions) ⇒ Object 



53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/teapot/generator.rb', line 53

def substitute(text, substitutions)
  return text unless substitutions

  if Hash === substitutions
    pattern = Regexp.new(substitutions.keys.map{|x| Regexp.escape(x)}.join('|'))

    text.gsub(pattern) {|key| substitutions[key]}
  else
    substitutions.call(text)
  end
end

#write(source, destination, substitutions = nil, mode = "a") ⇒ Object 



65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/teapot/generator.rb', line 65

def write(source, destination, substitutions = nil, mode = "a")
  source_path = Pathname(path) + source
  destination_path = Pathname(context.root) + destination

  destination_path.dirname.mkpath

  File.open(destination_path, mode) do |file|
    text = File.read(source_path)

    file.write substitute(text, substitutions)
  end
end