1 | require 'pathname'
|
---|
2 | require 'forwardable'
|
---|
3 | require 'tsort'
|
---|
4 |
|
---|
5 | module MRuby
|
---|
6 | module Gem
|
---|
7 | class << self
|
---|
8 | attr_accessor :current
|
---|
9 | end
|
---|
10 | LinkerConfig = Struct.new(:libraries, :library_paths, :flags, :flags_before_libraries, :flags_after_libraries)
|
---|
11 |
|
---|
12 | class Specification
|
---|
13 | include Rake::DSL
|
---|
14 | extend Forwardable
|
---|
15 | def_delegators :@build, :filename, :objfile, :libfile, :exefile
|
---|
16 |
|
---|
17 | attr_accessor :name, :dir, :build
|
---|
18 | alias mruby build
|
---|
19 | attr_accessor :build_config_initializer
|
---|
20 |
|
---|
21 | attr_accessor :version
|
---|
22 | attr_accessor :description, :summary
|
---|
23 | attr_accessor :homepage
|
---|
24 | attr_accessor :licenses, :authors
|
---|
25 | alias :license= :licenses=
|
---|
26 | alias :author= :authors=
|
---|
27 |
|
---|
28 | attr_accessor :rbfiles, :objs
|
---|
29 | attr_accessor :test_objs, :test_rbfiles, :test_args
|
---|
30 | attr_accessor :test_preload
|
---|
31 |
|
---|
32 | attr_accessor :bins
|
---|
33 |
|
---|
34 | attr_accessor :requirements
|
---|
35 | attr_reader :dependencies, :conflicts
|
---|
36 |
|
---|
37 | attr_accessor :export_include_paths
|
---|
38 |
|
---|
39 | attr_reader :generate_functions
|
---|
40 |
|
---|
41 | attr_block MRuby::Build::COMMANDS
|
---|
42 |
|
---|
43 | def initialize(name, &block)
|
---|
44 | @name = name
|
---|
45 | @initializer = block
|
---|
46 | @version = "0.0.0"
|
---|
47 | MRuby::Gem.current = self
|
---|
48 | end
|
---|
49 |
|
---|
50 | def setup
|
---|
51 | MRuby::Gem.current = self
|
---|
52 | MRuby::Build::COMMANDS.each do |command|
|
---|
53 | instance_variable_set("@#{command}", @build.send(command).clone)
|
---|
54 | end
|
---|
55 | @linker = LinkerConfig.new([], [], [], [], [])
|
---|
56 |
|
---|
57 | @rbfiles = Dir.glob("#{dir}/mrblib/**/*.rb").sort
|
---|
58 | @objs = Dir.glob("#{dir}/src/*.{c,cpp,cxx,cc,m,asm,s,S}").map do |f|
|
---|
59 | objfile(f.relative_path_from(@dir).to_s.pathmap("#{build_dir}/%X"))
|
---|
60 | end
|
---|
61 |
|
---|
62 | @generate_functions = !(@rbfiles.empty? && @objs.empty?)
|
---|
63 | @objs << objfile("#{build_dir}/gem_init") if @generate_functions
|
---|
64 |
|
---|
65 | @test_rbfiles = Dir.glob("#{dir}/test/**/*.rb")
|
---|
66 | @test_objs = Dir.glob("#{dir}/test/*.{c,cpp,cxx,cc,m,asm,s,S}").map do |f|
|
---|
67 | objfile(f.relative_path_from(dir).to_s.pathmap("#{build_dir}/%X"))
|
---|
68 | end
|
---|
69 | @custom_test_init = !@test_objs.empty?
|
---|
70 | @test_preload = nil # 'test/assert.rb'
|
---|
71 | @test_args = {}
|
---|
72 |
|
---|
73 | @bins = []
|
---|
74 |
|
---|
75 | @requirements = []
|
---|
76 | @dependencies, @conflicts = [], []
|
---|
77 | @export_include_paths = []
|
---|
78 | @export_include_paths << "#{dir}/include" if File.directory? "#{dir}/include"
|
---|
79 |
|
---|
80 | instance_eval(&@initializer)
|
---|
81 |
|
---|
82 | if !name || !licenses || !authors
|
---|
83 | fail "#{name || dir} required to set name, license(s) and author(s)"
|
---|
84 | end
|
---|
85 |
|
---|
86 | build.libmruby << @objs
|
---|
87 |
|
---|
88 | instance_eval(&@build_config_initializer) if @build_config_initializer
|
---|
89 |
|
---|
90 | compilers.each do |compiler|
|
---|
91 | compiler.define_rules build_dir, "#{dir}"
|
---|
92 | compiler.defines << %Q[MRBGEM_#{funcname.upcase}_VERSION=#{version}]
|
---|
93 | compiler.include_paths << "#{dir}/include" if File.directory? "#{dir}/include"
|
---|
94 | end
|
---|
95 |
|
---|
96 | define_gem_init_builder if @generate_functions
|
---|
97 | end
|
---|
98 |
|
---|
99 | def add_dependency(name, *requirements)
|
---|
100 | default_gem = requirements.last.kind_of?(Hash) ? requirements.pop : nil
|
---|
101 | requirements = ['>= 0.0.0'] if requirements.empty?
|
---|
102 | requirements.flatten!
|
---|
103 | @dependencies << {:gem => name, :requirements => requirements, :default => default_gem}
|
---|
104 | end
|
---|
105 |
|
---|
106 | def add_test_dependency(*args)
|
---|
107 | add_dependency(*args) if build.test_enabled?
|
---|
108 | end
|
---|
109 |
|
---|
110 | def add_conflict(name, *req)
|
---|
111 | @conflicts << {:gem => name, :requirements => req.empty? ? nil : req}
|
---|
112 | end
|
---|
113 |
|
---|
114 | def self.bin=(bin)
|
---|
115 | @bins = [bin].flatten
|
---|
116 | end
|
---|
117 |
|
---|
118 | def build_dir
|
---|
119 | "#{build.build_dir}/mrbgems/#{name}"
|
---|
120 | end
|
---|
121 |
|
---|
122 | def test_rbireps
|
---|
123 | "#{build_dir}/gem_test.c"
|
---|
124 | end
|
---|
125 |
|
---|
126 | def funcname
|
---|
127 | @funcname ||= @name.gsub('-', '_')
|
---|
128 | end
|
---|
129 |
|
---|
130 | def compilers
|
---|
131 | MRuby::Build::COMPILERS.map do |c|
|
---|
132 | instance_variable_get("@#{c}")
|
---|
133 | end
|
---|
134 | end
|
---|
135 |
|
---|
136 | def define_gem_init_builder
|
---|
137 | file objfile("#{build_dir}/gem_init") => [ "#{build_dir}/gem_init.c", File.join(dir, "mrbgem.rake") ]
|
---|
138 | file "#{build_dir}/gem_init.c" => [build.mrbcfile, __FILE__] + [rbfiles].flatten do |t|
|
---|
139 | FileUtils.mkdir_p build_dir
|
---|
140 | generate_gem_init("#{build_dir}/gem_init.c")
|
---|
141 | end
|
---|
142 | end
|
---|
143 |
|
---|
144 | def generate_gem_init(fname)
|
---|
145 | open(fname, 'w') do |f|
|
---|
146 | print_gem_init_header f
|
---|
147 | build.mrbc.run f, rbfiles, "gem_mrblib_irep_#{funcname}" unless rbfiles.empty?
|
---|
148 | f.puts %Q[void mrb_#{funcname}_gem_init(mrb_state *mrb);]
|
---|
149 | f.puts %Q[void mrb_#{funcname}_gem_final(mrb_state *mrb);]
|
---|
150 | f.puts %Q[]
|
---|
151 | f.puts %Q[void GENERATED_TMP_mrb_#{funcname}_gem_init(mrb_state *mrb) {]
|
---|
152 | f.puts %Q[ int ai = mrb_gc_arena_save(mrb);]
|
---|
153 | f.puts %Q[ mrb_#{funcname}_gem_init(mrb);] if objs != [objfile("#{build_dir}/gem_init")]
|
---|
154 | unless rbfiles.empty?
|
---|
155 | f.puts %Q[ mrb_load_irep(mrb, gem_mrblib_irep_#{funcname});]
|
---|
156 | f.puts %Q[ if (mrb->exc) {]
|
---|
157 | f.puts %Q[ mrb_print_error(mrb);]
|
---|
158 | f.puts %Q[ exit(EXIT_FAILURE);]
|
---|
159 | f.puts %Q[ }]
|
---|
160 | end
|
---|
161 | f.puts %Q[ mrb_gc_arena_restore(mrb, ai);]
|
---|
162 | f.puts %Q[}]
|
---|
163 | f.puts %Q[]
|
---|
164 | f.puts %Q[void GENERATED_TMP_mrb_#{funcname}_gem_final(mrb_state *mrb) {]
|
---|
165 | f.puts %Q[ mrb_#{funcname}_gem_final(mrb);] if objs != [objfile("#{build_dir}/gem_init")]
|
---|
166 | f.puts %Q[}]
|
---|
167 | end
|
---|
168 | end # generate_gem_init
|
---|
169 |
|
---|
170 | def print_gem_comment(f)
|
---|
171 | f.puts %Q[/*]
|
---|
172 | f.puts %Q[ * This file is loading the irep]
|
---|
173 | f.puts %Q[ * Ruby GEM code.]
|
---|
174 | f.puts %Q[ *]
|
---|
175 | f.puts %Q[ * IMPORTANT:]
|
---|
176 | f.puts %Q[ * This file was generated!]
|
---|
177 | f.puts %Q[ * All manual changes will get lost.]
|
---|
178 | f.puts %Q[ */]
|
---|
179 | end
|
---|
180 |
|
---|
181 | def print_gem_init_header(f)
|
---|
182 | print_gem_comment(f)
|
---|
183 | f.puts %Q[#include <stdlib.h>] unless rbfiles.empty?
|
---|
184 | f.puts %Q[#include "mruby.h"]
|
---|
185 | f.puts %Q[#include "mruby/irep.h"] unless rbfiles.empty?
|
---|
186 | end
|
---|
187 |
|
---|
188 | def print_gem_test_header(f)
|
---|
189 | print_gem_comment(f)
|
---|
190 | f.puts %Q[#include <stdio.h>]
|
---|
191 | f.puts %Q[#include <stdlib.h>]
|
---|
192 | f.puts %Q[#include "mruby.h"]
|
---|
193 | f.puts %Q[#include "mruby/irep.h"]
|
---|
194 | f.puts %Q[#include "mruby/variable.h"]
|
---|
195 | f.puts %Q[#include "mruby/hash.h"] unless test_args.empty?
|
---|
196 | end
|
---|
197 |
|
---|
198 | def test_dependencies
|
---|
199 | [@name]
|
---|
200 | end
|
---|
201 |
|
---|
202 | def custom_test_init?
|
---|
203 | @custom_test_init
|
---|
204 | end
|
---|
205 |
|
---|
206 | def version_ok?(req_versions)
|
---|
207 | req_versions.map do |req|
|
---|
208 | cmp, ver = req.split
|
---|
209 | cmp_result = Version.new(version) <=> Version.new(ver)
|
---|
210 | case cmp
|
---|
211 | when '=' then cmp_result == 0
|
---|
212 | when '!=' then cmp_result != 0
|
---|
213 | when '>' then cmp_result == 1
|
---|
214 | when '<' then cmp_result == -1
|
---|
215 | when '>=' then cmp_result >= 0
|
---|
216 | when '<=' then cmp_result <= 0
|
---|
217 | when '~>'
|
---|
218 | Version.new(version).twiddle_wakka_ok?(Version.new(ver))
|
---|
219 | else
|
---|
220 | fail "Comparison not possible with '#{cmp}'"
|
---|
221 | end
|
---|
222 | end.all?
|
---|
223 | end
|
---|
224 | end # Specification
|
---|
225 |
|
---|
226 | class Version
|
---|
227 | include Comparable
|
---|
228 | include Enumerable
|
---|
229 |
|
---|
230 | def <=>(other)
|
---|
231 | ret = 0
|
---|
232 | own = to_enum
|
---|
233 |
|
---|
234 | other.each do |oth|
|
---|
235 | begin
|
---|
236 | ret = own.next <=> oth
|
---|
237 | rescue StopIteration
|
---|
238 | ret = 0 <=> oth
|
---|
239 | end
|
---|
240 |
|
---|
241 | break unless ret == 0
|
---|
242 | end
|
---|
243 |
|
---|
244 | ret
|
---|
245 | end
|
---|
246 |
|
---|
247 | # ~> compare algorithm
|
---|
248 | #
|
---|
249 | # Example:
|
---|
250 | # ~> 2.2 means >= 2.2.0 and < 3.0.0
|
---|
251 | # ~> 2.2.0 means >= 2.2.0 and < 2.3.0
|
---|
252 | def twiddle_wakka_ok?(other)
|
---|
253 | gr_or_eql = (self <=> other) >= 0
|
---|
254 | still_minor = (self <=> other.skip_minor) < 0
|
---|
255 | gr_or_eql and still_minor
|
---|
256 | end
|
---|
257 |
|
---|
258 | def skip_minor
|
---|
259 | a = @ary.dup
|
---|
260 | a.slice!(-1)
|
---|
261 | a[-1] = a[-1].succ
|
---|
262 | a
|
---|
263 | end
|
---|
264 |
|
---|
265 | def initialize(str)
|
---|
266 | @str = str
|
---|
267 | @ary = @str.split('.').map(&:to_i)
|
---|
268 | end
|
---|
269 |
|
---|
270 | def each(&block); @ary.each(&block); end
|
---|
271 | def [](index); @ary[index]; end
|
---|
272 | def []=(index, value)
|
---|
273 | @ary[index] = value
|
---|
274 | @str = @ary.join('.')
|
---|
275 | end
|
---|
276 | def slice!(index)
|
---|
277 | @ary.slice!(index)
|
---|
278 | @str = @ary.join('.')
|
---|
279 | end
|
---|
280 | end # Version
|
---|
281 |
|
---|
282 | class List
|
---|
283 | include Enumerable
|
---|
284 |
|
---|
285 | def initialize
|
---|
286 | @ary = []
|
---|
287 | end
|
---|
288 |
|
---|
289 | def each(&b)
|
---|
290 | @ary.each(&b)
|
---|
291 | end
|
---|
292 |
|
---|
293 | def <<(gem)
|
---|
294 | unless @ary.detect {|g| g.dir == gem.dir }
|
---|
295 | @ary << gem
|
---|
296 | else
|
---|
297 | # GEM was already added to this list
|
---|
298 | end
|
---|
299 | end
|
---|
300 |
|
---|
301 | def empty?
|
---|
302 | @ary.empty?
|
---|
303 | end
|
---|
304 |
|
---|
305 | def generate_gem_table build
|
---|
306 | gem_table = @ary.reduce({}) { |res,v| res[v.name] = v; res }
|
---|
307 |
|
---|
308 | default_gems = []
|
---|
309 | each do |g|
|
---|
310 | g.dependencies.each do |dep|
|
---|
311 | unless gem_table.key? dep[:gem]
|
---|
312 | if dep[:default]; default_gems << dep
|
---|
313 | elsif File.exist? "#{MRUBY_ROOT}/mrbgems/#{dep[:gem]}" # check core
|
---|
314 | default_gems << { :gem => dep[:gem], :default => { :core => dep[:gem] } }
|
---|
315 | else # fallback to mgem-list
|
---|
316 | default_gems << { :gem => dep[:gem], :default => { :mgem => dep[:gem] } }
|
---|
317 | end
|
---|
318 | end
|
---|
319 | end
|
---|
320 | end
|
---|
321 |
|
---|
322 | until default_gems.empty?
|
---|
323 | def_gem = default_gems.pop
|
---|
324 |
|
---|
325 | spec = build.gem def_gem[:default]
|
---|
326 | fail "Invalid gem name: #{spec.name} (Expected: #{def_gem[:gem]})" if spec.name != def_gem[:gem]
|
---|
327 | spec.setup
|
---|
328 |
|
---|
329 | spec.dependencies.each do |dep|
|
---|
330 | unless gem_table.key? dep[:gem]
|
---|
331 | if dep[:default]; default_gems << dep
|
---|
332 | else default_gems << { :gem => dep[:gem], :default => { :mgem => dep[:gem] } }
|
---|
333 | end
|
---|
334 | end
|
---|
335 | end
|
---|
336 | gem_table[spec.name] = spec
|
---|
337 | end
|
---|
338 |
|
---|
339 | each do |g|
|
---|
340 | g.dependencies.each do |dep|
|
---|
341 | name = dep[:gem]
|
---|
342 | req_versions = dep[:requirements]
|
---|
343 | dep_g = gem_table[name]
|
---|
344 |
|
---|
345 | # check each GEM dependency against all available GEMs
|
---|
346 | if dep_g.nil?
|
---|
347 | fail "The GEM '#{g.name}' depends on the GEM '#{name}' but it could not be found"
|
---|
348 | end
|
---|
349 | unless dep_g.version_ok? req_versions
|
---|
350 | fail "#{name} version should be #{req_versions.join(' and ')} but was '#{dep_g.version}'"
|
---|
351 | end
|
---|
352 | end
|
---|
353 |
|
---|
354 | cfls = g.conflicts.select { |c|
|
---|
355 | cfl_g = gem_table[c[:gem]]
|
---|
356 | cfl_g and cfl_g.version_ok?(c[:requirements] || ['>= 0.0.0'])
|
---|
357 | }.map { |c| "#{c[:gem]}(#{gem_table[c[:gem]].version})" }
|
---|
358 | fail "Conflicts of gem `#{g.name}` found: #{cfls.join ', '}" unless cfls.empty?
|
---|
359 | end
|
---|
360 |
|
---|
361 | gem_table
|
---|
362 | end
|
---|
363 |
|
---|
364 | def tsort_dependencies ary, table, all_dependency_listed = false
|
---|
365 | unless all_dependency_listed
|
---|
366 | left = ary.dup
|
---|
367 | until left.empty?
|
---|
368 | v = left.pop
|
---|
369 | table[v].dependencies.each do |dep|
|
---|
370 | left.push dep[:gem]
|
---|
371 | ary.push dep[:gem]
|
---|
372 | end
|
---|
373 | end
|
---|
374 | end
|
---|
375 |
|
---|
376 | ary.uniq!
|
---|
377 | table.instance_variable_set :@root_gems, ary
|
---|
378 | class << table
|
---|
379 | include TSort
|
---|
380 | def tsort_each_node &b
|
---|
381 | @root_gems.each &b
|
---|
382 | end
|
---|
383 |
|
---|
384 | def tsort_each_child(n, &b)
|
---|
385 | fetch(n).dependencies.each do |v|
|
---|
386 | b.call v[:gem]
|
---|
387 | end
|
---|
388 | end
|
---|
389 | end
|
---|
390 |
|
---|
391 | begin
|
---|
392 | table.tsort.map { |v| table[v] }
|
---|
393 | rescue TSort::Cyclic => e
|
---|
394 | fail "Circular mrbgem dependency found: #{e.message}"
|
---|
395 | end
|
---|
396 | end
|
---|
397 |
|
---|
398 | def check(build)
|
---|
399 | gem_table = generate_gem_table build
|
---|
400 |
|
---|
401 | @ary = tsort_dependencies gem_table.keys, gem_table, true
|
---|
402 |
|
---|
403 | each do |g|
|
---|
404 | import_include_paths(g)
|
---|
405 | end
|
---|
406 | end
|
---|
407 |
|
---|
408 | def import_include_paths(g)
|
---|
409 | gem_table = @ary.reduce({}) { |res,v| res[v.name] = v; res }
|
---|
410 | g.dependencies.each do |dep|
|
---|
411 | dep_g = gem_table[dep[:gem]]
|
---|
412 | # We can do recursive call safely
|
---|
413 | # as circular dependency has already detected in the caller.
|
---|
414 | import_include_paths(dep_g)
|
---|
415 |
|
---|
416 | g.compilers.each do |compiler|
|
---|
417 | compiler.include_paths += dep_g.export_include_paths
|
---|
418 | g.export_include_paths += dep_g.export_include_paths
|
---|
419 | end
|
---|
420 | end
|
---|
421 | end
|
---|
422 | end # List
|
---|
423 | end # Gem
|
---|
424 |
|
---|
425 | GemBox = Object.new
|
---|
426 | class << GemBox
|
---|
427 | attr_accessor :path
|
---|
428 |
|
---|
429 | def new(&block); block.call(self); end
|
---|
430 | def config=(obj); @config = obj; end
|
---|
431 | def gem(gemdir, &block); @config.gem(gemdir, &block); end
|
---|
432 | end # GemBox
|
---|
433 | end # MRuby
|
---|