#!/bin/bash
# install promtail
# by zack.zhu@aloudata.com
# 202112301633

FILE_NAME="promtail-linux-amd64"
FILE_URL="http://10.5.20.30:8081/tools/promtail-linux-amd64"
PROMTAIL_HOME="/data/promtail"
PROMTAIL_CONF="$PROMTAIL_HOME/promtail-config.yaml"
IP_ADDR=$(ip a|grep inet|grep ' 10.[5|10].'|awk '{print $2}'|awk -F '/' '{print $1}')
LOKI_SERVER='loki.dev.aloudata.work'
# IP_ADDR=$(ifconfig $(route -n |awk '$1=="0.0.0.0" {print $NF}')|grep 'inet '|awk '{print $2}')
mkdir -p $PROMTAIL_HOME >/dev/null 2>&1
curl -s -o $PROMTAIL_HOME/$FILE_NAME $FILE_URL
if [ $? -ne 0 ]; then
        echo "download file is error."
        exit 1
fi
chmod +x $PROMTAIL_HOME/$FILE_NAME

cat > $PROMTAIL_CONF <<EOF
server:
  http_listen_port: 9080
  grpc_listen_port: 0

positions:
  filename: $PROMTAIL_HOME/positions.yaml

clients:
  - url: http://${LOKI_SERVER}:3100/loki/api/v1/push

scrape_configs:
- job_name: aloudsight
  static_configs:
  - targets:
      - localhost
    labels:
      job: aloudsight
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/aloudsight/*.log
  pipeline_stages:
  - match:
      selector: '{job="dws"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: dws
  static_configs:
  - targets:
      - localhost
    labels:
      job: dws
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/dws/*.log
  pipeline_stages:
  - match:
      selector: '{job="dws"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: das
  static_configs:
  - targets:
      - localhost
    labels:
      job: das
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/das/*.log
  pipeline_stages:
  - match:
      selector: '{job="das"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: cip
  static_configs:
  - targets:
      - localhost
    labels:
      job: cip
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/cip/*.log
  pipeline_stages:
  - match:
      selector: '{job="cip"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: afp
  static_configs:
  - targets:
      - localhost
    labels:
      job: afp
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/afp/*.log
  pipeline_stages:
  - match:
      selector: '{job="afp"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: asp
  static_configs:
  - targets:
      - localhost
    labels:
      job: asp
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/asp/*.log
  pipeline_stages:
  - match:
      selector: '{job="asp"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: bde
  static_configs:
  - targets:
      - localhost
    labels:
      job: dbe
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/dbe/*.log
  pipeline_stages:
  - match:
      selector: '{job="dbe"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: tse
  static_configs:
  - targets:
      - localhost
    labels:
      job: tse
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/tse/*.log
  pipeline_stages:
  - match:
      selector: '{job="tse"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: amq
  static_configs:
  - targets:
      - localhost
    labels:
      job: amq
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/amq/*.log
  pipeline_stages:
  - match:
      selector: '{job="amq"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: ame
  static_configs:
  - targets:
      - localhost
    labels:
      job: ame
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/ame/*.log
  pipeline_stages:
  - match:
      selector: '{job="ame"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
- job_name: airengine
  static_configs:
  - targets:
      - localhost
    labels:
      job: airengine
      host: $IP_ADDR
      environment: test
      __path__: /data/logs/airengine/*.log
  pipeline_stages:
  - match:
      selector: '{job="airengine"}'
      stages:
        - multiline:
            firstline: '^\d{4}-\d{2}-\d{2} \d{1,2}:\d{2}:\d{2}'
            max_wait_time: 3s
EOF

if [ $? -ne 0 ]; then
        echo 'config error.'
        exit 1
fi

cat > /usr/lib/systemd/system/promtail.service <<EOF
[Unit]
Description=promtail
After=network.target
[Service]
Type=simple
User=root
ExecStart=$PROMTAIL_HOME/$FILE_NAME -config.file=$PROMTAIL_CONF -config.expand-env=true
Restart=on-failure
[Install]
WantedBy=multi-user.target
EOF

if [ $? -ne 0 ]; then
        echo "systemd error."
        exit 1
fi

systemctl daemon-reload
systemctl enable promtail.service
systemctl start promtail.service