ref: a411870ee4640241e3c494367d922847da84f972
dir: purgatorio/appl/cmd/cddb.b
implement Cddb;
# this is a near transliteration of Plan 9 source, and subject to the Lucent Public License 1.02
include "sys.m";
sys: Sys;
include "draw.m";
include "bufio.m";
bufio: Bufio;
Iobuf: import bufio;
include "string.m";
str: String;
include "dial.m";
dial: Dial;
include "arg.m";
Cddb: module
{
init: fn(nil: ref Draw->Context, nil: list of string);
};
server := "freedb.freedb.org";
debug := 0;
tflag := 0;
Tflag := 0;
Track: adt {
n: int;
title: string;
};
Toc: adt {
diskid: int;
ntrack: int;
title: string;
track: array of Track;
};
DPRINT(fd: int, s: string)
{
if(debug)
sys->fprint(sys->fildes(fd), "%s", s);
}
dumpcddb(t: ref Toc)
{
sys->print("title %s\n", t.title);
for(i:=0; i<t.ntrack; i++){
if(tflag){
n := t.track[i+1].n;
if(i == t.ntrack-1)
n *= 75;
s := (n - t.track[i].n)/75;
sys->print("%d\t%s\t%d:%2.2d\n", i+1, t.track[i].title, s/60, s%60);
}
else
sys->print("%d\t%s\n", i+1, t.track[i].title);
}
if(Tflag){
s := t.track[i].n;
sys->print("Total time: %d:%2.2d\n", s/60, s%60);
}
}
cddbfilltoc(t: ref Toc): int
{
conn := dial->dial(dial->netmkaddr(server, "tcp", "888"), nil);
if(conn == nil){
sys->fprint(sys->fildes(2), "cddb: cannot dial %s: %r\n", server);
return -1;
}
bin := bufio->fopen(conn.dfd, Bufio->OREAD);
if((p:=getline(bin)) == nil || atoi(p)/100 != 2)
return died(p);
sys->fprint(conn.dfd, "cddb hello gre plan9 9cd 1.0\r\n");
if((p = getline(bin)) == nil || atoi(p)/100 != 2)
return died(p);
#
# Protocol level 6 is the same as level 5 except that
# the character set is now UTF-8 instead of ISO-8859-1.
#
sys->fprint(conn.dfd, "proto 6\r\n");
if((p = getline(bin)) == nil || atoi(p)/100 != 2)
return died(p);
DPRINT(2, sys->sprint("%s\n", p));
sys->fprint(conn.dfd, "cddb query %8.8ux %d", t.diskid, t.ntrack);
DPRINT(2, sys->sprint("cddb query %8.8ux %d", t.diskid, t.ntrack));
for(i:=0; i<t.ntrack; i++) {
sys->fprint(conn.dfd, " %d", t.track[i].n);
DPRINT(2, sys->sprint(" %d", t.track[i].n));
}
sys->fprint(conn.dfd, " %d\r\n", t.track[t.ntrack].n);
DPRINT(2, sys->sprint(" %d\r\n", t.track[t.ntrack].n));
if((p = getline(bin)) == nil || atoi(p)/100 != 2)
return died(p);
DPRINT(2, sys->sprint("cddb: %s\n", p));
(nf, fl) := sys->tokenize(p, " \t\n\r");
if(nf < 1)
return died(p);
categ, id: string;
case atoi(hd fl) {
200 => # exact match
if(nf < 3)
return died(p);
categ = hd tl fl;
id = hd tl tl fl;
210 or # exact matches
211 => # close matches
if((p = getline(bin)) == nil)
return died(nil);
if(p[0] == '.') # no close matches?
return died(nil);
# accept first match
(nsf, f) := sys->tokenize(p, " \t\n\r");
if(nsf < 2)
return died(p);
categ = hd f;
id = hd tl f;
# snarf rest of buffer
while(p[0] != '.') {
if((p = getline(bin)) == nil)
return died(p);
DPRINT(2, sys->sprint("cddb: %s\n", p));
}
202 or # no match
* =>
return died(p);
}
t.title = "";
for(i=0; i<t.ntrack; i++)
t.track[i].title = "";
# fetch results for this cd
sys->fprint(conn.dfd, "cddb read %s %s\r\n", categ, id);
do {
if((p = getline(bin)) == nil)
return died(nil);
DPRINT(2, sys->sprint("cddb %s\n", p));
if(len p >= 7 && p[0:7] == "DTITLE=")
t.title += p[7:];
else if(len p >= 7 && p[0:6] == "TTITLE"&& isdigit(p[6])) {
i = atoi(p[6:]);
if(i < t.ntrack) {
p = p[6:];
while(p != nil && isdigit(p[0]))
p = p[1:];
if(p != nil && p[0] == '=')
p = p[1:];
t.track[i].title += p;
}
}
} while(p[0] != '.');
sys->fprint(conn.dfd, "quit\r\n");
return 0;
}
getline(f: ref Iobuf): string
{
p := f.gets('\n');
while(p != nil && isspace(p[len p-1]))
p = p[0: len p-1];
return p;
}
isdigit(c: int): int
{
return c>='0' && c <= '9';
}
isspace(c: int): int
{
return c == ' ' || c == '\t' || c == '\n' || c == '\r';
}
died(p: string): int
{
sys->fprint(sys->fildes(2), "cddb: error talking to server\n");
if(p != nil){
p = p[0:len p-1];
sys->fprint(sys->fildes(2), "cddb: server says: %s\n", p);
}
return -1;
}
init(nil: ref Draw->Context, args: list of string)
{
sys = load Sys Sys->PATH;
bufio = load Bufio Bufio->PATH;
str = load String String->PATH;
dial = load Dial Dial->PATH;
arg := load Arg Arg->PATH;
arg->init(args);
arg->setusage("cddb [-DTt] [-s server] query diskid n ...");
while((o := arg->opt()) != 0)
case o {
'D' => debug = 1;
's' => server = arg->earg();
'T' => Tflag = 1; tflag = 1;
't' => tflag = 1;
* => arg->usage();
}
args = arg->argv();
argc := len args;
if(argc < 3 || hd args != "query")
arg->usage();
arg = nil;
ntrack := atoi(hd tl tl args);
toc := ref Toc(str->toint(hd tl args, 16).t0, ntrack, nil, array[ntrack+1] of Track);
if(argc != 3+toc.ntrack+1){
sys->fprint(sys->fildes(2), "cddb: argument count does not match given ntrack");
raise "fail:error";
}
args = tl tl tl args;
for(i:=0; i<=toc.ntrack; i++){ # <=?
toc.track[i].n = atoi(hd args);
args = tl args;
}
if(cddbfilltoc(toc) < 0)
raise "fail:whoops";
dumpcddb(toc);
}
atoi(s: string): int
{
return int s;
}