//
//  PBGitTree.m
//  GitTest
//
//  Created by Pieter de Bie on 15-06-08.
//  Copyright 2008 __MyCompanyName__. All rights reserved.
//
 
#import "PBGitTree.h"
#import "PBGitCommit.h"
#import "NSFileHandleExt.h"
#import "PBEasyPipe.h"
#import "PBEasyFS.h"
 
@implementation PBGitTree
 
@synthesize sha, path, repository, leaf, parent;
 
+ (PBGitTree*) rootForCommit:(id) commit
{
  PBGitCommit* c = commit;
  PBGitTree* tree = [[self alloc] init];
  tree.parent = nil;
  tree.leaf = NO;
  tree.sha = c.sha;
  tree.repository = c.repository;
  tree.path = @"";
  return tree;
}
 
+ (PBGitTree*) treeForTree: (PBGitTree*) prev andPath: (NSString*) path;
{
  PBGitTree* tree = [[self alloc] init];
  tree.parent = prev;
  tree.sha = prev.sha;
  tree.repository = prev.repository;
  tree.path = path;
  return tree;
}
 
- init
{
  children = nil;
  localFileName = nil;
  leaf = YES;
  return self;
}
 
- (NSString*) refSpec
{
  return [NSString stringWithFormat:@"%@:%@", self.sha, self.fullPath];
}
 
- (BOOL) isLocallyCached
{
  NSFileManager* fs = [NSFileManager defaultManager];
  if (localFileName && [fs fileExistsAtPath:localFileName])
  {
    NSDate* mtime = [[fs attributesOfItemAtPath:localFileName error: nil] objectForKey:NSFileModificationDate];
    if ([mtime compare:localMtime] == 0)
      return YES;
  }
  return NO;
}
 
- (NSString*) contents
{
  if (!leaf)
    return [NSString stringWithFormat:@"This is a tree with path %@", self];
 
  NSData* data = nil;
  
  if ([self isLocallyCached])
    data = [NSData dataWithContentsOfFile: localFileName];
  else {
    NSFileHandle* handle = [repository handleForArguments:[NSArray arrayWithObjects:@"show", [self refSpec], nil]];
    data = [handle readDataToEndOfFile];
  }
  
  return [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
}
 
- (void) saveToFolder: (NSString *) dir
{
  NSString* newName = [dir stringByAppendingPathComponent:path];
 
  if (leaf) {
    NSFileHandle* handle = [repository handleForArguments:[NSArray arrayWithObjects:@"show", [self refSpec], nil]];
    NSData* data = [handle readDataToEndOfFile];
    [data writeToFile:newName atomically:YES];
  } else { // Directory
    [[NSFileManager defaultManager] createDirectoryAtPath:newName attributes:nil];
    for (PBGitTree* child in children)
      [child saveToFolder: newName];
  }
}
 
- (NSString*) tmpDirWithContents
{
  if (leaf)
    return nil;
 
  if (!localFileName)
    localFileName = [PBEasyFS tmpDirWithPrefix: path];
 
  NSLog(@"Exporting children..");
 
  for (PBGitTree* child in [self children]) {
    NSLog(@"Telling %@ to save to %@!", [child fullPath], localFileName);
    [child saveToFolder: localFileName];
  }
  
  return localFileName;
}
 
  
 
- (NSString*) tmpFileNameForContents
{
  if (!leaf)
    return [self tmpDirWithContents];
  
  if ([self isLocallyCached])
    return localFileName;
  
  if (!localFileName)
    localFileName = [[PBEasyFS tmpDirWithPrefix: sha] stringByAppendingPathComponent:path];
  
  NSFileHandle* handle = [repository handleForArguments:[NSArray arrayWithObjects:@"show", [self refSpec], nil]];
  NSData* data = [handle readDataToEndOfFile];
  [data writeToFile:localFileName atomically:YES];
  
  NSFileManager* fs = [NSFileManager defaultManager];
  localMtime = [[fs attributesOfItemAtPath:localFileName error: nil] objectForKey:NSFileModificationDate];
 
  return localFileName;
}
 
- (NSArray*) children
{
  if (children != nil)
    return children;
  
  NSString* ref = [self refSpec];
  NSLog(@"Starting get for %@", ref);
 
  NSFileHandle* handle = [repository handleForArguments:[NSArray arrayWithObjects:@"show", ref, nil]];
  [handle readLine];
  [handle readLine];
  
  NSMutableArray* c = [NSMutableArray array];
  
  NSString* p = [handle readLine];
  while (p.length > 0) {
    BOOL isLeaf = ([p characterAtIndex:p.length - 1] != '/');
    if (!isLeaf)
      p = [p substringToIndex:p.length -1];
 
    PBGitTree* child = [PBGitTree treeForTree:self andPath:p];
    child.leaf = isLeaf;
    [c addObject: child];
    
    p = [handle readLine];
  }
  children = c;
  return c;
}
 
- (NSString*) fullPath
{
  if (!parent)
    return @"";
  
  if ([parent.fullPath isEqualToString:@""])
    return self.path;
  
  return [parent.fullPath stringByAppendingPathComponent: self.path];
}
 
- (void) finalize
{
  if (localFileName)
    [[NSFileManager defaultManager] removeFileAtPath:localFileName handler:nil];
  [super finalize];
}
@end