use strict;


FillInParentProductID->start();



package FillInParentProductID;
use lib '/app/tools/common/lib';
use lib '/app/tools/rps/lib';

use Common::Log;
use Common::Util;
use Common::Client;
use Common::RSApp;
use RPS::DB::Item::Product;
use RPS::DB::Item::Track;


use base 'RPS::Script';

sub _options
{
   {
      test => {
              short        => 't',
              description  => 'Test only, no changes',
              required     => 0,
      },
      alter => {
              short        => 'a',
              description  => 'Alter database (add new column)',
              required     => 0,
      },
   }
}



sub _process {
    my $self = shift;

    my $clientName = Common::Client::Current()->ClientName();
    print "$clientName\n";
    
    if ($self->{_test})
    {
        print "Running in test mode.  No changes will be saved.\n";
    }
    
    my $dbo = Common::RSApp::GetClientDB();

    # First we need to add the new column.
    if (!$self->{_test} && $self->{_alter})
    {
        $dbo->DoCmd('ALTER TABLE product ADD COLUMN parent_product_id int(10) unsigned default NULL AFTER product_id');
    }
    
    # Now we can fill it in for the digital track products.
    my $products = RPS::DB::Item::Product->GetAll();

    while (my $product = $products->next())
    {
        if ($product->product_type_id != RPS::DB::Item::Product::kProductTypeDigitalTrack)
        {
            next();
        }

        # Skip if we've already populated parent_product_id
        if ($product->parent_product_id)
        {
            next();
        }
        
        my $track = RPS::DB::Item::Track->Lookup(track_id => $product->asset_id);
        
        if ($track)
        {
            my $albumProducts = RPS::DB::Item::Product->GetAllDigitalAlbumProducts($track->album_id);
            
            if ($albumProducts->size() == 1)
            {
                if (!$self->{_test})
                {
                    my $albumProduct = $albumProducts->next();
                    $product->parent_product_id($albumProduct->product_id);
                    $product->save();
                }
            }
            elsif ($albumProducts->size() == 0)
            {
                Log->warn("Unable to find parent product for product ID: " . $product->product_id);
            }
            else
            {
                Log->warn("Found multiple parent products for product ID: " . $product->product_id);
            }
        }
    }    
    
}

sub _getProductionClientIDs {
	return Common::RSDB::GetAllRPSClientIDs();
}

sub _getLocalClientIDs {
	return 202; # RS Test
}

1;