fackup 0.0.1

data/bin/fackup

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+require 'fackup/cli'
+
+if Process.uid != 0
+  $stderr.puts "You should run the program as root,"
+  $stderr.puts "files without right permissions will be skipped"
+end
+
+FackUp::CLI.start(ARGV)

data/lib/fackup/cli.rb

@@ -0,0 +1,84 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of fackup.
+#
+# fackup is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# fackup is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with fackup. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'thor'
+require 'fackup'
+
+module FackUp
+  class CLI < Thor
+    class_option :help, type: :boolean, desc: 'Show help usage'
+
+    desc 'version', 'Show current version'
+    map '-v' => :version, '--version' => :version
+    def version
+      puts "fackup v#{FackUp::VERSION}"
+    end
+
+    desc 'add FILE... [OPTIONS]', 'add file to backup'
+      method_option :recursive, aliases: '-r', type: :boolean, default: false,
+        desc: 'Add files in directories recursively'
+    def add (*files)
+      files.each {|file|
+        if File.directory?(file)
+          if options[:recursive]
+            add(Dir["#{file}/*"])
+          else
+            say '* ', :red
+            say "#{file} is a directory."
+          end
+        elsif File.file?(file)
+          unless DB << file
+            say '* ', :red
+            say "Can't add #{file}, unknown error."
+          end
+        else
+          say '* ', :red
+          say "Can't add #{file}, unknown format."
+        end
+      }
+    end
+
+    desc 'backup [FILE]', 'backup files'
+    def backup (file=File.join(Dir.home, 'fackup_backup.img'))
+      FackUp.backup(file)
+    end
+
+    desc 'restore IMAGE [OPTIONS]', 'restore image file'
+      method_option :force, aliases: '-f', type: :boolean, default: false,
+        desc: 'Replace files without controls'
+    def restore (image)
+      FackUp.send((options[:force] ? :restore_f : :restore), image)
+    end
+
+    desc 'list [IMAGE]', 'list backup files, if given, list file in image files'
+    def list (image=nil)
+      if image and File.esists?(image)
+        FackUp::Image.new(image, 'r', true) {|img|
+          img.each {|file|
+            puts file.name
+          }
+        }
+      else
+        DB.each {|file|
+          puts file
+        } if File.exists?(DB.path)
+      end
+    end
+  end
+end

data/lib/fackup/db.rb

@@ -0,0 +1,76 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of fackup.
+#
+# fackup is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# fackup is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with fackup. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'csv'
+
+module FackUp
+  module DB
+    PATH = File.join(Dir.home, '.fackup_files')
+
+    class << self
+      def path= (p)
+        @path = File.realpath(p)
+      end
+
+      def path
+        @path ||= PATH
+      end
+
+      def all
+        touch(DB.path)
+        CSV.read(DB.path).map(&:first)
+      end
+
+      def each (&blk)
+        all.each(&blk)
+      end
+
+      def push (path)
+        path = File.realpath(path)
+
+        return false if all.include?(path)
+
+        CSV.open(DB.path, 'ab') {|csv|
+          csv << [path]
+        }
+        true
+      end
+      alias << push
+
+      def delete (path)
+        path = File.realpath(path)
+
+        a = all
+        return false unless a.include?(path)
+
+        CSV.open(DB.path, 'wb') {|csv|
+          (a - [path]).each {|p|
+            csv << [p]
+          }
+        }
+        true
+      end
+
+      private
+      def touch (file)
+        File.open(file, 'a') {}
+      end
+    end
+  end
+end

data/lib/fackup/image.rb

@@ -0,0 +1,120 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of fackup.
+#
+# fackup is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# fackup is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with fackup. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module FackUp
+  class Image
+    class File < Struct.new(:name, :io, :time, :start, :size, :uid, :gid, :mode)
+      def pack
+        "%s\0%s" % [name, [time.to_i, start, size, uid, gid, mode].pack(PACK_TEMPLATE)]
+      end
+
+      def content
+        io.seek(start)
+        io.read(size)
+      end
+    end
+
+    PACK_TEMPLATE = "LLLIII"
+    PACK_SIZE     = [0, 0, 0, 0, 0, 0].pack(PACK_TEMPLATE).bytesize
+
+    def initialize (file, mode='r', force=false)
+      @files = []
+
+      ::File.open(file, 'a'){} if force
+
+      if mode == 'r'
+        raise Errno::ENOENT, "File doesn't exist" unless ::File.exists?(file)
+
+        @fd = ::File.open(file, 'rb')
+
+        load_file
+
+        [:push, :<<, :save].each {|meth|
+          (class << self; self; end).class_eval { undef_method(meth) }
+        }
+      elsif mode == 'w'
+        @fd = ::File.open(file, 'wb')
+        (class << self; self; end).class_eval { undef_method(:each) }
+      else
+        raise "Unrecognized mode"
+      end
+
+      if block_given?
+        yield self
+        close
+      end
+    end
+
+    def push (path)
+      path = ::File.realpath(path)
+      io = ::File.open(path, 'rb')
+      stat = io.stat
+
+      @files << File.new(path, io, stat.mtime, 0, io.size, stat.uid, stat.gid, stat.mode)
+      self
+    end
+    alias << push
+
+    def each(&blk)
+      @files.each(&blk)
+    end
+
+    def save
+      @fd.write(header)
+      @fd.write("\0")
+      @files.each {|file|
+        @fd.write(file.content)
+      }
+    end
+
+    def close
+      if self.respond_to?(:save)
+        save
+      end
+
+      @fd.close
+    end
+
+    private
+    def header
+      last = @files.map {|x| x.name.bytesize }.inject(:+) + @files.size * (1 + PACK_SIZE) + 1
+      @files.map {|file|
+        file.dup.tap {|x| x.start = last }.pack.tap {
+          last += file.size
+        }
+      }.join
+    end
+
+    def load_file
+      @files = []
+      loop {
+        if @fd.gets("\0").size > 1
+          path = $_.strip
+
+          args = @fd.read(PACK_SIZE).unpack(PACK_TEMPLATE)
+          args[0] = Time.at(args[0])
+
+          @files << File.new(path, @fd, *args)
+        else
+          break
+        end
+      }
+    end
+  end
+end

data/lib/fackup.rb

@@ -0,0 +1,85 @@
+#--
+# Copyleft shura. [ shura1991@gmail.com ]
+#
+# This file is part of fackup.
+#
+# fackup is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# fackup is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with fackup. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'fileutils'
+require 'fackup/db'
+require 'fackup/image'
+
+module FackUp
+  VERSION = '0.0.1'
+
+  def self.restore (image)
+    FackUp::Image.new(image, 'r', true) {|img|
+      img.each {|file|
+        FileUtils.mkdir_p(File.dirname(file.name))
+
+        if File.exists?(file.name)
+          stat = File.stat(file.name)
+
+          next if file.time == stat.mtime and file.size == File.size(file.name) and
+            file.uid == stat.uid and file.gid == stat.gid and file.mode == stat.mod
+        end
+
+        begin
+          File.open(file.name, 'wb') {|f|
+            f.write(file.content)
+          }
+        rescue Errno::EACCES
+          next
+        end
+
+        File.chown(file.uid, file.gid, file.name)
+        File.chmod(file.mode, file.name)
+        File.utime(file.time, file.time, file.name)
+      }
+    }
+  end
+
+  def self.restore_f (image)
+    FackUp::Image.new(image, 'r', true) {|img|
+      img.each {|file|
+        FileUtils.mkdir_p(File.dirname(file.name))
+
+        begin
+          File.open(file.name, 'wb') {|f|
+            f.write(file.content)
+          }
+        rescue Errno::EACCES
+          next
+        end
+
+        File.chown(file.uid, file.gid, file.name)
+        File.chmod(file.mode, file.name)
+        File.utime(file.time, file.time, file.name)
+      }
+    }
+  end
+
+  def self.backup (file)
+    FackUp::Image.new(file, 'w', true) {|img|
+      DB.each {|file|
+        unless File.readable?(file)
+          next
+        end
+
+        img << file
+      }
+    }
+  end
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification 
+name: fackup
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- shura
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-05 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Simple tool to do simple backups
+email: shura1991@gmail.com
+executables: 
+- fackup
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/fackup/cli.rb
+- lib/fackup/image.rb
+- lib/fackup/db.rb
+- lib/fackup.rb
+- bin/fackup
+has_rdoc: true
+homepage: http://github.com/shurizzle/fackup
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Simple tool to do simple backups
+test_files: []
+