Compare commits
4 Commits
a4263ba25f
...
3c8918262b
| Author | SHA1 | Date | |
|---|---|---|---|
| 3c8918262b | |||
| a6bff434bd | |||
| 3c9d396af1 | |||
| 73c266e3f2 |
33
README.md
Normal file
33
README.md
Normal file
@ -0,0 +1,33 @@
|
||||
# bootstrap
|
||||
|
||||
Bootstrap an OS instance with bare essentials.
|
||||
|
||||
## Usage
|
||||
|
||||
To bootstrap a macOS instance:
|
||||
|
||||
```
|
||||
$ curl -s https://code.infektor.net/config/bootstrap/raw/master/bootstrap-macOS.sh | sh
|
||||
```
|
||||
|
||||
To bootstrap a Debain based Linux instance:
|
||||
|
||||
```
|
||||
$ wget https://code.infektor.net/config/bootstrap/raw/master/bootstrap-Debian.sh -O - | sh
|
||||
```
|
||||
|
||||
To bootstrap a Windows instance:
|
||||
|
||||
```
|
||||
$ bitsadmin /transfer bootstrap-Windows.cmd https://code.infektor.net/config/bootstrap/raw/master/bootstrap-Windows.cmd %cd%\bootstrap-Windows.cmd & %cd%\bootstrap-Windows.cmd
|
||||
```
|
||||
|
||||
## Package
|
||||
|
||||
Install as a pip package to set SSH keys on any of GitHub, GitLab, BitBucket
|
||||
Cloud, or Gogs servers:
|
||||
|
||||
```
|
||||
pip install git+https://code.infektor.net/config/bootstrap.git
|
||||
python -m 'import bootstrap; bootstrap.set_ssh_keys()'
|
||||
```
|
||||
105
bootstrap-Debian.sh
Executable file
105
bootstrap-Debian.sh
Executable file
@ -0,0 +1,105 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
|
||||
show_usage() {
|
||||
echo "usage: $0 [-h] [-y]"
|
||||
}
|
||||
|
||||
show_help() {
|
||||
show_usage
|
||||
echo
|
||||
echo "Bootstrap a Debian based distribution with:"
|
||||
echo
|
||||
echo "* update apt cache"
|
||||
echo "* upgrade apt packages"
|
||||
echo "* git - from apt"
|
||||
echo "* python - from apt"
|
||||
echo "* python-pip - from apt"
|
||||
echo "* virtualenv - from pip"
|
||||
echo "* SSH key - from ssh-keygen"
|
||||
echo "* GitHub public key - with SSH key"
|
||||
echo "* GitLab public key - with SSH key"
|
||||
echo "* BitBucket Cloud public key - with SSH key"
|
||||
echo "* Gogs Cloud public key - with SSH key"
|
||||
echo "* conduit - configuration manager"
|
||||
echo
|
||||
echo "If any already exist they will not be reinstalled."
|
||||
echo
|
||||
echo "optional arguments:"
|
||||
echo " -h show this help message and exit"
|
||||
echo " -y assume yes when prompted"
|
||||
}
|
||||
|
||||
yes=0
|
||||
|
||||
while getopts 'hy' opt; do
|
||||
case $opt in
|
||||
h) show_help; exit 0 ;;
|
||||
y) yes=1 ;;
|
||||
*) show_usage; exit 1 ;;
|
||||
esac
|
||||
done
|
||||
|
||||
missing() {
|
||||
which $1 &> /dev/null && return 1 || return 0
|
||||
}
|
||||
|
||||
agree() {
|
||||
local check=^[Nn]$
|
||||
[ $yes -eq 1 ] && [[ ! "$2" =~ $check ]] && return 0
|
||||
[[ "$2" =~ $check ]] && local default="[y/N]" || local default="[Y/n]"
|
||||
read -p "$1 $default? " answer
|
||||
case "$answer" in
|
||||
y|Y|yes) return 0 ;;
|
||||
n|N|no) return 1 ;;
|
||||
'') [[ "$2" =~ $check ]] && return 1 || return 0 ;;
|
||||
*) echo "invalid input: $answer" && return `agree "$1"` ;;
|
||||
esac
|
||||
}
|
||||
|
||||
apt_install() {
|
||||
sudo apt-get install --yes --install-recommends $1 > /dev/null
|
||||
}
|
||||
|
||||
pip_install() {
|
||||
pip install --user $1 > /dev/null
|
||||
}
|
||||
|
||||
export PATH=~/.local/bin:$PATH
|
||||
|
||||
agree "Update apt cache" && sudo apt-get update > /dev/null
|
||||
agree "Upgrade apt packages" "N" && sudo apt-get upgrade > /dev/null
|
||||
|
||||
missing git && agree "Install git" && apt_install git
|
||||
|
||||
if missing pip; then
|
||||
agree "Install python-pip" && apt_install python-pip
|
||||
agree "Upgrade pip with pip" && sudo -H pip install --upgrade pip > /dev/null
|
||||
fi
|
||||
|
||||
missing virtualenv && agree "Install virtualenv" && pip_install virtualenv
|
||||
|
||||
if [ ! -f ~/.ssh/id_rsa ] && agree "Generate SSH key"; then
|
||||
read -rp "SSH email: " email
|
||||
[ ! -d ~/.ssh ] && mkdir -p ~/.ssh
|
||||
ssh-keygen -t rsa -b 4096 -C "$email" -N "" -f ~/.ssh/id_rsa
|
||||
fi
|
||||
|
||||
if ! missing virtualenv && agree "Set SSH keys on remote Git servers"; then
|
||||
env=$(mktemp -d)
|
||||
virtualenv $env > /dev/null
|
||||
source $env/bin/activate
|
||||
pip install git+https://code.infektor.net/config/bootstrap.git > /dev/null
|
||||
python -c 'import bootstrap; bootstrap.set_ssh_keys()'
|
||||
deactivate
|
||||
rm -r $env
|
||||
fi
|
||||
|
||||
missing conduit && agree "Install conduit" && \
|
||||
pip_install git+ssh://git@github.com/kbenzie/conduit.git
|
||||
|
||||
echo "To use installed pip packages update your PATH:"
|
||||
echo 'export PATH=~/.local/bin:$PATH'
|
||||
|
||||
[ -f $0 ] && agree "Remove $0" "N" && rm $0 || exit 0
|
||||
193
bootstrap-Windows.cmd
Normal file
193
bootstrap-Windows.cmd
Normal file
@ -0,0 +1,193 @@
|
||||
@echo off
|
||||
setlocal
|
||||
|
||||
if [%1]==[/?] goto :help
|
||||
echo %* | find "/?" > nul
|
||||
if errorlevel 1 goto :main
|
||||
|
||||
:help
|
||||
echo usage: %0 [/?]
|
||||
echo
|
||||
echo Bootstrap a macOS instance with:
|
||||
echo
|
||||
echo * Windows SDK - from MSDN
|
||||
echo * Chocolatey - package manager
|
||||
echo * Cmder, including Git - from Chocolatey
|
||||
echo * Python - from Chocolatey
|
||||
echo * virtualenv - from pip
|
||||
echo * SSH key - from ssh-keygen
|
||||
echo * GitHub public key - with SSH key
|
||||
echo * GitLab public key - with SSH key
|
||||
echo * BitBucket Cloud public key - with SSH key
|
||||
echo * Gogs Cloud public key - with SSH key
|
||||
echo * conduit - configuration manager
|
||||
echo
|
||||
echo optional arguments:
|
||||
echo /? show this help message and exit
|
||||
goto end
|
||||
|
||||
:main
|
||||
|
||||
:: Check for admin permissions
|
||||
>nul 2>&1 "%SYSTEMROOT%\system32\cacls.exe" "%SYSTEMROOT%\system32\config\system"
|
||||
if errorlevel 1 (
|
||||
echo Administrator privilages required!
|
||||
exit /B 1
|
||||
)
|
||||
|
||||
:: Install Windows SDK
|
||||
:windows_sdk
|
||||
set choice=
|
||||
set /p choice="Install Windows SDK [Y/n]? "
|
||||
if "%choice%"=="" goto windows_sdk_yes
|
||||
if "%choice%"=="Y" goto windows_sdk_yes
|
||||
if "%choice%"=="y" goto windows_sdk_yes
|
||||
if "%choice%"=="N" goto windows_sdk_no
|
||||
if "%choice%"=="n" goto windows_sdk_no
|
||||
echo invalid value: %choice%
|
||||
goto windows_sdk
|
||||
:windows_sdk_yes
|
||||
bitsadmin.exe /transfer "Download Windows SDK" https://go.microsoft.com/fwlink/p/?linkid=845298 %~dp0\winsdksetup.exe
|
||||
if errorlevel 1 exit /B 1
|
||||
echo Installing Windows SDK...
|
||||
%~dp0\winsdksetup.exe /features + /q
|
||||
if errorlevel 1 exit /B 1
|
||||
del %~dp0\winsdksetup.exe
|
||||
:windows_sdk_no
|
||||
|
||||
:: Install Chocolatey
|
||||
:choco
|
||||
set choice=
|
||||
set /p choice="Install Chocolatey [Y/n]? "
|
||||
if "%choice%"=="" goto choco_yes
|
||||
if "%choice%"=="Y" goto choco_yes
|
||||
if "%choice%"=="y" goto choco_yes
|
||||
if "%choice%"=="N" goto choco_no
|
||||
if "%choice%"=="n" goto choco_no
|
||||
echo invalid value: %choice%
|
||||
goto choco
|
||||
:choco_yes
|
||||
@"%SystemRoot%\System32\WindowsPowerShell\v1.0\powershell.exe" -NoProfile -InputFormat None -ExecutionPolicy Bypass -Command "iex ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))" && SET "PATH=%PATH%;%ALLUSERSPROFILE%\chocolatey\bin"
|
||||
if errorlevel 1 exit /B 1
|
||||
:choco_no
|
||||
|
||||
:: Install Cmder, including Git
|
||||
:git
|
||||
set choice=
|
||||
set /p choice="Install Cmder, including Git [Y/n]? "
|
||||
if "%choice%"=="" goto git_yes
|
||||
if "%choice%"=="Y" goto git_yes
|
||||
if "%choice%"=="y" goto git_yes
|
||||
if "%choice%"=="N" goto git_no
|
||||
if "%choice%"=="n" goto git_no
|
||||
echo invalid value: %choice%
|
||||
goto git
|
||||
:git_yes
|
||||
choco install cmder --yes
|
||||
if errorlevel 1 exit /B 1
|
||||
:git_no
|
||||
|
||||
:: Install Python
|
||||
:python
|
||||
set choice=
|
||||
set /p choice="Install Python [Y/n]? "
|
||||
if "%choice%"=="" goto python_yes
|
||||
if "%choice%"=="Y" goto python_yes
|
||||
if "%choice%"=="y" goto python_yes
|
||||
if "%choice%"=="N" goto python_no
|
||||
if "%choice%"=="n" goto python_no
|
||||
echo invalid value: %choice%
|
||||
goto python
|
||||
:python_yes
|
||||
choco install python2 --yes
|
||||
if errorlevel 1 exit /B 1
|
||||
:python_no
|
||||
|
||||
:: Install virtualenv
|
||||
:virtualenv
|
||||
set choice=
|
||||
set /p choice="Install virtualenv [Y/n]? "
|
||||
if "%choice%"=="" goto virtualenv_yes
|
||||
if "%choice%"=="Y" goto virtualenv_yes
|
||||
if "%choice%"=="y" goto virtualenv_yes
|
||||
if "%choice%"=="N" goto virtualenv_no
|
||||
if "%choice%"=="n" goto virtualenv_no
|
||||
echo invalid value: %choice%
|
||||
goto virtualenv
|
||||
:virtualenv_yes
|
||||
"C:\Python27\Scripts\pip.exe" install virtualenv
|
||||
if errorlevel 1 exit /B 1
|
||||
:virtualenv_no
|
||||
|
||||
:: Generate SSH key
|
||||
:ssh_key
|
||||
set choice=
|
||||
set /p choice="Generate SSH key [Y/n]? "
|
||||
if "%choice%"=="" goto ssh_key_yes
|
||||
if "%choice%"=="Y" goto ssh_key_yes
|
||||
if "%choice%"=="y" goto ssh_key_yes
|
||||
if "%choice%"=="N" goto ssh_key_no
|
||||
if "%choice%"=="n" goto ssh_key_no
|
||||
echo invalid value: %choice%
|
||||
goto ssh_key
|
||||
:ssh_key_yes
|
||||
mkdir %USERPROFILE%\.ssh
|
||||
set /P email="SSH email: "
|
||||
"C:\tools\cmder\vendor\git-for-windows\usr\bin\ssh-keygen.exe" -t rsa -b 4096 -C "%email%" -N "" -f %USERPROFILE%\.ssh\id_rsa
|
||||
if errorlevel 1 exit /B 1
|
||||
:ssh_key_no
|
||||
|
||||
:: Set SSH keys on remote Git servers
|
||||
:remote_ssh_keys
|
||||
:: Add git to the PATH to pip can find it
|
||||
set PATH=C:\tools\cmder\vendor\git-for-windows\bin;%PATH%
|
||||
set choice=
|
||||
set /p choice="Set SSH keys on remote Git servers [Y/n]? "
|
||||
if "%choice%"=="" goto remote_ssh_keys_yes
|
||||
if "%choice%"=="Y" goto remote_ssh_keys_yes
|
||||
if "%choice%"=="y" goto remote_ssh_keys_yes
|
||||
if "%choice%"=="N" goto remote_ssh_keys_no
|
||||
if "%choice%"=="n" goto remote_ssh_keys_no
|
||||
echo invalid value: %choice%
|
||||
goto remote_ssh_keys
|
||||
:remote_ssh_keys_yes
|
||||
"C:\Python27\Scripts\virtualenv.exe" %~dp0\bootstrap_env
|
||||
if errorlevel 1 exit /B 1
|
||||
"%~dp0\bootstrap_env\Scripts\pip.exe" install git+https://code.infektor.net/config/bootstrap.git
|
||||
if errorlevel 1 exit /B 1
|
||||
"%~dp0\bootstrap_env\Scripts\python.exe" -c "import bootstrap; bootstrap.set_ssh_keys()"
|
||||
if errorlevel 1 exit /B 1
|
||||
:remote_ssh_keys_no
|
||||
|
||||
:: Install conduit
|
||||
:conduit
|
||||
set choice=
|
||||
set /p choice="Install conduit [Y/n]? "
|
||||
if "%choice%"=="" goto conduit_yes
|
||||
if "%choice%"=="Y" goto conduit_yes
|
||||
if "%choice%"=="y" goto conduit_yes
|
||||
if "%choice%"=="N" goto conduit_no
|
||||
if "%choice%"=="n" goto conduit_no
|
||||
echo invalid value: %choice%
|
||||
goto conduit
|
||||
:conduit_yes
|
||||
"C:\Python27\Scripts\pip.exe" install git+ssh://git@github.com/kbenzie/conduit.git
|
||||
rmdir /Q /S "%~dp0\bootstrap_env"
|
||||
:conduit_no
|
||||
|
||||
:remove_bootstrap
|
||||
set choice=
|
||||
set /p choice="Remove bootstrap-Windows.cmd [y/N]? "
|
||||
if "%choice%"=="" goto remove_bootstrap_no
|
||||
if "%choice%"=="Y" goto remove_bootstrap_yes
|
||||
if "%choice%"=="y" goto remove_bootstrap_yes
|
||||
if "%choice%"=="N" goto remove_bootstrap_no
|
||||
if "%choice%"=="n" goto remove_bootstrap_no
|
||||
echo invalid value: %choice%
|
||||
goto remove_bootstrap
|
||||
:remove_bootstrap_yes
|
||||
start /b "" cmd /c del "%~f0"&exit /b
|
||||
:remove_bootstrap_no
|
||||
|
||||
:end
|
||||
endlocal
|
||||
110
bootstrap-macOS.sh
Executable file
110
bootstrap-macOS.sh
Executable file
@ -0,0 +1,110 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
|
||||
show_usage() {
|
||||
echo "usage: $0 [-h] [-y]"
|
||||
}
|
||||
|
||||
show_help() {
|
||||
show_usage
|
||||
echo
|
||||
echo "Bootstrap a macOS instance with:"
|
||||
echo
|
||||
echo "* Xcode command line developer tools"
|
||||
echo "* Homebrew - package manager"
|
||||
echo "* python - from Homebrew"
|
||||
echo "* virtualenv - from pip"
|
||||
echo "* SSH key - from ssh-keygen"
|
||||
echo "* GitHub public key - with SSH key"
|
||||
echo "* GitLab public key - with SSH key"
|
||||
echo "* BitBucket Cloud public key - with SSH key"
|
||||
echo "* Gogs Cloud public key - with SSH key"
|
||||
echo "* conduit - configuration manager"
|
||||
echo
|
||||
echo "If any already exist they will not be reinstalled."
|
||||
echo
|
||||
echo "optional arguments:"
|
||||
echo " -h show this help message and exit"
|
||||
echo " -y assume yes when prompted"
|
||||
}
|
||||
|
||||
yes=0
|
||||
|
||||
while getopts 'hy' opt; do
|
||||
case $opt in
|
||||
h) show_help; exit 0 ;;
|
||||
y) yes=1 ;;
|
||||
*) show_usage; exit 1 ;;
|
||||
esac
|
||||
done
|
||||
|
||||
missing() {
|
||||
which $1 &> /dev/null && return 1 || return 0
|
||||
}
|
||||
|
||||
agree() {
|
||||
local check=^[Nn]$
|
||||
[ $yes -eq 1 ] && [[ ! "$2" =~ $check ]] && return 0
|
||||
[[ "$2" =~ $check ]] && local default="[y/N]" || local default="[Y/n]"
|
||||
read -p "$1 $default? " answer
|
||||
case "$answer" in
|
||||
y|Y|yes) return 0 ;;
|
||||
n|N|no) return 1 ;;
|
||||
'') [[ "$2" =~ $check ]] && return 1 || return 0 ;;
|
||||
*) echo "invalid input: $answer" && return `agree "$1"` ;;
|
||||
esac
|
||||
}
|
||||
|
||||
brew_install() {
|
||||
brew install $1 > /dev/null
|
||||
}
|
||||
|
||||
pip_install() {
|
||||
pip install --user $1 > /dev/null
|
||||
}
|
||||
|
||||
export PATH=~/.local/bin:$PATH
|
||||
export PYTHONUSERBASE=~/.local
|
||||
|
||||
if ! xcode-select --print-path &> /dev/null; then
|
||||
agree "Install Xcode command line developer tools" && xcode-select --install
|
||||
fi
|
||||
|
||||
missing brew && agree "Intalll Homebrew" && /usr/bin/ruby -e \
|
||||
"$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
|
||||
|
||||
agree "Update Homebrew packages" && brew update > /dev/null
|
||||
|
||||
if missing /usr/local/bin/python2 && agree "Install Homebrew python"; then
|
||||
brew_install python
|
||||
export PATH=/usr/local/opt/python/libexec/bin:$PATH
|
||||
echo "To use Homebrew python update your PATH:"
|
||||
echo 'export PATH=/usr/local/opt/python/libexec/bin:$PATH'
|
||||
fi
|
||||
|
||||
missing virtualenv && agree "Install virtualenv" && pip_install virtualenv
|
||||
|
||||
if [ ! -f ~/.ssh/id_rsa ] && agree "Generate SSH key"; then
|
||||
read -rp "SSH email: " email
|
||||
[ ! -d ~/.ssh ] && mkdir -p ~/.ssh
|
||||
ssh-keygen -t rsa -b 4096 -C "$email" -N "" -f ~/.ssh/id_rsa
|
||||
fi
|
||||
|
||||
if ! missing virtualenv && agree "Set SSH keys on remote Git servers"; then
|
||||
env=$(mktemp -d)
|
||||
virtualenv $env &> /dev/null
|
||||
source $env/bin/activate
|
||||
pip install git+https://code.infektor.net/config/bootstrap.git > /dev/null
|
||||
python -c 'import bootstrap; bootstrap.set_ssh_keys()'
|
||||
deactivate
|
||||
rm -r $env
|
||||
fi
|
||||
|
||||
! missing pip && missing conduit && agree "Install conduit" && \
|
||||
pip_install git+ssh://git@github.com/kbenzie/conduit.git
|
||||
|
||||
echo "To use installed pip packages update your PATH and PYTHONUSERBASE:"
|
||||
echo 'export PATH=~/.local/bin:$PATH && export PYTHONUSERBASE=~/.local'
|
||||
|
||||
[ -f $0 ] && agree "Remove $0" "N" && rm $0 || exit 0
|
||||
185
bootstrap/__init__.py
Normal file
185
bootstrap/__init__.py
Normal file
@ -0,0 +1,185 @@
|
||||
"""Interactively set SSH keys on remote Git servers."""
|
||||
|
||||
from __future__ import print_function
|
||||
from getpass import getpass
|
||||
from os import environ
|
||||
from os.path import join
|
||||
from platform import node, system
|
||||
|
||||
from requests import ConnectionError, get, post
|
||||
from requests.auth import HTTPBasicAuth
|
||||
from requests.compat import urlparse
|
||||
|
||||
try:
|
||||
input = raw_input
|
||||
except NameError:
|
||||
pass
|
||||
|
||||
|
||||
class BootstrapError(Exception):
|
||||
"""Bootstrap Exception."""
|
||||
pass
|
||||
|
||||
|
||||
def bootstrap_error(response):
|
||||
"""Create a BootstrapError from a Response."""
|
||||
return BootstrapError('%s %s' % (response.status_code, response.reason))
|
||||
|
||||
|
||||
def agree(question, default='Y'):
|
||||
"""Prompt user to answer a yes/no question."""
|
||||
valid = {
|
||||
'y': True,
|
||||
'Y': True,
|
||||
'yes': True,
|
||||
'n': False,
|
||||
'N': False,
|
||||
'no': False,
|
||||
'': {'Y': True,
|
||||
'N': False}[default]
|
||||
}
|
||||
answer = input('%s [%s]? ' % (question, {'Y': 'Y/n', 'N': 'y/N'}[default]))
|
||||
try:
|
||||
return valid[answer]
|
||||
except KeyError:
|
||||
print('invalid input: %s' % answer)
|
||||
return agree(question)
|
||||
|
||||
|
||||
def get_url(service, default):
|
||||
"""Get URL."""
|
||||
url = input('%s URL%s: ' % (service, ' (%s)' % default if default else ''))
|
||||
if url == '':
|
||||
if default:
|
||||
return default
|
||||
else:
|
||||
print('invalid input: %s' % url)
|
||||
return get_url(service, default)
|
||||
return url
|
||||
|
||||
|
||||
def get_username_password(service):
|
||||
"""Get username/password."""
|
||||
username = input('%s username: ' % service)
|
||||
password = getpass('%s password: ' % service)
|
||||
return (username, password)
|
||||
|
||||
|
||||
def get_local_key():
|
||||
"""Get local SSH key."""
|
||||
if system() == 'Windows':
|
||||
home = environ['userprofile']
|
||||
else:
|
||||
home = environ['HOME']
|
||||
with open(join(home, '.ssh', 'id_rsa.pub'), 'r') as key_file:
|
||||
return key_file.read().rstrip()
|
||||
|
||||
|
||||
def key_exists(keys, local_key):
|
||||
"""Check if local SSH key is already set."""
|
||||
for key in keys:
|
||||
if local_key.startswith(key['key']):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def set_github_ssh_key():
|
||||
"""Set GitHub SSH key."""
|
||||
url = urlparse(get_url('GitHub', 'https://github.com'))
|
||||
keys_url = '%s://api.%s/user/keys' % (url.scheme, url.netloc)
|
||||
username, password = get_username_password('GitHub')
|
||||
auth = HTTPBasicAuth(username, password)
|
||||
response = get(keys_url, auth=auth)
|
||||
if response.status_code != 200:
|
||||
raise bootstrap_error(response)
|
||||
keys = response.json()
|
||||
local_key = get_local_key()
|
||||
if not key_exists(keys, local_key):
|
||||
response = post(
|
||||
keys_url, auth=auth, json={'title': node(),
|
||||
'key': local_key})
|
||||
if response.status_code != 201:
|
||||
raise bootstrap_error(response)
|
||||
|
||||
|
||||
def set_gitlab_ssh_key():
|
||||
"""Set GitLab SSH key."""
|
||||
api_url = '%s/api/v4' % get_url('GitLab', 'https://gitlab.com')
|
||||
session_url = '%s/session' % api_url
|
||||
keys_url = '%s/user/keys' % api_url
|
||||
username, password = get_username_password('GitLab')
|
||||
response = post(session_url, {'login': username, 'password': password})
|
||||
if response.status_code != 201:
|
||||
raise bootstrap_error(response)
|
||||
auth = {'Private-Token': response.json()['private_token']}
|
||||
response = get(keys_url, headers=auth)
|
||||
if response.status_code != 200:
|
||||
raise bootstrap_error(response)
|
||||
keys = response.json()
|
||||
local_key = get_local_key()
|
||||
if not key_exists(keys, local_key):
|
||||
response = post(
|
||||
keys_url, headers=auth, json={'title': node(),
|
||||
'key': local_key})
|
||||
if response.status_code != 201:
|
||||
raise bootstrap_error(response)
|
||||
|
||||
|
||||
def set_bitbucket_cloud_ssh_key():
|
||||
"""Set BitBucket Cloud SSH key."""
|
||||
username, password = get_username_password('BitBucket Cloud')
|
||||
keys_url = 'https://api.bitbucket.org/1.0/users/%s/ssh-keys' % username
|
||||
auth = HTTPBasicAuth(username, password)
|
||||
response = get(keys_url, auth=auth)
|
||||
if response.status_code != 200:
|
||||
raise bootstrap_error(response)
|
||||
keys = response.json()
|
||||
local_key = get_local_key()
|
||||
if not key_exists(keys, local_key):
|
||||
response = post(
|
||||
keys_url, auth=auth, data={'label': node(),
|
||||
'key': local_key})
|
||||
if response.status_code != 200:
|
||||
raise bootstrap_error(response)
|
||||
|
||||
|
||||
def set_gogs_ssh_key():
|
||||
"""Set Gogs SSH key."""
|
||||
keys_url = '%s/api/v1/user/keys' % get_url('Gogs', None)
|
||||
username, password = get_username_password('Gogs')
|
||||
auth = HTTPBasicAuth(username, password)
|
||||
response = get(keys_url, auth=auth)
|
||||
if response.status_code != 200:
|
||||
raise bootstrap_error(response)
|
||||
keys = response.json()
|
||||
local_key = get_local_key()
|
||||
if not key_exists(keys, local_key):
|
||||
response = post(
|
||||
keys_url, auth=auth, json={'title': node(),
|
||||
'key': local_key})
|
||||
if response.status_code != 201:
|
||||
raise bootstrap_error(response)
|
||||
|
||||
|
||||
def set_ssh_keys():
|
||||
"""Interactively set SSH keys on remote Git servers."""
|
||||
try:
|
||||
for service in ['GitHub', 'GitLab', 'BitBucket Cloud', 'Gogs']:
|
||||
question = 'Set %s SSH key' % service
|
||||
default = 'Y'
|
||||
while agree(question, default):
|
||||
try:
|
||||
{
|
||||
'GitHub': set_github_ssh_key,
|
||||
'GitLab': set_gitlab_ssh_key,
|
||||
'BitBucket Cloud': set_bitbucket_cloud_ssh_key,
|
||||
'Gogs': set_gogs_ssh_key,
|
||||
}[service]()
|
||||
if service == 'BitBucket Cloud':
|
||||
break
|
||||
question = 'Set another %s SSH key' % service
|
||||
default = 'N'
|
||||
except (BootstrapError, ConnectionError) as error:
|
||||
print('error: %s' % error.message)
|
||||
except KeyboardInterrupt:
|
||||
exit(130)
|
||||
Loading…
x
Reference in New Issue
Block a user