caco 0.1.0

data/lib/caco/debian/apt_sources_list.rb

@@ -0,0 +1,15 @@
+module Caco::Debian
+  class AptSourcesList < Trailblazer::Operation
+    step ->(ctx, mirror_url: nil, **) {
+        ctx[:content] = Caco::Debian::Cell::SourcesList.(mirror_url: mirror_url).to_s
+      },
+      id: :generate_content
+
+    step ->(ctx, **) { ctx[:path] = '/etc/apt/sources.list' },
+      id: :build_path
+
+    step Subprocess(Caco::FileWriter),
+      input: [:path, :content],
+      output: { file_changed: :sources_updated }
+  end
+end

data/lib/caco/debian/apt_update.rb

@@ -0,0 +1,33 @@
+module Caco::Debian
+  class AptUpdate < Trailblazer::Operation
+    step :apt_needs_update,
+      Output(Trailblazer::Activity::Left, :failure) => End(:success)
+
+    step Subprocess(Caco::Executer),
+      input: ->(_ctx, **) {{
+        command: 'apt-get update'
+      }},
+      output: { exit_code: :command_exit_code, output: :command_output }
+
+    step :apt_updated
+
+    fail :command_failed
+
+    def apt_needs_update(ctx, force: false, **)
+      ctx[:apt_needs_update] = !Caco::Debian.apt_updated
+      ctx[:apt_needs_update] = true if force
+      ctx[:apt_needs_update]
+    end
+
+    def apt_updated(ctx, **)
+      ctx[:apt_updated] = true
+      Caco::Debian.apt_updated = true
+    end
+
+    def command_failed(ctx, command_exit_code:, command_output:, **)
+      ctx[:apt_updated] = false
+      Caco::Debian.apt_updated = false
+      true
+    end
+  end
+end

data/lib/caco/debian/cell/service.rb

@@ -0,0 +1,19 @@
+module Caco::Debian::Cell
+  class Service < Trailblazer::Cell
+    def description
+      property(:description) || "No Description Provided"
+    end
+
+    def environment_file
+      property(:environment_file)
+    end
+
+    def environment_vars
+      property(:environment_vars) || []
+    end
+
+    def command
+      property(:command)
+    end
+  end
+end

data/lib/caco/debian/cell/sources_list.rb

@@ -0,0 +1,7 @@
+module Caco::Debian::Cell
+  class SourcesList < Trailblazer::Cell
+    def mirror_url
+      property(:mirror_url) || Settings.debian.apt_default_repo.mirror_url
+    end
+  end
+end

data/lib/caco/debian/package_install.rb

@@ -0,0 +1,21 @@
+module Caco::Debian
+  class PackageInstall < Trailblazer::Operation
+    class PackageNameError < StandardError; end
+
+    step Subprocess(Caco::Debian::PackageInstalled),
+      id: "package_installed?",
+      input: [:package],
+      output: []
+
+    fail Subprocess(Caco::Executer),
+      Output(:success) => End(:success),
+      input: ->(_ctx, package:, **) {{
+        command: "apt-get install -y #{package}"
+      }},
+      output: { exit_code: :package_install_exit_code, output: :package_install_output },
+      id: "package_install"
+
+    step ->(ctx, **) { ctx[:already_installed] = true },
+      id: :package_already_installed!
+  end
+end

data/lib/caco/debian/package_installed.rb

@@ -0,0 +1,17 @@
+module Caco::Debian
+  class PackageInstalled < Trailblazer::Operation
+    step Subprocess(Caco::Executer),
+      input: ->(_ctx, package:, **) {{
+        command: "dpkg -s #{package}"
+      }},
+      output: { exit_code: :command_exit_code, output: :command_output },
+      id: "dpkg"
+
+    step Subprocess(Caco::Finder),
+      input: ->(_ctx, package:, **) {{
+        command: "dpkg-query -W -f='${Status} ${Version}\n' #{package}",
+        regexp: /^install/
+      }},
+      id: "dpkg_query"
+  end
+end

data/lib/caco/debian/service_enable.rb

@@ -0,0 +1,26 @@
+module Caco::Debian
+  class ServiceEnable < Trailblazer::Operation
+    step Subprocess(Caco::Executer),
+      input: ->(_ctx, service:, **) {{
+        command: "systemctl list-units --full -all | grep -Fq \"#{service}.service\""
+      }},
+      id: :check_service_exist
+
+    step Subprocess(Class.new(Caco::Executer)),
+      Output(:success) => End(:success),
+      Output(:failure) => Track(:success),
+      input: ->(_ctx, service:, **) {{
+        command: "systemctl is-enabled #{service}.service"
+      }},
+      id: :check_service_enabled
+
+    step Subprocess(Class.new(Caco::Executer)),
+      input: ->(_ctx, service:, **) {{
+        command: "systemctl enable #{service}.service"
+      }},
+      id: :enable_service
+
+    step ->(ctx, **) { ctx[:enabled] = true },
+      id: :mark_enabled
+  end
+end

data/lib/caco/debian/service_install.rb

@@ -0,0 +1,31 @@
+module Caco::Debian
+  class ServiceInstall < Trailblazer::Operation
+    step ->(ctx, name:, **) {
+        ctx[:service_path] = "/etc/systemd/system/#{name}.service"
+      },
+      id: :generate_service_path
+
+    step :generate_template_content
+
+    step Subprocess(Caco::FileWriter),
+      input: ->(_ctx, service_path:, template_content:, **) {{
+        path: service_path,
+        content: template_content
+      }}
+
+    step Subprocess(Caco::Executer),
+      input: ->(_ctx, **) {{
+        command: "systemctl daemon-reload",
+      }},
+      id: "systemctl_reload"
+
+    def generate_template_content(ctx, description: nil, environment_file: nil, environment_vars: nil, command:, **)
+      ctx[:template_content] = Caco::Debian::Cell::Service.(
+        command: command,
+        description: description,
+        environment_file: environment_file,
+        environment_vars: environment_vars,
+      ).to_s
+    end
+  end
+end

data/lib/caco/debian/user_home.rb

@@ -0,0 +1,17 @@
+module Caco::Debian
+  class UserHome < Trailblazer::Operation
+    step Subprocess(Caco::FileReader),
+      input: ->(_ctx, **) {{
+        path: "/etc/passwd",
+      }},
+      output: { output: :passwd_output }
+    step :find_user_home
+
+    def find_user_home(ctx, user:, passwd_output:, **)
+      match = passwd_output.match(/^#{user}:[^:]*:[^:]*:[^:]*:[^:]*:([^:]*):.*$/)
+      return false unless match
+
+      ctx[:user_home] = match[1]
+    end
+  end
+end

data/lib/caco/debian/view/service.erb

@@ -0,0 +1,18 @@
+# File Managed, Dot Not Edit
+[Unit]
+Description=<%= description %>
+After=syslog.target network.target network-online.target
+
+[Service]
+<% if environment_file %>
+EnvironmentFile=<%= environment_file %>
+<% end %>
+<% environment_vars.each do |env| %>
+Environment=<%= env %>
+<% end %>
+User=root
+Restart=on-failure
+ExecStart=<%= command %>
+
+[Install]
+WantedBy=multi-user.target

data/lib/caco/debian/view/sources_list.erb

@@ -0,0 +1,10 @@
+# File Managed, Dot Not Edit
+deb <%= mirror_url %> stretch main
+deb-src <%= mirror_url %> stretch main
+
+deb http://security.debian.org/debian-security stretch/updates main
+deb-src http://security.debian.org/debian-security stretch/updates main
+
+# stretch-updates, previously known as 'volatile'
+deb <%= mirror_url %> stretch-updates main
+deb-src <%= mirror_url %> stretch-updates main

data/lib/caco/downloader.rb

@@ -0,0 +1,41 @@
+require 'net/http'
+require "open-uri"
+
+module Caco
+  class Downloader < Trailblazer::Operation
+    Stubbed = Class.new(Trailblazer::Activity::Signal)
+
+    step ->(_ctx, stubbed_file: nil, **) {
+        stubbed_file ? Stubbed : true
+      },
+      Output(Stubbed, :stubbed) => Track(:stubbed),
+      id: :check_stubbed
+
+    step ->(ctx, url:, **) {
+        ctx[:tempfile] = Down.download(url)
+      },
+      id: :download_file
+    
+    step ->(ctx, stubbed_file:, **) {
+        ctx[:tempfile] = File.new(ctx[:stubbed_file])
+      },
+      magnetic_to: :stubbed,
+      id: :stubbed_download_file
+
+    step ->(_ctx, **) {
+        true
+      },
+      id: :check_md5
+
+    step :write_file
+
+    def write_file(ctx, tempfile:, dest:, **)
+      if Caco.config.write_files
+        FileUtils.mkdir_p(File.dirname(dest))
+        File.rename tempfile.path, dest
+      else
+        tempfile.unlink
+      end
+    end
+  end
+end

data/lib/caco/executer.rb

@@ -0,0 +1,33 @@
+module Caco
+  class Executer < Trailblazer::Operation
+    step :execute!
+
+    def execute!(ctx, command:, **)
+      s, e, o, error = self.class.send(:execute, command)
+      ctx[:signal] = [s, e, o, error]
+      ctx[:exit_code] = e
+      ctx[:output] = o
+      ctx[:stderr] = error
+
+      return s
+    end
+
+    module ClassMethods
+      private
+      def execute(command)
+        stdout = nil
+        stderr = nil
+        pid = nil
+        exit_status = nil
+        Open3.popen3(*command) do |i, o, e, t|
+          pid = t.pid
+          stdout = o.read
+          stderr = e.read
+          exit_status = t.value
+        end
+        return exit_status.success?, exit_status.exitstatus, stdout, stderr
+      end
+    end
+    extend ClassMethods
+  end
+end

data/lib/caco/facter.rb

@@ -0,0 +1,41 @@
+module Caco::Facter
+  class KeyNotFoundError < StandardError; end
+
+  module ClassMethods
+    @@fake_data = nil
+
+    def set_fake_data=(data)
+      @@fake_data = data
+    end
+
+    def use_fake(data, &block)
+      old_fake_data = @@fake_data
+      @@fake_data = data
+
+      yield
+
+      @@fake_data = old_fake_data
+    end
+
+    def call(*items)
+      value = json_data(*items).dig(*items)
+      raise KeyNotFoundError.new("#{items.join(":")} not found") unless value
+      value
+    end
+
+    private
+      def json_data(*items)
+        return @@fake_data unless @@fake_data.nil?
+
+        @@parsed_data ||= JSON.parse(external_facter_data)
+      end
+
+      def external_facter_data
+        facter_path = Caco::Executer.(command: "which facter")[:output].chomp!
+        result = Caco::Executer.(command: "#{facter_path} -j")
+        result[:output]
+      end
+  end
+
+  extend ClassMethods
+end

data/lib/caco/file_link.rb

@@ -0,0 +1,36 @@
+class Caco::FileLink < Trailblazer::Operation
+  pass :link_exist?
+  pass :target_exist?
+  step :link_same_target?,
+    Output(:success) => End(:success),
+    Output(:failure) => Track(:success)
+  step :ensure_target!
+  step :create_link!
+
+  def link_exist?(ctx, link:, **)
+    ctx[:link_exist] = !!File.lstat(link) rescue false
+    ctx[:link_realpath] = File.realdirpath(link) rescue false
+    ctx[:link_exist]
+  end
+
+  def target_exist?(ctx, target:, **)
+    ctx[:target_exist] = File.exist?(target)
+    ctx[:target_realpath] = File.realdirpath(target) rescue nil
+    ctx[:target_exist]
+  end
+
+  def link_same_target?(ctx, target_realpath:, link_realpath:, **)
+    ctx[:link_same_target] = (target_realpath == link_realpath)
+  end
+
+  def ensure_target!(ctx, target_exist:, ensure_target: false, **)
+    return false if !target_exist && ensure_target
+    true
+  end
+
+  def create_link!(ctx, target:, link:, link_exist:, link_same_target:, **)
+    ctx[:force] = (link_exist && !link_same_target)
+    FileUtils.ln_s target, link, force: ctx[:force]
+    ctx[:link_created] = true
+  end
+end

data/lib/caco/file_reader.rb

@@ -0,0 +1,24 @@
+class Caco::FileReader < Trailblazer::Operation
+  UseCustomRoot = Class.new(Trailblazer::Activity::Signal)
+
+  step :use_custom_root,
+    Output(UseCustomRoot, :use_custom_root) => Track(:success)
+
+  step :file_exist
+  step :read_file, Output(Trailblazer::Activity::Left, :failure) => End(:success)
+
+  def use_custom_root(ctx, path:, **)
+    return true unless Caco.config.write_files_root
+    ctx[:path] = "#{Caco.config.write_files_root}#{ctx[:path]}"
+
+    UseCustomRoot
+  end
+
+  def file_exist(ctx, path:, **)
+    ctx[:file_exist] = File.exist?(path)
+  end
+
+  def read_file(ctx, path:, **)
+    ctx[:output] = File.read(path)
+  end
+end

data/lib/caco/file_writer.rb

@@ -0,0 +1,57 @@
+class Caco::FileWriter < Trailblazer::Operation
+  SameMD5 = Class.new(Trailblazer::Activity::Signal)
+  DifferentMD5 = Class.new(Trailblazer::Activity::Signal)
+  UseCustomRoot = Class.new(Trailblazer::Activity::Signal)
+
+  step :use_custom_root,
+    Output(UseCustomRoot, :use_custom_root) => Track(:success)
+
+  pass :file_exist
+  step :calculate_md5
+  step :compare_md5,
+    Output(SameMD5, :same_md5) => End(:success),
+    Output(DifferentMD5, :success) => Track(:success)
+  step :mkdir_p
+  step :write_file
+
+  def use_custom_root(ctx, path:, **)
+    return true unless Caco.config.write_files_root
+    unless ctx[:path].start_with?(Caco.config.write_files_root.to_s)
+      ctx[:path] = "#{Caco.config.write_files_root}#{ctx[:path]}"
+    end
+    UseCustomRoot
+  end
+
+  def file_exist(ctx, path:, **)
+    ctx[:file_exist] = File.exist?(path)
+    ctx[:file_created] = !ctx[:file_exist]
+    ctx[:file_exist]
+  end
+
+  def calculate_md5(ctx, path:, file_exist:, content:, **)
+    ctx[:current_md5] = (file_exist ? Digest::MD5.hexdigest(File.read(path)) : "")
+    ctx[:content_md5] = Digest::MD5.hexdigest(content)
+  end
+
+  def compare_md5(ctx, content_md5:, current_md5:, **)
+    different_md5 = (content_md5 != current_md5)
+    ctx[:file_changed] = different_md5 ? true : false
+    different_md5 ? DifferentMD5 : SameMD5
+  end
+
+  def mkdir_p(ctx, path:, **)
+    dirname = File.dirname(path)
+    if Caco.config.write_files
+      FileUtils.mkdir_p(dirname) unless File.exist?(dirname)
+    end
+    true
+  end
+
+  def write_file(ctx, path:, content:, file_exist:, **)
+    if Caco.config.write_files
+      File.write(path, content)
+    end
+    ctx[:file_created] = !file_exist
+    ctx[:file_changed] = true
+  end
+end