From c79dfeb6eccd9d7da639276657419ca917407c9b Mon Sep 17 00:00:00 2001
From: Denys Havrysh <denis.gavrysh@zoomdata.com>
Date: Tue, 13 Sep 2016 16:42:36 +0300
Subject: [PATCH] Make setting `use_upstream_repo` install PostgreSQL 9.5 by
 default

---
 postgres/codenamemap.yaml |  3 ++-
 postgres/osmap.yaml       |  3 ++-
 postgres/upstream.sls     | 13 +++++++------
 3 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/postgres/codenamemap.yaml b/postgres/codenamemap.yaml
index 06ed39d..fbe9e06 100644
--- a/postgres/codenamemap.yaml
+++ b/postgres/codenamemap.yaml
@@ -5,7 +5,8 @@
   {% endif %}
 
 {{ name }}:
-  pkg_repo: deb http://apt.postgresql.org/pub/repos/apt/ {{ name }}-pgdg main
+  pkg_repo_humanname: PostgreSQL Official Repository
+  pkg_repo: deb http://apt.postgresql.org/pub/repos/apt/ {{ name }}-pgdg main {{ version }}
   pkg: postgresql-{{ version }}
   pkg_client: postgresql-client-{{ version }}
   conf_dir: /etc/postgresql/{{ version }}/main
diff --git a/postgres/osmap.yaml b/postgres/osmap.yaml
index 9a72ac5..77c9083 100644
--- a/postgres/osmap.yaml
+++ b/postgres/osmap.yaml
@@ -24,8 +24,9 @@ RedHat:
   {% set version = salt['pillar.get']('postgres:version', '9.5') %}
   {% set release = version|replace('.', '') %}
 
-  version: {{ version }}
   pkg_repo: pgdg{{ release }}
+  pkg_repo_humanname: PostgreSQL {{ version }} $releasever - $basearch
+  pkg_repo_url: https://download.postgresql.org/pub/repos/yum/{{ version }}/redhat/rhel-$releasever-$basearch
   pkg: postgresql{{ release }}-server
   pkg_client: postgresql{{ release }}
   conf_dir: /var/lib/pgsql/{{ version }}/data
diff --git a/postgres/upstream.sls b/postgres/upstream.sls
index b303fff..5ebb1fd 100644
--- a/postgres/upstream.sls
+++ b/postgres/upstream.sls
@@ -1,11 +1,11 @@
 {%- from "postgres/map.jinja" import postgres with context %}
 
-{%- if grains['os_family'] == 'Debian' %}
+{%- if grains['os_family'] == 'Debian' -%}
 
 install-postgresql-repo:
   pkgrepo.managed:
-    - humanname: PostgreSQL Official Repository
-    - name: {{ postgres.pkg_repo }} {{ postgres.version }}
+    - humanname: {{ postgres.pkg_repo_humanname }}
+    - name: {{ postgres.pkg_repo }}
     - keyid: B97B0AFCAA1A47F044F244A07FCC7D46ACCC4CF8
     - keyserver: keyserver.ubuntu.com
     - file: {{ postgres.pkg_repo_file }}
@@ -21,12 +21,13 @@ install-postgresql-repo:
     - source_hash: md5=78b5db170d33f80ad5a47863a7476b22
   pkgrepo.managed:
     - name: {{ postgres.pkg_repo }}
-    - order: 1
-    - humanname: PostgreSQL {{ postgres.version }} $releasever - $basearch
-    - baseurl: https://download.postgresql.org/pub/repos/yum/{{ postgres.version }}/redhat/rhel-$releasever-$basearch
+    - humanname: {{ postgres.pkg_repo_humanname }}
+    - baseurl: {{ postgres.pkg_repo_url }}
     - gpgcheck: 1
     - gpgkey: file:///etc/pki/rpm-gpg/RPM-GPG-KEY-PGDG
     - require:
       - file: install-postgresql-repo
+    - require_in:
+      - pkg: postgresql-installed
 
 {%- endif %}