init.sh

#!/bin/bash

set -aeux

SETUP_REPO_DIR=/tmp/isucon-setup
GOLANG_VERSION=latest
MEMBERS=(tesso57 toshi-pono Ras96)

# 環境変数が設定されてなかったら終了
echo "PROJECT_ROOT=$PROJECT_ROOT" >> ~/.bashrc
echo "REPO_SSH_URL=$REPO_SSH_URL" >> ~/.bashrc
echo "APP_NAME=$APP_NAME" >> ~/.bashrc
echo "SERVICE_NAME=$SERVICE_NAME" >> ~/.bashrc
echo "DASHBOARD_HOST=$DASHBOARD_HOST" >> ~/.bashrc
echo "SERVER_ENV_PATH=$SERVER_ENV_PATH" >> ~/.bashrc
echo "SERVER_NUMBER=$SERVER_NUMBER" >> ~/.bashrc

# その他bashrcの設定
echo "export PATH=$PROJECT_ROOT/bin:\$PATH" >> ~/.bashrc
echo "export GOPATH=\"\"" >> ~/.bashrc
echo "export GOROOT=\"\"" >> ~/.bashrc

# 存在確認
[ ! -d $SETUP_REPO_DIR ] && echo "SETUP_REPO_DIR is not found" && exit 1
[ ! -d $PROJECT_ROOT ] && echo "PROJECT_ROOT is not found" && exit 1
[ ! -f $SERVER_ENV_PATH ] && echo "SERVER_ENV_PATH is not found" && exit 1

set +u

# aptからインストール
sudo apt install -y build-essential percona-toolkit htop git curl wget vim graphviz cmake flex bison

# 公開鍵を登録
mkdir -p ~/.ssh
for member in ${MEMBERS[@]}; do
  curl https://github.com/$member.keys >> ~/.ssh/authorized_keys
done

# isucon-setupから設定ファイルをコピー
cp -r $SETUP_REPO_DIR/bin $PROJECT_ROOT
cp -r $SETUP_REPO_DIR/fluent-bit $PROJECT_ROOT
sed -i "1i@SET dashboard_host=$DASHBOARD_HOST" $PROJECT_ROOT/fluent-bit/fluent-bit.conf

# Gitの設定
git config --global user.name "server"
git config --global user.email "github-actions[bot]@users.noreply.github.com"
git config --global core.editor "vim"
git config --global push.default current
git config --global init.defaultbranch main
git config --global fetch.prune true
git config --global alias.lo "log --oneline"
# git branch -M main
if [ ! -d $PROJECT_ROOT/.git ]; then
  cd $PROJECT_ROOT
  git init
  git remote add origin $REPO_SSH_URL
  if [ -d $PROJECT_ROOT/.git/logs ]; then
    git fetch origin
    git reset --hard origin/main
  fi
fi

# env.sh,.bashrcにシンボリックリンクを貼る
confdir=$PROJECT_ROOT/isu$SERVER_NUMBER
mkdir -p $confdir
if [ ! -L $SERVER_ENV_PATH ]; then
  mv $SERVER_ENV_PATH $confdir
  ln -sf $confdir/env.sh $SERVER_ENV_PATH
fi
if [ ! -L ~/.bashrc ]; then
  mv ~/.bashrc $confdir
  ln -sf $confdir/.bashrc ~/.bashrc
fi

# asdfをインストール
if [ ! -d ~/.asdf ]; then
  git clone https://github.com/asdf-vm/asdf.git ~/.asdf
  echo "source ~/.asdf/asdf.sh" >> ~/.bashrc
  echo "source ~/.asdf/completions/asdf.bash" >> ~/.bashrc
  source ~/.asdf/asdf.sh

  # asdfからgoをインストール
  asdf plugin add golang
  asdf install golang $GOLANG_VERSION
  asdf global golang $GOLANG_VERSION
fi

# MySQLTuner-perl をインストール
if [ ! -d /usr/local/src/MySQLTuner-perl ]; then
  sudo git clone --depth 1 -b master https://github.com/major/MySQLTuner-perl.git /usr/local/src/MySQLTuner-perl
  sudo ln -sf /usr/local/src/MySQLTuner-perl/mysqltuner.pl /usr/local/bin/mysqltuner.pl
fi

# aptかソースからfluent-bitをインストール
# TODO: https://github.com/fluent/fluent-bit/issues/5628
if [ "$(cat /etc/issue | awk '{print $2}')" == "22.04" ]; then
  [ ! -d /tmp/fluent-bit ] && git clone --depth 1 https://github.com/fluent/fluent-bit.git /tmp/fluent-bit
  cd /tmp/fluent-bit/build
  cmake ../ -DFLB_CONFIG_YAML=Off
  make
  sudo cp /tmp/fluent-bit/build/bin/fluent-bit /usr/local/bin
else
  curl https://raw.githubusercontent.com/fluent/fluent-bit/master/install.sh | sh
  sudo cp /opt/fluent-bit/bin/fluent-bit /usr/local/bin
fi

# fluent-bitを常時動かす
sudo rm -rf /etc/fluent-bit
sudo ln -sf $PROJECT_ROOT/fluent-bit /etc
$PROJECT_ROOT/bin/restart-fluent-bit

set +x

echo "Done!!! (you should restart shell & push diff to github)"
echo ""
echo "- source ~/.bashrc"
echo "- vim .gitignore"
echo "- git commit -a -m 'initial commit'"