source: bookmarks/trunk/bkmk @ 101

#!/usr/bin/perl -w
use strict;

use FindBin;
use lib "$FindBin::RealBin/lib";

use YAML;
use Getopt::Long;

use Bookmarks;

GetOptions(
    'file|f=s' => \my $DBNAME,
    'title=s' => \my $TITLE,
);

my $dbname = $DBNAME || $ENV{BKMK_DBNAME};
die "Usage: $0 --file <dbname> <command>\n" unless $dbname;

my $bookmarks = Bookmarks->new({
    dbname => $dbname,
});

my $command = shift;

my %action_for = (
    init => sub {
        my $src_file = shift;
        $bookmarks->create_tables;
        load_bookmarks($src_file) if $src_file;
    },
    
    get => sub {
        my $identifier = shift;
        my $bookmark = find_bookmark($identifier);
        print $bookmark ? Dump($bookmark->to_hashref) : "Not Found\n";
    },
    
    add => sub {
        my ($uri, @tags) = @_;
        my $title = defined $TITLE ? $TITLE : fetch_title($uri);
        my $bookmark = $bookmarks->add({ uri => $uri, title => $title, tags => \@tags });
        print Dump($bookmark->to_hashref);
    },

    list => sub {
        my @tags = @_;
        my $resources = $bookmarks->search({
            tags => \@tags
        });
        # TODO: list by tags, date, etc.
        # TODO: coordinate this commandline script with the CGI app
        print Dump([ map { $_->to_hashref } @{ $resources->results } ]);
    },

    tag => sub {
        my ($identifier, @tags) = @_;
        my $bookmark = find_bookmark($identifier);
        if ($bookmark) {
            $bookmark->tags(\@tags);
            $bookmarks->update($bookmark);
            print Dump($bookmark->to_hashref);
        } else {
            die "Not found\n";
        }
    },

    #TODO: interactive editing of a bookmark

    # bulk loading
    load => sub {
        my ($src_file) = @_;
        load_bookmarks($src_file);
    },

    # bulk dumping
    dump => sub {
        my ($dump_file) = @_;
        my $dump = [ map { $_->to_hashref } @{ $bookmarks->search->results } ];
        $dump_file ? YAML::DumpFile($dump_file, $dump) : print Dump($dump);
    },
);

$action_for{$command}->(@ARGV);

sub find_bookmark {
    my $identifier = shift;
    my $query = $identifier =~ /^\d+$/ ? { id => $identifier } : { uri => $identifier };
    return $bookmarks->get_bookmark($query);
}

sub fetch_title {
    my $uri = shift;
    require WWW::Mechanize;
    my $mech = WWW::Mechanize->new;
    $mech->get($uri);
    return $mech->title || $uri;
}

sub load_bookmarks {
    my $src_file = shift;
    my $src_bookmarks = YAML::LoadFile($src_file);
    for my $bookmark (@{ $src_bookmarks }) {
        $bookmarks->add($bookmark);
    }
}