IJInventoryWindowController.m 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360
  1. //
  2. // IJInventoryWindowController.m
  3. // InsideJob
  4. //
  5. // Created by Adam Preble on 10/7/10.
  6. // Copyright 2010 Adam Preble. All rights reserved.
  7. //
  8. #import "IJInventoryWindowController.h"
  9. #import "IJMinecraftLevel.h"
  10. #import "IJInventoryItem.h"
  11. #import "IJItemPickerWindowController.h"
  12. @implementation IJInventoryWindowController
  13. @synthesize outlineView;
  14. @synthesize worldSelectionControl;
  15. @synthesize statusTextField;
  16. - (void)awakeFromNib
  17. {
  18. armorItem = [NSMutableArray array];
  19. quickItem = [NSMutableArray array];
  20. inventoryItem = [NSMutableArray array];
  21. rootItems = [[NSArray alloc] initWithObjects:armorItem, quickItem, inventoryItem, nil];
  22. statusTextField.stringValue = @"";
  23. }
  24. - (void)dealloc
  25. {
  26. [inventory release];
  27. [rootItems release];
  28. [level release];
  29. [super dealloc];
  30. }
  31. #pragma mark -
  32. #pragma mark World Selection
  33. - (void)loadWorldAtIndex:(int)worldIndex
  34. {
  35. [armorItem removeAllObjects];
  36. [quickItem removeAllObjects];
  37. [inventoryItem removeAllObjects];
  38. // Reload data here because we have just invalidated all of the items used in the outline view.
  39. [outlineView reloadData];
  40. [self willChangeValueForKey:@"worldTime"];
  41. [level release];
  42. level = nil;
  43. [inventory release];
  44. inventory = nil;
  45. [self didChangeValueForKey:@"worldTime"];
  46. if (![IJMinecraftLevel worldExistsAtIndex:worldIndex])
  47. {
  48. NSBeginCriticalAlertSheet(@"No world exists in that slot.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Please create a new single player world in this slot using Minecraft and try again.");
  49. return;
  50. }
  51. sessionLockValue = [IJMinecraftLevel writeToSessionLockAtIndex:worldIndex];
  52. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  53. {
  54. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable obtain the session lock.");
  55. return;
  56. }
  57. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  58. NSData *fileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:levelPath]];
  59. if (!fileData)
  60. {
  61. // Error loading
  62. [outlineView reloadData];
  63. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", levelPath);
  64. return;
  65. }
  66. [self willChangeValueForKey:@"worldTime"];
  67. level = [[IJMinecraftLevel nbtContainerWithData:fileData] retain];
  68. inventory = [[level inventory] retain];
  69. [self didChangeValueForKey:@"worldTime"];
  70. // Add placeholder inventory items:
  71. for (int i = 0; i < IJInventorySlotQuickLast + 1 - IJInventorySlotQuickFirst; i++)
  72. [quickItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotQuickFirst + i]];
  73. for (int i = 0; i < IJInventorySlotNormalLast + 1 - IJInventorySlotNormalFirst; i++)
  74. [inventoryItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotNormalFirst + i]];
  75. for (int i = 0; i < IJInventorySlotArmorLast + 1 - IJInventorySlotArmorFirst; i++)
  76. [armorItem addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotArmorFirst + i]];
  77. // Overwrite the placeholders with actual inventory:
  78. for (IJInventoryItem *item in inventory)
  79. {
  80. if (IJInventorySlotQuickFirst <= item.slot && item.slot <= IJInventorySlotQuickLast)
  81. {
  82. [quickItem replaceObjectAtIndex:item.slot - IJInventorySlotQuickFirst withObject:item];
  83. }
  84. else if (IJInventorySlotNormalFirst <= item.slot && item.slot <= IJInventorySlotNormalLast)
  85. {
  86. [inventoryItem replaceObjectAtIndex:item.slot - IJInventorySlotNormalFirst withObject:item];
  87. }
  88. else if (IJInventorySlotArmorFirst <= item.slot && item.slot <= IJInventorySlotArmorLast)
  89. {
  90. [armorItem replaceObjectAtIndex:item.slot - IJInventorySlotArmorFirst withObject:item];
  91. }
  92. }
  93. [outlineView reloadData];
  94. [outlineView expandItem:nil expandChildren:YES];
  95. dirty = NO;
  96. statusTextField.stringValue = @"";
  97. }
  98. - (void)saveToWorldAtIndex:(int)worldIndex
  99. {
  100. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  101. {
  102. NSBeginCriticalAlertSheet(@"Another application has modified this world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"The session lock was changed by another application.");
  103. return;
  104. }
  105. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  106. NSMutableArray *newInventory = [NSMutableArray array];
  107. for (NSArray *items in rootItems)
  108. {
  109. for (IJInventoryItem *item in items)
  110. {
  111. if (item.count > 0 && item.itemId > 0)
  112. [newInventory addObject:item];
  113. }
  114. }
  115. [level setInventory:newInventory];
  116. NSString *backupPath = [levelPath stringByAppendingPathExtension:@"insidejobbackup"];
  117. BOOL success;
  118. NSError *error = nil;
  119. success = [[NSFileManager defaultManager] removeItemAtPath:backupPath error:&error];
  120. success = [[NSFileManager defaultManager] copyItemAtPath:levelPath
  121. toPath:backupPath
  122. error:&error];
  123. if (!success)
  124. {
  125. NSBeginCriticalAlertSheet(@"An error occurred while saving.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable to create a backup of the existing level file.");
  126. return;
  127. }
  128. [[level writeData] writeToURL:[NSURL fileURLWithPath:levelPath] atomically:NO];
  129. dirty = NO;
  130. statusTextField.stringValue = @"Saved.";
  131. }
  132. - (void)markDirty
  133. {
  134. dirty = YES;
  135. statusTextField.stringValue = @"World has unsaved changes.";
  136. }
  137. #pragma mark -
  138. #pragma mark Actions
  139. - (IBAction)worldSelectionChanged:(id)sender
  140. {
  141. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  142. [self loadWorldAtIndex:worldIndex];
  143. }
  144. - (void)saveDocument:(id)sender
  145. {
  146. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  147. [self saveToWorldAtIndex:worldIndex];
  148. }
  149. - (void)delete:(id)sender
  150. {
  151. IJInventoryItem *item = [outlineView itemAtRow:[outlineView selectedRow]];
  152. item.count = 0;
  153. item.itemId = 0;
  154. item.damage = 0;
  155. [self markDirty];
  156. [outlineView reloadItem:item];
  157. }
  158. - (BOOL)validateUserInterfaceItem:(id <NSValidatedUserInterfaceItem>)anItem
  159. {
  160. if (anItem.action == @selector(delete:))
  161. {
  162. return [outlineView selectedRow] != -1 && ![rootItems containsObject:[outlineView itemAtRow:[outlineView selectedRow]]];
  163. }
  164. return YES;
  165. }
  166. - (NSNumber *)worldTime
  167. {
  168. return [level worldTimeContainer].numberValue;
  169. }
  170. - (void)setWorldTime:(NSNumber *)number
  171. {
  172. [self willChangeValueForKey:@"worldTime"];
  173. [level worldTimeContainer].numberValue = number;
  174. [self didChangeValueForKey:@"worldTime"];
  175. [self markDirty];
  176. }
  177. #pragma mark -
  178. #pragma mark Inventory Outline View
  179. - (id)outlineView:(NSOutlineView *)theOutlineView child:(NSInteger)index ofItem:(id)item
  180. {
  181. if (item == nil)
  182. {
  183. return [rootItems objectAtIndex:index];
  184. }
  185. else
  186. {
  187. return [item objectAtIndex:index];
  188. }
  189. }
  190. - (BOOL)outlineView:(NSOutlineView *)theOutlineView isItemExpandable:(id)item
  191. {
  192. return item == nil || [rootItems containsObject:item];
  193. }
  194. - (NSInteger)outlineView:(NSOutlineView *)theOutlineView numberOfChildrenOfItem:(id)item
  195. {
  196. if (item == nil)
  197. {
  198. return 3;
  199. }
  200. else if ([rootItems containsObject:item])
  201. {
  202. return [(NSArray *)item count];
  203. }
  204. else
  205. {
  206. return 0;
  207. }
  208. }
  209. - (id)outlineView:(NSOutlineView *)theOutlineView objectValueForTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
  210. {
  211. if ([rootItems containsObject:item])
  212. {
  213. if ([tableColumn.identifier isEqual:@"slot"])
  214. {
  215. if (item == armorItem)
  216. return @"Armor";
  217. else if (item == quickItem)
  218. return @"Quick Inventory";
  219. else if (item == inventoryItem)
  220. return @"Inventory";
  221. }
  222. else
  223. {
  224. return nil;
  225. }
  226. }
  227. IJInventoryItem *invItem = item;
  228. if ([tableColumn.identifier isEqual:@"slot"])
  229. {
  230. return [NSString stringWithFormat:@"%d", invItem.slot];
  231. }
  232. else if ([tableColumn.identifier isEqual:@"id"])
  233. {
  234. if (invItem.itemId)
  235. return [NSNumber numberWithShort:invItem.itemId];
  236. else
  237. return nil;
  238. }
  239. else if ([tableColumn.identifier isEqual:@"item"])
  240. {
  241. if (invItem.itemId)
  242. return invItem.itemName;
  243. else
  244. return @"";
  245. }
  246. else if ([tableColumn.identifier isEqual:@"count"])
  247. {
  248. if (invItem.count)
  249. return [NSNumber numberWithUnsignedChar:invItem.count];
  250. else
  251. return nil;
  252. }
  253. else if ([tableColumn.identifier isEqual:@"damage"])
  254. {
  255. if (invItem.damage)
  256. return [NSNumber numberWithShort:invItem.damage];
  257. else
  258. return nil;
  259. }
  260. return nil;
  261. }
  262. - (void)outlineView:(NSOutlineView *)theOutlineView setObjectValue:(id)object forTableColumn:(NSTableColumn *)tableColumn byItem:(id)item
  263. {
  264. IJInventoryItem *invItem = item;
  265. if ([tableColumn.identifier isEqual:@"id"])
  266. {
  267. invItem.itemId = [object shortValue];
  268. [self markDirty];
  269. }
  270. else if ([tableColumn.identifier isEqual:@"count"])
  271. {
  272. invItem.count = [object unsignedCharValue];
  273. if (invItem.count > 64)
  274. invItem.count = 64;
  275. [self markDirty];
  276. }
  277. else if ([tableColumn.identifier isEqual:@"damage"])
  278. {
  279. invItem.damage = [object shortValue];
  280. [self markDirty];
  281. }
  282. }
  283. - (BOOL)outlineView:(NSOutlineView *)theOutlineView shouldEditTableColumn:(NSTableColumn *)tableColumn item:(id)item
  284. {
  285. if ([rootItems containsObject:item])
  286. return NO;
  287. else if ([tableColumn.identifier isEqual:@"item"])
  288. {
  289. IJInventoryItem *invItem = item;
  290. [[IJItemPickerWindowController sharedController] showPickerWithInitialItemId:invItem.itemId completionBlock:^(uint16_t itemId) {
  291. invItem.itemId = itemId;
  292. [outlineView reloadItem:item];
  293. [self markDirty];
  294. }];
  295. return NO;
  296. }
  297. else
  298. {
  299. return [tableColumn.identifier isEqual:@"slot"] == NO;
  300. }
  301. }
  302. @end