#!/bin/bash

# Define the log format
LOG_FORMAT='log_format domain_combined '\''$host $remote_addr - $remote_user [$time_local] '\
''\"$request\" $status $body_bytes_sent '\
''\"$http_referer\" \"$http_user_agent\"'\'';'

# Check if we're root
if [ "$EUID" -ne 0 ]; then 
    echo "Please run as root"
    exit 1
fi

# Backup the original config
cp /etc/nginx/nginx.conf /etc/nginx/nginx.conf.bak

# Check if format already exists
if grep -q "log_format domain_combined" /etc/nginx/nginx.conf; then
    echo "domain_combined format already exists"
    exit 0
fi

# Add the format after the first 'http {' line
sed -i "/^http {/a \    ${LOG_FORMAT}" /etc/nginx/nginx.conf

# Test the nginx config
nginx -t

if [ $? -eq 0 ]; then
    echo "Successfully added domain_combined log format"
else
    echo "Error in nginx config, reverting..."
    mv /etc/nginx/nginx.conf.bak /etc/nginx/nginx.conf
    exit 1
fi