server_maint 0.0.6 → 0.0.7

data/lib/cookbooks/postgresql/definitions/pg_database_extensions.rb

@@ -0,0 +1,67 @@
+define :pg_database_extensions, :action => :create do
+
+  dbname = params[:name]
+  languages = [params[:languages] || []].flatten # Allow single value or array of values
+  extensions = [params[:extensions] || []].flatten
+  postgis = params[:postgis]
+
+  postgresql_version = node["postgresql"]["version"]
+  postgis_version = node["postgis"]["version"]
+
+  case params[:action]
+  when :create
+
+    languages.each do |language|
+      execute "createlang #{language} #{dbname}" do
+        user "postgres"
+        not_if "psql -c 'SELECT lanname FROM pg_catalog.pg_language' #{dbname} | grep '^ #{language}$'", :user => "postgres"
+      end
+    end
+
+    extensions.each do |extension|
+      execute "psql -c 'CREATE EXTENSION IF NOT EXISTS #{extension}' #{dbname}" do
+        user "postgres"
+      end
+    end
+
+    if postgis
+      include_recipe 'postgresql::postgis'
+
+      execute "psql -d #{dbname} -f /usr/share/postgresql/#{postgresql_version}/contrib/postgis-#{postgis_version}/postgis.sql" do
+        user "postgres"
+        not_if "psql -c \"SELECT proname FROM pg_catalog.pg_proc WHERE proname = 'st_area'\" #{dbname} | grep 'st_area$'", :user => "postgres"
+      end
+
+      execute "psql -d #{dbname} -f /usr/share/postgresql/#{postgresql_version}/contrib/postgis-#{postgis_version}/spatial_ref_sys.sql" do
+        user "postgres"
+        only_if "psql -c 'SELECT count(1) FROM spatial_ref_sys' #{dbname} | grep '0$'", :user => "postgres"
+      end
+
+      [:geometry_columns, :geography_columns, :spatial_ref_sys].each do |table|
+        execute "psql -c 'GRANT ALL ON #{table} TO PUBLIC' #{dbname}" do
+          user "postgres"
+        end
+      end
+    end
+
+  when :drop
+
+    languages.each do |language|
+      execute "droplang #{language} #{dbname}" do
+        user "postgres"
+        only_if "psql -c 'SELECT lanname FROM pg_catalog.pg_language' #{dbname} | grep '^ #{language}$'", :user => "postgres"
+      end
+    end
+
+    extensions.each do |extension|
+      execute "psql -c 'DROP EXTENSION IF EXISTS #{extension}' #{dbname}" do
+        user "postgres"
+      end
+    end
+
+    if postgis
+      Chef::Log.warn("Postgis support dropping isn't supported")
+    end
+
+  end
+end

data/lib/cookbooks/postgresql/definitions/pg_user.rb

@@ -0,0 +1,45 @@
+define :pg_user, :action => :create do
+  case params[:action]
+  when :create
+    privileges = {
+      :superuser => false,
+      :createdb => false,
+      :login => true
+    }
+    privileges.merge! params[:privileges] if params[:privileges]
+
+    sql = [params[:name]]
+
+    sql.push privileges.to_a.map! { |p,b| (b ? '' : 'NO') + p.to_s.upcase }.join ' '
+
+    if params[:encrypted_password]
+      sql.push "ENCRYPTED PASSWORD '#{params[:encrypted_password]}'"
+    elsif params[:password]
+      sql.push "PASSWORD '#{params[:password]}'"
+    end
+
+    sql = sql.join ' '
+
+    exists = ["psql -c \"SELECT usename FROM pg_user WHERE usename='#{params[:name]}'\""]
+    exists.push "| grep #{params[:name]}"
+    exists = exists.join ' '
+
+    execute "altering pg user #{params[:name]}" do
+      user "postgres"
+      command "psql -c \"ALTER ROLE #{sql}\""
+      only_if exists, :user => "postgres"
+    end
+
+    execute "creating pg user #{params[:name]}" do
+      user "postgres"
+      command "psql -c \"CREATE ROLE #{sql}\""
+      not_if exists, :user => "postgres"
+    end
+
+  when :drop
+    execute "dropping pg user #{params[:name]}" do
+      user "postgres"
+      command "psql -c \"DROP ROLE IF EXISTS #{params[:name]}\""
+    end
+  end
+end

data/lib/cookbooks/postgresql/files/default/pgdg.pref

@@ -0,0 +1,3 @@
+Package: *
+Pin: release o=apt.postgresql.org
+Pin-Priority: 500

data/lib/cookbooks/postgresql/metadata.rb

@@ -0,0 +1,22 @@
+name             "postgresql"
+maintainer       "Phil Cohen"
+maintainer_email "github@phlippers.net"
+license          "MIT"
+description      "Installs PostgreSQL, The world's most advanced open source database."
+long_description IO.read(File.join(File.dirname(__FILE__), "README.md"))
+version          "0.9.0"
+
+recipe "postgresql",           "Set up the apt repository and install dependent packages"
+recipe "postgresql::client",   "Front-end programs for PostgreSQL 9.x"
+recipe "postgresql::server",   "Object-relational SQL database, version 9.x server"
+recipe "postgresql::contrib",  "Additional facilities for PostgreSQL"
+recipe "postgresql::dbg",      "Debug symbols for the server daemon"
+recipe "postgresql::doc",      "Documentation for the PostgreSQL database management system"
+recipe "postgresql::libpq",    "PostgreSQL C client library and header files for libpq5 (PostgreSQL library)"
+recipe "postgresql::postgis",  "Geographic objects support for PostgreSQL 9.x"
+
+%w[ubuntu debian].each do |os|
+  supports os
+end
+
+depends "apt"

data/lib/cookbooks/postgresql/recipes/client.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: client
+#
+
+include_recipe "postgresql"
+
+package "postgresql-client-#{node["postgresql"]["version"]}"

data/lib/cookbooks/postgresql/recipes/contrib.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: contrib
+#
+
+include_recipe "postgresql"
+
+package "postgresql-contrib-#{node["postgresql"]["version"]}"

data/lib/cookbooks/postgresql/recipes/dbg.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: dbg
+#
+
+include_recipe "postgresql"
+
+package "postgresql-#{node["postgresql"]["version"]}-dbg"

data/lib/cookbooks/postgresql/recipes/default.rb

@@ -0,0 +1,50 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: default
+#
+
+case node["platform"]
+when "ubuntu"
+
+  apt_repository "pitti-postgresql" do
+    uri "http://ppa.launchpad.net/pitti/postgresql/ubuntu"
+    distribution node["lsb"]["codename"]
+    components ["main"]
+    keyserver "keyserver.ubuntu.com"
+    key "8683D8A2"
+    action :add
+  end
+
+  # install common files
+  package "postgresql-common"
+
+when "debian"
+
+  # backports for initial support
+  apt_repository "debian-backports" do
+    uri "http://backports.debian.org/debian-backports"
+    distribution "#{node["lsb"]["codename"]}-backports"
+    components ["main"]
+    action :add
+  end
+
+  cookbook_file "/etc/apt/preferences.d/pgdg.pref" do
+    source "pgdg.pref"
+  end
+
+  # backports support for debian
+  %w[libpq5 postgresql-common].each do |pkg|
+    package pkg do
+      options "-t #{node["lsb"]["codename"]}-backports"
+    end
+  end
+
+  # use `apt.postgresql.org` for primary package installation support
+  apt_repository "apt.postgresql.org" do
+    uri "http://apt.postgresql.org/pub/repos/apt"
+    distribution "#{node["lsb"]["codename"]}-pgdg"
+    components ["main"]
+    key "http://apt.postgresql.org/pub/repos/apt/ACCC4CF8.asc"
+    action :add
+  end
+end

data/lib/cookbooks/postgresql/recipes/doc.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: doc
+#
+
+include_recipe "postgresql"
+
+package "postgresql-doc-#{node["postgresql"]["version"]}"

data/lib/cookbooks/postgresql/recipes/libpq.rb

@@ -0,0 +1,9 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: libpq
+#
+
+include_recipe "postgresql"
+
+package "libpq5"
+package "libpq-dev"

data/lib/cookbooks/postgresql/recipes/postgis.rb

@@ -0,0 +1,8 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: postgis
+#
+
+include_recipe "postgresql"
+
+package "postgresql-#{node["postgresql"]["version"]}-postgis"

data/lib/cookbooks/postgresql/recipes/server.rb

@@ -0,0 +1,118 @@
+#
+# Cookbook Name:: postgresql
+# Recipe:: server
+#
+
+include_recipe "postgresql"
+
+pg_version = node["postgresql"]["version"]
+
+# install the package
+package "postgresql-#{pg_version}"
+
+
+# ensure data directory exists
+directory node["postgresql"]["data_directory"] do
+  owner  "postgres"
+  group  "postgres"
+  mode   "0700"
+  not_if "test -f #{node["postgresql"]["data_directory"]}/PG_VERSION"
+end
+
+# initialize the data directory if necessary
+bash "postgresql initdb" do
+  user "postgres"
+  code <<-EOC
+  /usr/lib/postgresql/#{pg_version}/bin/initdb \
+    #{node["postgresql"]["initdb_options"]} \
+    -U postgres \
+    -D #{node["postgresql"]["data_directory"]}
+  EOC
+  creates "#{node["postgresql"]["data_directory"]}/PG_VERSION"
+end
+
+# environment
+template "/etc/postgresql/#{pg_version}/main/environment" do
+  source "environment.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0644"
+  notifies :restart, "service[postgresql]"
+end
+
+# pg_ctl
+template "/etc/postgresql/#{pg_version}/main/pg_ctl.conf" do
+  source "pg_ctl.conf.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0644"
+  notifies :restart, "service[postgresql]"
+end
+
+# pg_hba
+template node["postgresql"]["hba_file"] do
+  source "pg_hba.conf.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0640"
+  notifies :restart, "service[postgresql]"
+end
+
+# pg_ident
+template node["postgresql"]["ident_file"] do
+  source "pg_ident.conf.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0640"
+  notifies :restart, "service[postgresql]"
+end
+
+# postgresql
+pg_template_source = node["postgresql"]["conf"].any? ? "custom" : "standard"
+template "/etc/postgresql/#{pg_version}/main/postgresql.conf" do
+  source "postgresql.conf.#{pg_template_source}.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0644"
+  variables(:configuration => node["postgresql"]["conf"])
+  notifies :restart, "service[postgresql]"
+end
+
+# start
+template "/etc/postgresql/#{pg_version}/main/start.conf" do
+  source "start.conf.erb"
+  owner  "postgres"
+  group  "postgres"
+  mode   "0644"
+  notifies :restart, "service[postgresql]", :immediately
+end
+
+# setup users
+node["postgresql"]["users"].each do |user|
+  pg_user user["username"] do
+    privileges :superuser => user["superuser"], :createdb => user["createdb"], :login => user["login"]
+    password user["password"]
+  end
+end
+
+# setup databases
+node["postgresql"]["databases"].each do |database|
+  pg_database database["name"] do
+    owner database["owner"]
+    encoding database["encoding"]
+    template database["template"]
+    locale database["locale"]
+  end
+
+  pg_database_extensions database["name"] do
+    extensions database["extensions"]
+    languages database["languages"]
+    postgis database["postgis"]
+  end
+end
+
+# define the service
+service "postgresql" do
+  supports :restart => true
+  action [:enable, :start]
+end

data/lib/cookbooks/postgresql/templates/default/environment.erb

@@ -0,0 +1,11 @@
+# environment variables for postmaster process
+# This file has the same syntax as postgresql.conf:
+#  VARIABLE = simple_value
+#  VARIABLE2 = 'any value!'
+# I. e. you need to enclose any value which does not only consist of letters,
+# numbers, and '-', '_', '.' in single quotes. Shell commands are not
+# evaluated.
+
+<% node["postgresql"]["environment_variables"].each_pair do |k, v| %>
+<%= "#{k} = #{v}" %>
+<% end %>

data/lib/cookbooks/postgresql/templates/default/pg_ctl.conf.erb

@@ -0,0 +1,5 @@
+# Automatic pg_ctl configuration
+# This configuration file contains cluster specific options to be passed to
+# pg_ctl(1).
+
+pg_ctl_options = '<%= node["postgresql"]["pg_ctl_options"] %>'

data/lib/cookbooks/postgresql/templates/default/pg_hba.conf.erb

@@ -0,0 +1,100 @@
+# PostgreSQL Client Authentication Configuration File
+# ===================================================
+#
+# Refer to the "Client Authentication" section in the PostgreSQL
+# documentation for a complete description of this file.  A short
+# synopsis follows.
+#
+# This file controls: which hosts are allowed to connect, how clients
+# are authenticated, which PostgreSQL user names they can use, which
+# databases they can access.  Records take one of these forms:
+#
+# local      DATABASE  USER  METHOD  [OPTIONS]
+# host       DATABASE  USER  ADDRESS  METHOD  [OPTIONS]
+# hostssl    DATABASE  USER  ADDRESS  METHOD  [OPTIONS]
+# hostnossl  DATABASE  USER  ADDRESS  METHOD  [OPTIONS]
+#
+# (The uppercase items must be replaced by actual values.)
+#
+# The first field is the connection type: "local" is a Unix-domain
+# socket, "host" is either a plain or SSL-encrypted TCP/IP socket,
+# "hostssl" is an SSL-encrypted TCP/IP socket, and "hostnossl" is a
+# plain TCP/IP socket.
+#
+# DATABASE can be "all", "sameuser", "samerole", "replication", a
+# database name, or a comma-separated list thereof. The "all"
+# keyword does not match "replication". Access to replication
+# must be enabled in a separate record (see example below).
+#
+# USER can be "all", a user name, a group name prefixed with "+", or a
+# comma-separated list thereof.  In both the DATABASE and USER fields
+# you can also write a file name prefixed with "@" to include names
+# from a separate file.
+#
+# ADDRESS specifies the set of hosts the record matches.  It can be a
+# host name, or it is made up of an IP address and a CIDR mask that is
+# an integer (between 0 and 32 (IPv4) or 128 (IPv6) inclusive) that
+# specifies the number of significant bits in the mask.  A host name
+# that starts with a dot (.) matches a suffix of the actual host name.
+# Alternatively, you can write an IP address and netmask in separate
+# columns to specify the set of hosts.  Instead of a CIDR-address, you
+# can write "samehost" to match any of the server's own IP addresses,
+# or "samenet" to match any address in any subnet that the server is
+# directly connected to.
+#
+# METHOD can be "trust", "reject", "md5", "password", "gss", "sspi",
+# "krb5", "ident", "peer", "pam", "ldap", "radius" or "cert".  Note that
+# "password" sends passwords in clear text; "md5" is preferred since
+# it sends encrypted passwords.
+#
+# OPTIONS are a set of options for the authentication in the format
+# NAME=VALUE.  The available options depend on the different
+# authentication methods -- refer to the "Client Authentication"
+# section in the documentation for a list of which options are
+# available for which authentication methods.
+#
+# Database and user names containing spaces, commas, quotes and other
+# special characters must be quoted.  Quoting one of the keywords
+# "all", "sameuser", "samerole" or "replication" makes the name lose
+# its special character, and just match a database or username with
+# that name.
+#
+# This file is read on server startup and when the postmaster receives
+# a SIGHUP signal.  If you edit the file on a running system, you have
+# to SIGHUP the postmaster for the changes to take effect.  You can
+# use "pg_ctl reload" to do that.
+
+# Put your actual configuration here
+# ----------------------------------
+#
+# If you want to allow non-local connections, you need to add more
+# "host" records.  In that case you will also need to make PostgreSQL
+# listen on a non-local interface via the listen_addresses
+# configuration parameter, or via the -i or -h command line switches.
+
+<% node["postgresql"]["pg_hba"].each do |hba| %>
+<%= hba['type'] %> <%= hba['db'] %> <%= hba['user'] %> <%= hba['addr'] %> <%= hba['method'] %>
+<% end %>
+
+# DO NOT DISABLE!
+# If you change this first entry you will need to make sure that the
+# database superuser can access the database using some other method.
+# Noninteractive access to all databases is required during automatic
+# maintenance (custom daily cronjobs, replication, and similar tasks).
+#
+# TYPE  DATABASE        USER            ADDRESS                 METHOD
+<% if node[:postgresql][:pg_hba_defaults] %>
+# Database administrative login by Unix domain socket
+local   all             postgres                                peer
+# "local" is for Unix domain socket connections only
+local   all             all                                     peer
+# IPv4 local connections:
+host    all             all             127.0.0.1/32            md5
+# IPv6 local connections:
+host    all             all             ::1/128                 md5
+# Allow replication connections from localhost, by a user with the
+# replication privilege.
+#local   replication     postgres                                peer
+#host    replication     postgres        127.0.0.1/32            md5
+#host    replication     postgres        ::1/128                 md5
+<% end %>