#!/bin/sh -e

# This script is run if an optical drive lacks a rule for persistent naming.
#
# It adds symlinks for optical drives based on the device class determined
# by cdrom_id and used ID_PATH or ID_SERIAL or ID_MODEL and ID_REVISION
# to identify the device.
#
# (C) 2006 Marco d'Itri <md@Linux.IT>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation version 2 of the License.

# debug, if UDEV_LOG=<debug>
if [ "$UDEV_LOG" ]; then
	if [ "$UDEV_LOG" -ge 7 ]; then
		set -x
	fi
fi

RULES_FILE="/etc/udev/rules.d/70-persistent-cd.rules"

. /lib/udev/rule_generator.functions

find_next_available() {
	raw_find_next_available "$(find_all_rules 'SYMLINK\+=' "$1")"
}

write_rule() {
	local match="$1"
	local link="$2"
	local comment="$3"

	{
	if [ "$PRINT_HEADER" ]; then
		PRINT_HEADER=
		echo "# This file was automatically generated by the $0"
		echo "# program, run by the cd-aliases-generator.rules rules file."
		echo "#"
		echo "# You can modify it, as long as you keep each rule on a single"
		echo "# line, and set the \$GENERATED variable."
		echo ""
	fi

	[ "$comment" ] && echo "# $comment"
	echo "$match, SYMLINK+=\"$link\", ENV{GENERATED}=\"1\""
	} >> $RULES_FILE
	SYMLINKS="$SYMLINKS $link"
}

if [ ! "$DEVPATH" ]; then
	echo "Missing \$DEVPATH." >&2
	exit 1
fi
if [ ! "$ID_CDROM" ]; then
	echo "$DEVPATH is not a CD reader." >&2
	exit 1
fi

if [ "$1" ]; then
	METHOD="$1"
else
	METHOD='by-path'
fi

case "$METHOD" in
	by-path)
	if [ ! "$ID_PATH" ]; then
		echo "$DEVPATH not supported by path_id. by-id may work." >&2
		exit 1
	fi
	RULE="ENV{ID_PATH}==\"$ID_PATH\""
	;;

	by-id)
	if [ "$ID_SERIAL" ]; then
		RULE="ENV{ID_SERIAL}==\"$ID_SERIAL\""
	elif [ "$ID_MODEL" -a "$ID_REVISION" ]; then
		RULE="ENV{ID_MODEL}==\"$ID_MODEL\", ENV{ID_REVISION}==\"$ID_REVISION\""
	else
		echo "$DEVPATH not supported by ata_id. by-path may work." >&2
		exit 1
	fi
	;;

	*)
	echo "Invalid argument (must be either by-path or by-id)." >&2
	exit 1
	;;
esac

# Prevent concurrent processes from modifying the file at the same time.
lock_rules_file

# Check if the rules file is writeable.
choose_rules_file

link_num=$(find_next_available 'cdrom[0-9]*')
[ ! "$link_num"  ] && link_num=0

match="ENV{ID_CDROM}==\"?*\", $RULE"

comment="$ID_MODEL ($ID_PATH)"
	write_rule "$match" "cdrom$link_num" "$comment"
	if [ ! "$(find_all_rules 'SYMLINK+=' cdrom)" ]; then
		write_rule "$match" "cdrom"
	fi

	if [ "$ID_CDROM_CD_R" ]; then
	       	write_rule "$match" "cdr$link_num"
		if [ ! "$(find_all_rules 'SYMLINK+=' cdr)" ]; then
			write_rule "$match" "cdr"
		fi

		write_rule "$match" "cdwriter$link_num"
		if [ ! "$(find_all_rules 'SYMLINK+=' cdwriter)" ]; then
			write_rule "$match" "cdwriter"
		fi

		if [ "$ID_CDROM_CD_RW" ]; then
			write_rule "$match" "cdrw$link_num"
			if [ ! "$(find_all_rules 'SYMLINK+=' cdrw)" ]; then
				write_rule "$match" "cdrw"
			fi
		fi

		if [ ! "$(find_all_rules 'SYMLINK+=' writer)" ]; then
			write_rule "$match" "writer"
		fi
	fi

	if [  "$ID_CDROM_DVD" ]; then
		write_rule "$match" "dvd$link_num"
		if [ ! "$(find_all_rules 'SYMLINK+=' dvd)" ]; then
			write_rule "$match" "dvd"
		fi

		if [ "$ID_CDROM_DVD_R" -o "$ID_CDROM_DVD_RW" -o "$ID_CDROM_DVD_RAM" ]; then
			write_rule "$match" "dvdrw$link_num"
			if [ ! "$(find_all_rules 'SYMLINK+=' dvdrw)" ]; then
				write_rule "$match" "dvdrw"
			fi

			write_rule "$match" "dvdwriter$link_num"
			if [ ! "$(find_all_rules 'SYMLINK+=' dvdwriter)" ]; then
				write_rule "$match" "dvdwriter"
			fi
		fi
	fi



unlock_rules_file

echo $SYMLINKS

exit 0