#!/usr/bin/perl
#-----------------------------------------------------------------------------
#
#	Copyright (C) 2011 - 2016
#		Michael Theall (mtheall)
#		Dave Murphy (WinterMute)
#
#	This software is provided 'as-is', without any express or implied
#	warranty.  In no event will the authors be held liable for any
#	damages arising from the use of this software.
#
#	Permission is granted to anyone to use this software for any
#	purpose, including commercial applications, and to alter it and
#	redistribute it freely, subject to the following restrictions:
#
#	1.	The origin of this software must not be misrepresented; you
#		must not claim that you wrote the original software. If you use
#		this software in a product, an acknowledgment in the product
#		documentation would be appreciated but is not required.
#	2.	Altered source versions must be plainly marked as such, and
#		must not be misrepresented as being the original software.
#	3.	This notice may not be removed or altered from any source
#		distribution.
#
#-----------------------------------------------------------------------------
  use strict;

  my $dir = "$ENV{HOME}/devkitPro";
  my $downloader;
  my $archname;

  if($ENV{"DEVKITPRO"} ne "")
  {
    $dir = $ENV{"DEVKITPRO"};
  }

  if($#ARGV eq 0)
  {
    $dir = $ARGV[0];
  }

  # Ensure full pathname
  if(!($dir =~ /^\//))
  {
    my $pwd = `pwd`;
    chomp($pwd);
    $dir = "$pwd/$dir";
  }

  printf("devkitPPC Updater/Installer\n");
  printf("Installing to %s\n", $dir);

  # Get OS information
  my $os = `uname`;
  my $arch = `uname -m`;
  chomp($os);
  chomp($arch);

  # Check OS information
  if($os eq "Linux" and ($arch eq "i686" or $arch eq "x86_64"))
  {
	 $downloader = "wget -q";
	 $archname = $arch . "-linux";
  }
  elsif($os eq "Darwin" and ($arch eq "i386" or $arch eq "x86_64"))
  {
	 $downloader = "curl -L -O -s";
	 $archname = $arch . "-osx";
  }
  else
  {
    printf(STDERR "Not on Linux i686/x86_64 or Darwin i386/x86_64!\n");
    exit(1);
  }

  # Set up directories
  if(!(-d "$dir"))
  {
    mkdir("$dir") or die $!;
  }

  if(!(-d "$dir/libogc"))
  {
    mkdir("$dir/libogc") or die $!;
  }
  if(!(-d "$dir/examples"))
  {
    mkdir("$dir/examples") or die $!;
  }
  if(!(-d "$dir/examples/wii"))
  {
    mkdir("$dir/examples/wii") or die $!;
  }
  if(!(-d "$dir/examples/gamecube"))
  {
    mkdir("$dir/examples/gamecube") or die $!;
  }

  # Grab update file
  if(-e "devkitProUpdatePPCr29.ini")
  {
	unlink("devkitProUpdatePPCr29.ini") or die $!;
  }
  printf("Downloading update file...");
  system($downloader . " https://raw.githubusercontent.com/Maschell/controller_patcher/master/other/devkitProUpdatePPCr29.ini") and die "Failed to download!";
  printf("OK!\n");

  # Initialize versions & newVersions
  my %versions =
    (
      'devkitPPC'    => 0,
      'libogc'       => 0,
      'libogcfat'    => 0,
      'wiiexamples'  => 0,
      'cubeexamples' => 0,
    );
  my %newVersions = %versions;

  my %files    = ();
  my $current  = "";

  if(-e "$dir/dkppc-update.ini")
  {
    open(MYFILE, "<$dir/dkppc-update.ini") or die $!;
    while(<MYFILE>)
    {
      chomp;
      if($_ =~ /\[(.*)\]/)
      {
        $current = $1;
      }
      elsif($_ =~ /Version=(.*)/ and defined($versions{$current}))
      {
        $versions{$current} = $1;
      }
      elsif($_ =~ /File=(.*)/)
      {
        $files{$current} = $1;
      }
    }
    close(MYFILE);
  }

  my %newFiles = ();

  open(MYFILE, "<devkitProUpdatePPCr29.ini") or die $!;
  while(<MYFILE>)
  {
    chomp;
    if($_ =~ /\[(.*)\]/)
    {
      $current = $1;
    }
    elsif($_ =~ /Version=(.*)/ and defined($newVersions{$current}))
    {
      $newVersions{$current} = $1;
    }
    elsif($_ =~ /File=(.*)/)
    {
      $newFiles{$current} = $1;
    }
  }
  close(MYFILE);
  unlink("devkitProUpdatePPCr29.ini") or die $!;

  # see what to update
  my %updates = ();
  foreach my $key (keys %versions)
  {
    if($versions{$key} ne $newVersions{$key} and $newVersions{$key} ne 0)
    {
      $newFiles{$key} =~ s/win32\.exe/$archname.tar.bz2/;
      $updates{$key} = $newFiles{$key};
    }
    else
    {
      printf("%s is up-to-date\n", $key);
    }
  }
  
  # Download files
  foreach my $key (keys %updates)
  {
    printf("Update %s with %s\n", $key, $updates{$key});
    if(-e $updates{$key})
    {
      unlink($updates{$key});
    }

    my $cmd = sprintf("%s http://download.sourceforge.net/devkitpro/%s", $downloader, $updates{$key});
    printf("  Downloading...");
    system($cmd) and die "Failed to download $updates{$key}\n";
    printf("OK!\n");
  }

  # Install files
  my %install =
    (
      'devkitPPC'    => '',
      'libogc'       => 'libogc',
      'libogcfat'    => 'libogc',
      'wiiexamples'  => 'examples/wii',
      'cubeexamples' => 'examples/gamecube',
    );

  foreach my $key (keys %updates)
  {
    my $cmd = sprintf("tar -xjf %s -C $dir/%s", $updates{$key}, $install{$key});
    printf("Extracting %s...", $updates{$key});
    system($cmd) and die "Failed\n";
    printf("OK!\n");
  }

  # Output update info
  open(MYFILE, ">$dir/dkppc-update.ini") or die $!;
  foreach my $key (keys %newVersions)
  {
    printf(MYFILE "[%s]\n", $key);
    printf(MYFILE "Version=%s\n", $newVersions{$key});
    printf(MYFILE "File=%s\n", $newFiles{$key});
    printf(MYFILE "\n");
  }
  close(MYFILE);

  # Check environment variables
  printf("Checking DEVKITPRO...");
  my $env = `echo \$DEVKITPRO`;
  chomp($env);
  if($env ne "$dir")
  {
    printf("Please set DEVKITPRO in your environment as $dir\n");
  }
  else
  {
    printf("OK!\n");
  }

  printf("Checking DEVKITPPC...");
  $env = `echo \$DEVKITPPC`;
  chomp($env);
  if($env ne "$dir/devkitPPC")
  {
    printf("Please set DEVKITPPC in your environment as \${DEVKITPRO}/devkitPPC\n");
  }
  else
  {
    printf("OK!\n");
  }

  exit(0);