#!/bin/sh
# original: http://memorandum.yamasnet.com/archives/Post-3124.html
cd "/hoge/radiko"

playerurl="http://radiko.jp/player/swf/player_3.0.0.01.swf"
swfextract_path="swfextract"
rtmpdump_path="rtmpdump"
MP4Box_path="MP4Box"
neroAacTag_path="neroAacTag"
playerfile="player.swf"
keyfile="authkey.png"

if [ $# -eq 3 ]; then
  station="$1"
  filename="/hoge/radio/$2"
  DURATION="$3"
elif [ $# -eq 4 ]; then
  station="$1"
  filename="/hoge/radio/$2"
  DURATION="$3"
  title="$2"
  album=`echo $2 | sed -e "s/ 201.*分//g"`
  artist="$4"
else
  echo "usage : $0 station name duration [artist]"
  exit 1
fi

#
# get player
#
if [ ! -f $playerfile ]; then
  wget -q -O $playerfile $playerurl

  if [ $? -ne 0 ]; then
    echo "failed get player"
    exit 1
  fi
fi

#
# get keydata (need swftool)
#
if [ ! -f $keyfile ]; then
  $swfextract_path -b 14 $playerfile -o $keyfile

  if [ ! -f $keyfile ]; then
    echo "failed get keydata"
    exit 1
  fi
fi

if [ -f auth1_fms ]; then
  rm -f auth1_fms
fi

#
# access auth1_fms
#
wget -q \
     --header="pragma: no-cache" \
     --header="X-Radiko-App: pc_1" \
     --header="X-Radiko-App-Version: 2.0.1" \
     --header="X-Radiko-User: test-stream" \
     --header="X-Radiko-Device: pc" \
     --post-data='\r\n' \
     --no-check-certificate \
     --save-headers \
     https://radiko.jp/v2/api/auth1_fms

if [ $? -ne 0 ]; then
  echo "failed auth1 process"
  exit 1
fi

#
# get partial key
#
authtoken=`perl -ne 'print $1 if(/x-radiko-authtoken: ([\w-]+)/i)' auth1_fms`
offset=`perl -ne 'print $1 if(/x-radiko-keyoffset: (\d+)/i)' auth1_fms`
length=`perl -ne 'print $1 if(/x-radiko-keylength: (\d+)/i)' auth1_fms`

partialkey=`dd if=$keyfile bs=1 skip=${offset} count=${length} 2> /dev/null | base64`
echo "authtoken: ${authtoken} \noffset: ${offset} length: ${length} \npartialkey: 

$partialkey"

rm -f auth1_fms

if [ -f auth2_fms ]; then
  rm -f auth2_fms
fi

#
# access auth2_fms
#
wget -q \
     --header="pragma: no-cache" \
     --header="X-Radiko-App: pc_1" \
     --header="X-Radiko-App-Version: 2.0.1" \
     --header="X-Radiko-User: test-stream" \
     --header="X-Radiko-Device: pc" \
     --header="X-Radiko-Authtoken: ${authtoken}" \
     --header="X-Radiko-Partialkey: ${partialkey}" \
     --post-data='\r\n' \
     --no-check-certificate \
     https://radiko.jp/v2/api/auth2_fms

if [ $? -ne 0 -o ! -f auth2_fms ]; then
  echo "failed auth2 process"
  exit 1
fi

echo "authentication success"
areaid=`perl -ne 'print $1 if(/^([^,]+),/i)' auth2_fms`
echo "areaid: $areaid"

rm -f auth2_fms

#
# rtmpdump
#
$rtmpdump_path -v \
         -r "rtmpe://w-radiko.smartstream.ne.jp" \
         --playpath "simul-stream.stream" \
         --app "${station}/_definst_" \
         -W $playerurl \
         -C S:"" -C S:"" -C S:"" -C S:$authtoken \
         --live \
         --stop $DURATION \
         -o "$filename.flv"

ffmpeg -y -i "$filename.flv" -acodec copy "$filename.aac"
$MP4Box_path -sbr -add "$filename.aac" -new "$filename.m4a"

rm "$filename.flv" "$filename.aac"

if [ "$title" ]; then
  $neroAacTag_path -meta:"album=$album" -meta:"artist=$artist" -meta:"genre=ラジオ" -meta-user:"itunespodcast=1" -meta:"title=$title" "$filename.m4a"
fi