#!/bin/bash
set -e

function usage {
	echo "${0##*/} <pv|hvm|baremetal> <guest> [vdisk]"
	exit 1
}

[[ -z $2 ]] && usage

platform=$1
[[ $platform != pv &&
   $platform != hvm &&
   $platform != baremetal ]] && usage
guest=$2

vdisk=$3

[[ ! -f network.conf ]] && echo network.conf missing && exit 1
source network.conf

[[ -z $sets ]] && echo \$sets not defined -- tune network.conf first && exit 1

[[ ! -f $sets/base.tar.xz ]] && echo $sets/base.tar.xz not found && exit 1
[[ $platform != pv && ! -f $sets/netbsd-INSTALL.gz ]] && echo $set/netbsd-INSTALL.gz not found && exit 1

[[ ! -z `mount | grep lala` ]] && echo lala/ already mounted && exit 1

tmp=`xl li | grep $guest` || true
[[ ! -z $tmp ]] && echo XEN guest is running && echo "$tmp" && exit 1
unset tmp

tmp=`losetup --list | grep $guest | awk '{print $1}'`
[[ ! -z $tmp ]] && echo already looped && echo "$tmp" && exit 1
unset tmp

[[ -f $guest.sparse ]] && echo $guest.sparse already exists && exit 1
if [[ $platform != baremetal ]]; then
	rm -f $guest.sparse $guest.sfdisk
	dd if=/dev/zero of=$guest.sparse bs=1000k count=0 seek=4000
fi
echo

#need more than 512MB for pkg_add to be happy?  getting core dumps because of... what?
[[ $platform = pv ]] && echo -n XEN/PV guest setup... && cat > $guest <<EOF && echo done
#kernel = "$kernel/netbsd-INSTALL_XEN3_DOMU.gz"
kernel = "$kernel/netbsd-XEN3_DOMU.gz"
root = "xbd0a"
#extra = "-v -s"
name = "$guest"
memory = 1024
vcpus = 3
#disk = [ 'tap:tapdisk:aio:/data/guests/$guest/$guest.sparse,xvda,w' ]
disk = [ 'phy:$vdisk,xvda,w' ]
vif = [ 'bridge=br0,vifname=$guest.0' ]
EOF

[[ $platform = hvm ]] && echo XEN/HVM guest setup... \\c
[[ $platform = hvm ]] && cat > $guest <<EOF
#type = "hvm"
builder = "hvm"
#on_reboot = "preserve"
on_crash = "preserve"
sdl = 1
name = "$guest"
memory = 1024
vcups = 3
disk = [ 'tap:tapdisk:aio:/data/guests/$guest/$guest.sparse,xvda,w' ]
vif = [ 'bridge=xenbr0,vifname=ext2fs.0' ]
#nographics=1
#serial='pty'
EOF

if [[ $platform = hvm ]]; then
	disk=`losetup --find --partscan --show $guest.sparse`
	partition=${disk}p1
elif [[ $platform = baremetal ]]; then
	disk=/dev/sda
	partition=/dev/sda1
elif [[ $platform = pv ]]; then
	# no need to define disk
	if [[ -n $vdisk ]]; then
		partition=$vdisk
	else
		partition=$guest.sparse
	fi
	echo partition is $partition
fi
echo

[[ $platform != pv ]] && total=`fdisk -l $disk | head -1 | awk '{print $7}'`
[[ $platform != pv ]] && cat > $guest.sfdisk <<EOF
label: dos
label-id: 0x8ae$((RANDOM % 10))$((RANDOM % 10))3cc
device: $disk
unit: sectors

$partition : start=        2048, size=     $((total-2048)), type=a9, bootable
EOF
[[ $platform != pv ]] && sfdisk $disk < $guest.sfdisk
[[ $platform != pv ]] && partprobe $disk && sleep 1

mkfs.ext2 -O^dir_index,^ext_attr $partition

mkdir -p lala/
mount -t ext2 $partition lala/

#no debug nor tests
for set in base comp etc man misc modules text; do
	echo -n extracting $set...
	tar xJph --numeric-owner -f $sets/$set.tar.xz -C lala/ && echo done
	# tar xzph
done; unset set
#ls -lkF lala/sbin/fsck_ext2fs # NOT 75064 --> 71136 --> 75672

if [[ $platform != pv ]]; then
	echo -n adding kern-GENERIC and netbsd-INSTALL.gz for baremetal or hvm...
	tar xJph --numeric-owner -f $sets/kern-GENERIC.tar.xz -C lala/ && echo done
	cp -v $kernel/netbsd-INSTALL.gz lala/
fi

OWD=`pwd`
echo working directory is $OWD/
ls -alkF lala/dev/
cd lala/dev/
#grep ' -F netbsd' MAKEDEV
#grep ' -R"' MAKEDEV
#grep ' -r"' MAKEDEV
#grep ' -g ' MAKEDEV
sed -r 's/ -F netbsd//;
	s/ -R"/"/;
	s/ -r"/"/;
	s/ -g [^ ]+ -u [^ ]+//;
	' MAKEDEV > MAKEDEV.linux
#diff grep exists non-zero
#diff -bu MAKEDEV MAKEDEV.linux 
#grep ' -F netbsd' MAKEDEV.linux
#grep ' -R"' MAKEDEV.linux
#grep ' -r"' MAKEDEV.linux
#grep ' -g ' MAKEDEV.linux
chmod +x MAKEDEV.linux
./MAKEDEV.linux all 2>/dev/null
./MAKEDEV.linux opty
cd $OWD/
unset OWD

#does not need to be executable
mv lala/etc/rc.local lala/etc/rc.local.dist
cat > lala/etc/rc.local <<EOF
if [ ! -f /.makedev_done ]; then
	# CWD is /
	cd /dev/
	echo -n MAKEDEV all...
	./MAKEDEV all && date > /.makedev_done && echo done
	echo -n MAKEDEV opty...
	./MAKEDEV opty && echo done
	cd /
	echo OWNERSHIPS FIXED E.G.
	ls -lkF /dev/cgd0
	ls -lkF /dev/altq/altq
fi
EOF

mv -i lala/etc/passwd lala/etc/passwd.dist
sed '1s@/bin/sh@/bin/ksh@' lala/etc/passwd.dist > lala/etc/passwd
#diff -u lala/etc/passwd.dist lala/etc/passwd

mv -i lala/etc/master.passwd lala/etc/master.passwd.dist
sed '1s@/bin/sh@/bin/ksh@' lala/etc/master.passwd.dist > lala/etc/master.passwd
#diff -u lala/etc/master.passwd.dist lala/etc/master.passwd

echo writing fstab ...

[[ $platform != pv ]] && cat > lala/etc/fstab <<EOF && echo done
/dev/wd0e / ext2fs rw 0 1
ptyfs /dev/pts ptyfs rw 0 0
tmpfs /tmp tmpfs rw 0 0
#procfs /proc procfs ro,linux 0 0
EOF

[[ $platform = pv ]] && cat > lala/etc/fstab <<EOF && echo done
/dev/xbd0a / ext2fs rw 0 1
EOF

[[ $platform != pv ]] && wscons=yes
[[ $platform = pv ]] && wscons=no
mv -i lala/etc/rc.conf lala/etc/rc.conf.dist
cat > lala/etc/rc.conf <<EOF
. /etc/defaults/rc.conf
rc_silent=no # /var/run/rc.log
rc_configured=yes
inetd=no
wscons=$wscons
no_swap=yes
savecore=no
powerd=yes
EOF
unset wscons

#enabling E1,2,3
[[ $platform != pv ]] && cp -pi lala/etc/ttys lala/etc/ttys.dist
[[ $platform != pv ]] && cat > lala/etc/ttys <<EOF
console	"/usr/libexec/getty Pc"		vt100	on secure
constty	"/usr/libexec/getty Pc"		vt100	off secure
ttyE0	"/usr/libexec/getty Pc"		wsvt25	off secure
ttyE1	"/usr/libexec/getty Pc"		wsvt25	on secure
ttyE2	"/usr/libexec/getty Pc"		wsvt25	on secure
ttyE3	"/usr/libexec/getty Pc"		wsvt25	on secure
tty00	"/usr/libexec/getty std.9600"	unknown off secure
tty01	"/usr/libexec/getty std.9600"	unknown off secure
tty02	"/usr/libexec/getty std.9600"	unknown off secure
tty03	"/usr/libexec/getty std.9600"	unknown off secure
tty04	"/usr/libexec/getty std.9600"	unknown off secure
tty05	"/usr/libexec/getty std.9600"	unknown off secure
tty06	"/usr/libexec/getty std.9600"	unknown off secure
tty07	"/usr/libexec/getty std.9600"	unknown off secure
EOF

[[ $platform != pv ]] && grub-install --no-floppy --boot-directory=lala $disk
[[ $platform != pv ]] && cat > lala/grub/grub.cfg <<EOF
set default=0
set timeout=5

menuentry "netbsd -v" {
        insmod ext2
        set root=(hd0,msdos1)
        knetbsd /netbsd -v --root=wd0e
}

menuentry "netbsd -v -s" {
        insmod ext2
        set root=(hd0,msdos1)
        knetbsd /netbsd -v -s --root=wd0e
}

menuentry "netbsd -v -a" {
        insmod ext2
        set root=(hd0,msdos1)
        knetbsd /netbsd -v -a
}

menuentry "netbsd-INSTALL.gz -v" {
        insmod ext2
        set root=(hd0,msdos1)
        knetbsd /netbsd-INSTALL.gz -v
}
EOF


#
# NETWORK SETUP
#

echo -n myname...
echo $host > lala/etc/myname && echo done

echo -n $netif...
echo inet $cidr up > lala/etc/ifconfig.$netif && echo done

echo -n hosts...
cat > lala/etc/hosts <<EOF && echo done
::1             localhost localhost.
127.0.0.1       localhost localhost.
${cidr%/*} $host
EOF

if [[ -n $gw ]]; then
        echo -n mygate...
        echo $gw > lala/etc/mygate && echo done
	echo -n hosts/gw...
	cat >> lala/etc/hosts <<-EOF && echo done
	$gw gw
	EOF
else
        echo no gateway
fi

rm -f lala/etc/resolv.conf
for ip in $dns; do
        echo -n nameserver $ip...
        cat >> lala/etc/resolv.conf <<EOF && echo done
nameserver $ip
EOF
done; unset ip

echo -n authorized_keys...
if [[ -f lala/root/.ssh/authorized_keys ]]; then
        echo already:
        cat lala/root/.ssh/authorized_keys
else
        mkdir lala/root/.ssh/
        chmod 700 lala/root/.ssh/
        cat > lala/root/.ssh/authorized_keys <<EOF9
$pubkey
EOF9
        chmod 600 lala/root/.ssh/authorized_keys && echo done
fi

echo -n sshd_config...
mv lala/etc/ssh/sshd_config lala/etc/ssh/sshd_config.dist
sed -r '/^[[:space:]]*#/d; /^[[:space:]]*$/d; /UsePam /d' lala/etc/ssh/sshd_config.dist \
	> lala/etc/ssh/sshd_config
cat >> lala/etc/ssh/sshd_config <<EOF9 && echo done

AddressFamily inet
#ListenAddress ${cidr%/*}
Protocol 2
Port 22
AllowGroups wheel
PermitRootLogin without-password
PasswordAuthentication no
PermitEmptyPasswords no
StrictModes yes
X11Forwarding no
ChallengeResponseAuthentication no
UsePAM no
UseDNS no
HostKey /etc/ssh/ssh_host_ed25519_key
PrintMotd no
EOF9

echo -n sshd=yes...
echo sshd=yes >> lala/etc/rc.conf && echo done

umount lala/
[[ $platform = hvm ]] && losetup --detach $disk
echo

echo all done-- $disk $partition is ready

[[ $platform != baremetal ]] && cat <<EOF

you can now run

	xl create $guest -c

EOF