//
//  MainViewController.m
//  Lexikon
//
//  Created by Caleb Jaffa
//  Copyright 2008-2009 Caleb Jaffa, MIT licensed
//
 
#import "MainViewController.h"
#import "Word.h"
#import "LexikonAppDelegate.h"
#import "DetailViewController.h"
#import "AboutViewController.h"
#import "SearchSuggestionsController.h"
 
@implementation MainViewController
 
@synthesize tableView, indexLetters, mySearchBar, suggestionsController;
 
- (void)awakeFromNib {
  // we need to get at properties of our application delegate
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  
  // make it so the index is always displayed
  tableView.sectionIndexMinimumDisplayRowCount = 1;
  [self changeIndexLetters: appDelegate.swedishToEnglish];
  
  searching = NO; // keep track if we are in the middle of a search or not
}
 
- (void)dealloc {
  [mySearchBar release];
  [detailViewController release];
  [indexLetters release];
  [tableView release];
  [suggestionsController release];
  [super dealloc];
}
 
- (void)didReceiveMemoryWarning {
  [super didReceiveMemoryWarning]; // Releases the view if it doesn't have a superview
  // Release anything that's not essential, such as cached data
}
 
- (void)viewWillAppear:(BOOL)animated {
  if (searching) {
    [mySearchBar becomeFirstResponder];
  }
  
  [self shiftOverlaysOff:NO];
  
  // update the app delegate that we are no longer viewing a specific word
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  appDelegate.currentWord = nil;
  
  [super viewWillAppear:animated];
}
 
- (void)viewWillDisappear:(BOOL)animated {
  [super viewWillDisappear:animated];
  [self shiftOverlaysOff:YES];
}
 
- (void)changeIndexLetters:(BOOL) swedish {
  // first time through set it up
  if(self.indexLetters == nil) {
    self.indexLetters = [NSMutableArray arrayWithObjects: @"{search}", @"A", @"B", @"C", @"D", @"E", @"F", @"G", @"H", @"I", @"J", @"K", @"L", @"M", @"N", @"O", @"P", @"Q", @"R", @"S", @"T", @"U", @"V", @"W", @"X", @"Y", @"Z", nil];
  }
  
  if(swedish) {
    [self.indexLetters addObject: @"Å"];
    [self.indexLetters addObject: @"Ä"];
    [self.indexLetters addObject: @"Ö"];
    languageSwitcherButton.title = NSLocalizedString(@"Swe to Eng", @"Language toggle button Swedish to English");
  }
  else {
    [self.indexLetters removeObjectsInArray: [NSArray arrayWithObjects: @"Å", @"Ä", @"Ö", nil]];
    languageSwitcherButton.title = NSLocalizedString(@"Eng to Swe", @"Lanuage toggle button English to Swedish");
  }
}
 
- (IBAction)switchLanguage:(id)sender {
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  
  BOOL swedish = [appDelegate toggleSwedishToEnglish];
  
  [self changeIndexLetters: swedish];
  
  // if we are searching we might need to update the suggestions/filter list
  if (searching) {
    [self searchBar:mySearchBar textDidChange:mySearchBar.text];
  }
  
  [self.tableView reloadData];    
}
 
- (IBAction)showAbout:(id)sender {
  AboutViewController *aboutViewController = [[AboutViewController alloc] initWithNibName:@"AboutView" bundle:nil];
  [[self navigationController] pushViewController:aboutViewController animated: YES];
  self.navigationController.navigationBarHidden = NO;
  [aboutViewController release];
}
 
 
#pragma mark TableView Datasource
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
  if(section == 0) {
    return 1; // return 1 for section 0 all other will be dynamic 
  }
  else {
    // we need to get at properties in our app delegate
    LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
    NSString *sectionLetter = [self.indexLetters objectAtIndex:section];
    NSMutableArray *wordsForSection = [appDelegate.currentWords objectForKey:sectionLetter];
    
    return [wordsForSection count];
  }
}
 
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
  return [self.indexLetters count];
}
 
- (NSArray *)sectionIndexTitlesForTableView:(UITableView *)tableView {
  return self.indexLetters;
}
 
- (NSString *)tableView:(UITableView *)tableView titleForHeaderInSection:(NSInteger)section {
  // return nil for the searchBar (1st section)
  if(section == 0) {
    return nil;
  }
  else {
    //    NSLog(@"section # %d", section);
    LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
    
    // if there are no words for a letter return a blank string so the UI isn't cluttered
    NSString *sectionLetter = [self.indexLetters objectAtIndex:section];
    NSMutableArray *wordsForSection = [appDelegate.currentWords objectForKey:sectionLetter];
    if(wordsForSection == nil || [wordsForSection count] == 0) {
      return @"";
    }
    else {
      return sectionLetter;
    }
  }
}
 
 
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {  
  // searchBar cell gets its own identifier
  NSString *identifier = (indexPath.section == 0) ? @"search" : @"cell";
  
  UITableViewCell *cell = [self.tableView dequeueReusableCellWithIdentifier:identifier];
  if(cell == nil) {
    cell = [[[UITableViewCell alloc] initWithFrame:CGRectZero reuseIdentifier:identifier] autorelease];
  }
  
  //NSLog(@"Row: %i,%i ID: %@", indexPath.section, indexPath.row, identifier);
  if(indexPath.section == 0) {
    // we don't need to redo the cell contents if they already exist
    if (self.mySearchBar == nil) {
      // TODO: navigation bar or image to fill in the gap so our seach bar can be 290 pixels and not over the index
      UINavigationBar *myNavigationBar = [[UINavigationBar alloc] initWithFrame:CGRectZero];
      [myNavigationBar setTintColor: [UIColor colorWithRed:0.769 green:0.80 blue:0.824 alpha:1.0]];
      [myNavigationBar sizeToFit];
      [cell.contentView addSubview:myNavigationBar];
      [myNavigationBar release];
      
      // Now add our search bar
      UISearchBar *aSearchBar = [[UISearchBar alloc] initWithFrame:CGRectZero];
      [aSearchBar setTintColor: [UIColor colorWithRed:0.769 green:0.80 blue:0.824 alpha:1.0]];
      [aSearchBar sizeToFit];
      aSearchBar.autocapitalizationType = UITextAutocapitalizationTypeNone;
      aSearchBar.placeholder = NSLocalizedString(@"Search", @"Search placeholder");
      aSearchBar.delegate = self;
      
      aSearchBar.frame = CGRectMake(0, 0, 290, 44);
      [cell.contentView addSubview:aSearchBar];
      self.mySearchBar = [aSearchBar retain];
      [aSearchBar release];
    }
  }
  else {
    LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
    NSString *sectionLetter = [self.indexLetters objectAtIndex:indexPath.section];
    NSMutableArray *wordsForSection = [appDelegate.currentWords objectForKey:sectionLetter];
    
    cell.text = [[wordsForSection objectAtIndex:indexPath.row] word];
  }
  
  return cell;
}
 
- (void)tableView:(UITableView *)aTableView commitEditingStyle:(UITableViewCellEditingStyle)editingStyle forRowAtIndexPath:(NSIndexPath *)indexPath {
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  
  if (editingStyle == UITableViewCellEditingStyleDelete) {    
    // first remove from our word list and the database
    [appDelegate removeWordAtSectionLetter:[self.indexLetters objectAtIndex:indexPath.section] index:indexPath.row];
    // remove from the tableview
    [tableView deleteRowsAtIndexPaths:[NSArray arrayWithObject:indexPath] withRowAnimation:UITableViewRowAnimationFade];
  }
}
 
- (BOOL)tableView:(UITableView *)tableView canEditRowAtIndexPath:(NSIndexPath *)indexPath {
  if(indexPath.section == 0) {
    return NO;
  }
  else {
    return YES;
  }
}
#pragma mark TableView Delegate
 
- (void)viewWord:(Word *) word {
  if (detailViewController == nil) {
    detailViewController = [[DetailViewController alloc] init];
  }
  
  // update the app delegate with what word we are viewing
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  appDelegate.currentWord = word.word;  
  
  detailViewController.word = word.word;
  detailViewController.html = [[NSString stringWithContentsOfFile:[[[NSBundle mainBundle] resourcePath] 
                                                                   stringByAppendingPathComponent:@"translationTemplate.html"]]
                               stringByReplacingOccurrencesOfString:@"{yield}" withString:word.translation];
  
  [self.navigationController pushViewController:detailViewController animated:YES];
}
 
- (NSIndexPath *)tableView:(UITableView *)tableView willSelectRowAtIndexPath:(NSIndexPath *)indexPath {
  if (indexPath.section == 0) {
    return nil;
  }
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  NSMutableArray *wordsForSection = [appDelegate.currentWords objectForKey:[self.indexLetters objectAtIndex:indexPath.section]];
  Word *myWord = [wordsForSection objectAtIndex:indexPath.row];
  
  [self viewWord: myWord];
  
  return nil;
}
 
- (void)tableView:(UITableView *)tableView willBeginEditingRowAtIndexPath:(NSIndexPath *)indexPath {
  [self hideIndex:YES];
}
 
- (void)tableView:(UITableView *)tableView didEndEditingRowAtIndexPath:(NSIndexPath *)indexPath {
  [self hideIndex:NO];
}
 
- (UITableViewCellEditingStyle)tableView:(UITableView *)tableView editingStyleForRowAtIndexPath:(NSIndexPath *)indexPath {
  if(indexPath.section == 0) {
    return UITableViewCellEditingStyleNone;
  }
  else {
    return UITableViewCellEditingStyleDelete;
  }
}
 
#pragma mark UISearchBarDelegate
 
- (void)searchBarTextDidBeginEditing:(UISearchBar *)searchBar {
  if (! searching) {
    // if we pop back to this view we might already be in editing mode, only activate when needed
    searching = YES;
    [self hideIndex:YES];
  }
}
 
- (void)shiftOverlaysOff:(BOOL) shiftOff {
  // shift the cancelSearchTableCover and suggestionsController.view off by sliding off the screen but by
  // doing this manually we have more control than making these a subview of the MainViewController's tableView
  int newX = (shiftOff) ? 0-tableView.window.frame.size.width : 0;
 
  [UIView beginAnimations:nil context:nil];
  [UIView setAnimationDuration:0.35];
  
  cancelSearchTableCover.frame = CGRectMake(newX,
                                            cancelSearchTableCover.frame.origin.y,
                                            cancelSearchTableCover.frame.size.width,
                                            cancelSearchTableCover.frame.size.height);
  suggestionsController.view.frame = CGRectMake(newX,
                                                suggestionsController.view.frame.origin.y,
                                                suggestionsController.view.frame.size.width,
                                                suggestionsController.view.frame.size.height);
  
  [UIView commitAnimations];
}
 
- (void)showOverlays:(BOOL) show {
  // this handles showing and hiding the different overlays
  // show is NO hide cancelCoverButton and suggestions
  // show is YES show cancelCoverButton is mySearchBar.text.length == 0
  // or show suggestionsController.view
  tableView.scrollEnabled = ! show; // make sure the user can't scroll the table by catching an edge of the search field
 
  [UIView beginAnimations:nil context:nil];
  [UIView setAnimationDuration:0.2];
  
  if (! show) {
    cancelSearchTableCover.alpha = 0.0f;
    suggestionsController.view.hidden = YES;
  }
  else {
    if (mySearchBar.text.length > 0) {
      suggestionsController.view.hidden = NO;
    }
    else {
      suggestionsController.view.hidden = YES;
      cancelSearchTableCover.alpha = 0.8f;
    }
  }
 
  [UIView commitAnimations];
}
 
- (void)searchBar:(UISearchBar *)searchBar textDidChange:(NSString *)searchText {
  if (suggestionsController == nil) {
    suggestionsController = [[SearchSuggestionsController alloc] initWithNibName:@"SearchSuggestions" bundle:nil];
    //suggestionsController.view.alpha = 1.0f;
    suggestionsController.main = self;
    suggestionsController.view.frame = CGRectMake(suggestionsController.view.frame.origin.x,
                                             suggestionsController.view.frame.origin.y+88,
                                             suggestionsController.view.frame.size.width,
                                             suggestionsController.view.frame.size.height);
    [self.navigationController.view addSubview:suggestionsController.view];
  }
  
  [self showOverlays:YES];
  
  if (searchBar.text.length > 0) {    
    LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
    
    // find the filtered list of words that match and put them into the search suggestion table
    NSMutableArray *suggestions = [[NSMutableArray alloc] init];
    
    for (NSString *letter in self.indexLetters) {
      for (Word *aWord in [appDelegate.currentWords objectForKey:letter]) {
        if([aWord.word rangeOfString:searchBar.text options:(NSDiacriticInsensitiveSearch | NSCaseInsensitiveSearch)].location != NSNotFound) {
          [suggestions addObject:aWord];
        }
      }
    }
 
    suggestionsController.suggestions = suggestions;
    [suggestions release];
    [suggestionsController.tableView reloadData];
  }
}
 
 
- (void)cancelSearching {
  [self hideIndex:NO];
  searching = NO;
  self.mySearchBar.text = @"";
  [self.mySearchBar resignFirstResponder];
  [self showOverlays:NO];
}
 
- (void) toggleButtonItem {
  UIBarButtonItem *buttonItem;
  // change the About button to a cancel button
  if (self.navigationItem.rightBarButtonItem.action == @selector(showAbout:)) {
    tableView.sectionIndexMinimumDisplayRowCount = NSIntegerMax;
    buttonItem = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemCancel target:self action:@selector(cancelSearching)];
  }
  else {
    tableView.sectionIndexMinimumDisplayRowCount = 1;
    buttonItem = [[UIBarButtonItem alloc] initWithTitle:NSLocalizedString(@"About", @"About button") style:UIBarButtonItemStylePlain target:self action:@selector(showAbout:)];
  }
  [self.navigationItem setRightBarButtonItem: buttonItem animated:YES];
  [buttonItem release];
}
 
- (void)hideIndex:(BOOL) hide {
  CGFloat factor = (hide) ? 30.0 : -30.0;
  
  if (cancelSearchTableCover == nil) {
    cancelSearchTableCover = [UIButton buttonWithType:UIButtonTypeCustom];
    [cancelSearchTableCover addTarget:self action:@selector(cancelSearching) forControlEvents:UIControlEventTouchDown];
    cancelSearchTableCover.backgroundColor = [UIColor blackColor];
    cancelSearchTableCover.alpha = 0.0f;
    cancelSearchTableCover.frame = CGRectMake(0, 88, 320, 320);
    [self.navigationController.view addSubview:cancelSearchTableCover];
  }
  
  if (searching) {
    [self toggleButtonItem];
    [self showOverlays:hide];
  }
  
  // setup the animation
  [UIView beginAnimations:nil context:nil];
  [UIView setAnimationDuration:0.2];
 
  mySearchBar.frame = CGRectMake(mySearchBar.frame.origin.x, 
                                 mySearchBar.frame.origin.y, 
                                 mySearchBar.frame.size.width + factor, 
                                 mySearchBar.frame.size.height); 
  
  // change the UISearchBar's size
  for ( UIView *view in mySearchBar.subviews) {
    if ([view isKindOfClass:NSClassFromString(@"UISearchBarTextField")]) {
      view.frame = CGRectMake(view.frame.origin.x, 
                              view.frame.origin.y, 
                              view.frame.size.width + factor, 
                              view.frame.size.height);
    }
  }
  
  // move the index out of the way
//  for ( UIView *view in tableView.subviews ) {
//    if ([view isKindOfClass:NSClassFromString(@"UITableViewIndex")]) {
//      view.center = CGPointMake(view.center.x + factor, view.center.y);
//    }
//  }
  [tableView setIndexHidden:hide animated:NO];
  [UIView commitAnimations];
}
 
- (void)searchBarSearchButtonClicked:(UISearchBar *)searchBar {  
  LexikonAppDelegate *appDelegate = (LexikonAppDelegate *)[[UIApplication sharedApplication] delegate];
  [UIApplication sharedApplication].networkActivityIndicatorVisible = YES;
  
  NSString *searchWord = [searchBar.text capitalizedString];
  NSString *translationString;
  NSString *translation;
  NSURL *translationURL;
  NSRange start, end;
  int encoding;
 
  [self cancelSearching];
 
  if(appDelegate.swedishToEnglish) {
    translationString = [[NSString stringWithFormat:@"http://lexin.nada.kth.se/Lexin/?dict=sve-eng&lang=source&word=%@", searchWord] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding];
    encoding = NSUTF8StringEncoding;
  }
  else {
    translationString = [[NSString stringWithFormat:@"http://lexin.nada.kth.se/cgi-bin/sve-eng?:%@", searchWord] stringByAddingPercentEscapesUsingEncoding: NSUTF8StringEncoding];
    encoding = NSISOLatin1StringEncoding;
  }
 
  // clean up the special chars  
  translationURL = [NSURL URLWithString:translationString];
  
  // get the html contents in a string
  NSError *myError;
  translation = [NSString stringWithContentsOfURL:translationURL encoding:encoding error:&myError];
  
  if(translation != nil) {
    start = [translation rangeOfString:@"<DL>"];
    end = [translation rangeOfString:@"</DL>" options:NSBackwardsSearch];
    if(! (start.location == NSNotFound && end.location == NSNotFound)) {
      NSRange myRange;
      myRange.location = start.location;
      myRange.length = end.location - start.location;
      translation = [translation substringWithRange: myRange];
      
      Word *newWord = [[Word alloc] init];
      newWord.word = searchWord;
      newWord.lang = (appDelegate.swedishToEnglish) ? SWE_LANGUAGE : ENG_LANGUAGE;
      newWord.translation = translation;
      
      [appDelegate addWordToDictionary:appDelegate.currentWords word:newWord andDatabase:YES];
      [self viewWord:newWord];
      [tableView reloadData];
    }
    else {
      // display error message that the word was not found
      [self searchFailed:[NSString stringWithFormat:NSLocalizedString(@"%@ not found", @"word not found"), searchWord]];
    }
  }
  else {
    // display error message alerting the user that we were not able to contact the Lexin website
    [self searchFailed:NSLocalizedString(@"Unable to reach the Lexin website", @"Error message when unable to contact Lexin")];
    //[NSString stringWithFormat:@"%@ %@", [myError localizedDescription], [myError localizedFailureReason]]];
  }  
  
  // hide the activity indicator
  [UIApplication sharedApplication].networkActivityIndicatorVisible = NO;
  //[searchWord release];
  searchWord = nil;
}
 
- (void)searchFailed:(NSString *)message {
  UIAlertView *alert = [[UIAlertView alloc] initWithTitle:NSLocalizedString(@"Search Error", @"Search Error alert title")
                                                  message:message
                                                 delegate:nil 
                                        cancelButtonTitle:NSLocalizedString(@"OK", @"Alert OK button")
                                        otherButtonTitles: nil];
	[alert show];
	[alert release];  
}
 
@end