From bcf8cda4be8ef21860960d1cff749fe853fc8937 Mon Sep 17 00:00:00 2001
From: Timothy Redaelli <timothy.redaelli@gmail.com>
Date: Tue, 30 Jun 2015 17:02:29 +0200
Subject: [PATCH] Add a command line option to specify custom mirror(s)

---
 vps2arch | 42 ++++++++++++++++++++++++++++++++++++------
 1 file changed, 36 insertions(+), 6 deletions(-)

diff --git a/vps2arch b/vps2arch
index a5d07f3..4224852 100755
--- a/vps2arch
+++ b/vps2arch
@@ -13,10 +13,6 @@
 # GNU General Public License at <http://www.gnu.org/licenses/> for
 # more details.
 
-: ${ARCH_MIRRORS="http://mirror.rackspace.com/archlinux http://dgix.ru/mirrors/archlinux"}
-
-# You don't need to edit anything below this comment
-
 set -e
 
 # Gathering informations about actual environment.
@@ -31,10 +27,12 @@ fi
 
 cpu_type=$(uname -m)
 
+is_openvz() { [ -d /proc/vz -a ! -d /proc/bc ]; }
+
 download() {
 	local path="$1" x=
 	shift
-	for x in $ARCH_MIRRORS; do
+	for x in $mirrors; do
 		_download "$x/$path" && return 0
 	done
 	return 1
@@ -65,7 +63,7 @@ chroot_exec() {
 
 configure_chroot() {
 	local m
-	for m in $ARCH_MIRRORS; do
+	for m in $mirrors; do
 		echo 'Server = '"$m"'/$repo/os/$arch'
 	done >> "/root.$cpu_type/etc/pacman.d/mirrorlist"
 	# Install and initialize haveged if needed
@@ -194,6 +192,38 @@ finalize() {
 	EOF
 }
 
+mirrors=
+
+while getopts ":m:h" opt; do
+	case $opt in
+	m)
+		mirrors="${mirrors:+$mirrors }$OPTARG"
+		;;
+	h)
+		cat <<-EOF
+			usage: ${0##*/} [options]
+
+			  Options:
+			    -m mirror                    Use the provided mirror (you can specify this option more than once).
+
+			    -h                           Print this help message
+		EOF
+		exit 0
+		;;
+	:)
+		printf "%s: option requires an argument -- '%s'\n" "${0##*/}" "$OPTARG" >&2
+		exit 1
+		;;
+	?)
+		printf "%s: invalid option -- '%s'\n" "${0##*/}" "$OPTARG" >&2
+		exit 1
+		;;
+	esac
+done
+shift $((OPTIND - 1))
+
+: ${mirrors:="http://mirror.rackspace.com/archlinux http://dgix.ru/mirrors/archlinux"}
+
 cd /
 download_and_extract_bootstrap
 configure_chroot