rbtune 1.0

data/lib/rbtune/radiko_premium.rb

@@ -0,0 +1,70 @@
+#radiko.rb
+# coding: utf-8
+
+require "rbtune/radiko"
+
+class RadikoPremium < Radiko
+
+	def self.set_authentication(kc, account)
+		begin
+			password = kc.query('radikoプレミアムのパスワードを入力してください', account)
+			radio = RadikoPremium.new
+			radio.login account, password
+			kc.set account, password
+			puts "Radikoプレミアムのアカウントが正しく登録されました"
+
+		rescue Radio::HTTPForbiddenException
+			$stderr.puts "アカウント情報が正しくありません"
+
+		rescue RuntimeError
+			$stderr.puts $!
+
+		ensure
+			radio && radio.close
+		end
+	end
+
+
+	def headers
+		{
+			'pragma'           => 'no-cache',
+			'Cache-Control'    => 'no-cache',
+			'Expires'          => 'Thu, 01 Jan 1970 00:00:00 GMT',
+			'Accept-Language'  => 'ja-jp',
+			'Accept-Encoding'  => 'gzip, deflate',
+			'Accept'           => 'application/json, text/javascript, */*; q=0.01',
+			'X-Requested-With' => 'XMLHttpRequest'
+		}
+	end
+
+
+	def login(account, password)
+		res = agent.post 'https://radiko.jp/ap/member/login/login', {
+			mail: account, pass: password
+		}
+
+		begin
+			params = []
+			referer = nil
+			@logged_in = true
+			agent.get 'https://radiko.jp/ap/member/webapi/member/login/check', params, referer, headers
+		rescue Mechanize::ResponseCodeError => ex
+			raise HTTPForbiddenException if ex.message.include?('400 => Net::HTTPBadRequest')
+		end
+	end
+
+
+	def close
+		params = []
+		referer = nil
+		agent.get 'https://radiko.jp/ap/member/webapi/member/logout', params, referer, headers
+		@logged_in = false
+	end
+
+	def stations_uri
+		"http://radiko.jp/v3/station/region/full.xml"
+	end
+
+
+
+end

data/lib/rbtune/radio.rb

@@ -0,0 +1,207 @@
+# Radio仮想基底クラス
+# Radiko, Radiru等で継承して使う
+#
+# radio.login account, password (RadikoPremium 等、必要な場合のみ)
+# radio.open
+# radio.tune channel
+# radio.play または radio.record
+# radio.close
+
+require "date"
+require "benchmark"
+require "net/http"
+require "rexml/document"
+
+class Radio
+	attr_accessor :outdir
+	attr_reader   :ext
+	attr_reader   :area_id # for Radiko(Premium)
+	attr_reader   :area_ja # for Radiko(Premium)
+	attr_reader   :area_en # for Radiko(Premium)
+
+	def Radio.inherited(subclass)
+		@@bands ||= []
+		@@bands << subclass
+	end
+
+
+	def initialize
+		@outdir = '.'
+		@ext = 'm4a'
+	end
+
+
+	def self.db
+		@@db ||= Station::pstore_db
+	end
+
+
+	def self.stations
+		@stations ||= self.db.transaction(true) { self.db[name] }
+	end
+
+
+	def self.channels
+		@channels ||= self.stations.map {|st| [st.id, st.uri]}.to_h
+	end
+
+
+	def self.bands
+		@@bands
+	end
+
+
+	def self.search(channel)
+		radio_class, station = self.find(channel) || self.match(channel)
+	end
+
+
+	# Radioクラスのリストから、id と一致する放送局を探す
+	# return: [Radioクラス, 放送局] or nil
+	def self.find(id)
+		Radio.bands.each do |tuner|
+			if tuner.stations
+				station = tuner.stations.find {|station| station.id == id}
+				return [tuner, station] if station
+			end
+		end
+		nil
+	end
+
+
+	# Radioクラスのリストから、name を含む放送局を探す
+	# return: [Radioクラス, 放送局] or nil
+	def self.match(name)
+		matcher = /#{name}/i
+		Radio.bands.each do |tuner|
+			next unless tuner.stations
+			found = tuner.stations.find { |station| station.name.match?(matcher) || station.ascii_name.match?(matcher) }
+			return [tuner, found] if found
+		end
+		nil
+	end
+
+
+	def agent
+		@agent ||= Mechanize.new
+	end
+
+
+	def create_player(uri)
+		# rtmpdumpのコマンドラインを生成する(playから呼ばれる)
+	end
+
+
+	def login(account=nil, password=nil)
+		# ラジオサービスにログイン
+	end
+
+	def open
+	end
+
+	def close
+	end
+
+
+	def tune(channel)
+		@channel = channel
+	end
+
+
+	def channel_to_uri
+		self.class::channels[@channel] || @channel
+	end
+
+
+	def record(filename, sec, quiet: false, dt: DateTime.now)
+		begin
+			uri = channel_to_uri
+			raise 'not tuned yet.' unless uri
+
+			puts "record: #{uri}"
+			# $stderr.puts "play: #{sec}, #{filename}, #{quiet}"
+			player      = create_player uri
+			remain_sec  = sec
+			rtime       = 0
+			minimum_sec = 60   # 残り録音時間がこれ以下ならば、録音が中断してもやり直さない
+			datetimes   = []
+			begin
+				rtime += Benchmark.realtime do
+					dt = datetime dt
+					datetimes << dt
+					tmpfile = make_tmpfile @channel, dt
+					player.rec tmpfile, remain_sec, quiet
+				end
+				remain_sec -= rtime
+				dt = DateTime.now
+			end while remain_sec >= minimum_sec
+
+		rescue Interrupt, Errno::EPIPE
+			# do nothing
+
+		ensure
+			# 最後にまとめて convert する
+			datetimes.each do |dt|
+				tmpfile = make_tmpfile @channel, dt
+				recfile = make_recfile(filename, dt)
+				convert tmpfile, recfile
+			end
+		end
+	end
+
+
+	def play
+		uri = channel_to_uri
+		raise 'not tuned yet.' unless uri
+		puts "play: #{uri}"
+		create_player(uri).play
+	end
+
+
+	def convert(tmpfile, recfile)
+		FileUtils.mv tmpfile, recfile
+	end
+
+
+	def convert_ffmpeg(tmpfile, recfile)
+		ffmpeg = FFMpeg.new
+		ffmpeg['loglevel'] = 'quiet'
+		ffmpeg['i'] = %Q("#{tmpfile}")
+		ffmpeg['b:a'] = '70k'
+		stdout, stderr, status = ffmpeg.rec recfile, nil
+		FileUtils.rm tmpfile if status.success?
+	end
+
+
+	def datetime(dt)
+		dt.to_s[0..15].gsub(/:/, '=')
+	end
+
+
+	def make_tmpfile(channel, datetime)
+		File.join outdir, "#{channel}.#{datetime}.#{$$}.#{ext}"
+	end
+
+
+	def out_ext
+		@out_ext || ext
+	end
+
+	def make_recfile(title, datetime)
+		File.join outdir, "#{title}.#{datetime}.#{out_ext}"
+	end
+
+
+	def fetch_stations
+		body = agent.get stations_uri
+		stations = parse_stations body
+	end
+
+
+	class HTTPBadRequestException < StandardError; end
+	class HTTPForbiddenException < StandardError; end
+
+end
+
+
+

data/lib/rbtune/radiru.rb

@@ -0,0 +1,52 @@
+# radiru.rb
+# coding: utf-8
+
+require "rbtune/radio"
+require "player/ffmpeg"
+require "fileutils"
+
+class Radiru < Radio
+
+
+	def create_player(uri)
+		ffmpeg           = FFMpeg.new
+		ffmpeg['i']      = uri # input stream
+		ffmpeg['acodec'] = 'copy'  # acodecオプションはiオプションのあとに置かないとエラー
+		ffmpeg
+	end
+
+
+  def stations_uri
+    "https://www.nhk.or.jp/radio/config/config_web.xml"
+  end
+
+
+  def parse_stations(body)
+    stations = body.search '//data'
+    stationsjp = {
+      'r1' => "ラジオ第1",
+      'r2' => "ラジオ第2",
+      'fm' => "FM",
+    }
+    stations = stations.map do |station|
+      areajp = station.at('areajp').text
+      area   = station.at('area').text
+      # 地区ごとに第1, 第2, FM を登録する
+      r1, r2, fm = %w(r1 r2 fm).map do |v|
+        hls  = "#{v}hls"
+        id   = "nhk#{v}-#{area}".upcase
+        uri  = station.at(hls).text
+        name = "NHK#{stationsjp[v]}-#{areajp}"
+        Station.new(id, uri, name: name, ascii_name: id)
+      end
+    end
+    stations.flatten!
+    # id: NHKR1, NHKR2, NHKFM を東京局に割り当てる　
+    stations.find_all { |station| station.id =~ /-TOKYO/}.reverse.map do |station|
+      id = station.id.sub(/-TOKYO/, '')
+      stations.unshift Station.new(id, station.uri, name: station.name, ascii_name: id)
+    end
+    stations
+  end
+
+end

data/lib/rbtune/simul.rb

@@ -0,0 +1,93 @@
+#!/usr/bin/env ruby
+
+require "rbtune/radio"
+require "player/mplayer"
+require "rbtune/station"
+
+class Simul < Radio
+
+	def initialize
+		super
+		@ext     = 'asf'
+		@out_ext = 'm4a'
+	end
+
+
+	def channel_to_uri
+		ch = super
+		if ch.end_with?('.asx')
+			parse_asx ch
+		else
+			ch
+		end
+	end
+
+
+	def parse_asx(uri)
+		asx = Net::HTTP::get URI::parse(uri)
+		asx.force_encoding "Shift_JIS"
+		asx.encode! 'utf-8'
+		asx.downcase!
+		if asx.gsub!(%r(</ask>), '</asx>')
+			$stderr.puts 'fix!! </ask> to <asx>'
+		end
+		doc = REXML::Document.new(asx)
+		ref = doc.get_elements('//entry/ref')[0]
+		ref.attribute('href').value
+	end
+
+
+	def create_player(uri)
+		mplayer = Mplayer.new uri
+		mplayer.merge! 'benchmark' => '',	'vo' => 'null'
+		mplayer
+	end
+
+
+	def convert(tmpfile, recfile)
+		convert_ffmpeg(tmpfile, recfile)
+	end
+
+
+	def link_to_station_id(link)
+		link =~ %r{(/asx/([\w-]+).asx|nkansai.tv/(\w+)/?\Z|(flower|redswave|fm-tanba|darazfm|AmamiFM|comiten|fm-shimabara|fm-kitaq))}
+		id = ($2 || $3 || $4).sub(/fm[-_]/, 'fm').sub(/[-_]fm/, 'fm')
+	end
+
+
+	def stations_uri
+		'http://www.simulradio.info'
+	end
+
+
+	def parse_stations(body)
+		radioboxes = body / 'div.radiobox'
+		radioboxes.map do |station|
+			rows   = station / 'tr'
+			cols   = rows[1] / 'td'
+			ankers = station / 'a'
+
+			player = ankers.select {|a|
+				img = a.at 'img'
+				img && img['alt'] == '放送を聴く'
+			}
+
+			title  = station.at('td > p > strong > a').text.strip
+			links0 = player.map! {|e| e['href']}
+			links  = links0.select { |uri| uri =~ %r{\.asx\Z|nkansai.tv} }
+			if links.empty?
+				# $stderr.puts "#{title}: #{links0 * ', '}"
+			else
+				link = links[0]
+				id   = link_to_station_id(link)
+				if id
+					Station.new(id, link, name: title, ascii_name: id)
+				else
+					$stderr.puts "!!! #{title}: #{link}"
+				end
+			end
+		end.compact
+	end
+
+
+end

data/lib/rbtune/station.rb

@@ -0,0 +1,91 @@
+require "pstore"
+
+class Station
+  attr_reader :name
+  attr_reader :id
+  attr_reader :ascii_name
+  attr_reader :description
+  attr_reader :uri
+
+  def self.pstore_db
+    @db ||= PStore.new(File.join(ENV['HOME'], '.rbtune.db'))
+  end
+
+
+  def self.list_stations
+      Radio.bands.each do |radio|
+        name = radio.to_s
+        stations = radio.stations
+        if stations.nil? || stations.empty?
+          $stderr.puts "warning: #{name} に放送局が登録されていません。"
+          $stderr.puts "         rbtune --fetch-stations を実行して、放送局情報を取得してください。"
+        else
+          puts "* #{name}"
+          stations.each { |station| puts "    #{station}" }
+          puts ''
+        end
+      end
+  end
+
+
+  def self.fetch_stations
+    db = Station::pstore_db
+    Radio.bands.each do |radio_class|
+      begin
+        $stderr.puts ">>> fetching #{radio_class} stations..."
+        radio = radio_class.new
+        radio.open
+        stations = radio.fetch_stations
+        if stations.empty?
+          $stderr.puts "    warning: no station found."
+        else
+          db.transaction { db[radio_class.to_s] = stations }
+          $stderr.puts "    #{stations.size} stations fetched."
+        end
+
+      rescue SocketError
+        $stderr.puts $!
+
+      rescue Net::HTTPNotFound
+        $stderr.puts $!
+
+      rescue REXML::ParseException
+        $stderr.puts $!
+
+      rescue
+        # 例外を握りつぶしてすべてのクラスの放送局情報の取得を試みる
+        $stderr.puts $!
+      end
+    end
+  end
+
+
+  def initialize(id, uri, name: '', ascii_name: '', description: '')
+    @id          = id.to_s.upcase
+    @uri         = uri
+    @name        = normalize_name name
+    @ascii_name  = ascii_name
+    @description = description
+  end
+
+
+  def normalize_name(name)
+    # NHKの局名はnormalizeしない (NHKFM-* の-が取れてしまうから)
+    return name if name.include?('NHK')
+    name.strip
+    .sub(/[ -]?FM[ -]?/i, 'FM')
+    .sub(/fm|ＦＭ|エフエム|えふえむ/, 'FM')
+    .tr('　！＠', ' !@')
+  end
+
+  def to_s
+    "#{id}:\t\t#{name}\t\t#{ascii_name}"
+  end
+
+  alias :inspect :to_s
+
+  def ==(other)
+    id == other.id
+  end
+
+end