From e6b4539b92b6a9168c80402d90088a1295fc00cd Mon Sep 17 00:00:00 2001
From: Alex Polvi <alex@polvi.net>
Date: Mon, 17 Mar 2014 14:53:28 -0700
Subject: [PATCH] fix(etcd.service): refactor to support cloudinit and easy
 c10n slaughtering

---
 .../oem-ami/files/cloud-config.yml            | 25 ++++---------
 .../coreos-base/oem-ami/files/coreos-c10n     | 14 +++++---
 .../oem-ami/files/coreos-setup-environment    |  2 +-
 .../coreos-base/oem-ami/oem-ami-0.0.1.ebuild  |  2 +-
 .../dev-db/etcd/etcd-0.3.0-r6.ebuild          |  1 +
 ...{etcd-0.3.0-r5.ebuild => etcd-9999.ebuild} |  9 +++--
 .../dev-db/etcd/etcd-99999.ebuild             | 35 -------------------
 .../dev-db/etcd/files/etcd.service            | 15 +++-----
 8 files changed, 31 insertions(+), 72 deletions(-)
 create mode 120000 sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r6.ebuild
 rename sdk_container/src/third_party/coreos-overlay/dev-db/etcd/{etcd-0.3.0-r5.ebuild => etcd-9999.ebuild} (82%)
 delete mode 100644 sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-99999.ebuild

diff --git a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/cloud-config.yml b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/cloud-config.yml
index f0137b00ef..d312630dd8 100644
--- a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/cloud-config.yml
+++ b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/cloud-config.yml
@@ -11,7 +11,7 @@ coreos:
           [Service]
           Type=oneshot
           StandardOutput=journal+console
-          ExecStart=/usr/share/oem/usr/bin/ec2-ssh-key
+          ExecStart=/usr/share/oem/bin/ec2-ssh-key
       - name: ec2-cloudinit.service
         runtime: yes
         content: |
@@ -32,21 +32,10 @@ coreos:
           [Service]
           Type=oneshot
           RemainAfterExit=no
-          ExecStart=/usr/share/oem/usr/bin/coreos-c10n
-      - name: etcd.service
-        runtime: yes
-        content: |
-          [Unit]
-          Description=etcd with ec2 auto-bootstrap
-          Requires=ec2-c10n.service
-          After=ec2-c10n.service
+          ExecStart=/usr/share/oem/bin/coreos-c10n
 
-          [Service]
-          User=etcd
-          PermissionsStartOnly=true
-          ExecStart=/usr/share/oem/usr/bin/etcd-bootstrap
-          Restart=always
-          RestartSec=10s
-
-          [Install]
-          WantedBy=multi-user.target
+write_files:
+    - path: /etc/systemd/system/etcd.service.d/10-oem.conf
+      contents: |
+        [Service]
+        Environment=ETCD_PEER_ELECTION_TIMEOUT=1200
diff --git a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-c10n b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-c10n
index 0789f2e523..b97a07e779 100755
--- a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-c10n
+++ b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-c10n
@@ -9,8 +9,8 @@ depwarn() {
 C10N_ENDPOINT=https://core-api.appspot.com/v1/c10n/group
 META_URL="http://169.254.169.254/latest"
 
-BOOTSTRAP_PEERS="/var/run/etcd/bootstrap.config"
-BOOTSTRAP_DISCOVERY="/var/run/etcd/bootstrap.disco"
+ETCD_SERVICE_D="/run/systemd/system/etcd.service.d"
+mkdir -p $ETCD_SERVICE_D
 
 /usr/bin/block-until-url $META_URL
 
@@ -64,7 +64,7 @@ if  head -n 1 $TMP | grep -q '^#!'; then
 		START_FLEET=1
 	fi
 
-# If user-data is a cloud-config, hand it off to the new hotness
+# If user-data is a cloud-config, do not do c10n
 elif head -n 1 $TMP | grep -q '^#cloud-config'; then
 	# If an etcd discovery URL was found, we can assume the new hotness
 	# will handle bootstrapping, allowing us to clear the c10n token
@@ -87,7 +87,8 @@ fi
 # After all that, if a script happened to set the discovery URL, write it to disk for etcd-bootstrap
 if [ -n "${ETCD_DISCOVERY_URL}"  ]; then
 	echo "Using '${ETCD_DISCOVERY_URL}' as etcd discovery url"
-	echo "${ETCD_DISCOVERY_URL}" > $BOOTSTRAP_DISCOVERY
+	echo "[Service]
+Environment=ETCD_DISCOVERY=${ETCD_DISCOVERY}" > ${ETCD_SERVICE_D}/11-c10n.conf
 # ...otherwise, we treat the provided user-data as a legacy c10n URL
 elif [ -n "${C10N_TOKEN}" ]; then
 	depwarn "Bootstrapping etcd with the c10n service is deprecated."
@@ -98,7 +99,10 @@ elif [ -n "${C10N_TOKEN}" ]; then
 	/usr/bin/block-until-url $C10N_ENDPOINT
 
 	IP=$(curl -s $META_URL/meta-data/local-ipv4)
-	curl -s $C10N_ENDPOINT -d "c10n_url=$C10N_TOKEN" -d"ip_list=$IP" > $BOOTSTRAP_PEERS
+	# XXX: change to append a valid ETCD_PEERS to /usr/share/oem/etcd.env
+	ETCD_PEERS=$(curl -s $C10N_ENDPOINT -d "c10n_url=$C10N_TOKEN" -d"ip_list=$IP")
+	echo "[Service]
+Environment=ETCD_PEERS=${ETCD_PEERS}" > ${ETCD_SERVICE_D}/12-c10n.conf
 fi
 
 if [ "$START_FLEET" = "1" ]; then
diff --git a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-setup-environment b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-setup-environment
index dfbae79f9e..c148d23fe3 100755
--- a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-setup-environment
+++ b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/files/coreos-setup-environment
@@ -1,4 +1,4 @@
-#!/bin/bash +x
+#!/bin/bash
 
 ENV=$1
 
diff --git a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/oem-ami-0.0.1.ebuild b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/oem-ami-0.0.1.ebuild
index 1d9ff6def6..66f8a7d16e 100644
--- a/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/oem-ami-0.0.1.ebuild
+++ b/sdk_container/src/third_party/coreos-overlay/coreos-base/oem-ami/oem-ami-0.0.1.ebuild
@@ -16,7 +16,7 @@ IUSE=""
 S="${WORKDIR}"
 
 src_install() {
-	into "/usr"
+	into "/"
 	dobin ${FILESDIR}/ec2-ssh-key
 	dobin ${FILESDIR}/coreos-setup-environment
 	dobin ${FILESDIR}/coreos-c10n
diff --git a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r6.ebuild b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r6.ebuild
new file mode 120000
index 0000000000..f972b32dc1
--- /dev/null
+++ b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r6.ebuild
@@ -0,0 +1 @@
+etcd-9999.ebuild
\ No newline at end of file
diff --git a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r5.ebuild b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-9999.ebuild
similarity index 82%
rename from sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r5.ebuild
rename to sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-9999.ebuild
index 312ed6aa7f..ea8c96f046 100644
--- a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-0.3.0-r5.ebuild
+++ b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-9999.ebuild
@@ -9,16 +9,21 @@ EAPI=4
 CROS_WORKON_PROJECT="coreos/etcd"
 CROS_WORKON_LOCALNAME="etcd"
 CROS_WORKON_REPO="git://github.com"
-CROS_WORKON_COMMIT="040c1f591efa735e24e794d5084ad07782123e3c" # v0.3.0 + v1 API fix
 inherit toolchain-funcs cros-workon systemd
 
+if [[ "${PV}" == 9999 ]]; then
+    KEYWORDS="~amd64"
+else
+    CROS_WORKON_COMMIT="040c1f591efa735e24e794d5084ad07782123e3c"
+    KEYWORDS="amd64"
+fi
+
 DESCRIPTION="etcd"
 HOMEPAGE="https://github.com/coreos/etcd"
 SRC_URI=""
 
 LICENSE="Apache-2.0"
 SLOT="0"
-KEYWORDS="amd64 x86"
 IUSE=""
 
 DEPEND=">=dev-lang/go-1.2"
diff --git a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-99999.ebuild b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-99999.ebuild
deleted file mode 100644
index 57e4e6541a..0000000000
--- a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/etcd-99999.ebuild
+++ /dev/null
@@ -1,35 +0,0 @@
-#
-# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
-# Copyright (c) 2013 CoreOS, Inc.. All rights reserved.
-# Distributed under the terms of the GNU General Public License v2
-# $Header:$
-#
-
-EAPI=4
-CROS_WORKON_PROJECT="coreos/etcd"
-CROS_WORKON_LOCALNAME="etcd"
-CROS_WORKON_REPO="git://github.com"
-inherit toolchain-funcs cros-workon systemd
-
-DESCRIPTION="etcd"
-HOMEPAGE="https://github.com/coreos/etcd"
-SRC_URI=""
-
-LICENSE="Apache-2.0"
-SLOT="0"
-KEYWORDS="~amd64"
-IUSE=""
-
-DEPEND=">=dev-lang/go-1.2"
-
-src_compile() {
-	./build
-}
-
-src_install() {
-	dobin ${S}/bin/${PN}
-
-	systemd_dounit "${FILESDIR}"/${PN}.service
-	systemd_enable_service multi-user.target ${PN}.service
-	systemd_dotmpfilesd "${FILESDIR}"/${PN}.conf
-}
diff --git a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/files/etcd.service b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/files/etcd.service
index 30b3999314..1f903ecd1e 100644
--- a/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/files/etcd.service
+++ b/sdk_container/src/third_party/coreos-overlay/dev-db/etcd/files/etcd.service
@@ -1,19 +1,14 @@
 [Unit]
 Description=etcd
-RequiresMountsFor=/usr/share/oem
-# oem will setup etcd for auto bootstrapping
-ConditionPathExists=!/usr/share/oem/cloud-config.yml
+# this will setup cloudinit and user envs
+Wants=coreos-cloudinit-oem.service
+After=coreos-cloudinit-oem.service
 
 [Service]
 User=etcd
 PermissionsStartOnly=true
-# Default, this will be overwritten with a cloud-config
-ExecStart=/usr/bin/etcd -f -data-dir /var/lib/etcd -bind-addr 0.0.0.0 -n default
-
-Restart=always
-# Set a longish timeout in case this machine isn't behaving
-# nicely and bothering the rest of the cluster
-RestartSec=10s
+Environment=ETCD_DATA_DIR=/var/lib/etcd ETCD_NAME=default
+ExecStart=/usr/bin/etcd 
 
 [Install]
 WantedBy=multi-user.target