#!/bin/bash

# Usage: ./ensure_header_footer.sh filename

FILE="$1"
HEADER=$(cat <<'EOF'
// Copyright (C) 2025 vx_clutch ( owestness@gmail.com )
// See end of file for extended copyright information.
EOF
)
HEADER+=$'\n'
FOOTER=$(cat <<'EOF'

/* yait is yet another init tool.
 * Copyright (C) 2025 vx-clutch
 *
 * This file is part of yait.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions, and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions, and the following disclaimer in the documentation or
 * other materials provided with the distribution.
 * 3. Neither the name of vx-clutch nor the names of its contributors may be
 * used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */
EOF
)

if [[ -z "$FILE" ]]; then
  echo "Usage: $0 <filename>"
  exit 1
fi

if [[ ! -f "$FILE" ]]; then
  echo "Error: File '$FILE' does not exist."
  exit 1
fi

TMP_FILE="$(mktemp)"

read -r FIRST_LINE < "$FILE"
if [[ "$FIRST_LINE" != "$HEADER" ]]; then
  echo "$HEADER" > "$TMP_FILE"
  cat "$FILE" >> "$TMP_FILE"
  mv "$TMP_FILE" "$FILE"
else
  rm "$TMP_FILE"
fi

if [[ "$(tail -n 1 "$FILE")" != "$FOOTER" ]]; then
  echo "$FOOTER" >> "$FILE"
fi

echo "Checked '$FILE' for header/footer."