1 # id3fs - a FUSE-based filesystem for browsing audio metadata
2 # Copyright (C) 2010 Ian Beckwith <ianb@erislabs.net>
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
14 # You should have received a copy of the GNU General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 use ID3FS::Path::Node;
23 our ($STATE_INVALID, $STATE_ROOT, $STATE_TAG, $STATE_BOOLEAN,
24 $STATE_ALBUMS, $STATE_TRACKLIST, $STATE_FILE, $STATE_ALL)=(0..7);
27 our %priorities=( "OR" => 0, "AND" => 1, "NOT" => 2 );
29 our $PATH_ALLTRACKS= "TRACKS";
30 our $PATH_NOARTIST = "NOARTIST";
31 our $PATH_NOALBUM = "NOALBUM";
33 our $ENABLE_FILTER = 0;
38 my $class=ref($proto) || $proto;
45 $self->{verbose}=shift;
46 $self->{maxtagdepth}=shift;
47 $self->{curtagdepth}=0;
48 $self->{path} =~ s/\/\//\//g; # drop doubled slashes
51 # print "STATE: ", $self->state(), "\n";
58 if(($self->state() == $STATE_FILE) ||
59 ($self->state() == $STATE_INVALID))
69 return($self->state() == $STATE_FILE);
75 return($self->state() != $STATE_INVALID);
80 my($self, $mountpoint)=@_;
81 if($self->state() == $STATE_FILE)
83 return $self->filename($mountpoint);
85 return "ERROR"; #should never happen?
93 my $state=$self->state();
94 # print "DIRENTS: STATE: $state\n";
95 # print "DIRENTS: FILE: $self->{path}\n";
96 if($state==$STATE_ALL)
98 @dents=($self->filter($PATH_ALLTRACKS, $PATH_NOARTIST), $self->artists());
100 elsif($state==$STATE_TAG)
102 my $tail=$self->tail();
103 if($self->is($TYPE_TAG, $tail) && $self->{db}->tag_has_values($tail->id()))
105 @dents=$self->tags();
109 if($self->{maxtagdepth} && ($self->{curtagdepth} < $self->{maxtagdepth}))
113 push(@dents, $self->filter($PATH_ALLTRACKS, $PATH_NOARTIST), $self->artists());
116 elsif($state==$STATE_BOOLEAN)
118 my $parent=$self->tail();
119 unless($self->is($TYPE_BOOL, $parent) && $parent->name() eq "NOT")
123 push(@dents,$self->tags());
125 elsif($state==$STATE_ROOT)
127 @dents=(qw(ALL NOT), $self->tags());
129 elsif($state==$STATE_ALBUMS)
131 @dents=$self->filter($PATH_ALLTRACKS, $PATH_NOALBUM, $self->albums());
133 elsif($state==$STATE_TRACKLIST)
135 @fents=$self->tracks();
139 print "id3fsd: INTERNAL ERROR: DIRENTS: UNHANDLED STATE: $state\n";
141 return(\@dents, \@fents);
144 # State Machine Of Doom
148 $self->state($STATE_ROOT);
149 return if($self->{path} eq "/");
150 @{$self->{components}}=split(/\//, $self->{path});
151 shift @{$self->{components}}; # drop empty field before leading /
152 # print "PATH: $self->{path}\n";
153 my @parts=@{$self->{components}};
155 $self->{elements}=[];
159 while(defined(my $name=shift @parts))
161 # print "NAME: $name\n";
162 my $state=$self->state();
163 if($state==$STATE_INVALID)
165 # print "SM: INVALID: $name\n";
168 elsif($state==$STATE_ROOT)
170 # print "SM: ROOT: $name\n";
174 $self->state($STATE_ALL);
176 elsif($name eq "NOT")
179 push(@{$self->{elements}}, ID3FS::Path::Node->new($self->{db}, $TYPE_BOOL, $name));
180 $self->state($STATE_BOOLEAN);
184 $tag=ID3FS::Path::Node->new($self->{db}, $TYPE_TAG, $name);
187 push(@{$self->{elements}}, $tag);
189 $self->state($STATE_TAG);
193 $self->state($STATE_INVALID);
197 elsif($state==$STATE_TAG)
199 my $tag=$self->tail();
200 # print "SM: TAG/TAGVAL($state): $name\n";
201 if($self->is($TYPE_TAG, $tag) && $self->{db}->tag_has_values($tag->id()))
203 # print "Parsing: parent: $tag->id()\n";
204 my $tagval=ID3FS::Path::Node->new($self->{db}, $TYPE_TAG, $name, $tag->id());
208 push(@{$self->{elements}}, $tagval);
212 $self->state($STATE_INVALID);
215 elsif($name eq $PATH_ALLTRACKS)
217 $self->state($STATE_TRACKLIST);
219 elsif($name eq $PATH_NOARTIST)
221 $self->state($STATE_TRACKLIST);
223 elsif($name eq "AND")
225 $self->state($STATE_BOOLEAN);
226 push(@{$self->{elements}}, ID3FS::Path::Node->new($self->{db}, $TYPE_BOOL, $name));
230 $self->state($STATE_BOOLEAN);
231 push(@{$self->{elements}}, ID3FS::Path::Node->new($self->{db}, $TYPE_BOOL, $name));
235 my $artist=ID3FS::Path::Node->new($self->{db}, $TYPE_ARTIST, $name);
238 push(@{$self->{elements}}, $artist);
239 $self->state($STATE_ALBUMS);
243 $self->state($STATE_INVALID);
247 elsif($state==$STATE_BOOLEAN)
249 # print "SM: BOOLEAN: $name\n";
250 my $parent=$self->tail();
252 if($self->is($TYPE_BOOL, $parent) &&
253 $parent->name() eq "NOT")
257 if($allownot && $name eq "NOT")
259 $self->state($STATE_BOOLEAN);
260 push(@{$self->{elements}}, ID3FS::Path::Node->new($self->{db}, $TYPE_BOOL, $name));
264 my $tag=ID3FS::Path::Node->new($self->{db}, $TYPE_TAG, $name);
267 push(@{$self->{elements}}, $tag);
269 $self->state($STATE_TAG);
273 $self->state($STATE_INVALID);
277 elsif($state==$STATE_ALBUMS)
279 # print "SM: ALBUM: $name\n";
280 if($name eq $PATH_ALLTRACKS)
282 $self->state($STATE_TRACKLIST);
284 elsif($name eq $PATH_NOALBUM)
286 $self->state($STATE_TRACKLIST);
290 my $album=ID3FS::Path::Node->new($self->{db}, $TYPE_ALBUM, $name);
293 push(@{$self->{elements}}, $album);
294 $self->state($STATE_TRACKLIST);
298 $self->state($STATE_INVALID);
302 elsif($state==$STATE_TRACKLIST)
304 # print "SM: TRACKLIST: $name\n";
305 my $track=ID3FS::Path::Node->new($self->{db}, $TYPE_FILE, $name);
308 push(@{$self->{elements}}, $track);
309 $self->state($STATE_FILE);
313 $self->state($STATE_INVALID);
316 elsif($state==$STATE_FILE)
318 # print "SM: FILE: $name\n";
319 # Can't have anything after a filename
320 $self->state($STATE_INVALID);
322 elsif($state==$STATE_ALL)
324 if($name eq $PATH_ALLTRACKS)
326 $self->state($STATE_TRACKLIST);
328 elsif($name eq $PATH_NOARTIST)
330 $self->state($STATE_TRACKLIST);
334 my $artist=ID3FS::Path::Node->new($self->{db}, $TYPE_ARTIST, $name);
337 push(@{$self->{elements}}, $artist);
338 $self->state($STATE_ALBUMS);
342 $self->state($STATE_INVALID);
348 print "SM: ERROR: UNKNOWN STATE: $self->{state}\n";
349 $self->state($STATE_INVALID);
353 my @elements=@{$self->{elements}};
354 # remove trailing boolean
355 while(@elements && $self->is($TYPE_BOOL, $elements[$#elements]))
359 # print "\nELEMENTS: ", join(' ', map { $_->name(); } @elements), "\n";
360 my @joins=$self->number_joins(@elements);
361 @joins=qw(INNER) unless(@joins);
362 $self->{joins}=\@joins;
363 # print "AFTER: ", join(' ', map { $_->name() . "(" . $_->{table} . ")"; } @elements), "\n";
364 # print "JOINS: ", join(', ', @joins), "\n";
365 # sort elements by precedence
366 @elements=$self->sort_elements(@elements);
367 $self->{tagtree}=$self->elements_to_tree(\@elements);
372 my($self, @elements)=@_;
377 return (@joins) unless(@elements);
378 while(my $node=shift @elements)
380 if($node->type() == $TYPE_BOOL)
382 my $op=$node->name();
393 # NOT/foo/OR needs an extra join
394 $nextjoin="INNER" if($lastop && $lastop eq "NOT");
402 # if tag has a value, eat the tag, shifting to the value
403 if($elements[0]->type() == $TYPE_TAG)
406 $node=shift(@elements);
413 elsif($self->{db}->tag_has_values($node->id()))
415 # if the expression ends in a tag that has a value
416 # (ie we have the tag and want the value)
417 # use an INNER join even if we were in a NOT
418 $nextjoin="INNER" if($nextjoin);
423 push(@joins, $nextjoin);
426 $node->table($table);
434 my($self, $newstate)=@_;
435 if(defined($newstate))
437 $self->{state}=$newstate;
438 $self->{curtagdepth}++ if($newstate == $STATE_TAG);
440 return $self->{state};
443 # link up precedence-sorted list into a binary tree
446 my($self, $elements)=@_;
447 return undef unless(@$elements);
448 my ($left, $right, $op)=(undef, undef, undef);
449 my $node=pop @$elements;
450 if($self->is($TYPE_BOOL, $node))
452 $right=$self->elements_to_tree($elements);
453 if($node->name() ne "NOT")
455 $left=$self->elements_to_tree($elements);
458 $node->right($right);
463 # Dijkstra's shunting-yard algorithm
466 my ($self, @input)=@_;
469 while(my $node = shift @input)
471 if($self->is($TYPE_TAG, $node))
473 # Handle tag values by dropping parent
474 if(@input && $self->is($TYPE_TAG, $input[0]))
478 push(@output, $node);
480 elsif($self->is($TYPE_BOOL, $node))
483 ($priorities{$node->name()} <= $priorities{$opstack[$#opstack]->name()}))
485 push(@output, pop(@opstack));
487 push(@opstack, $node);
492 push(@output, pop(@opstack));
500 return() unless(defined($self->{tagtree}));
501 return($self->{tagtree}->used_tags());
507 my $tail=$self->tail();
508 if($self->is($TYPE_TAG, $tail))
510 return($self->{db}->tag_has_values($tail->id()));
517 my $tail=$self->tail();
518 if($self->is($TYPE_TAG, $tail))
525 sub trailing_tag_parent
528 my $tail=$self->tail();
529 if($self->is($TYPE_TAG, $tail))
531 return($tail->{parents_id});
539 return($self->{elements}->[$#{$self->{elements}}]);
544 my($self, $type, $node)=@_;
545 return 0 unless($node);
546 return 0 unless($node->type());
547 return 1 if($type == $node->type());
551 # the one before last
555 return($self->{elements}->[($#{$self->{elements}}) - 1]);
558 ######################################################################
563 if(!$self->{tagtree}) # / or /NOT
565 my $sql="SELECT DISTINCT name FROM tags WHERE parents_id='';";
566 return($self->{db}->cmd_firstcol($sql));
568 my $sql="SELECT tags.name FROM ";
569 if($self->want_all_tags())
571 $sql .= "files_x_tags\n";
576 $self->tags_subselect() .
578 "INNER JOIN files_x_tags ON subselect.files_id=files_x_tags.files_id\n");
580 $sql .= "INNER JOIN tags ON files_x_tags.tags_id=tags.id\n";
582 my $id=$self->trailing_tag_id();
584 my $parentclause= "tags.parents_id='";
585 $parentclause .= $id if($self->expecting_values());
586 $parentclause .= "'";
587 push(@andclauses, $parentclause);
589 my @used=$self->used_tags();
592 push(@andclauses, "tags.id NOT IN (" . join(', ', @used) . ")");
596 $sql .= "WHERE " . join(' AND ', @andclauses) . "\n";
599 $sql .= "GROUP BY tags.name;";
600 print "SQL(TAGS): $sql\n" if($self->{verbose});
601 my @tagnames=$self->{db}->cmd_firstcol($sql);
602 print("SUBNAMES: ", join(', ', @tagnames), "\n") if($self->{verbose});
609 if(!@{$self->{elements}}) # /ALL
611 my $sql="SELECT DISTINCT name FROM artists WHERE name!='';";
612 return($self->{db}->cmd_firstcol($sql));
614 my $sql=$self->sql_start("artists.name");
615 $sql .= ("INNER JOIN artists ON files.artists_id=artists.id\n" .
616 "WHERE artists.name != ''\n" .
617 "GROUP BY artists.name;");
618 print "SQL(ARTISTS): $sql\n" if($self->{verbose});
619 my @tagnames=$self->{db}->cmd_firstcol($sql);
620 print("ARTISTS: ", join(', ', @tagnames), "\n") if($self->{verbose});
627 my $tail=$self->tail();
628 if($self->is($TYPE_ARTIST, $tail))
630 return $self->artist_albums($tail->id());
632 my $sql=$self->sql_start("albums.name");
633 $sql .= ("INNER JOIN albums ON files.albums_id=albums.id\n" .
634 "WHERE albums.name != ''\n" .
635 "GROUP BY albums.name;");
636 print "SQL(ALBUMS): \n$sql\n" if($self->{verbose});
637 my @names=$self->{db}->cmd_firstcol($sql);
638 print("ALBUMS: ", join(', ', @names), "\n") if($self->{verbose});
644 my($self, $artist_id)=@_;
645 my $sql=$self->sql_start("albums.name");
646 $sql .= ("INNER JOIN albums ON albums.id=files.albums_id\n" .
647 "INNER JOIN artists ON artists.id=files.artists_id\n" .
648 "WHERE artists.id=? and albums.name <> ''\n" .
649 "GROUP BY albums.name\n");
650 print "ARTIST_ALBUMS SQL: $sql\n" if($self->{verbose});
651 my @albums=$self->{db}->cmd_firstcol($sql, $artist_id);
652 print("ALBUMS: ", join(', ', @albums), "\n") if($self->{verbose});
658 my($self, $artist_id)=@_;
659 my $sql=$self->sql_start("files.name");
660 $sql .= ("INNER JOIN artists ON artists.id=files.artists_id\n" .
661 "INNER JOIN albums ON albums.id=files.albums_id\n" .
662 "WHERE artists.id=? AND albums.name=''\n" .
663 "GROUP BY files.name\n");
664 print "ARTIST_TRACKS SQL: $sql\n" if($self->{verbose});
665 my @names=$self->{db}->cmd_firstcol($sql, $artist_id);
666 print("ARTISTTRACKS: ", join(', ', @names), "\n") if($self->{verbose});
672 my($self, $artist_id, $album_id)=@_;
673 my $sql=("SELECT files.name FROM files\n" .
674 "INNER JOIN albums ON albums.id=files.albums_id\n" .
675 "INNER JOIN artists ON artists.id=files.artists_id\n" .
676 "WHERE artists.id=? AND albums.id=?\n" .
677 "GROUP BY files.name\n");
678 print "ALBUM_TRACKS SQL($artist_id, $album_id): $sql\n" if($self->{verbose});
679 my @names=$self->{db}->cmd_firstcol($sql, $artist_id, $album_id);
680 print("TRACKS: ", join(', ', @names), "\n") if($self->{verbose});
687 my $tail=$self->tail();
688 if($self->is($TYPE_ARTIST, $tail))
690 return $self->artist_tracks($tail->id());
692 elsif($self->is($TYPE_ALBUM, $tail))
695 my $artist=$self->tail_parent();
696 if($self->is($TYPE_ARTIST, $artist))
698 # should always happen
699 $artist_id=$artist->id();
701 return $self->album_tracks($artist_id, $tail->id());
703 my $sql=$self->sql_start("files.name");
704 $sql .= "INNER JOIN artists ON files.artists_id=artists.id\n";
705 if($self->{components}->[$#{$self->{components}}] eq $PATH_NOARTIST)
707 $sql .= "WHERE artists.name =''\n";
709 $sql .= "GROUP BY files.name;";
710 print "TRACKS SQL($self->{path}): $sql\n" if($self->{verbose});
711 my @names=$self->{db}->cmd_firstcol($sql);
712 print("TRACKS: ", join(', ', @names), "\n") if($self->{verbose});
718 my($self, $mountpoint)=@_;
719 my $tail=$self->tail();
720 if($self->is($TYPE_FILE, $tail))
723 my $sql=("SELECT paths.name, files.name FROM files\n" .
724 "INNER JOIN paths ON files.paths_id=paths.id\n" .
725 "WHERE files.id=?\n" .
726 "GROUP BY paths.name, files.name");
727 print "FILENAME SQL: $sql\n" if($self->{verbose});
728 my ($path, $name)=$self->{db}->cmd_onerow($sql, $id);
729 return($self->{db}->relativise($path, $name, $mountpoint, $self->{path}));
731 # should never happen
738 my $tree=$self->{tagtree};
740 # print Dumper $tree;
741 my ($sqlclause, @joins)=$tree->to_sql() if($tree);
742 my $sql="\tSELECT fxt1.files_id FROM tags t1";
745 @joins=@{$self->{joins}};
746 for(my $i=0; $i <= $#joins; $i++)
750 my $inner=("\t$join JOIN files_x_tags fxt$cnt ON " .
751 "t${cnt}.id=fxt${cnt}.tags_id");
754 push(@crosses, "CROSS JOIN tags t$cnt");
755 $inner .= " AND fxt1.files_id=fxt${cnt}.files_id";
757 push(@inners, $inner);
759 $sql .= ("\n\t" . join(" ", @crosses)) if(@crosses);
760 $sql .= ("\n" . join("\n", @inners)) if(@inners);
761 $sql .= "\n\tWHERE $sqlclause" if($sqlclause);
762 $sql .= "\n\tGROUP BY fxt1.files_id\n";
768 my($self, $tables)=@_;
769 my $sql="SELECT $tables FROM ";
777 $self->tags_subselect() .
779 "INNER JOIN files ON subselect.files_id=files.id\n");
784 # we just filter $ALLTRACKS, $NOARTIST and $NOALBUM
785 # filtering tags properly requires up to four levels of recursion
786 # (tag/tagval/AND/NOT) and is too slow
790 return(@dirs) unless($ENABLE_FILTER);
791 my $base=$self->{path};
795 push(@outdirs, $dir) unless($self->empty("$base/$dir"));
803 my $path=ID3FS::Path->new($self->{db}, $dir, $self->{verbose},
804 ($self->{maxtagdepth} - $self->{curtagdepth}));
805 return 1 unless($path->isvalid());
806 my($subdirs,$subfiles)=$path->dirents();
807 return 0 if(@$subfiles || @$subdirs);
811 # if path is .../OR/ or .../OR/NOT or .../AND/NOT
815 my $tail=$self->tail();
816 return 0 unless($tail);
817 my $parent=$self->tail_parent();
818 my $parent_valid = ($parent && $parent->type() == $TYPE_BOOL);
819 if($tail->type() == $TYPE_BOOL)
821 return 1 if($tail->name() eq "OR");
822 return 0 unless($tail->name() eq "NOT");
823 return 0 unless($parent_valid);
824 return 1 if($parent->name() eq "OR");
825 return 1 if($parent->name() eq "AND");
827 elsif($tail->type() == $TYPE_TAG)
829 return 0 unless($parent_valid);
830 return 1 if($parent->name() eq "NOT");