Scalaでニコ生アラート(コメントサーバに接続するところまで)書いてみた

Scalaでニコ生アラート

rubyとかjavaとかでニコ生アラートつくってみた、という記事はありますがScalaでニコ生アラートつくってみた、というのがなかった(or 見つけられなかった)ので作ってみました。
Web APIとの通信はDispatcherを使っているのでBuild.scalaでDependenciesに追加する必要があります

NicoliveAlert.scala

/**
 * author: crexista
 */
class NicoLiveAlertClient {

	val mailAddress = ""//ここにニコニコ動画に登録してるメールアドレスを入れます

	val passWord = ""//ここにニコニコ動画にログインするさいのパスワードをいれます

	val authURL = "https://secure.nicovideo.jp/secure/login?site=nicolive_antenna"

	val userAgent = "NicoLiveAlert 1.0.0"

	val apiURL = "http://live.nicovideo.jp/api/"

	val getAlertAPI = "getalertstatus"

	def start() = {
		val http = new Http
		val param = Map[String, String]("mail" -> mailAddress, "password" -> passWord)
		val userAgentMap = Map[String,String]("User-Agent" -> userAgent)

		//ユーザー認証トークンです
		val token = (http(url(authURL).POST <:< userAgentMap << param >> ((a:InputStream) => XML.load(a))) \\ "ticket").text

		//ニコ生アラートに接続するためのスレッド情報を取得します
		val alertStatus = http(url(apiURL + getAlertAPI).POST << Map("ticket" -> token) >> ((a:InputStream) => XML.load(a)))

		val host = (alertStatus \\ "ms" \\ "addr").text
		val port = (alertStatus \\ "ms" \\ "port").text.toInt
		val thread = (alertStatus \\ "ms" \\ "thread").text.toInt

		val socket = new Socket(host, port)
		val echoMessage = "<thread thread=\"" + thread.toString + "\" version=\"20061206\" res_from=\"-1\"/>\0"
		val recieveData:ArrayBuffer[Byte] = new ArrayBuffer[Byte](200)

		socket.getOutputStream.write(echoMessage.getBytes("UTF-8"))
		socket.getOutputStream.flush()

		while(true) {
			val onData = socket.getInputStream.read.toByte
			recieveData += onData
			if (onData == 0){
				println(new String(recieveData.toArray, "UTF-8"))
				recieveData.clear()
			}
		}
	}

}

はまりどころ

ニコ生のコメントサーバはXmlSocketというプロトコルを使っていて
xmlデータの最後には改行コードではなくNULL文字をつけて通信しているということ。
詳しくはここ
http://d.hatena.ne.jp/rivertop/20070622/1182529919

なので、getlineとかつかうとずっと待ち受け状態になってしまう。

Build.scala

ニコニコ生放送のWeb APIをDispatcherを使って叩いているのでDispatcher(classic)をDependenciesで呼んであげる必要がある

Build.scala

import sbt._
import sbt.Keys._


object NicoLiveAlertBuild extends Build {

	val dispatch = "net.databinder" %% "dispatch-http" % "0.8.9"

	val typeSafeResolver = "Typesafe Repository" at "http://repo.typesafe.com/typesafe/releases/"


	lazy val root = Project(
		id = "NicoliveAlert",
		base = file("."),
		settings = Project.defaultSettings ++ Seq(
			name := "NicoliveAlert",
			organization := "st.crexi",
			scalaVersion := "2.10.0",
			libraryDependencies ++= Seq(dispatch),
			resolvers ++= Seq(typeSafeResolver)
		)
	)

}