/*

 * Hedgewars-iOS, a Hedgewars port for iOS devices

 * Copyright (c) 2009-2012 Vittorio Giovara <vittorio.giovara@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA.

 */

#import "SchemeWeaponConfigViewController.h"

#import <QuartzCore/QuartzCore.h>

#define DISABLED_GAME_STYLES @[@"Frenzy.lua", @"Gravity.lua", @"HedgeEditor.lua", @"Continental_supplies.lua", @"Space_Invasion.lua", @"Tumbler.lua"]

#define LABEL_TAG 57423

#define TABLE_TAG 45657

@implementation SchemeWeaponConfigViewController

@synthesize listOfSchemes, listOfWeapons, listOfScripts, lastIndexPath_sc, lastIndexPath_we, lastIndexPath_lu,

            selectedScheme, selectedWeapon, selectedScript, scriptCommand, topControl, sectionsHidden;

- (BOOL)shouldAutorotateToInterfaceOrientation:(UIInterfaceOrientation)interfaceOrientation {

    return rotationManager(interfaceOrientation);

}

#pragma mark -

#pragma mark custom setters/getters

- (NSString *)selectedScheme {

    if (selectedScheme == nil)

        self.selectedScheme = @"Default.plist";

    return selectedScheme;

}

- (NSString *)selectedWeapon {

    if (selectedWeapon == nil)

        self.selectedWeapon = @"Default.plist";

    return selectedWeapon;

}

- (NSString *)selectedScript {

    if (selectedScript == nil)

        self.selectedScript = @"";

    return selectedScript;

}

- (NSString *)scriptCommand {

    if (scriptCommand == nil)

        self.scriptCommand = @"";

    return scriptCommand;

}

-(NSArray *)listOfSchemes {

    if (listOfSchemes == nil)

        self.listOfSchemes = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:SCHEMES_DIRECTORY() error:NULL];

    return listOfSchemes;

}

-(NSArray *)listOfWeapons {

    if (listOfWeapons == nil)

        self.listOfWeapons = [[NSFileManager defaultManager] contentsOfDirectoryAtPath:WEAPONS_DIRECTORY() error:NULL];

    return listOfWeapons;

}

-(NSArray *)listOfScripts {

    if (listOfScripts == nil)

        self.listOfScripts = [[[NSFileManager defaultManager] contentsOfDirectoryAtPath:SCRIPTS_DIRECTORY() error:NULL]

                              filteredArrayUsingPredicate:[NSPredicate predicateWithFormat:@"SELF ENDSWITH '.lua' AND NOT (SELF IN %@)", DISABLED_GAME_STYLES]];

    return listOfScripts;

}

-(UISegmentedControl *)topControl {

    if (topControl == nil) {

        NSArray *array = [[NSArray alloc] initWithObjects:

                          NSLocalizedString(@"Scheme",@""),

                          NSLocalizedString(@"Weapon",@""),

                          NSLocalizedString(@"Style",@""),nil];

        UISegmentedControl *controller = [[UISegmentedControl alloc] initWithItems:array];

        controller.segmentedControlStyle = UISegmentedControlStyleBar;

        controller.tintColor = [UIColor lightGrayColor];

        controller.selectedSegmentIndex = 0;

        self.topControl = controller;

    }

    return topControl;

}

#pragma mark -

#pragma mark View lifecycle

- (void)viewDidLoad {

    self.sectionsHidden = NO;

    NSInteger topOffset = IS_IPAD() ? 45 : 0;

    NSInteger bottomOffset = IS_IPAD() ? 3 : 0;

    UITableView *aTableView = [[UITableView alloc] initWithFrame:CGRectMake(0,

                                                                            topOffset,

                                                                            self.view.frame.size.width,

                                                                            self.view.frame.size.height - topOffset - bottomOffset)

                                                           style:UITableViewStyleGrouped];

    aTableView.delegate = self;

    aTableView.dataSource = self;

    if (IS_IPAD()) {

        [aTableView setBackgroundColorForAnyTable:[UIColor clearColor]];

        UILabel *background = [[UILabel alloc] initWithFrame:CGRectMake(0, 0, self.view.frame.size.width, self.view.frame.size.height)

                                                    andTitle:nil

                                             withBorderWidth:2.7f];

        background.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;

        [self.view insertSubview:background atIndex:0];

        self.topControl.frame = CGRectMake(0, 4, self.view.frame.size.width * 80/100, 30);

        self.topControl.autoresizingMask = UIViewAutoresizingFlexibleLeftMargin | UIViewAutoresizingFlexibleRightMargin;

        self.topControl.center = CGPointMake(self.view.frame.size.width/2, 24);

        [self.topControl addTarget:aTableView action:@selector(reloadData) forControlEvents:UIControlEventValueChanged];

        [self.view addSubview:self.topControl];

    } else {

        UIImage *backgroundImage = [[UIImage alloc] initWithContentsOfFile:@"background~iphone.png"];

        UIImageView *background = [[UIImageView alloc] initWithImage:backgroundImage];

        background.contentMode = UIViewContentModeScaleAspectFill;

        background.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;

        [self.view addSubview:background];

        [aTableView setBackgroundColorForAnyTable:[UIColor clearColor]];

    }

    aTableView.tag = TABLE_TAG;

    aTableView.indicatorStyle = UIScrollViewIndicatorStyleWhite;

    aTableView.separatorColor = [UIColor whiteColor];

    aTableView.separatorStyle = UITableViewCellSeparatorStyleNone;

    aTableView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;

    [self.view addSubview:aTableView];

    [super viewDidLoad];

    // display or hide the lists, driven by MapConfigViewController

    [[NSNotificationCenter defaultCenter] addObserver:self

                                             selector:@selector(fillSections)

                                                 name:@"fillsections"

                                               object:nil];

    [[NSNotificationCenter defaultCenter] addObserver:self

                                             selector:@selector(emptySections)

                                                 name:@"emptysections"

                                               object:nil];

}

#pragma mark -

#pragma mark Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)aTableView {

    return (self.sectionsHidden ? 0 : 1);

}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {

    if (self.topControl.selectedSegmentIndex == 0)

        return [self.listOfSchemes count];

    else if (self.topControl.selectedSegmentIndex == 1)

        return [self.listOfWeapons count];

    else

        return [self.listOfScripts count] + 1; // +1 for fake 'Normal'

}

// Customize the appearance of table view cells.

-(UITableViewCell *)tableView:(UITableView *)aTableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {

    static NSString *CellIdentifier = @"Cell";

    NSInteger index = self.topControl.selectedSegmentIndex;

    NSInteger row = [indexPath row];

    UITableViewCell *cell = [aTableView dequeueReusableCellWithIdentifier:CellIdentifier];

    if (cell == nil)

        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleSubtitle reuseIdentifier:CellIdentifier];

    cell.accessoryView = nil;

    if (0 == index) {

        cell.textLabel.text = [[self.listOfSchemes objectAtIndex:row] stringByDeletingPathExtension];

        NSString *str = [NSString stringWithFormat:@"%@/%@",SCHEMES_DIRECTORY(),[self.listOfSchemes objectAtIndex:row]];

        NSDictionary *dict = [[NSDictionary alloc] initWithContentsOfFile:str];

        cell.detailTextLabel.text = [dict objectForKey:@"description"];

        if ([[self.listOfSchemes objectAtIndex:row] isEqualToString:self.selectedScheme]) {

            UIImageView *checkbox = [[UIImageView alloc] initWithImage:[UIImage imageWithContentsOfFile:@"checkbox.png"]];

            cell.accessoryView = checkbox;

            self.lastIndexPath_sc = indexPath;

        }

    } else if (1 == index) {

        cell.textLabel.text = [[self.listOfWeapons objectAtIndex:row] stringByDeletingPathExtension];

        NSString *str = [NSString stringWithFormat:@"%@/%@",WEAPONS_DIRECTORY(),[self.listOfWeapons objectAtIndex:row]];

        NSDictionary *dict = [[NSDictionary alloc] initWithContentsOfFile:str];

        cell.detailTextLabel.text = [dict objectForKey:@"description"];

        if ([[self.listOfWeapons objectAtIndex:row] isEqualToString:self.selectedWeapon]) {

            UIImageView *checkbox = [[UIImageView alloc] initWithImage:[UIImage imageWithContentsOfFile:@"checkbox.png"]];

            cell.accessoryView = checkbox;

            self.lastIndexPath_we = indexPath;

        }

    } else {

        if (row == 0)

        {

            cell.textLabel.text = @"Normal";

            if ([self.selectedScript isEqualToString:@""])

            {

                UIImageView *checkbox = [[UIImageView alloc] initWithImage:[UIImage imageWithContentsOfFile:@"checkbox.png"]];

                cell.accessoryView = checkbox;

                self.lastIndexPath_lu = indexPath;

            }

        }

        else

        {

            row--;

            cell.textLabel.text = [[[self.listOfScripts objectAtIndex:row] stringByDeletingPathExtension]

                                   stringByReplacingOccurrencesOfString:@"_" withString:@" "];

            //cell.detailTextLabel.text = ;

            if ([[self.listOfScripts objectAtIndex:row] isEqualToString:self.selectedScript]) {

                UIImageView *checkbox = [[UIImageView alloc] initWithImage:[UIImage imageWithContentsOfFile:@"checkbox.png"]];

                cell.accessoryView = checkbox;

                self.lastIndexPath_lu = indexPath;

            }

        }

    }

    cell.backgroundColor = [UIColor blackColorTransparent];

    cell.textLabel.textColor = [UIColor lightYellowColor];

    cell.detailTextLabel.textColor = [UIColor whiteColor];

    cell.textLabel.adjustsFontSizeToFitWidth = YES;

    cell.detailTextLabel.adjustsFontSizeToFitWidth = YES;

    return cell;

}

- (CGFloat)tableView:(UITableView *)aTableView heightForHeaderInSection:(NSInteger)section {

    return IS_IPAD() ? 0 : 50;

}

-(UIView *)tableView:(UITableView *)aTableView viewForHeaderInSection:(NSInteger)section {

    if (IS_IPAD())

        return nil;

    UIView *theView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, self.view.frame.size.width, 30)];

    theView.autoresizingMask = UIViewAutoresizingFlexibleWidth;

    self.topControl.frame = CGRectMake(0, 0, self.view.frame.size.width * 80/100, 30);

    self.topControl.center = CGPointMake(self.view.frame.size.width/2, 24);

    [self.topControl addTarget:aTableView action:@selector(reloadData) forControlEvents:UIControlEventValueChanged];

    [theView addSubview:self.topControl];

    return theView;

}

- (CGFloat)tableView:(UITableView *)aTableView heightForFooterInSection:(NSInteger)section {

    return 40;

}

-(UIView *)tableView:(UITableView *)aTableView viewForFooterInSection:(NSInteger)section {

    NSInteger height = 40;

    UIView *footer = [[UIView alloc] initWithFrame:CGRectMake(0, 0, aTableView.frame.size.width, height)];

    footer.backgroundColor = [UIColor clearColor];

    footer.autoresizingMask = UIViewAutoresizingFlexibleWidth;

    UILabel *label = [[UILabel alloc] initWithFrame:CGRectMake(0, 0, aTableView.frame.size.width*90/100, height)];

    label.center = CGPointMake(aTableView.frame.size.width/2, height/2);

    label.textAlignment = NSTextAlignmentCenter;

    label.font = [UIFont italicSystemFontOfSize:12];

    label.textColor = [UIColor whiteColor];

    label.numberOfLines = 2;

    label.backgroundColor = [UIColor clearColor];

    label.autoresizingMask = UIViewAutoresizingFlexibleLeftMargin | UIViewAutoresizingFlexibleRightMargin | UIViewAutoresizingFlexibleWidth;

    label.text = NSLocalizedString(@"Setting a Style might force a particular Scheme or Weapon configuration.",@"");

    [footer addSubview:label];

    return footer;

}

#pragma mark -

#pragma mark Table view delegate

- (void)tableView:(UITableView *)aTableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath {

    NSIndexPath *lastIndexPath;

    NSInteger index = self.topControl.selectedSegmentIndex;

    if (index == 0)

        lastIndexPath = self.lastIndexPath_sc;

    else if (index == 1)

        lastIndexPath = self.lastIndexPath_we;

    else

        lastIndexPath = self.lastIndexPath_lu;

    NSInteger newRow = [indexPath row];

    NSInteger oldRow = (lastIndexPath != nil) ? [lastIndexPath row] : -1;

    if (newRow != oldRow) {

        //TODO: this code works only for a single section table

        UITableViewCell *newCell = [aTableView cellForRowAtIndexPath:indexPath];

        UIImageView *checkbox = [[UIImageView alloc] initWithImage:[UIImage imageWithContentsOfFile:@"checkbox.png"]];

        newCell.accessoryView = checkbox;

        UITableViewCell *oldCell = [aTableView cellForRowAtIndexPath:lastIndexPath];

        oldCell.accessoryView = nil;

        if (index == 0) {

            self.lastIndexPath_sc = indexPath;

            self.selectedScheme = [self.listOfSchemes objectAtIndex:newRow];

            // also set weaponset when selecting scheme, if set

            NSUserDefaults *settings = [NSUserDefaults standardUserDefaults];

            if ([[settings objectForKey:@"sync_ws"] boolValue]) {

                for (NSString *str in self.listOfWeapons) {

                    if ([str isEqualToString:self.selectedScheme]) {

                        NSInteger row = [self.listOfSchemes indexOfObject:str];

                        self.selectedWeapon = str;

                        self.lastIndexPath_we = [NSIndexPath indexPathForRow:row inSection:1];

                        break;

                    }

                }

            }

        } else if (index == 1) {

            self.lastIndexPath_we = indexPath;

            self.selectedWeapon = [self.listOfWeapons objectAtIndex:newRow];

        } else {

            self.lastIndexPath_lu = indexPath;

            if (newRow == 0)

            {

                self.selectedScript = nil;

                self.scriptCommand = nil;

                self.selectedScheme = @"Default.plist";

                [self.topControl setEnabled:YES forSegmentAtIndex:0];

                self.selectedWeapon = @"Default.plist";

                [self.topControl setEnabled:YES forSegmentAtIndex:1];

            }

            else

            {

                newRow--;

                self.selectedScript = [self.listOfScripts objectAtIndex:newRow];

                // some styles disable or force the choice of a particular scheme/weaponset

                NSString *path = [[NSString alloc] initWithFormat:@"%@/%@.cfg",SCRIPTS_DIRECTORY(),[self.selectedScript stringByDeletingPathExtension]];

                NSString *configFile = [[NSString alloc] initWithContentsOfFile:path];

                NSArray *scriptOptions = [configFile componentsSeparatedByString:@"\n"];

                self.scriptCommand = [NSString stringWithFormat:@"escript Scripts/Multiplayer/%@",self.selectedScript];

                NSString *scheme = [scriptOptions objectAtIndex:0];

                if ([scheme isEqualToString:@"locked"])

                {

                    self.selectedScheme = @"Default.plist";

                    [self.topControl setEnabled:NO forSegmentAtIndex:0];

                }

                else

                {

                    if (scheme && ![scheme isEqualToString:@"*"])

                    {

                        NSString *correctScheme = [scheme stringByReplacingOccurrencesOfString:@"_" withString:@" "];

                        self.selectedScheme = [NSString stringWithFormat:@"%@.plist", correctScheme];

                    }

                    [self.topControl setEnabled:YES forSegmentAtIndex:0];

                }

                NSString *weapon = [scriptOptions objectAtIndex:1];

                if ([weapon isEqualToString:@"locked"])

                {

                    self.selectedWeapon = @"Default.plist";

                    [self.topControl setEnabled:NO forSegmentAtIndex:1];

                }

                else

                {

                    if (weapon && ![weapon isEqualToString:@"*"])

                    {

                        NSString *correctWeapon = [weapon stringByReplacingOccurrencesOfString:@"_" withString:@" "];

                        self.selectedWeapon = [NSString stringWithFormat:@"%@.plist", correctWeapon];

                    }

                    [self.topControl setEnabled:YES forSegmentAtIndex:1];

                }

            }

        }

        [aTableView selectRowAtIndexPath:indexPath animated:YES scrollPosition:UITableViewScrollPositionNone];

    }

    [aTableView deselectRowAtIndexPath:indexPath animated:YES];

}

#pragma mark -

#pragma mark called by an NSNotification to empty or fill the sections completely

- (void)fillSections {

    if (self.sectionsHidden == YES) {

        self.sectionsHidden = NO;

        NSIndexSet *sections = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, 1)];

        UITableView *aTableView = (UITableView *)[self.view viewWithTag:TABLE_TAG];

        [aTableView insertSections:sections withRowAnimation:UITableViewRowAnimationFade];

        aTableView.scrollEnabled = YES;

        [[self.view viewWithTag:LABEL_TAG] removeFromSuperview];

    }

}

- (void)emptySections {

    if (self.sectionsHidden == NO) {

        self.sectionsHidden = YES;

        NSIndexSet *sections = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0, 1)];

        UITableView *aTableView = (UITableView *)[self.view viewWithTag:TABLE_TAG];

        [aTableView deleteSections:sections withRowAnimation:UITableViewRowAnimationFade];

        aTableView.scrollEnabled = NO;

        CGRect frame = CGRectMake(0, 0, self.view.frame.size.width * 80/100, 60);

        UILabel *theLabel = [[UILabel alloc] initWithFrame:frame

                                                  andTitle:NSLocalizedString(@"Missions don't need further configuration",@"")];

        theLabel.center = CGPointMake(self.view.frame.size.width/2, self.view.frame.size.height/2);

        theLabel.numberOfLines = 2;

        theLabel.tag = LABEL_TAG;

        theLabel.autoresizingMask = UIViewAutoresizingFlexibleWidth |

                                    UIViewAutoresizingFlexibleTopMargin |

                                    UIViewAutoresizingFlexibleBottomMargin;

        [self.view addSubview:theLabel];

    }

}

#pragma mark -

#pragma mark Memory management

- (void)didReceiveMemoryWarning {

    self.listOfSchemes = nil;

    self.listOfWeapons = nil;

    self.listOfScripts = nil;

    MSG_MEMCLEAN();

    [super didReceiveMemoryWarning];

}

- (void)dealloc

{

    [[NSNotificationCenter defaultCenter] removeObserver:self];

}

@end