p2b-backup 0.0.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: de162570feb13aa0e5cc913160a12dceeebee89eaf9892e4730620c0ee9a68a9
+  data.tar.gz: 2ddd2e1faba8f6af9b58ef7ce7a604f5ad32834be82b1aa4bacc49583a34019d
+SHA512:
+  metadata.gz: 2ef0af368c87a81623e4a67679bd10d89d63b23cc93728b25df2a7b51d3ee073e8caae13f0d1a69dad8d0250c088ece8f45b2be58ac6501daa7eee221e5255b8
+  data.tar.gz: b32868e417622690edfdc70178bf9ba388982529b5ded208d5b7c3617c7115222af1f204d93f5d33c97792aceb80864d52606cdae30daeb64f1d41f6815bbad3

data/Rakefile

@@ -0,0 +1 @@
+require 'rubygems'

data/bin/p2bbackup

@@ -0,0 +1,86 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'yaml'
+require 'optparse'
+require 'p2b-backup'
+
+options = { command: :new_backup }
+
+options[:config] = '/etc/.postgresql.backups.yml' if File.exist?('/etc/.postgresql.backups.yml')
+
+# Build a parser for the command line arguments
+opts = OptionParser.new do |opts|
+  opts.version = '0.0.4'
+
+  opts.banner = 'Usage: p2bbackup [-flag] [argument]'
+  opts.define_head 'p2bbackup: backing up your shit since way back when...'
+  opts.separator '*' * 80
+
+  opts.on('-l', '--list-backup DATABASE', 'List mysql backups for DATABASE') do |db|
+    options[:db] = (db || 'all')
+    options[:command] = :list
+  end
+
+  opts.on('-n', '--new-backup', 'Create new mysql backup') do
+    options[:command] = :new_backup
+  end
+
+  opts.on('-c', '--config CONFIG', 'Use config file.') do |config|
+    options[:config] = config
+  end
+
+  opts.on('-d', '--download BACKUP_INDEX',
+          'download the backup specified by index. Run p2bbackup -l to get the index.') do |index|
+    options[:command] = :download
+    options[:index] = index
+  end
+
+  opts.on('-e', '--engine DATABASE_ENGINE', 'The database engine. ex: mysql, postgresql.') do |engine|
+    options[:engine] = engine || if File.exist?('/etc/.mysql.backups.yml')
+                                   'mysql'
+                                 elsif File.exist?('/etc/.postgresql.backups.yml')
+                                   'postgresql'
+                                 end
+
+    options[:config] ||= "/etc/.#{options[:engine]}.backups.yml"
+  end
+
+  opts.on('-r', '--restore BACKUP_INDEX',
+          'Download and apply the backup specified by index WARNING! will overwrite the current db with the backup. Run eybackup -l to get the index.') do |index|
+    options[:command] = :restore
+    options[:index] = index
+  end
+end
+
+opts.parse!
+options[:config] ||= (File.exist?('/etc/.postgresql.backups.yml') ? '/etc/.postgresql.backups.yml' : '/etc/.mysql.backups.yml')
+options[:config] = File.expand_path(options[:config])
+
+options[:engine] ||= if File.exist?('/etc/.mysql.backups.yml')
+                       'mysql'
+                     elsif File.exist?('/etc/.postgresql.backups.yml')
+                       'postgresql'
+                     end
+
+backup_klass = nil
+if File.exist?(options[:config])
+  backup_klass = case options[:engine]
+                 when 'postgresql', 'postgres' then P2bBackup::PostgresqlBackup.new(YAML.load(IO.read(options[:config]),
+                                                                                              symbolize_names: true))
+                 when 'mysql', NilClass        then P2bBackup::MysqlBackup.new(YAML.load(IO.read(options[:config]),
+                                                                                         symbolize_names: true))
+                 else raise "Invalid engine: #{options[:engine]}"
+                 end
+end
+
+case options[:command]
+when :list
+  backup_klass.list options[:db], true
+when :new_backup
+  backup_klass.new_backup
+when :download
+  backup_klass.download(options[:index])
+when :restore
+  backup_klass.restore(options[:index])
+end
+# backup_klass.cleanup

data/lib/p2b-backup/backup.rb

@@ -0,0 +1,52 @@
+# require 'aws/s3'
+require 'date'
+require 'digest'
+require 'net/http'
+require 'tmpdir'
+require 'fileutils'
+require 'p2b-backup/providers/s3'
+require 'p2b-backup/providers/sftp'
+
+module P2bBackup
+  class Backup
+    def initialize(opts = {})
+      @dbuser = opts[:dbuser]
+      @dbpass = opts[:dbpass]
+      @databases = opts[:databases]
+      @keep = opts[:keep]
+      @tmpname = "#{Time.now.strftime('%Y-%m-%dT%H:%M:%S').gsub(/:/, '-')}.sql.gz"
+
+      @provider = case opts[:provider]
+                  when 's3' then P2bBackup::Providers::S3.new(opts[:provider_opts])
+                  when 'sftp' then P2bBackup::Providers::Sftp.new(opts[:provider_opts])
+                  else raise "Unknown provider #{opts[:provider]}"
+                  end
+    end
+
+    def backup_database(database)
+      if system(backup_command(database))
+        @provider.upload("#{Dir.tmpdir}/#{database}.#{@tmpname}", filename: @tmpname)
+        FileUtils.rm "#{Dir.tmpdir}/#{database}.#{@tmpname}"
+        puts "successful backup: #{Dir.tmpdir}/#{database}.#{@tmpname}"
+      else
+        raise "Unable to dump database '#{database}' wtf?"
+      end
+    end
+
+    def new_backup
+      @databases.each do |db|
+        backup_database(db)
+      end
+    end
+
+    def normalize_name(obj)
+      obj.key.gsub(%r{^.*?/}, '')
+    end
+
+    private
+
+    def backup_command(*)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/p2b-backup/backups/mysql_backup.rb

@@ -0,0 +1,13 @@
+require 'tmpdir'
+require 'fileutils'
+require 'p2b-backup/providers/s3'
+
+module P2bBackup
+  class MysqlBackup < Backup
+    private
+
+    def backup_command(database)
+      "mysqldump -u #{@dbuser} -p'#{@dbpass}' #{database} | gzip - > #{Dir.tmpdir}/#{database}.#{@tmpname}"
+    end
+  end
+end

data/lib/p2b-backup/backups/postgresql_backup.rb

@@ -0,0 +1,13 @@
+require 'tmpdir'
+require 'fileutils'
+require 'p2b-backup/providers/s3'
+
+module P2bBackup
+  class PostgresqlBackup < Backup
+    private
+
+    def backup_command(database)
+      "PGPASSWORD='#{@dbpass}' pg_dump --clean --no-owner --no-privileges -U#{@dbuser} #{database} | gzip - > #{Dir.tmpdir}/#{database}.#{@tmpname}"
+    end
+  end
+end

data/lib/p2b-backup/providers/s3.rb

@@ -0,0 +1,27 @@
+require 'aws-sdk-core'
+require 'aws-sdk-s3'
+module P2bBackup
+  module Providers
+    class S3
+      def initialize(opts = {})
+        @client = Aws::S3::Client.new(
+          region: opts[:region],
+          access_key_id: opts[:access_key_id],
+          secret_access_key: opts[:secret_access_key]
+        )
+        @bucket = opts[:bucket]
+      end
+
+      def upload(file_or_io)
+        @client.put_object(
+          {
+            body: file_or_io,
+            bucket: @bucket,
+            key: 'HappyFace.jpg'
+          }
+        )
+        puts "Uploaded to #{@bucket}"
+      end
+    end
+  end
+end

data/lib/p2b-backup/providers/sftp.rb

@@ -0,0 +1,19 @@
+require 'net/sftp'
+
+module P2bBackup
+  module Providers
+    class Sftp
+      def initialize(opts = {})
+        @opts = opts
+      end
+
+      def upload(path, filename: 'file.sql.gz')
+        Net::SFTP.start(@opts[:host], @opts[:username], password: @opts[:password]) do |sftp|
+          # upload a file or directory to the remote host
+          sftp.upload!(path, "#{@opts[:path]}/#{filename}")
+        end
+        puts "Uploaded to #{@bucket}"
+      end
+    end
+  end
+end

data/lib/p2b-backup.rb

@@ -0,0 +1,3 @@
+require 'p2b-backup/backup'
+require 'p2b-backup/backups/mysql_backup'
+require 'p2b-backup/backups/postgresql_backup'

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification
+name: p2b-backup
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+platform: ruby
+authors:
+- Frederik Spang Thomsen
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-02-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-s3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: net-sftp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: hello
+email: fst@place2book.com
+executables:
+- p2bbackup
+extensions: []
+extra_rdoc_files: []
+files:
+- Rakefile
+- bin/p2bbackup
+- lib/p2b-backup.rb
+- lib/p2b-backup/backup.rb
+- lib/p2b-backup/backups/mysql_backup.rb
+- lib/p2b-backup/backups/postgresql_backup.rb
+- lib/p2b-backup/providers/s3.rb
+- lib/p2b-backup/providers/sftp.rb
+homepage: http://place2book.com
+licenses: []
+metadata:
+  allowed_push_host: https://rubygems.org
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.22
+signing_key:
+specification_version: 4
+summary: hello
+test_files: []