check_bsto_series.rb

#!/usr/bin/env ruby

# Coypright (C) 2018 Christian Mayer <christian@fox21.at>


require 'net/http'
require 'json'
require 'optparse'
require 'cgi'

API_BASE_URL = 'https://bs.to/serie/%s/%d'
STATES = ['OK', 'WARNING', 'CRITICAL', 'UNKNOWN']

@options = {
	:series_name => nil,
	:series_season => nil,
	:warning_episode => nil,
	:critical_episode => nil,
	:lang => 'en',
	:unknown_state => 3,
}
opts = OptionParser.new do |o|
	o.banner = 'Usage: --series <name> --season <number> -w <episode> -c <episode> --lang en|de'
	o.separator('')
	
	o.on('-n', '--series <name>', 'Series Name') do |series_name|
		@options[:series_name] = series_name
	end
	
	o.on('-s', '--season <number>', 'Season Number') do |season|
		@options[:series_season] = season.to_i
	end
	
	o.on('-w', '--warning <episode>', 'Warning Episode') do |episode|
		@options[:warning_episode] = episode.to_i
	end
	
	o.on('-c', '--critical <episode>', 'Critical Episode') do |episode|
		@options[:critical_episode] = episode.to_i
	end
	
	o.on('-l', '--lang <string>', 'Language: English or German') do |lang|
		@options[:lang] = lang.downcase
	end
	
	o.on('-u', '--unknown <int>', 'Treat UNKNOWN state with different status code. Default: 3') do |state|
		@options[:unknown_state] = state.to_i
	end
	
	o.on_tail('-h', '--help', 'Show this message.') do
		puts o
		puts
		exit 3
	end
end
ARGV << '-h' if ARGV.count == 0
commands = opts.parse(ARGV)

# Build URL.
url = API_BASE_URL % [@options[:series_name], @options[:series_season]]

uri = URI(url)

# Make response to API.
response = Net::HTTP.get(uri)

# This happens when you enter a season that does not exist (yet).
not_found_pos = response.index('Staffel nicht gefunden')
if not not_found_pos.nil?
	state_name = STATES[@options[:unknown_state]]
	perf_data = [
		state_name, @options[:series_season],
		@options[:series_name].downcase,
		@options[:series_season], @options[:warning_episode],
		@options[:series_season], @options[:critical_episode],
	]
	puts "%s: Season %d not found | '%s'=U;%d%03d;%d%03d" % perf_data
	exit @options[:unknown_state]
end

# Search the beginning position of the episodes table.
table_beginning_pos = response.index('<table class="episodes">')

# Cut everything above the table beginning to get the table end.
table = response[table_beginning_pos..-1]

# Search the end position of the table.
# Hopefully there is no other table inside the table.
table_end_pos = table.index('</table>')

# Cut everything below the table.
table = table[0...table_end_pos]

regex_s = [
	'<tr>',
		'<td>',
			'<a href="[^"]{6,}" title="([^"]{1,})">(\d{1,3})<.a>', # Title, Episode Number
		'<.td>',
		'<td>',
			'<a [^>]{20,}>[^<]{1,}<s([tp]).{10}', # Language based on the HTML tag.
].join('\s{0,20}') # Whitespace between HTML tags.

regex = Regexp.new(regex_s)

episodes = table.scan(regex)

state = 0 # OK
title = 'N/A'
etitle = nil
id = 0
eid = nil
lang = 'N/A'
elang = nil
episodes.each do |episode|
	etitle, eid, etag = episode
	
	etitle = CGI.unescapeHTML(etitle)
		.strip
		.gsub(/[^ \-A-Za-z0-9']/, '')
	eid = eid.to_i
	elang = etag == 't' ? 'de' : 'en'
	
	case @options[:lang]
	when 'en'
		# puts "episode: #%d %s (%s) '%s'" % [eid, etag, elang, etitle]
		
		if eid >= @options[:critical_episode]
			state = 2
			title = etitle
			id = eid
			lang = elang
			break
		elsif eid >= @options[:warning_episode]
			state = 1
			title = etitle
			id = eid
			lang = elang
			# contine loop
		end
	when 'de'
		if elang == 'de'
			# puts "episode: #%d %s (%s) '%s'" % [eid, etag, elang, etitle]
			
			if eid >= @options[:critical_episode]
				state = 2
				title = etitle
				id = eid
				lang = elang
				break
			elsif eid >= @options[:warning_episode]
				state = 1
				title = etitle
				id = eid
				lang = elang
				# contine loop
			end
		end
	end
end

if state == 0
	title = etitle
	id = eid
	lang = elang
end

state_name = STATES[state]

perf_data = [
	state_name, id, title, @options[:series_season], lang, # Normal Output
	@options[:series_name].downcase, id,
	@options[:series_season], @options[:warning_episode],
	@options[:series_season], @options[:critical_episode],
]
puts "%s: #%d %s (s=%d,l=%s) | '%s'=%d;%d%03d;%d%03d" % perf_data

exit state