webif/var/mongoose/lib/ts.class

if {![exists -proc class]} { package require oo }
if {![exists -proc pack]} { package require pack }
source /mod/webif/lib/setup
require tdelete system.class

set dmsfile /mnt/hd2/dms_cds.db

set tsgroup {ts nts hmt thm}

class ts {
	file		""
	base		""
	title		""
	synopsis	""
	definition	""
	channel_num	0
	channel_name	""
	start		0
	end		0
	flags		""
	error		""
	guidance	""
	bookmarks	0
	schedstart	0
	scheddur	0
	genre		0
}

ts method duration {{raw 0}} {
	set d [expr $end - $start]
	if {!$raw} { set d $($d / 60) }
	return $d
}

ts method size {} {
	file stat $file st
	return $st(size)
}

ts method _parse {line} {
	set vars [split $line "\t"]

	lassign [split $line "\t"] \
	    title synopsis definition channel_num channel_name \
	    start end flags_list guidance bookmarks schedstart scheddur \
	    genre

	set flags [split [string range $flags_list 0 end-1] ,]
}

ts method inuse {} {
	if {[system inuse [file rootname $file]]} { return 1 }
	return 0
}

ts method bookmarks {} {
	return [exec /mod/bin/hmt -bookmarks $file]
}

ts method flag {f} {
	if {$f in $flags} {return 1} else {return 0}
}

ts method unlock {} {
	set cmd [list /mod/bin/hmt -lock $file]
	exec {*}$cmd
	return 1
}

ts method lock {} {
	set cmd [list /mod/bin/hmt +lock $file]
	exec {*}$cmd
	return 1
}

ts method set_shrunk {} {
	set cmd [list /mod/bin/hmt +shrunk $file]
	exec {*}$cmd
	lappend flags "Shrunk"
	return 1
}

ts method set_deduped {} {
	set cmd [list /mod/bin/hmt +dedup $file]
	exec {*}$cmd
	return 1
}

ts method unenc {} {
	set cmd [list /mod/bin/hmt -protect $file]
	exec {*}$cmd
	return 1
}

ts method enc {} {
	set cmd [list /mod/bin/hmt +protect $file]
	exec {*}$cmd
	return 1
}

ts method set_new {} {
	set cmd [list /mod/bin/hmt +new $file]
	exec {*}$cmd
	return 1
}

ts method set_watched {} {
	set cmd [list /mod/bin/hmt -new $file]
	exec {*}$cmd
	return 1
}

ts method setfile {f} { set file $f }

proc {ts parse} {file line} {
	set e [ts new]
	$e setfile $file
	$e _parse $line
	return $e
}

proc {ts exec} {file} {
	set raw 0
	set cmd [list /mod/bin/hmt]
	lappend cmd "-p"
	lappend cmd $file

	#puts "CMD -$cmd-"

	return [exec {*}$cmd]
}

ts method fileset {} {
	global tsgroup

	set root [file rootname $file]
	set fset {}
	foreach ext $tsgroup {
		if {[file exists "$root.$ext"]} {
			lappend fset "$root.$ext"
		}
	}
	return $fset
}

proc {ts fetch} {file {checked 0}} {
	# Check that this is a .ts file which has at least one sidecar
	# file (.nts)
	if {!$checked} {
		if {[file extension $file] ne ".ts"} { return 0 }
		if {![file exists "[file rootname $file].nts"]} { return 0 }
	}

	return [ts parse $file [ts exec $file]]
}

ts method delete {} {
	foreach f [$self fileset] {
		tdelete $f
		puts "Removed $f<br>"
	}
	return 1
}

ts method move {dst {touch 0} {force 0}} {
	foreach f [$self fileset] {
		set nf "$dst/[file tail $f]"
		while {[file exists $nf]} {
			set nf "$dst/_[file tail $nf]"
		}
		file rename $f $nf
		if {$touch} {
			exec /mod/bin/busybox/touch $nf
		}
	}
	return 1
}

ts method copy {dst} {
	foreach f [$self fileset] {
		file copy $f "$dst/[file tail $f]"
	}
	return 1
}

ts method settitle {newtitle} {
	if {[string length newtitle] > 48} { return }

	exec /mod/bin/hmt "+settitle=${newtitle}" $file
}

ts method setsynopsis {newsynopsis} {
	if {[string length newsynopsis] > 252} { return }

	exec /mod/bin/hmt "+setsynopsis=${newsynopsis}" $file
}

ts method setguidance {newguidance} {
	if {[string length newguidance] > 48} { return }

	if {$newguidance eq ""} {
		exec /mod/bin/hmt "-guidance" $file
	} else {
		exec /mod/bin/hmt "+setguidance=${newguidance}" $file
	}
}

ts method setgenre {newgenre} {
	exec /mod/bin/hmt "+setgenre=-${newgenre}" $file
}

ts method dlnaloc {{urlbase 0}} {
	set mime "video/ts"
	if {![file exists $::dmsfile]} { return {} }
	if {[catch {set db [sqlite3.open $::dmsfile]}]} { return {} }
	set url ""
	set rfile [file normalize $file]
	if {[catch {set muri [$db query "
	    select tblresource.mimetype, contenturi
	    from tblresource join tblmedia using (mediaid)
	    where localurl = '%s'" $rfile]}]} {
		set muri ""
	}
	if {$muri ne ""} {
		set rec [lindex $muri 0]
		set xuri [lindex $rec 3]
		set mime [lindex $rec 1]
		if {$urlbase ne "0"} {
			set url "http://$urlbase:9000/web/$xuri"
		} else {
			set url "http://[system ip]:9000/web/$xuri"
		}
	}
	$db close

	if {$url ne "" } { return [list $url $mime] } else { return {} }
}

ts method cleanbmp {} {
	set bfile [file rootname $file]
	foreach f [glob -nocomplain "${bfile}*.bmp"] {
		file delete $f
	}
}

ts method mkbmps {{offset 0}} {
	set bfile [file rootname $file]
	if {[catch {
	    exec /mod/bin/ffmpeg -loglevel fatal -ss $offset -i $file \
	    -vf fps=fps=2 -frames 5 \
	    -pix_fmt argb -vf vflip -s 140x78 "${bfile}%d.bmp"
	} msg]} {
		puts "ERROR: $msg"
		return 0
	}
	return 1
}

ts method mkbmp {{offset 0} {ext ""}} {
	set bfile [file rootname $file]
	if {[catch {
	    exec /mod/bin/ffmpeg -loglevel fatal -ss $offset -i $file \
	    -frames 1 -pix_fmt argb -vf vflip -s 140x78 "$bfile$ext.bmp"
	} msg]} {
		puts "ERROR: $msg"
		return 0
	}
	return 1
}

ts method mkthm {{offset 0}} {
	if {![$self mkbmp $offset]} { return 0 }
	set bfile [file rootname $file]
	if {[catch {
	    exec /bin/dd if=$bfile.bmp of=$bfile.thm~ bs=54 skip=1
	    } msg]} {
		puts "ERROR: $msg"
		return 0
	}
	exec /bin/echo -n " " >> $bfile.thm~
	file rename -force $bfile.thm~ $bfile.thm
	tdelete $bfile.bmp
	return 1
}

proc {ts renamegroup} {from to} {
	global tsgroup

	set dir [file dirname $from]
	set root [file rootname $from]

	# Catch from string without a . character in it
	if {$root eq $from} { return }

	foreach ext $tsgroup {
		set f "$root.$ext"
		if {![file exists $f]} continue
		file rename $f "${dir}/${to}.${ext}"
	}

	exec /mod/bin/hmt "+setfilename=$to" "${dir}/${to}.hmt"

	set ndir [file normalize $dir]

	if {![catch {set db [sqlite3.open $::dmsfile]}]} {
	    catch {
		set x [lindex [$db query {select mediaid from tblMedia
		    where localUrl = '%s'} [file normalize $from]] 0]
		lassign $x key mediaid
		if {$mediaid ne ""} {
			$db query {update tblMedia set localUrl = '%s'
			    where mediaid = %s} "${ndir}/{$to}.ts" $mediaid
			$db query {update tblMedia set title = '%s'
			    where mediaid = %s} "{$to}.ts" $mediaid
		}
	    }
	$db close
	}
}

proc {ts genrelist} {} {
	return {
		0	{ Unclassified		Unclassified }
		16	{ Film 			Movie }
		32	{ "News & Factual"	News }
		48	{ Entertainment		Special }
		64	{ Sport			Sports }
		80	{ Children		Children }
		96	{ Entertainment		Special }
		144	{ Education		Education }
		160	{ Lifestyle		Leisure }
		240	{ Drama			Show }
	}
}

proc {ts resetnew} {dir} {
	if {![file isdirectory $dir]} return
	if {![file exists "$dir/.series"]} {
		set fd [open "$dir/.series" "w"]
		puts -nonewline $fd [string repeat "\x0" 276]
		close $fd
	}

	set tot 0
	set watched 0
	foreach file [readdir -nocomplain $dir] {
		if {![string match {*.ts} $file]} { continue }
		incr tot
		if {[set ts [ts fetch "$dir/$file"]] != 0} {
			if {![$ts flag "New"]} { incr watched }
		}
	}

	set fd [open "$dir/.series"]
	set bytes [read $fd]
	close $fd
	set recs [unpack $bytes -uintle 0 32]
	set played [unpack $bytes -uintle 32 32]

	#puts "Current: $played/$recs"
	#hexdump $bytes
	#puts "Calculated: $watched/$tot"

	pack bytes $tot -intle 32 0 
	pack bytes $watched -intle 32 32

	#hexdump $bytes

	set fd [open "$dir/.series" "w"]
	puts -nonewline $fd $bytes
	close $fd
}