1
0

init.sh 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
  1. #!/bin/bash
  2. #======================================
  3. # Author: ulyc
  4. #======================================
  5. set -eo pipefail
  6. red='\e[91m'
  7. green='\e[92m'
  8. yellow='\e[93m'
  9. magenta='\e[95m'
  10. cyan='\e[96m'
  11. none='\e[0m'
  12. _red() { echo -e "${red}$*${none}"; }
  13. _green() { echo -e "${green}$*${none}"; }
  14. _yellow() { echo -e "${yellow}$*${none}"; }
  15. _magenta() { echo -e "${magenta}$*${none}"; }
  16. _cyan() { echo -e "${cyan}$*${none}"; }
  17. domain_name="http://127.0.0.1"
  18. modules=""
  19. meta_module="meta.sr.ht"
  20. git_module="git.sr.ht"
  21. hg_module="hg.sr.ht"
  22. build_module="builds.sr.ht"
  23. list_module="lists.sr.ht"
  24. man_module="man.sr.ht"
  25. paste_module="paste.sr.ht"
  26. todo_module="todo.sr.ht"
  27. function add_module() {
  28. read -r input
  29. echo
  30. if [[ "$input" -le 1 ]]; then
  31. modules=$modules" "$1
  32. echo -e "$yellow $1 is selected${none}"
  33. fi
  34. }
  35. function progress() {
  36. echo "$2 IS RUNNING..."
  37. printf "[▓"
  38. while kill -0 "$1" 2>/dev/null; do
  39. printf "▓"
  40. sleep 0.05
  41. done
  42. wait "$1"
  43. printf "▓] done!"
  44. }
  45. function generate_config() {
  46. service_key=$(grep <genkeys "Service" | awk -F: '{gsub(/[[:blank:]]*/,"");print $2}')
  47. network_key=$(grep <genkeys "Network" | awk -F: '{gsub(/[[:blank:]]*/,"");print $2}')
  48. webhook_key=$(grep <genkeys "Webhook Private" | awk -F: '{gsub(/[[:blank:]]*/,"");print $2}')
  49. sed -i "s/{{SERVICE_KEY}}/$service_key/" config.ini
  50. sed -i "s/{{NETWORK_KEY}}/$network_key/" config.ini
  51. sed -i "s@{{WEBHOOK_KEY}}@$webhook_key@" config.ini
  52. sed -i "s@{{DOMAIN}}@$domain_name@" config.ini
  53. }
  54. function generate_keys() {
  55. docker run sr.ht-base:dev sh -c 'srht-keygen service && srht-keygen network && srht-keygen webhook' | awk '{n[1]="Service";n[2]="Network";n[3]=n[4]="Webhook";print n[NR]" "$0 > "genkeys"}'
  56. }
  57. function build_base_image() {
  58. docker build -t sr.ht-base:dev ./base/ && echo
  59. }
  60. function generate_launch_shell() {
  61. md="$(echo "$meta_module $1" | awk '{ gsub(/\./,""); print $0 }')"
  62. md_count=$(echo "$md" | awk '{print NF}')
  63. cp ./template/config.ini.template config.ini
  64. echo "" >"start.sh"
  65. for ((i = 1; i <= "$md_count"; i++)); do
  66. m=$(echo "$md" | awk -v i="$i" '{print $i }')
  67. mp="500$i"
  68. echo "/usr/bin/gunicorn $m.app:app -b 0.0.0.0:$mp -D" >>"start.sh"
  69. m_domain="http://$m.$domain_name"
  70. if [[ "$domain_name" == "http://localhost" ]]; then
  71. m_domain=$domain_name:$mp
  72. fi
  73. tp="{{"$m"_domain}}"
  74. sed -i "s@$tp@$m_domain@" config.ini
  75. done
  76. sed "s/{{PORTS}}/5001-500$md_count:5001-500$md_count/" ./template/docker-compose.yml.template >docker-compose.yml
  77. echo "tail -f /dev/null" >>start.sh
  78. }
  79. function select_version_control() {
  80. # Git or Mercurial or Both
  81. echo -e "Select your distributed version control system ${cyan}1.Git${none} or ${cyan}2.Mercurial${none} or ${cyan}3.Both${none}?"
  82. read -r version_control_system_input
  83. echo
  84. if [[ "$version_control_system_input" -le 1 ]]; then
  85. modules=$modules" "$git_module
  86. echo -e "$yellow $modules is selected${none}"
  87. elif [[ "$version_control_system_input" -le 2 ]]; then
  88. modules=$modules" "$hg_module
  89. echo -e "$yellow $modules is selected${none}"
  90. else
  91. modules=$modules' '$git_module' '$hg_module
  92. echo -e "$yellow $modules are selected${none}"
  93. fi
  94. }
  95. function advance_config() {
  96. # CI
  97. # builds.sr.ht
  98. echo -e "Do you want to use ${cyan} Sourcehut CI ${none}? ${cyan}1.Yes${none} or ${cyan}2.No${none}?"
  99. add_module "$build_module"
  100. echo -e "Do you want to use ${cyan}Mailing list service${none}? ${cyan}1.Yes${none} or ${cyan}2.No${none}?"
  101. add_module "$list_module"
  102. # wiki service
  103. # man.sr.ht
  104. echo -e "Do you want to use ${cyan}Wiki service${none}? ${cyan}1.Yes${none} or ${cyan}2.No${none}?"
  105. add_module "$man_module"
  106. # Syntax highlighting
  107. # paste.sr.ht
  108. echo -e "Do you want to use ${cyan}Syntax highlighting${none}? ${cyan}1.Yes${none} or ${cyan}2.No${none}?"
  109. add_module "$paste_module"
  110. # issue and bug tracker service
  111. # todo.sr.ht
  112. echo -e "Do you want to use ${cyan}issue and bug tracker service${none}? ${cyan}1.Yes${none} or ${cyan}2.No${none}?"
  113. add_module "$todo_module"
  114. }
  115. function select_mode() {
  116. min_mode="Minimal Installation(Install only required modules)"
  117. max_mod="Maximize Installation(Install all modules)"
  118. adv_mod="Advanced Mode(Choose your own personalized modules)"
  119. quit="Quit"
  120. mode=("$min_mode" "$max_mod" "$adv_mod" "$quit")
  121. PS3="Select the installation mode: "
  122. select item in "${mode[@]}"; do
  123. case $item in
  124. "$min_mode")
  125. echo "$item"
  126. select_version_control
  127. break
  128. ;;
  129. "$max_mod")
  130. echo "$item"
  131. modules=$modules"$git_module $hg_module $build_module $list_module $man_module $paste_module $todo_module"
  132. break
  133. ;;
  134. "$adv_mod")
  135. echo "$item"
  136. advance_config
  137. break
  138. ;;
  139. "$quit")
  140. exit
  141. ;;
  142. quit)
  143. break
  144. ;;
  145. *)
  146. echo "Invalid option $REPLY"
  147. ;;
  148. esac
  149. done
  150. }
  151. ## Start
  152. select_mode
  153. # Set Domain
  154. echo -e "Set your ${cyan} Domain name (no http(s) prefix )${none} or ${cyan}s (skip) to use localhost${none}?"
  155. read -r domain_input
  156. echo
  157. echo "domain_input :$domain_input"
  158. if [[ "$domain_input" == 's' || -z "$domain_input" ]]; then
  159. echo -e "$yellow skip${none}"
  160. else
  161. domain_name=$domain_input
  162. fi
  163. echo "domain: $domain_name"
  164. echo
  165. echo
  166. generate_launch_shell "$modules" &
  167. progress $! "🤖 Generate Launch Shell"
  168. echo
  169. echo
  170. sed "s/{{MODULES}}/$modules/" ./template/Dockerfile.template >Dockerfile &
  171. progress $! "🐋 Generate Dockerfile"
  172. echo
  173. echo
  174. build_base_image &
  175. progress $! "🐋 Build Base Image"
  176. echo
  177. echo
  178. generate_keys &
  179. progress $! "🔒 Generate Keys"
  180. echo
  181. echo
  182. echo -e "$(<genkeys)"
  183. echo -e "${yellow}Distribute the webhook public key to anyone who would want to verify ${none}"
  184. echo -e "${yellow}webhook payloads from your service.${none}"
  185. echo -e "${yellow}you can see generated keys in the 'genkeys' file ${none}"
  186. echo
  187. echo
  188. generate_config
  189. progress $! "🔧 Generate Config"
  190. echo
  191. echo
  192. # TODO modules config
  193. # TODO Database secret