cupper 0.0.1 → 0.1.0

data/lib/cupper/plugins/ohai/init_system.rb

@@ -0,0 +1,14 @@
+
+Ohai.plugin(:InitSystem) do
+  provides 'init_system'
+
+  def from_cmd(cmd)
+    so = shell_out(cmd)
+    so.stdout.strip
+  end
+
+  collect_data(:linux) do
+    init_system Mash.new
+    init_system["init"] = from_cmd("cat /proc/1/comm")
+  end
+end

data/lib/cupper/plugins/ohai/pacman.rb

@@ -0,0 +1,31 @@
+
+Ohai.plugin(:Pacman) do
+  provides 'pacman'
+
+  def from_cmd(cmd)
+    so = shell_out(cmd)
+    so.stdout.lines
+  end
+
+  def extract_dependencies(pkg)
+    pkg_infos = from_cmd("pacman -Qi #{pkg}")
+    infos = pkg_infos.stdout.lines
+    infos.each do |info|
+      info.slice! /Depends On\s+:/ if info.include? "Depends On"
+    end
+    info
+  end
+
+  collect_data(:default) do
+    pacman Mash.new
+    pkgs = from_cmd('pacman -Q')
+
+    pkgs.each do |pkg|
+      name, version = pkg.split
+      pacman[name] = {
+        "version" => version,
+        "dependecies" => info,
+      }
+    end
+  end
+end

data/lib/cupper/plugins/ohai/pkg_deps.rb

@@ -0,0 +1,30 @@
+Ohai.plugin(:PkgDeps) do
+  provides 'pkg_deps'
+  depends 'platform_family'
+
+  def from_cmd(cmd)
+    so = shell_out(cmd)
+    so.stdout.lines
+  end
+
+  def all_packages
+    if %w{debian}.include? platform_family
+      from_cmd("dpkg-query -W")
+    end
+  end
+
+  def extract_dependecies(pkg)
+    pkg_infos = from_cmd("apt-cache showpkg #{pkg}")
+    pkg_infos.keep_if { |item| item.strip!.match(/^Depends:/)}
+    pkg_infos.each { |item| item.slice!("Depends: ") }
+  end
+
+  collect_data(:linux) do
+    pkg_deps Mash.new
+    if %w{debian}.include? platform_family
+      all_packages.each do |pkg|
+        pkg_deps[pkg.split.first] = extract_dependecies(pkg.split.first)
+      end
+    end
+  end
+end

data/lib/cupper/plugins/ohai/pkg_manager.rb

@@ -0,0 +1,31 @@
+
+Ohai.plugin(:PkgManager) do
+  provides 'pkg_manager'
+
+  def return_version(cmd)
+    out = shell_out("#{cmd} || true")
+    value out.stdout.strip
+    value.match(/(\d+\.)(\d+\.)(\d+)/)
+  end
+
+  collect_data(:linux) do
+    pkg_manager Mash.new
+
+    # HACK: it may be a better way to collect the default pkg manages
+    DEFAULT_PKG_MANAGES = [
+      'pacman',
+      'dpkg',
+      'apt',
+      'gem',
+      'pip',
+      'pip3',
+    ]
+
+    DEFAULT_PKG_MANAGES.each do |manager|
+      pkg_manager[manager] = {
+        "version" => return_version("#{manager} --version")
+      }
+    end
+
+  end
+end

data/lib/cupper/plugins/ohai/services.rb

@@ -0,0 +1,20 @@
+Ohai.plugin(:Services) do
+  provides 'services'
+
+  def from_cmd(cmd)
+    so = shell_out(cmd)
+    so.stdout.lines
+  end
+
+  collect_data(:default) do
+    services Mash.new
+    srvs from_cmd('systemctl list-units | grep loaded | grep active | grep running')
+
+    srvs.each do |srv|
+      name = srv.split.first
+      services[name] = {
+        "action" => 'restart',
+      }
+    end
+  end
+end

data/lib/cupper/project.rb

@@ -0,0 +1,40 @@
+# This module prepare the Cupper project with the
+#   defaults files and dirs. All the files created are
+#   just samples and must to be changed by the user
+require 'cupper/entity'
+
+module Cupper
+  class Structure
+    include Entity
+    def initialize(name,dest_path, erb_file = nil, type = nil)
+      super(name, dest_path, erb_file, type)
+    end
+  end
+
+  class Project
+    attr_reader :name
+    attr_reader :dir
+
+    def initialize(project_name, directory = nil)
+      @name = project_name
+      @dir = directory.nil? ? Dir.getwd : directory
+      @subdirs = [
+        'cookbooks'
+      ]
+      @files = [
+        'CupperFile'
+      ]
+    end
+
+    def create
+      # Root project directory
+      struct = Structure.new(@name, @dir, nil, Entity::DIR)
+      struct.create
+
+      @subdirs.zip(@files).each do |dir, file|
+        Structure.new(dir, "#{@dir}/#{@name}", nil, Entity::DIR).create
+        Structure.new(file, "#{@dir}/#{@name}", file).create
+      end
+    end
+  end
+end

data/lib/cupper/recipe.rb

@@ -0,0 +1,240 @@
+
+module Cupper
+  # Represents the recipe of the cookbook
+  class Recipe
+    include Entity
+    def initialize(dest_path, collector, erb_file = nil, recipe_name = 'default', recipe_deps = nil)
+      @recipe_deps  = recipe_deps
+      @packages     = Array.new
+      @services     = Array.new
+      @templates    = Array.new
+      @users        = Array.new
+      @groups       = Array.new
+      @execute      = Array.new
+      @links        = Array.new
+      @directories  = Array.new
+      @files        = Array.new
+      @collector    = collector
+      super(recipe_name, dest_path, erb_file, nil, '.rb')
+    end
+
+    def create
+      @sources_list  = expand_sources_list(@collector.extract 'files')
+      @packages = expand_packages(@collector.extract 'packages')
+      @services = expand_services(@collector.extract 'services')
+      @users    = expand_users(@collector.extract 'users')
+      @groups   = expand_groups(@collector.extract 'groups')
+      @links    = expand_links(@collector.extract 'files')
+      @files    = expand_files(@collector.extract 'files')
+      super
+    end
+
+    def expand_sources_list(files)
+      att = Array.new
+      files.each do |attr|
+        # TODO: Doesn't works for arch, this should be a plugin responsability
+        if attr[0].include?("/etc/apt/sources.list") and (convert_mode(attr[1]['mode']) != 'Unknown') and text_type?(attr)
+          path = attr[0]
+          group = attr[1]['group']
+          mode = attr[1]['mode']
+          owner = attr[1]['owner']
+          att.push(new_file(group, mode, owner, path))
+        end
+      end
+      att
+    end
+
+    def expand_packages(packages)
+      att = Array.new
+      packages.each do |attr|
+        pkg = attr[0]
+        version = attr[1]['version']
+
+        att.push(new_package(pkg,version))
+      end
+      att
+    end
+
+    def link_type?(file)
+      (file[1]['type'].split.first(2).join(' ').match('symbolic link'))
+    end
+
+    def dir_type?(file)
+      file[1]['type'].match('directory')
+    end
+
+    def text_type?(file)
+      file[1]['type'].match('text') or file[1]['type'].match('ASCII')
+    end
+
+    # TODO: this should be on ohai plugin
+    def convert_mode(mode)
+      return 'ERROR' if not mode
+      result = case mode.split('').last(9).join # Common modes
+               when 'rwxrwxrwx' then '777'
+               when 'rwxr-xr-x' then '755'
+               when 'rwx------' then '700'
+               when 'rw-rw-rw-' then '666'
+               when 'rw-r--r--' then '644'
+               when 'rw-------' then '600'
+               else 'Unknown'
+               end
+      result
+    end
+
+    def expand_services(services)
+      att = Array.new
+      services.each do |attr|
+        srv = attr[0]
+        action = attr[1]['action']
+
+        att.push(new_service(srv,action))
+      end
+      att
+    end
+
+    def expand_links(links)
+      att = Array.new
+      links.each do |attr|
+        if link_type?(attr)
+          target = attr[0]
+          to = attr[1]['type'].split.last(1).join
+          group = attr[1]['group']
+          mode = attr[1]['mode']
+          owner = attr[1]['owner']
+
+          att.push(new_link(group, mode, owner, target, to))
+        end
+      end
+      att
+    end
+
+    def expand_users(users)
+      att = Array.new
+      users.each do |attr|
+        usr = attr[0]
+        uid = attr[1]['uid']
+        gid = attr[1]['gid']
+        dir = attr[1]['dir']
+        shell = attr[1]['shell']
+
+        att.push(new_user(usr, uid, gid, dir, shell))
+      end
+      att
+    end
+
+    def expand_groups(groups)
+      att = Array.new
+      groups.each do |attr|
+        grp = attr[0]
+        gid = attr[1]['gid']
+        members = attr[1]['members']
+
+        att.push(new_group(grp, gid, members))
+      end
+      att
+    end
+
+    def expand_files(files)
+      att = Array.new
+      files.each do |attr|
+        if text_type?(attr) and !(attr[1]['related'].nil?)
+          path = attr[0]
+          group = attr[1]['group']
+          mode = attr[1]['mode']
+          owner = attr[1]['owner']
+
+          att.push(new_file(group, mode, owner, path))
+        end
+      end
+      att
+    end
+
+    # Every attribute object is created dynamic
+    def new_package(name, version)
+      package = Attribute.new
+      class << package
+        attr_accessor :name
+        attr_accessor :version
+      end
+      package.name = name
+      package.version = version
+      package
+    end
+
+    def new_service(name, action)
+      service = Attribute.new
+      class << service
+        attr_accessor :name
+        attr_accessor :action
+      end
+      service.name = name
+      service.action = action
+      service
+    end
+
+    def new_link(group, mode, owner, target_file, to)
+      link = Attribute.new
+      class << link
+        attr_accessor :group
+        attr_accessor :mode
+        attr_accessor :owner
+        attr_accessor :target_file
+        attr_accessor :to
+      end
+      link.group        = group
+      link.mode         = convert_mode(mode)
+      link.owner        = owner
+      link.target_file  = target_file
+      link.to           = to
+      link
+    end
+
+    def new_user(name, uid, gid, dir, shell)
+      user = Attribute.new
+      class << user
+        attr_accessor :name
+        attr_accessor :uid
+        attr_accessor :gid
+        attr_accessor :dir
+        attr_accessor :shell
+      end
+      user.name         = name
+      user.uid          = uid 
+      user.gid          = gid
+      user.dir          = dir
+      user.shell        = shell 
+      user
+    end
+
+    def new_group(name, gid, members)
+      group = Attribute.new
+      class << group
+        attr_accessor :name
+        attr_accessor :gid
+        attr_accessor :members
+      end
+      group.name         = name
+      group.gid          = gid
+      group.members      = members
+      group
+    end
+    
+    def new_file(group, mode, owner, path, source='')
+      file = Attribute.new
+      class << file
+        attr_accessor :path
+        attr_accessor :source
+        attr_accessor :group
+        attr_accessor :mode
+        attr_accessor :owner
+      end
+      file.path         = path
+      file.source       = source
+      file.group        = group
+      file.mode         = convert_mode(mode)
+      file.owner        = owner
+      file
+    end
+  end
+end

data/lib/cupper/templates/CupperFile.erb

@@ -0,0 +1 @@
+# Cupper config file

data/lib/cupper/templates/_cookbook_file.erb

@@ -0,0 +1,9 @@
+# Cookbook_file Session
+
+<% for @cbf in @files %>
+cookbook_file '<%= @cbf.path %>'
+  owner '<%= @cbf.owner %>'
+  group '<%= @cbf.group %>'
+  mode '<%= @cbf.mode %>'
+end
+<% end %>

data/lib/cupper/templates/_groups.erb

@@ -0,0 +1,10 @@
+# Group session
+
+<% for @group in @groups %>
+group '<%= @group.name%>' do
+  action :create
+  gid  '<%= @group.gid%>'
+  members <%= @group.members%>
+  append true
+end
+<% end %>

data/lib/cupper/templates/_links.erb

@@ -0,0 +1,10 @@
+# Links Session
+
+<% for @link in @links %>
+link '<%= @link.target_file %>' do
+  group '<%= @link.group  %>'
+  mode '<%= @link.mode %>'
+  owner '<%= @link.owner %>'
+  to '<%= @link.to %>'
+end
+<% end %>