package MT::Plugin::CloudNine;

use CloudCreator;
use Data::Dumper;
use POSIX qw(strftime);

use MT::Entry;
use MT::ObjectTag;
use MT::Request;
use MT::Tag;
use MT::Template::Context;

use vars qw( $VERSION );

$VERSION = 3.3;

my $plugin = MT::Plugin->new( {
	name => 'CloudNine',
	description => 'Create weighted tag clouds for your blog.',
	doc_link => 'http://blog.socklabs.com/cloudnine/',
	author_name => 'Nick Gerakines',
	author_link => 'http://www.socklabs.com/',
	config_template => \&configuration_template,
	settings => new MT::PluginSettings( [
		['smallest'],
		['largest'],
		['start'],
		['last'],
		['variation'],
		['scheme'],
		['minentries'],
		['reverse'],
		['unit'],
		['enabgrad'],
	] )
});
MT->add_plugin($plugin);

MT::Template::Context->add_container_tag('CloudNine' => \&cloudnine);
MT::Template::Context->add_tag('CloudWeight' => \&_catcloud_weight);
MT::Template::Context->add_tag('CloudLabel' => \&_catcloud_label);
MT::Template::Context->add_tag('CloudStub' => \&_catcloud_stub);
MT::Template::Context->add_tag('CloudCSS' => \&_catcloud_css);

sub instance { $plugin; }

sub configuration_template {
	my ($plugin, $param, $scope) = @_;
    $param->{'smallest'} ||= 8;
    $param->{'scheme'} ||= 'mono';
    $param->{'variation'} ||= 'default';
	$param->{'reverse'} ||= 0;
    $param->{'largest'} ||= 16;
	$param->{'unit'} ||= 'pt';
    $param->{'start'} ||= '#C8C8FF';
    $param->{'last'} ||= '';
    $param->{'minentries'} ||= 0;
    $param->{'enabgrad'} ||= 'enable';
	return qq{
<p>A quick note about this plugin: This plugin will attempt to create nice looking tag clouds based on a number of things. It does this pretty well, however there are some caveats. The colors you expect from the gradients you put in, may not be what you think. Because of this, getting what you want may take some tweaking and adjusting here and there. Just try not to get real frustrated.</p>
<p>&nbsp;</p>
<div class="setting">
<p>This field represents the minimum entry count that category must have to show up in the cloud.</p>
<div class="label"><label for="cloudnine_minentries">Minimum Entry Count:</label></div>
<div class="field">
<input type="text" name="minentries" id="cloudnine_minentries" value="<TMPL_VAR NAME=MINENTRIES>" style="width: 15%" />
</div>
</div>

<div class="setting">
<p>Set the font unit: Defaults to 'pt'.</p>
<div class="label"><label for="cloudnine_minentries">Font unit:</label></div>
<div class="field">
<input type="text" name="unit" id="cloudnine_unit" value="<TMPL_VAR NAME=UNIT>" style="width: 15%" />
</div>
</div>

<div class="setting">
<p>Set the smallest size unit of the cloud. Defaults to 8</p>
<div class="label"><label for="cloudnine_smallest">Smallest:</label></div>
<div class="field">
<input type="text" name="smallest" id="cloudnine_smallest" value="<TMPL_VAR NAME=SMALLEST>" style="width: 10%" />
</div>
</div>

<div class="setting">
<p>Set the largest size unit of the cloud. Defaults to 16</p>
<div class="label"><label for="cloudnine_largest">Largest:</label></div>
<div class="field">
<input type="text" name="largest" id="cloudnine_largest" value="<TMPL_VAR NAME=LARGEST>" style="width: 10%" />
</div>
</div>

<p>&nbsp;</p>
<p>Select this option if you would like to enable color gradients in the cloud.</p>
<div class="setting">
<div class="label">Enable Gradients:</div>
<div class="field">
<label><input type="checkbox" name="enabgrad" value="enable" <TMPL_IF NAME=ENABGRAD>checked="checked"</TMPL_IF>/></label>
</div>
<p>&nbsp;</p>
<p>Set the color value for the smallest entries in the cloud. Note that this is the starting point of the gradient creaetor and by default the cloud creator will automatically find a contrasting color to swing to. This must be a full 6 character hex value. aka 006400</p>
<div class="label"><label for="cloudnine_start">Light:</label></div>
<div class="field">
<input type="text" name="start" id="cloudnine_start" value="<TMPL_VAR NAME=START>" style="width: 10%" />
</div>
<p>&nbsp;</p>
<p>Set the color value for the largest entries in the cloud. If you would like CloudNine to find a contrasting color and do its own voodoo, leave this blank.</p>
<div class="setting">
<div class="label"><label for="cloudnine_last">Heavy:</label></div>
<div class="field">
<input type="text" name="last" id="cloudnine_last" value="<TMPL_VAR NAME=LAST>" style="width: 10%" />
</div>
</div>

<p>Set the color scheme to use when finding contrasting and gradiant colors.</p>
<p>Currently set to <TMPL_VAR NAME=SCHEME></p>
<div class="setting">
<div class="label">Scheme:</div>
<div class="field">
<select name="scheme">
<option value="mono">Default (mono)</option>
<option value="contrast">Contrats</option>
<option value="triade">Triade</option>
<option value="tetrade">Tetrade</option>
<option value="analogic">Analogic</option>
</select>
</div>
<p>&nbsp;</p>
<p>Set the color scheme variation to use when finding contrasting and gradiant colors.</p>
<p>Currently set to <TMPL_VAR NAME=VARIATION></p>
<div class="label">Variation:</div>
<div class="field">
<select name="variation">
<option value="default">Default</option>
<option value="pastel">Pastel</option>
<option value="soft">Soft</option>
<option value="light">Light</option>
<option value="hard">Hard</option>
<option value="pale">Pale</option>
</select>
</div>
<p>&nbsp;</p>
<p>Select this option if you would like to reverse the color scheme that CloudNine finds for you. <em>This is experimental and may not work the way you want it to.</em></p>
<div class="label">Reverse:</div>
<div class="field">
<label><input type="checkbox" name="reverse" value="enable" <TMPL_IF NAME=REVERSE>checked="checked"</TMPL_IF>/></label>
</div>
</div>
	};
}

sub config {
    my $config = {};
    if ($plugin) {
        my ($scope) = (@_);
        $config = MT::Request->instance->cache('catcloud_config_'.$scope);
        if (!$config) {
            $config = $plugin->get_config_hash($scope);
            MT::Request->instance->cache('catcloud_config_'.$scope, $config);
        }
    }
    $config;
}

sub cloudnine {
	my ($ctx, $args, $cond) = @_;
	my $blog_id = $ctx->stash('blog_id');
	my $config = config('blog:' . $blog_id);
	my $cloud = CloudCreator->new(
		'smallest'  => $config->{'smallest'},
		'largest'   => $config->{'largest'},
		'start'     => $config->{'start'},
		'last'      => $config->{'last'},
		'variation' => $config->{'variation'},
		'scheme'    => $config->{'scheme'},
		'reverse'   => $config->{'reverse'},
		'unit'      => $config->{'unit'},
		'enabgrad'  => $config->{'enabgrad'},
	);
	my $builder = $ctx->stash('builder');
	my $tokens = $ctx->stash('tokens');
    my $parts;
    if ( ! $args->{'type'} ) { $parts = blog_tags($blog_id); }
    if ( $args->{'type'} eq 'blog_tags' ) { $parts = blog_tags($blog_id); }
    if ( $args->{'type'} eq 'blog_categories' ) { $parts = blog_categories($blog_id); }
    if ( $args->{'type'} eq 'top_tags' ) { $parts = top_tags($blog_id); }
    if ( $args->{'type'} eq 'recent_tags' ) { $parts = recent_tags($blog_id); }
    if ( $args->{'type'} eq 'css' ) {
        my $span_one = $config->{'smallest'};
        my $span_two = $config->{'largest'};
        my @span = ($span_one .. $span_two);
        $parts = [ map { [$_, $_, $_] } @span ];
    }
    for my $part (@{ $parts }) {
        $cloud->add( $part->[0], $part->[1], $part->[2] );
    }
	my @tags = $cloud->gencloud;
	undef $cloud;
	for my $tag ( @tags ) {
		local $ctx->{__stash}{'CloudLabel'} = $tag->[0];
		local $ctx->{__stash}{'CloudWeight'} = $tag->[4];
		local $ctx->{__stash}{'CloudCSS'} = $tag->[2];
		local $ctx->{__stash}{'CloudStub'} = $tag->[3];
		defined(my $out = $builder->build( $ctx, $tokens )) or return $ctx->error( $ctx->errstr );
		push @res, $out;
	}
	my $glue = $args->{'glue'} || '';
	join $glue, @res;
}

sub blog_tags {
    my ($blog_id) = @_;
    my $type = MT::Entry->datasource;
    my @tags = MT::Tag->load_by_datasource( $type,  { 'is_private' => 0, 'blog_id' => $blog_id }, );
    my @out;
    for my $tag (@tags) {
        my $count = MT::Entry->count(
            { 'blog_id' => $blog_id, 'status' => MT::Entry::RELEASE() },
            { 'join' => [ 'MT::ObjectTag', 'object_id', { 'tag_id' => $tag->id, 'object_datasource' => $type }], 'unique' => 1 }
        );
        push @out, [ $tag->name, $count, $tag->name ];
    }
    return \@out
}

sub blog_categories {
    my ($blog_id) = @_;
    my @blog_categories = MT::Category->load( { 'blog_id' => $blog_id } );
    my @out;
    for my $cat ( @blog_categories ) {
        my $count = MT::Entry->count(
            { 'blog_id' => $blog_id, 'status' => MT::Entry::RELEASE() },
            { 'join' => [ 'MT::Placement', 'entry_id', { 'category_id' => $cat->id }, { 'unique' => 1 } ],}
        );
        push @out, [ $cat->label, $count, $cat->basename ];
    }
    return \@out;
}

sub top_tags {
    my ($blog_id) = @_;
    my $type = MT::Entry->datasource;
    my @tags = MT::Tag->load_by_datasource( $type,  { 'is_private' => 0, 'blog_id' => $blog_id }, );
    my (@out, @top);
    for my $tag (@tags) {
        my $count = MT::Entry->count(
            { 'blog_id' => $blog_id, 'status' => MT::Entry::RELEASE() },
            { 'join' => [ 'MT::ObjectTag', 'object_id', { 'tag_id' => $tag->id, 'object_datasource' => $type }], 'unique' => 1 }
        );
        push @top, [ $tag->name, $count, $tag->name ];
    }
    @top = sort { $b->[1] <=> $a->[1] } @top;
    return [ @top[ 0 .. 25 ] ];
}


sub recent_tags {
    my ($blog_id) = @_;
    my @entries = MT::Entry->load(
		{ 'blog_id' => $blog_id, 'status' => MT::Entry::RELEASE(), },
		{ 'sort' => 'created_on', 'direction' => 'descend', 'limit' => 10 }
	);
    my %entry_tags = map { $_ => 1 } map { $_->tags } @entries;
    my $type = MT::Entry->datasource;
    my @tags = MT::Tag->load_by_datasource( $type,  { 'name' => [ keys %entry_tags ], 'blog_id' => $blog_id } );
    my @out;
    for my $tag (@tags) {
        my $count = MT::Entry->count(
            { 'blog_id' => $blog_id, 'status' => MT::Entry::RELEASE() },
            { 'join' => [ 'MT::ObjectTag', 'object_id', { 'tag_id' => $tag->id, 'object_datasource' => $type }], 'unique' => 1 }
        );
        push @out, [ $tag->name, $count, $tag->name ];
    }
    return \@out;
}

sub _catcloud_weight {
	$_[0]->stash('CloudWeight') || 0;
}

sub _catcloud_label {
	$_[0]->stash('CloudLabel') || '';
}

sub _catcloud_stub {
	$_[0]->stash('CloudStub') || '';
}

sub _catcloud_css {
	$_[0]->stash('CloudCSS') || '';
}

1;