lib.sh 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149
  1. #!/bin/bash
  2. # NOTE: This file is sourced in CI across different repos (e.g. snuba),
  3. # thus, renaming this file or any functions can break CI!
  4. #
  5. # Module containing code shared across various shell scripts
  6. # Execute functions from this module via the script do.sh
  7. # shellcheck disable=SC2034 # Unused variables
  8. # shellcheck disable=SC2001 # https://github.com/koalaman/shellcheck/wiki/SC2001
  9. # This block is a safe-guard since in CI calling tput will fail and abort scripts
  10. if [ -z "${CI+x}" ]; then
  11. bold="$(tput bold)"
  12. red="$(tput setaf 1)"
  13. green="$(tput setaf 2)"
  14. yellow="$(tput setaf 3)"
  15. reset="$(tput sgr0)"
  16. fi
  17. venv_name=".venv"
  18. # XDG paths' standardized defaults:
  19. # (see https://specifications.freedesktop.org/basedir-spec/basedir-spec-latest.html#variables )
  20. export XDG_DATA_HOME="${XDG_DATA_HOME:-$HOME/.local/share}"
  21. export XDG_CONFIG_HOME="${XDG_CONFIG_HOME:-$HOME/.config}"
  22. export XDG_STATE_HOME="${XDG_STATE_HOME:-$HOME/.local/state}"
  23. export XDG_DATA_DIRS="${XDG_DATA_DIRS:-/usr/local/share/:/usr/share/}"
  24. export XDG_CONFIG_DIRS="${XDG_CONFIG_DIRS:-/etc/xdg}"
  25. export XDG_CACHE_HOME="${XDG_CACHE_HOME:-$HOME/.cache}"
  26. export XDG_RUNTIME_DIR="${XDG_RUNTIME_DIR:-/var/run}"
  27. # Check if a command is available
  28. require() {
  29. command -v "$1" >/dev/null 2>&1
  30. }
  31. query-valid-python-version() {
  32. python_version=$(python3 -V 2>&1 | awk '{print $2}')
  33. if [[ -n "${SENTRY_PYTHON_VERSION:-}" ]]; then
  34. if [ "$python_version" != "$SENTRY_PYTHON_VERSION" ]; then
  35. cat <<EOF
  36. ${red}${bold}
  37. ERROR: You have explicitly set a non-recommended Python version (${SENTRY_PYTHON_VERSION}),
  38. but it doesn't match the value of python's version: ${python_version}
  39. You should create a new ${SENTRY_PYTHON_VERSION} virtualenv by running "rm -rf ${venv_name} && devenv sync".
  40. ${reset}
  41. EOF
  42. return 1
  43. else
  44. cat <<EOF
  45. ${yellow}${bold}
  46. You have explicitly set a non-recommended Python version (${SENTRY_PYTHON_VERSION}). You're on your own.
  47. ${reset}
  48. EOF
  49. return 0
  50. fi
  51. else
  52. minor=$(echo "${python_version}" | sed 's/[0-9]*\.\([0-9]*\)\.\([0-9]*\)/\1/')
  53. patch=$(echo "${python_version}" | sed 's/[0-9]*\.\([0-9]*\)\.\([0-9]*\)/\2/')
  54. if [ "$minor" -ne 12 ] || [ "$patch" -lt 1 ]; then
  55. cat <<EOF
  56. ${red}${bold}
  57. ERROR: You're running a virtualenv with Python ${python_version}.
  58. We only support >= 3.12.1, < 3.13.
  59. Either run "rm -rf ${venv_name} && direnv allow" to
  60. OR set SENTRY_PYTHON_VERSION=${python_version} to an .env file to bypass this check."
  61. EOF
  62. return 1
  63. fi
  64. fi
  65. }
  66. sudo-askpass() {
  67. if [ -z "${sudo-askpass-x}" ]; then
  68. sudo --askpass "$@"
  69. else
  70. sudo "$@"
  71. fi
  72. }
  73. init-config() {
  74. sentry init --dev --no-clobber
  75. }
  76. run-dependent-services() {
  77. sentry devservices up
  78. }
  79. create-db() {
  80. container_name=${POSTGRES_CONTAINER:-sentry_postgres}
  81. echo "--> Creating 'sentry' database"
  82. docker exec "${container_name}" createdb -h 127.0.0.1 -U postgres -E utf-8 sentry || true
  83. echo "--> Creating 'control', 'region' and 'secondary' database"
  84. docker exec "${container_name}" createdb -h 127.0.0.1 -U postgres -E utf-8 control || true
  85. docker exec "${container_name}" createdb -h 127.0.0.1 -U postgres -E utf-8 region || true
  86. docker exec "${container_name}" createdb -h 127.0.0.1 -U postgres -E utf-8 secondary || true
  87. }
  88. apply-migrations() {
  89. create-db
  90. echo "--> Applying migrations"
  91. sentry upgrade --noinput
  92. }
  93. create-superuser() {
  94. echo "--> Creating a superuser account"
  95. if [[ -n "${GITHUB_ACTIONS+x}" ]]; then
  96. sentry createuser --superuser --email foo@tbd.com --no-password --no-input
  97. else
  98. sentry createuser --superuser --email admin@sentry.io --password admin --no-input
  99. echo "Password is admin."
  100. fi
  101. }
  102. build-platform-assets() {
  103. echo "--> Building platform assets"
  104. python3 -m sentry.build._integration_docs
  105. # make sure this didn't silently do nothing
  106. test -f src/sentry/integration-docs/android.json
  107. }
  108. clean() {
  109. echo "--> Cleaning static cache"
  110. rm -rf dist/* src/sentry/static/sentry/dist/*
  111. echo "--> Cleaning integration docs cache"
  112. rm -rf src/sentry/integration-docs
  113. echo "--> Cleaning pyc files"
  114. find . -name "*.pyc" -delete
  115. echo "--> Cleaning python build artifacts"
  116. rm -rf build/ dist/ src/sentry/assets.json
  117. echo ""
  118. }
  119. drop-db() {
  120. container_name=${POSTGRES_CONTAINER:-sentry_postgres}
  121. echo "--> Dropping existing 'sentry' database"
  122. docker exec "${container_name}" dropdb --if-exists -h 127.0.0.1 -U postgres sentry
  123. echo "--> Dropping 'control' and 'region' database"
  124. docker exec "${container_name}" dropdb --if-exists -h 127.0.0.1 -U postgres control
  125. docker exec "${container_name}" dropdb --if-exists -h 127.0.0.1 -U postgres region
  126. docker exec "${container_name}" dropdb --if-exists -h 127.0.0.1 -U postgres secondary
  127. }
  128. reset-db() {
  129. drop-db
  130. apply-migrations
  131. create-superuser
  132. echo 'Finished resetting database. To load mock data, run `./bin/load-mocks`'
  133. }