summaryrefslogtreecommitdiff
path: root/tool/checksum.rb
blob: 40f444efd299d4c4ae692b637e9c4b7ce4d8edb6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!ruby

require_relative 'vpath'

class Checksum
  def initialize(vpath)
    @vpath = vpath
  end

  def source=(source)
    @source = source
    @checksum = File.basename(source, ".*") + ".chksum"
  end

  def update?
    src = @vpath.read(@source)
    @len = src.length
    @sum = src.sum
    begin
      data = @vpath.read(@checksum)
    rescue
      return false
    else
      return false unless data[/src="([0-9a-z_.-]+)",/, 1] == @source
      return false unless @len == data[/\blen=(\d+)/, 1].to_i
      return false unless @sum == data[/\bchecksum=(\d+)/, 1].to_i
      return true
    end
  end

  def update!
    open(@checksum, "wb") {|f|
      f.puts("src=\"#{@source}\", len=#{@len}, checksum=#{@sum}")
    }
  end

  def update
    return true if update?
    update! if ret = yield(self)
    ret
  end

  def copy(name)
    @vpath.open(name, "rb") {|f|
      IO.copy_stream(f, name)
    }
    true
  end

  def make(arg)
    system([@make, arg].compact.join(' '))
  end

  def def_options(opt = (require 'optparse'; OptionParser.new))
    @vpath.def_options(opt)
    opt.on("--make=PATH") {|v| @make = v}
    opt
  end

  def self.update(argv)
    k = new(VPath.new)
    k.source, *argv = k.def_options.parse(*argv)
    k.update {|k| yield(k, *argv)}
  end
end