lib.sh 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050
  1. #!/usr/bin/env bash
  2. # -*- coding: utf-8; mode: sh indent-tabs-mode: nil -*-
  3. # SPDX-License-Identifier: AGPL-3.0-or-later
  4. # shellcheck disable=SC2059,SC1117
  5. # ubuntu, debian, arch, fedora ...
  6. DIST_ID=$(source /etc/os-release; echo $ID);
  7. # shellcheck disable=SC2034
  8. DIST_VERS=$(source /etc/os-release; echo $VERSION_ID);
  9. ADMIN_NAME="${ADMIN_NAME:-$(git config user.name)}"
  10. ADMIN_NAME="${ADMIN_NAME:-$USER}"
  11. ADMIN_EMAIL="${ADMIN_EMAIL:-$(git config user.email)}"
  12. ADMIN_EMAIL="${ADMIN_EMAIL:-$USER@$(hostname)}"
  13. if [[ -z "${REPO_ROOT}" ]]; then
  14. REPO_ROOT=$(dirname "${BASH_SOURCE[0]}")
  15. while [ -h "${REPO_ROOT}" ] ; do
  16. REPO_ROOT=$(readlink "${REPO_ROOT}")
  17. done
  18. REPO_ROOT=$(cd "${REPO_ROOT}/.." && pwd -P )
  19. fi
  20. if [[ -z ${TEMPLATES} ]]; then
  21. TEMPLATES="${REPO_ROOT}/utils/templates"
  22. fi
  23. if [[ -z "$CACHE" ]]; then
  24. CACHE="${REPO_ROOT}/cache"
  25. fi
  26. if [[ -z ${DIFF_CMD} ]]; then
  27. DIFF_CMD="diff -u"
  28. if command -v colordiff >/dev/null; then
  29. DIFF_CMD="colordiff -u"
  30. fi
  31. fi
  32. DOT_CONFIG="${DOT_CONFIG:-${REPO_ROOT}/.config.sh}"
  33. source_dot_config() {
  34. if [[ ! -e "${DOT_CONFIG}" ]]; then
  35. err_msg "configuration does not extsts at: ${DOT_CONFIG}"
  36. return 42
  37. fi
  38. # shellcheck disable=SC1090
  39. source "${DOT_CONFIG}"
  40. }
  41. sudo_or_exit() {
  42. # usage: sudo_or_exit
  43. if [ ! "$(id -u)" -eq 0 ]; then
  44. err_msg "this command requires root (sudo) privilege!" >&2
  45. exit 42
  46. fi
  47. }
  48. required_commands() {
  49. # usage: required_commands [cmd1 ...]
  50. local exit_val=0
  51. while [ -n "$1" ]; do
  52. if ! command -v "$1" &>/dev/null; then
  53. err_msg "missing command $1"
  54. exit_val=42
  55. fi
  56. shift
  57. done
  58. return $exit_val
  59. }
  60. # colors
  61. # ------
  62. # shellcheck disable=SC2034
  63. set_terminal_colors() {
  64. _colors=8
  65. _creset='\e[0m' # reset all attributes
  66. _Black='\e[0;30m'
  67. _White='\e[1;37m'
  68. _Red='\e[0;31m'
  69. _Green='\e[0;32m'
  70. _Yellow='\e[0;33m'
  71. _Blue='\e[0;34m'
  72. _Violet='\e[0;35m'
  73. _Cyan='\e[0;36m'
  74. _BBlack='\e[1;30m'
  75. _BWhite='\e[1;37m'
  76. _BRed='\e[1;31m'
  77. _BGreen='\e[1;32m'
  78. _BYellow='\e[1;33m'
  79. _BBlue='\e[1;34m'
  80. _BPurple='\e[1;35m'
  81. _BCyan='\e[1;36m'
  82. }
  83. if [ ! -p /dev/stdout ]; then
  84. set_terminal_colors
  85. fi
  86. # reST
  87. # ----
  88. if command -v fmt >/dev/null; then
  89. export FMT="fmt -u"
  90. else
  91. export FMT="cat"
  92. fi
  93. rst_title() {
  94. # usage: rst_title <header-text> [part|chapter|section]
  95. case ${2-chapter} in
  96. part) printf "\n${_BGreen}${1//?/=}\n${_BCyan}${1}${_BGreen}\n${1//?/=}${_creset}\n";;
  97. chapter) printf "\n${_BCyan}${1}\n${_BGreen}${1//?/=}${_creset}\n";;
  98. section) printf "\n${_BCyan}${1}\n${_BGreen}${1//?/-}${_creset}\n";;
  99. *)
  100. err_msg "invalid argument '${2}' in line $(caller)"
  101. return 42
  102. ;;
  103. esac
  104. }
  105. rst_para() {
  106. # usage: RST_INDENT=1 rst_para "lorem ipsum ..."
  107. local prefix=''
  108. if [[ -n $RST_INDENT ]] && [[ $RST_INDENT -gt 0 ]]; then
  109. prefix="$(for i in $(seq 1 "$RST_INDENT"); do printf " "; done)"
  110. echo -en "\n$*\n" | $FMT | prefix_stdout "$prefix"
  111. else
  112. echo -en "\n$*\n" | $FMT
  113. fi
  114. }
  115. err_msg() { echo -e "${_BRed}ERROR:${_creset} $*" >&2; }
  116. warn_msg() { echo -e "${_BBlue}WARN:${_creset} $*" >&2; }
  117. info_msg() { echo -e "${_BYellow}INFO:${_creset} $*" >&2; }
  118. clean_stdin() {
  119. if [[ $(uname -s) != 'Darwin' ]]; then
  120. while read -r -n1 -t 0.1; do : ; done
  121. fi
  122. }
  123. wait_key(){
  124. # usage: waitKEY [<timeout in sec>]
  125. clean_stdin
  126. local _t=$1
  127. local msg="${MSG}"
  128. [[ -z "$msg" ]] && msg="${_Green}** press any [${_BCyan}KEY${_Green}] to continue **${_creset}"
  129. [[ -n $FORCE_TIMEOUT ]] && _t=$FORCE_TIMEOUT
  130. [[ -n $_t ]] && _t="-t $_t"
  131. printf "$msg"
  132. # shellcheck disable=SC2086
  133. read -r -s -n1 $_t
  134. echo
  135. clean_stdin
  136. }
  137. ask_yn() {
  138. # usage: ask_yn <prompt-text> [Ny|Yn] [<timeout in sec>]
  139. local EXIT_YES=0 # exit status 0 --> successful
  140. local EXIT_NO=1 # exit status 1 --> error code
  141. local _t=$3
  142. [[ -n $FORCE_TIMEOUT ]] && _t=$FORCE_TIMEOUT
  143. [[ -n $_t ]] && _t="-t $_t"
  144. case "${FORCE_SELECTION:-${2}}" in
  145. Y) return ${EXIT_YES} ;;
  146. N) return ${EXIT_NO} ;;
  147. Yn)
  148. local exit_val=${EXIT_YES}
  149. local choice="[${_BGreen}YES${_creset}/no]"
  150. local default="Yes"
  151. ;;
  152. *)
  153. local exit_val=${EXIT_NO}
  154. local choice="[${_BGreen}NO${_creset}/yes]"
  155. local default="No"
  156. ;;
  157. esac
  158. echo
  159. while true; do
  160. clean_stdin
  161. printf "$1 ${choice} "
  162. # shellcheck disable=SC2086
  163. read -r -n1 $_t
  164. if [[ -z $REPLY ]]; then
  165. printf "$default\n"; break
  166. elif [[ $REPLY =~ ^[Yy]$ ]]; then
  167. exit_val=${EXIT_YES}
  168. printf "\n"
  169. break
  170. elif [[ $REPLY =~ ^[Nn]$ ]]; then
  171. exit_val=${EXIT_NO}
  172. printf "\n"
  173. break
  174. fi
  175. _t=""
  176. err_msg "invalid choice"
  177. done
  178. clean_stdin
  179. return $exit_val
  180. }
  181. tee_stderr () {
  182. # usage::
  183. # tee_stderr 1 <<EOF | python -i
  184. # print("hello")
  185. # EOF
  186. # ...
  187. # >>> print("hello")
  188. # hello
  189. local _t="0";
  190. if [[ -n $1 ]] ; then _t="$1"; fi
  191. (while read -r line; do
  192. # shellcheck disable=SC2086
  193. sleep $_t
  194. echo -e "$line" >&2
  195. echo "$line"
  196. done)
  197. }
  198. prefix_stdout () {
  199. # usage: <cmd> | prefix_stdout [prefix]
  200. local prefix="${_BYellow}-->|${_creset}"
  201. if [[ -n $1 ]] ; then prefix="${_BYellow}$1${_creset}"; fi
  202. # shellcheck disable=SC2162
  203. (while IFS= read line; do
  204. echo -e "${prefix}$line"
  205. done)
  206. }
  207. append_line() {
  208. # usage: append_line <line> <file>
  209. #
  210. # Append line if not exists, create file if not exists. E.g::
  211. #
  212. # append_line 'source ~/.foo' ~/bashrc
  213. local LINE=$1
  214. local FILE=$2
  215. grep -qFs -- "$LINE" "$FILE" || echo "$LINE" >> "$FILE"
  216. }
  217. cache_download() {
  218. # usage: cache_download <url> <local-filename>
  219. local exit_value=0
  220. if [[ -n ${SUDO_USER} ]]; then
  221. sudo -u "${SUDO_USER}" mkdir -p "${CACHE}"
  222. else
  223. mkdir -p "${CACHE}"
  224. fi
  225. if [[ -f "${CACHE}/$2" ]] ; then
  226. info_msg "already cached: $1"
  227. info_msg " --> ${CACHE}/$2"
  228. fi
  229. if [[ ! -f "${CACHE}/$2" ]]; then
  230. info_msg "caching: $1"
  231. info_msg " --> ${CACHE}/$2"
  232. if [[ -n ${SUDO_USER} ]]; then
  233. sudo -u "${SUDO_USER}" wget --progress=bar -O "${CACHE}/$2" "$1" ; exit_value=$?
  234. else
  235. wget --progress=bar -O "${CACHE}/$2" "$1" ; exit_value=$?
  236. fi
  237. if [[ ! $exit_value = 0 ]]; then
  238. err_msg "failed to download: $1"
  239. fi
  240. fi
  241. }
  242. backup_file() {
  243. # usage: backup_file /path/to/file.foo
  244. local stamp=$(date +"_%Y%m%d_%H%M%S")
  245. info_msg "create backup: ${1}${stamp}"
  246. cp -a "${1}" "${1}${stamp}"
  247. }
  248. choose_one() {
  249. # usage:
  250. #
  251. # DEFAULT_SELECT= 2 \
  252. # choose_one <name> "your selection?" "Coffee" "Coffee with milk"
  253. local default=${DEFAULT_SELECT-1}
  254. local REPLY
  255. local env_name=$1 && shift
  256. local choice=$1;
  257. local max="${#@}"
  258. local _t
  259. [[ -n $FORCE_TIMEOUT ]] && _t=$FORCE_TIMEOUT
  260. [[ -n $_t ]] && _t="-t $_t"
  261. list=("$@")
  262. echo -e "${_BGreen}Menu::${_creset}"
  263. for ((i=1; i<= $((max -1)); i++)); do
  264. if [[ "$i" == "$default" ]]; then
  265. echo -e " ${_BGreen}$i.${_creset}) ${list[$i]} [default]"
  266. else
  267. echo -e " $i.) ${list[$i]}"
  268. fi
  269. done
  270. while true; do
  271. clean_stdin
  272. printf "$1 [${_BGreen}$default${_creset}] "
  273. if (( 10 > max )); then
  274. # shellcheck disable=SC2086
  275. read -r -n1 $_t
  276. else
  277. # shellcheck disable=SC2086,SC2229
  278. read -r $_t
  279. fi
  280. # selection fits
  281. [[ $REPLY =~ ^-?[0-9]+$ ]] && (( REPLY > 0 )) && (( REPLY < max )) && break
  282. # take default
  283. [[ -z $REPLY ]] && REPLY=$default && break
  284. _t=""
  285. err_msg "invalid choice"
  286. done
  287. eval "$env_name"='${list[${REPLY}]}'
  288. echo
  289. clean_stdin
  290. }
  291. install_template() {
  292. # usage:
  293. #
  294. # install_template [--no-eval] [--variant=<name>] \
  295. # {file} [{owner} [{group} [{chmod}]]]
  296. #
  297. # E.g. the origin of variant 'raw' of /etc/updatedb.conf is::
  298. #
  299. # ${TEMPLATES}/etc/updatedb.conf:raw
  300. #
  301. # To install variant 'raw' of /etc/updatedb.conf without evaluated
  302. # replacements you can use::
  303. #
  304. # install_template --variant=raw --no-eval \
  305. # /etc/updatedb.conf root root 644
  306. local _reply=""
  307. local do_eval=1
  308. local variant=""
  309. local pos_args=("$0")
  310. for i in "$@"; do
  311. case $i in
  312. --no-eval) do_eval=0; shift ;;
  313. --variant=*) variant=":${i#*=}"; shift ;;
  314. *) pos_args+=("$i") ;;
  315. esac
  316. done
  317. local dst="${pos_args[1]}"
  318. local template_origin="${TEMPLATES}${dst}${variant}"
  319. local template_file="${TEMPLATES}${dst}"
  320. local owner="${pos_args[2]-$(id -un)}"
  321. local group="${pos_args[3]-$(id -gn)}"
  322. local chmod="${pos_args[4]-644}"
  323. info_msg "install (eval=$do_eval): ${dst}"
  324. [[ -n $variant ]] && info_msg "variant: ${variant}"
  325. if [[ ! -f "${template_origin}" ]] ; then
  326. err_msg "${template_origin} does not exists"
  327. err_msg "... can't install $dst"
  328. wait_key 30
  329. return 42
  330. fi
  331. if [[ "$do_eval" == "1" ]]; then
  332. template_file="${CACHE}${dst}${variant}"
  333. info_msg "BUILD template ${template_file}"
  334. if [[ -n ${SUDO_USER} ]]; then
  335. sudo -u "${SUDO_USER}" mkdir -p "$(dirname "${template_file}")"
  336. else
  337. mkdir -p "$(dirname "${template_file}")"
  338. fi
  339. # shellcheck disable=SC2086
  340. eval "echo \"$(cat ${template_origin})\"" > "${template_file}"
  341. if [[ -n ${SUDO_USER} ]]; then
  342. chown "${SUDO_USER}:${SUDO_USER}" "${template_file}"
  343. fi
  344. else
  345. template_file=$template_origin
  346. fi
  347. mkdir -p "$(dirname "${dst}")"
  348. if [[ ! -f "${dst}" ]]; then
  349. info_msg "install: ${template_file}"
  350. sudo -H install -v -o "${owner}" -g "${group}" -m "${chmod}" \
  351. "${template_file}" "${dst}" | prefix_stdout
  352. return $?
  353. fi
  354. if [[ -f "${dst}" ]] && cmp --silent "${template_file}" "${dst}" ; then
  355. info_msg "file ${dst} allready installed"
  356. return 0
  357. fi
  358. info_msg "diffrent file ${dst} allready exists on this host"
  359. while true; do
  360. choose_one _reply "choose next step with file $dst" \
  361. "replace file" \
  362. "leave file unchanged" \
  363. "interactiv shell" \
  364. "diff files"
  365. case $_reply in
  366. "replace file")
  367. info_msg "install: ${template_file}"
  368. sudo -H install -v -o "${owner}" -g "${group}" -m "${chmod}" \
  369. "${template_file}" "${dst}" | prefix_stdout
  370. break
  371. ;;
  372. "leave file unchanged")
  373. break
  374. ;;
  375. "interactiv shell")
  376. echo "// edit ${dst} to your needs"
  377. echo -e "// exit with [${_BCyan}CTRL-D${_creset}]"
  378. sudo -H -u "${owner}" -i
  379. $DIFF_CMD "${dst}" "${template_file}"
  380. echo
  381. echo "${_BBlack}did you edit file ...${_creset}"
  382. printf " ${template_file}"
  383. if ask_yn "... to your needs?"; then
  384. break
  385. fi
  386. ;;
  387. "diff files")
  388. $DIFF_CMD "${dst}" "${template_file}" | prefix_stdout
  389. esac
  390. done
  391. }
  392. service_is_available() {
  393. # usage: service_is_available <URL>
  394. local URL="$1"
  395. if [[ -z $URL ]]; then
  396. err_msg "service_is_available: missing arguments"
  397. return 42
  398. fi
  399. http_code=$(curl -H 'Cache-Control: no-cache' \
  400. --silent -o /dev/null --head --write-out '%{http_code}' --insecure \
  401. "${URL}")
  402. exit_val=$?
  403. if [[ $exit_val = 0 ]]; then
  404. info_msg "got $http_code from ${URL}"
  405. fi
  406. case "$http_code" in
  407. 404|410|423) exit_val=$http_code;;
  408. esac
  409. return "$exit_val"
  410. }
  411. # golang
  412. # ------
  413. go_is_available() {
  414. # usage: go_is_available $SERVICE_USER && echo "go is installed!"
  415. sudo -i -u "${1}" which go &>/dev/null
  416. }
  417. install_go() {
  418. # usage: install_go "${GO_PKG_URL}" "${GO_TAR}" "${SERVICE_USER}"
  419. local _svcpr=" |${3}| "
  420. rst_title "Install Go in user's HOME" section
  421. rst_para "download and install go binary .."
  422. cache_download "${1}" "${2}"
  423. tee_stderr 0.1 <<EOF | sudo -i -u "${3}" | prefix_stdout "$_svcpr"
  424. echo \$PATH
  425. echo \$GOPATH
  426. mkdir -p \$HOME/local
  427. rm -rf \$HOME/local/go
  428. tar -C \$HOME/local -xzf ${CACHE}/${2}
  429. EOF
  430. sudo -i -u "${3}" <<EOF | prefix_stdout
  431. ! which go >/dev/null && echo "ERROR - Go Installation not found in PATH!?!"
  432. which go >/dev/null && go version && echo "congratulations -- Go installation OK :)"
  433. EOF
  434. }
  435. # system accounts
  436. # ---------------
  437. service_account_is_available() {
  438. # usage: service_account_is_available "$SERVICE_USER" && echo "OK"
  439. sudo -i -u "$1" echo \$HOME &>/dev/null
  440. }
  441. drop_service_account() {
  442. # usage: drop_service_account "${SERVICE_USER}"
  443. rst_title "Drop ${1} HOME" section
  444. if ask_yn "Do you really want to drop ${1} home folder?"; then
  445. userdel -r -f "${1}" 2>&1 | prefix_stdout
  446. else
  447. rst_para "Leave HOME folder $(du -sh "${1}") unchanged."
  448. fi
  449. }
  450. interactive_shell(){
  451. # usage: interactive_shell "${SERVICE_USER}"
  452. echo -e "// exit with [${_BCyan}CTRL-D${_creset}]"
  453. sudo -H -u "${1}" -i
  454. }
  455. # systemd
  456. # -------
  457. SYSTEMD_UNITS="${SYSTEMD_UNITS:-/lib/systemd/system}"
  458. systemd_install_service() {
  459. # usage: systemd_install_service "${SERVICE_NAME}" "${SERVICE_SYSTEMD_UNIT}"
  460. rst_title "Install System-D Unit ${1}" section
  461. echo
  462. install_template "${2}" root root 644
  463. wait_key
  464. systemd_activate_service "${1}"
  465. }
  466. systemd_remove_service() {
  467. # usage: systemd_remove_service "${SERVICE_NAME}" "${SERVICE_SYSTEMD_UNIT}"
  468. if ! ask_yn "Do you really want to deinstall systemd unit ${1}?"; then
  469. return 42
  470. fi
  471. systemd_deactivate_service "${1}"
  472. rm "${2}" 2>&1 | prefix_stdout
  473. }
  474. systemd_activate_service() {
  475. # usage: systemd_activate_service "${SERVICE_NAME}"
  476. rst_title "Activate ${1} (service)" section
  477. echo
  478. tee_stderr <<EOF | bash 2>&1
  479. systemctl enable ${1}.service
  480. systemctl restart ${1}.service
  481. EOF
  482. tee_stderr <<EOF | bash 2>&1
  483. systemctl status --no-pager ${1}.service
  484. EOF
  485. }
  486. systemd_deactivate_service() {
  487. # usage: systemd_deactivate_service "${SERVICE_NAME}"
  488. rst_title "De-Activate ${1} (service)" section
  489. echo
  490. tee_stderr <<EOF | bash 2>&1 | prefix_stdout
  491. systemctl stop ${1}.service
  492. systemctl disable ${1}.service
  493. EOF
  494. }
  495. systemd_restart_service() {
  496. # usage: systemd_restart_service "${SERVICE_NAME}"
  497. rst_title "Restart ${1} (service)" section
  498. echo
  499. tee_stderr <<EOF | bash 2>&1
  500. systemctl restart ${1}.service
  501. EOF
  502. tee_stderr <<EOF | bash 2>&1
  503. systemctl status --no-pager ${1}.service
  504. EOF
  505. }
  506. # Apache
  507. # ------
  508. # FIXME: Arch Linux & RHEL should be added
  509. if [[ -z "${APACHE_SITES_AVAILABE}" ]]; then
  510. APACHE_SITES_AVAILABE="/etc/apache2/sites-available"
  511. fi
  512. apache_is_installed() {
  513. (command -v apachectl \
  514. && command -v a2ensite \
  515. && command -v a2dissite ) &>/dev/null
  516. }
  517. apache_reload() {
  518. info_msg "reload apache .."
  519. echo
  520. sudo -H apachectl configtest
  521. sudo -H service apache2 force-reload
  522. }
  523. apache_install_site() {
  524. # usage: apache_install_site [<template option> ...] <mysite.conf>
  525. #
  526. # <template option>: see install_template
  527. local template_opts=()
  528. local pos_args=("$0")
  529. for i in "$@"; do
  530. case $i in
  531. -*) template_opts+=("$i");;
  532. *) pos_args+=("$i");;
  533. esac
  534. done
  535. install_template "${template_opts[@]}" \
  536. "${APACHE_SITES_AVAILABE}/${pos_args[1]}" \
  537. root root 644
  538. apache_enable_site "${pos_args[1]}"
  539. info_msg "installed apache site: ${pos_args[1]}"
  540. }
  541. apache_remove_site() {
  542. # usage: apache_remove_site <mysite.conf>
  543. info_msg "remove apache site: $1"
  544. apache_dissable_site "$1"
  545. rm -f "${APACHE_SITES_AVAILABE}/$1"
  546. }
  547. apache_enable_site() {
  548. # usage: apache_enable_site <mysite.conf>
  549. info_msg "enable apache site: $1"
  550. sudo -H a2ensite -q "$1"
  551. apache_reload
  552. }
  553. apache_dissable_site() {
  554. # usage: apache_disable_site <mysite.conf>
  555. info_msg "disable apache site: $1"
  556. sudo -H a2dissite -q "$1"
  557. apache_reload
  558. }
  559. # uWSGI
  560. # -----
  561. uWSGI_SETUP="${uWSGI_SETUP:=/etc/uwsgi}"
  562. uWSGI_USER=
  563. uWSGI_GROUP=
  564. # How distros manage uWSGI apps is very different. From uWSGI POV read:
  565. # - https://uwsgi-docs.readthedocs.io/en/latest/Management.html
  566. case $DIST_ID-$DIST_VERS in
  567. ubuntu-*|debian-*)
  568. # init.d --> /usr/share/doc/uwsgi/README.Debian.gz
  569. # For uWSGI debian uses the LSB init process, this might be changed
  570. # one day, see https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=833067
  571. uWSGI_APPS_AVAILABLE="${uWSGI_SETUP}/apps-available"
  572. uWSGI_APPS_ENABLED="${uWSGI_SETUP}/apps-enabled"
  573. ;;
  574. arch-*)
  575. # systemd --> /usr/lib/systemd/system/uwsgi@.service
  576. # For uWSGI archlinux uses systemd template units, see
  577. # - http://0pointer.de/blog/projects/instances.html
  578. # - https://uwsgi-docs.readthedocs.io/en/latest/Systemd.html#one-service-per-app-in-systemd
  579. uWSGI_APPS_AVAILABLE="${uWSGI_SETUP}/apps-archlinux"
  580. uWSGI_APPS_ENABLED="${uWSGI_SETUP}"
  581. ;;
  582. fedora-*)
  583. # systemd --> /usr/lib/systemd/system/uwsgi.service
  584. # The unit file starts uWSGI in emperor mode (/etc/uwsgi.ini), see
  585. # - https://uwsgi-docs.readthedocs.io/en/latest/Emperor.html
  586. uWSGI_APPS_AVAILABLE="${uWSGI_SETUP}/apps-available"
  587. uWSGI_APPS_ENABLED="${uWSGI_SETUP}.d"
  588. uWSGI_USER="uwsgi"
  589. uWSGI_GROUP="uwsgi"
  590. ;;
  591. *)
  592. info_msg "$DIST_ID-$DIST_VERS: uWSGI not yet implemented"
  593. ;;
  594. esac
  595. uWSGI_restart() {
  596. # usage: uWSGI_restart() <myapp.ini>
  597. local CONF="$1"
  598. if [[ -z $CONF ]]; then
  599. err_msg "uWSGI_restart: missing arguments"
  600. return 42
  601. fi
  602. info_msg "restart uWSGI service"
  603. case $DIST_ID-$DIST_VERS in
  604. ubuntu-*|debian-*)
  605. # the 'service' method seems broken in that way, that it (re-)starts
  606. # the whole uwsgi process.
  607. service uwsgi restart "${CONF%.*}"
  608. ;;
  609. arch-*)
  610. # restart systemd template instance
  611. if uWSGI_app_available "${CONF}"; then
  612. systemctl restart "uwsgi@${CONF%.*}"
  613. else
  614. info_msg "in systemd template mode: ${CONF} not installed (nothing to restart)"
  615. fi
  616. ;;
  617. fedora-*)
  618. # in emperor mode, just touch the file to restart
  619. if uWSGI_app_enabled "${CONF}"; then
  620. touch "${uWSGI_APPS_ENABLED}/${CONF}"
  621. else
  622. info_msg "in uWSGI emperor mode: ${CONF} not installed (nothing to restart)"
  623. fi
  624. ;;
  625. *)
  626. err_msg "$DIST_ID-$DIST_VERS: uWSGI not yet implemented"
  627. return 42
  628. ;;
  629. esac
  630. }
  631. uWSGI_app_available() {
  632. # usage: uWSGI_app_available <myapp.ini>
  633. local CONF="$1"
  634. if [[ -z $CONF ]]; then
  635. err_msg "uWSGI_app_available: missing arguments"
  636. return 42
  637. fi
  638. [[ -f "${uWSGI_APPS_AVAILABLE}/${CONF}" ]]
  639. }
  640. uWSGI_install_app() {
  641. # usage: uWSGI_install_app [<template option> ...] <myapp.ini>
  642. #
  643. # <template option>: see install_template
  644. local pos_args=("$0")
  645. for i in "$@"; do
  646. case $i in
  647. -*) template_opts+=("$i");;
  648. *) pos_args+=("$i");;
  649. esac
  650. done
  651. mkdir -p "${uWSGI_APPS_AVAILABLE}"
  652. install_template "${template_opts[@]}" \
  653. "${uWSGI_APPS_AVAILABLE}/${pos_args[1]}" \
  654. root root 644
  655. uWSGI_enable_app "${pos_args[1]}"
  656. uWSGI_restart "${pos_args[1]}"
  657. info_msg "uWSGI app: ${pos_args[1]} is installed"
  658. }
  659. uWSGI_remove_app() {
  660. # usage: uWSGI_remove_app <myapp.ini>
  661. local CONF="$1"
  662. info_msg "remove uWSGI app: ${CONF}"
  663. uWSGI_disable_app "${CONF}"
  664. uWSGI_restart "${CONF}"
  665. rm -f "${uWSGI_APPS_AVAILABLE}/${CONF}"
  666. }
  667. uWSGI_app_enabled() {
  668. # usage: uWSGI_app_enabled <myapp.ini>
  669. local CONF="$1"
  670. local exit_val=0
  671. if [[ -z $CONF ]]; then
  672. err_msg "uWSGI_app_enabled: missing arguments"
  673. return 42
  674. fi
  675. case $DIST_ID-$DIST_VERS in
  676. ubuntu-*|debian-*)
  677. [[ -f "${uWSGI_APPS_ENABLED}/${CONF}" ]]
  678. exit_val=$?
  679. ;;
  680. arch-*)
  681. systemctl -q is-enabled "uwsgi@${CONF%.*}"
  682. exit_val=$?
  683. ;;
  684. fedora-*)
  685. [[ -f "${uWSGI_APPS_ENABLED}/${CONF}" ]]
  686. exit_val=$?
  687. ;;
  688. *)
  689. # FIXME
  690. err_msg "$DIST_ID-$DIST_VERS: uWSGI not yet implemented"
  691. exit_val=1
  692. ;;
  693. esac
  694. return $exit_val
  695. }
  696. # shellcheck disable=SC2164
  697. uWSGI_enable_app() {
  698. # usage: uWSGI_enable_app <myapp.ini>
  699. local CONF="$1"
  700. if [[ -z $CONF ]]; then
  701. err_msg "uWSGI_enable_app: missing arguments"
  702. return 42
  703. fi
  704. case $DIST_ID-$DIST_VERS in
  705. ubuntu-*|debian-*)
  706. mkdir -p "${uWSGI_APPS_ENABLED}"
  707. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  708. ln -s "${uWSGI_APPS_AVAILABLE}/${CONF}" "${uWSGI_APPS_ENABLED}/${CONF}"
  709. info_msg "enabled uWSGI app: ${CONF} (restart required)"
  710. ;;
  711. arch-*)
  712. mkdir -p "${uWSGI_APPS_ENABLED}"
  713. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  714. ln -s "${uWSGI_APPS_AVAILABLE}/${CONF}" "${uWSGI_APPS_ENABLED}/${CONF}"
  715. info_msg "enabled uWSGI app: ${CONF} (restart required)"
  716. ;;
  717. fedora-*)
  718. mkdir -p "${uWSGI_APPS_ENABLED}"
  719. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  720. ln -s "${uWSGI_APPS_AVAILABLE}/${CONF}" "${uWSGI_APPS_ENABLED}/${CONF}"
  721. chown "${uWSGI_USER}:${uWSGI_GROUP}" "${uWSGI_APPS_ENABLED}/${CONF}"
  722. info_msg "enabled uWSGI app: ${CONF}"
  723. ;;
  724. *)
  725. # FIXME
  726. err_msg "$DIST_ID-$DIST_VERS: uWSGI not yet implemented"
  727. ;;
  728. esac
  729. }
  730. uWSGI_disable_app() {
  731. # usage: uWSGI_disable_app <myapp.ini>
  732. local CONF="$1"
  733. if [[ -z $CONF ]]; then
  734. err_msg "uWSGI_disable_app: missing arguments"
  735. return 42
  736. fi
  737. case $DIST_ID-$DIST_VERS in
  738. ubuntu-*|debian-*)
  739. service uwsgi stop "${CONF%.*}"
  740. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  741. info_msg "disabled uWSGI app: ${CONF} (restart uWSGI required)"
  742. ;;
  743. arch-*)
  744. systemctl stop "uwsgi@${CONF%.*}"
  745. systemctl disable "uwsgi@${CONF%.*}"
  746. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  747. ;;
  748. fedora-*)
  749. # in emperor mode, just remove the app.ini file
  750. rm -f "${uWSGI_APPS_ENABLED}/${CONF}"
  751. ;;
  752. *)
  753. # FIXME
  754. err_msg "$DIST_ID-$DIST_VERS: uWSGI not yet implemented"
  755. ;;
  756. esac
  757. }
  758. # distro's package manager
  759. # ------------------------
  760. pkg_install() {
  761. # usage: TITEL='install foobar' pkg_install foopkg barpkg
  762. rst_title "${TITLE:-installation of packages}" section
  763. echo -en "\npackage(s)::\n\n $*\n" | $FMT
  764. if ! ask_yn "Should packages be installed?" Yn 30; then
  765. return 42
  766. fi
  767. case $DIST_ID in
  768. ubuntu|debian)
  769. # shellcheck disable=SC2068
  770. apt-get install -m -y $@
  771. ;;
  772. arch)
  773. # shellcheck disable=SC2068
  774. pacman -S --noconfirm $@
  775. ;;
  776. fedora)
  777. # shellcheck disable=SC2068
  778. dnf install -y $@
  779. ;;
  780. esac
  781. }
  782. pkg_remove() {
  783. # usage: TITEL='remove foobar' pkg_remove foopkg barpkg
  784. rst_title "${TITLE:-remove packages}" section
  785. echo -en "\npackage(s)::\n\n $*\n" | $FMT
  786. if ! ask_yn "Should packages be removed (purge)?" Yn 30; then
  787. return 42
  788. fi
  789. case $DIST_ID in
  790. ubuntu|debian)
  791. # shellcheck disable=SC2068
  792. apt-get purge --autoremove --ignore-missing -y $@
  793. ;;
  794. arch)
  795. # shellcheck disable=SC2068
  796. pacman -R --noconfirm $@
  797. ;;
  798. fedora)
  799. # shellcheck disable=SC2068
  800. dnf remove -y $@
  801. ;;
  802. esac
  803. }
  804. pkg_is_installed() {
  805. # usage: pkg_is_install foopkg || pkg_install foopkg
  806. case $DIST_ID in
  807. ubuntu|debian)
  808. dpkg -l "$1" &> /dev/null
  809. return $?
  810. ;;
  811. arch)
  812. pacman -Qsq "$1" &> /dev/null
  813. return $?
  814. ;;
  815. fedora)
  816. dnf list -q --installed "$1" &> /dev/null
  817. return $?
  818. ;;
  819. esac
  820. }
  821. # git tooling
  822. # -----------
  823. # shellcheck disable=SC2164
  824. git_clone() {
  825. # usage:
  826. #
  827. # git_clone <url> <name> [<branch> [<user>]]
  828. # git_clone <url> <path> [<branch> [<user>]]
  829. #
  830. # First form uses $CACHE/<name> as destination folder, second form clones
  831. # into <path>. If repository is allready cloned, pull from <branch> and
  832. # update working tree (if needed, the caller has to stash local changes).
  833. #
  834. # git clone https://github.com/asciimoo/searx searx-src origin/master searxlogin
  835. #
  836. local url="$1"
  837. local dest="$2"
  838. local branch="$3"
  839. local user="$4"
  840. local bash_cmd="bash"
  841. local remote="origin"
  842. if [[ ! "${dest:0:1}" = "/" ]]; then
  843. dest="$CACHE/$dest"
  844. fi
  845. [[ -z $branch ]] && branch=master
  846. [[ -z $user ]] && [[ -n "${SUDO_USER}" ]] && user="${SUDO_USER}"
  847. [[ -n $user ]] && bash_cmd="sudo -H -u $user -i"
  848. if [[ -d "${dest}" ]] ; then
  849. info_msg "already cloned: $dest"
  850. tee_stderr 0.1 <<EOF | $bash_cmd 2>&1 | prefix_stdout " |$user| "
  851. cd "${dest}"
  852. git checkout -m -B "$branch" --track "$remote/$branch"
  853. git pull --all
  854. EOF
  855. else
  856. info_msg "clone into: $dest"
  857. tee_stderr 0.1 <<EOF | $bash_cmd 2>&1 | prefix_stdout " |$user| "
  858. mkdir -p "$(dirname "$dest")"
  859. cd "$(dirname "$dest")"
  860. git clone --branch "$branch" --origin "$remote" "$url" "$(basename "$dest")"
  861. EOF
  862. fi
  863. }