Class: Restic::Service::AutoUpdate

Inherits:
Object
  • Object
show all
Defined in:
lib/restic/service/auto_update.rb

Defined Under Namespace

Classes: FailedUpdate

Constant Summary collapse

RESTIC_RELEASE_VERSION = 
"0.9.2"
RCLONE_RELEASE_VERSION = 
"1.41"

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(binary_path) ⇒ AutoUpdate

Returns a new instance of AutoUpdate.



24
25
26
27
28
29
30
31
 
# File 'lib/restic/service/auto_update.rb', line 24

def initialize(binary_path)
    @root = File.dirname(File.dirname(binary_path))
    unless File.file?(File.join(@root, "Gemfile"))
        raise FailedUpdate, "cannot guess installation path (tried #{@root})"
    end

    @gem_home = ENV['GEM_HOME']
end

Class Method Details

.rclone_release_url(platform) ⇒ Object 



17
18
19
20
21
 
# File 'lib/restic/service/auto_update.rb', line 17

def self.rclone_release_url(platform)
    basename = "rclone-v#{RCLONE_RELEASE_VERSION}-#{platform}"
    url = "https://github.com/ncw/rclone/releases/download/v#{RCLONE_RELEASE_VERSION}/#{basename}.zip"
    [basename, url]
end

.restic_release_url(platform) ⇒ Object 



11
12
13
 
# File 'lib/restic/service/auto_update.rb', line 11

def self.restic_release_url(platform)
    "https://github.com/restic/restic/releases/download/v#{RESTIC_RELEASE_VERSION}/restic_#{RESTIC_RELEASE_VERSION}_#{platform}.bz2"
end

Instance Method Details

#current_gem_version(gem_name) ⇒ Object

Raises:



47
48
49
50
51
52
53
54
 
# File 'lib/restic/service/auto_update.rb', line 47

def current_gem_version(gem_name)
    gemfile_lock = File.join(@root, "Gemfile.lock")
    File.readlines(gemfile_lock).each do |line|
        match = /^\s+#{gem_name} \((.*)\)$/.match(line)
        return match[1] if match
    end
    raise FailedUpdate, "cannot find the version line for #{gem_name} in #{gemfile_lock}"
end

#download_file(temp_path, url) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/restic/service/auto_update.rb', line 72

def download_file(temp_path, url)
    binary = nil
    while !binary
        response = Net::HTTP.get_response(URI(url))
        case response
        when Net::HTTPSuccess
            binary = response.body
        when Net::HTTPRedirection
            url = response['location']
        else
            raise FailedUpdate, "failed to fetch #{url}: #{response}"
        end
    end

    File.open(temp_path, 'w') do |io|
        io.write binary
    end
end

#patch_binstubs ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/restic/service/auto_update.rb', line 33

def patch_binstubs
    bindir = File.join(@root, 'bin')
    Dir.new(bindir).each do |entry|
        entry = File.join(bindir, entry)
        if File.file?(entry)
            patched_contents = File.readlines(entry)
            patched_contents.insert(1, "ENV['GEM_HOME'] = '#{@gem_home}'\n")
            File.open(entry, 'w') do |io|
                io.write patched_contents.join("")
            end
        end
    end
end

#update_rclone(platform, target_path) ⇒ Object 



119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
 
# File 'lib/restic/service/auto_update.rb', line 119

def update_rclone(platform, target_path)
    basename, release_url = self.class.rclone_release_url(platform)

    tmpdir  = Dir.mktmpdir
    tmppath = File.join(tmpdir, "rclone.zip")
    download_file(tmppath, release_url)

    if !system("unzip", tmppath, chdir: tmpdir, out: '/dev/null')
        raise FailedUpdate, "failed to uncompress the rclone release file"
    end

    rclone_path = File.join(tmpdir, basename, 'rclone')
    if File.file?(target_path)
        current = File.read(target_path)
        new     = File.read(rclone_path)
        return if current == new
    end

    FileUtils.mv rclone_path, target_path
    FileUtils.chmod 0755, target_path
    true

ensure
    FileUtils.rm_rf tmpdir if tmpdir
end

#update_release_if_needed(target_path, temp_path) ⇒ Object 



91
92
93
94
95
96
97
98
99
100
101
 
# File 'lib/restic/service/auto_update.rb', line 91

def update_release_if_needed(target_path, temp_path)
    if File.file?(target_path)
        current = File.read(target_path)
        new     = File.read(temp_path)
        return false if current == new
    end

    FileUtils.mv temp_path, target_path
    FileUtils.chmod 0755, target_path
    true
end

#update_restic(platform, target_path) ⇒ Object 



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

def update_restic(platform, target_path)
    release_url = self.class.restic_release_url(platform)

    tmpdir = Dir.mktmpdir
    restic_path = File.join(tmpdir, "restic")
    download_file("#{restic_path}.bz2", release_url)

    if !system("bzip2", "-d", "#{restic_path}.bz2")
        raise FailedUpdate, "failed to uncompress the restic release file"
    end
    update_release_if_needed(target_path, restic_path)

ensure
    FileUtils.rm_rf tmpdir if tmpdir
end

#update_restic_service ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'lib/restic/service/auto_update.rb', line 56

def update_restic_service
    current_version = current_gem_version "restic-service"
    reader, writer = IO.pipe
    if !system("bundle", "update", out: writer, err: writer)
        writer.close
        puts reader.read(1024)
        raise FailedUpdate, "failed to run bundle update"
    end
    patch_binstubs
    new_version = current_gem_version "restic-service"
    [current_version, new_version]
ensure
    writer.close if writer && !writer.closed?
    reader.close if reader && !reader.closed?
end