IJInventoryWindowController.m 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486
  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 "IJInventoryView.h"
  12. #import "IJItemPropertiesViewController.h"
  13. #import "MAAttachedWindow.h"
  14. @interface IJInventoryWindowController ()
  15. @end
  16. @implementation IJInventoryWindowController
  17. @synthesize worldSelectionControl;
  18. @synthesize statusTextField;
  19. @synthesize inventoryView, armorView, quickView;
  20. @synthesize itemSearchField, itemTableView;
  21. - (void)awakeFromNib
  22. {
  23. armorInventory = [[NSMutableArray alloc] init];
  24. quickInventory = [[NSMutableArray alloc] init];
  25. normalInventory = [[NSMutableArray alloc] init];
  26. statusTextField.stringValue = @"";
  27. [inventoryView setRows:3 columns:9 invert:NO];
  28. [quickView setRows:1 columns:9 invert:NO];
  29. [armorView setRows:4 columns:1 invert:YES];
  30. inventoryView.delegate = self;
  31. quickView.delegate = self;
  32. armorView.delegate = self;
  33. // Item Table View setup
  34. NSArray *keys = [[IJInventoryItem itemIdLookup] allKeys];
  35. keys = [keys sortedArrayUsingSelector:@selector(compare:)];
  36. allItemIds = [[NSArray alloc] initWithArray:keys];
  37. filteredItemIds = [allItemIds retain];
  38. }
  39. - (void)dealloc
  40. {
  41. [propertiesViewController release];
  42. [armorInventory release];
  43. [quickInventory release];
  44. [normalInventory release];
  45. [inventory release];
  46. [level release];
  47. [super dealloc];
  48. }
  49. #pragma mark -
  50. #pragma mark World Selection
  51. - (void)loadWorldAtIndex:(int)worldIndex
  52. {
  53. [armorInventory removeAllObjects];
  54. [quickInventory removeAllObjects];
  55. [normalInventory removeAllObjects];
  56. [self willChangeValueForKey:@"worldTime"];
  57. [level release];
  58. level = nil;
  59. [inventory release];
  60. inventory = nil;
  61. [self didChangeValueForKey:@"worldTime"];
  62. [inventoryView setItems:normalInventory];
  63. [quickView setItems:quickInventory];
  64. [armorView setItems:armorInventory];
  65. statusTextField.stringValue = @"No world loaded.";
  66. if (![IJMinecraftLevel worldExistsAtIndex:worldIndex])
  67. {
  68. 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.");
  69. return;
  70. }
  71. sessionLockValue = [IJMinecraftLevel writeToSessionLockAtIndex:worldIndex];
  72. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  73. {
  74. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable obtain the session lock.");
  75. return;
  76. }
  77. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  78. NSData *fileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:levelPath]];
  79. if (!fileData)
  80. {
  81. // Error loading
  82. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", levelPath);
  83. return;
  84. }
  85. [self willChangeValueForKey:@"worldTime"];
  86. level = [[IJMinecraftLevel nbtContainerWithData:fileData] retain];
  87. inventory = [[level inventory] retain];
  88. [self didChangeValueForKey:@"worldTime"];
  89. // Add placeholder inventory items:
  90. for (int i = 0; i < IJInventorySlotQuickLast + 1 - IJInventorySlotQuickFirst; i++)
  91. [quickInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotQuickFirst + i]];
  92. for (int i = 0; i < IJInventorySlotNormalLast + 1 - IJInventorySlotNormalFirst; i++)
  93. [normalInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotNormalFirst + i]];
  94. for (int i = 0; i < IJInventorySlotArmorLast + 1 - IJInventorySlotArmorFirst; i++)
  95. [armorInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotArmorFirst + i]];
  96. // Overwrite the placeholders with actual inventory:
  97. for (IJInventoryItem *item in inventory)
  98. {
  99. if (IJInventorySlotQuickFirst <= item.slot && item.slot <= IJInventorySlotQuickLast)
  100. {
  101. [quickInventory replaceObjectAtIndex:item.slot - IJInventorySlotQuickFirst withObject:item];
  102. }
  103. else if (IJInventorySlotNormalFirst <= item.slot && item.slot <= IJInventorySlotNormalLast)
  104. {
  105. [normalInventory replaceObjectAtIndex:item.slot - IJInventorySlotNormalFirst withObject:item];
  106. }
  107. else if (IJInventorySlotArmorFirst <= item.slot && item.slot <= IJInventorySlotArmorLast)
  108. {
  109. [armorInventory replaceObjectAtIndex:item.slot - IJInventorySlotArmorFirst withObject:item];
  110. }
  111. }
  112. // NSLog(@"normal: %@", normalInventory);
  113. // NSLog(@"quick: %@", quickInventory);
  114. [inventoryView setItems:normalInventory];
  115. [quickView setItems:quickInventory];
  116. [armorView setItems:armorInventory];
  117. dirty = NO;
  118. statusTextField.stringValue = @"";
  119. }
  120. - (void)saveToWorldAtIndex:(int)worldIndex
  121. {
  122. if (inventory == nil)
  123. return; // no world loaded, nothing to save
  124. if (![IJMinecraftLevel checkSessionLockAtIndex:worldIndex value:sessionLockValue])
  125. {
  126. NSBeginCriticalAlertSheet(@"Another application has modified this world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"The session lock was changed by another application.");
  127. return;
  128. }
  129. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtIndex:worldIndex];
  130. NSMutableArray *newInventory = [NSMutableArray array];
  131. for (NSArray *items in [NSArray arrayWithObjects:armorInventory, quickInventory, normalInventory, nil])
  132. {
  133. for (IJInventoryItem *item in items)
  134. {
  135. if (item.count > 0 && item.itemId > 0)
  136. [newInventory addObject:item];
  137. }
  138. }
  139. [level setInventory:newInventory];
  140. NSString *backupPath = [levelPath stringByAppendingPathExtension:@"insidejobbackup"];
  141. BOOL success = NO;
  142. NSError *error = nil;
  143. success = [[NSFileManager defaultManager] removeItemAtPath:backupPath error:&error];
  144. if (!success)
  145. {
  146. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  147. NSBeginCriticalAlertSheet(@"An error occurred while saving.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable to remove the prior backup of this level file:\n%@", [error localizedDescription]);
  148. return;
  149. }
  150. success = [[NSFileManager defaultManager] copyItemAtPath:levelPath toPath:backupPath error:&error];
  151. if (!success)
  152. {
  153. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  154. 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:\n%@", [error localizedDescription]);
  155. return;
  156. }
  157. success = [[level writeData] writeToURL:[NSURL fileURLWithPath:levelPath] options:0 error:&error];
  158. if (!success)
  159. {
  160. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  161. NSError *restoreError = nil;
  162. success = [[NSFileManager defaultManager] copyItemAtPath:backupPath toPath:levelPath error:&restoreError];
  163. if (!success)
  164. {
  165. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [restoreError localizedDescription]);
  166. NSBeginCriticalAlertSheet(@"An error occurred while saving.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable to save to the existing level file, and the backup could not be restored.\n%@\n%@", [error localizedDescription], [restoreError localizedDescription]);
  167. }
  168. else
  169. {
  170. NSBeginCriticalAlertSheet(@"An error occurred while saving.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"Inside Job was unable to save to the existing level file, and the backup was successfully restored.\n%@", [error localizedDescription]);
  171. }
  172. return;
  173. }
  174. dirty = NO;
  175. statusTextField.stringValue = @"Saved.";
  176. }
  177. - (void)markDirty
  178. {
  179. dirty = YES;
  180. statusTextField.stringValue = @"World has unsaved changes.";
  181. }
  182. #pragma mark -
  183. #pragma mark Actions
  184. - (IBAction)menuSelectWorld:(id)sender
  185. {
  186. int worldIndex = [sender tag];
  187. [self loadWorldAtIndex:worldIndex];
  188. [worldSelectionControl setSelectedSegment:worldIndex - 1];
  189. }
  190. - (IBAction)worldSelectionChanged:(id)sender
  191. {
  192. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  193. [self loadWorldAtIndex:worldIndex];
  194. }
  195. - (void)saveDocument:(id)sender
  196. {
  197. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  198. [self saveToWorldAtIndex:worldIndex];
  199. }
  200. - (void)delete:(id)sender
  201. {
  202. // IJInventoryItem *item = [outlineView itemAtRow:[outlineView selectedRow]];
  203. // item.count = 0;
  204. // item.itemId = 0;
  205. // item.damage = 0;
  206. // [self markDirty];
  207. // [outlineView reloadItem:item];
  208. }
  209. - (IBAction)makeSearchFieldFirstResponder:(id)sender
  210. {
  211. [itemSearchField becomeFirstResponder];
  212. }
  213. - (BOOL)validateUserInterfaceItem:(id <NSValidatedUserInterfaceItem>)anItem
  214. {
  215. if (anItem.action == @selector(saveDocument:))
  216. return inventory != nil;
  217. return YES;
  218. }
  219. - (NSNumber *)worldTime
  220. {
  221. return [level worldTimeContainer].numberValue;
  222. }
  223. - (void)setWorldTime:(NSNumber *)number
  224. {
  225. [self willChangeValueForKey:@"worldTime"];
  226. [level worldTimeContainer].numberValue = number;
  227. [self didChangeValueForKey:@"worldTime"];
  228. [self markDirty];
  229. }
  230. #pragma mark -
  231. #pragma mark IJInventoryViewDelegate
  232. - (NSMutableArray *)itemArrayForInventoryView:(IJInventoryView *)theInventoryView slotOffset:(int*)slotOffset
  233. {
  234. if (theInventoryView == inventoryView)
  235. {
  236. if (slotOffset) *slotOffset = IJInventorySlotNormalFirst;
  237. return normalInventory;
  238. }
  239. else if (theInventoryView == quickView)
  240. {
  241. if (slotOffset) *slotOffset = IJInventorySlotQuickFirst;
  242. return quickInventory;
  243. }
  244. else if (theInventoryView == armorView)
  245. {
  246. if (slotOffset) *slotOffset = IJInventorySlotArmorFirst;
  247. return armorInventory;
  248. }
  249. return nil;
  250. }
  251. - (void)inventoryView:(IJInventoryView *)theInventoryView removeItemAtIndex:(int)itemIndex
  252. {
  253. int slotOffset = 0;
  254. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  255. if (itemArray)
  256. {
  257. IJInventoryItem *item = [IJInventoryItem emptyItemWithSlot:slotOffset + itemIndex];
  258. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  259. [theInventoryView setItems:itemArray];
  260. }
  261. [self markDirty];
  262. }
  263. - (void)inventoryView:(IJInventoryView *)theInventoryView setItem:(IJInventoryItem *)item atIndex:(int)itemIndex
  264. {
  265. int slotOffset = 0;
  266. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  267. if (itemArray)
  268. {
  269. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  270. item.slot = slotOffset + itemIndex;
  271. [theInventoryView setItems:itemArray];
  272. }
  273. [self markDirty];
  274. }
  275. - (void)inventoryView:(IJInventoryView *)theInventoryView selectedItemAtIndex:(int)itemIndex
  276. {
  277. // Show the properties window for this item.
  278. IJInventoryItem *lastItem = propertiesViewController.item;
  279. NSPoint itemLocationInView = [theInventoryView pointForItemAtIndex:itemIndex];
  280. NSPoint point = [theInventoryView convertPoint:itemLocationInView toView:nil];
  281. point.x += 16 + 8;
  282. point.y -= 16;
  283. NSArray *items = [self itemArrayForInventoryView:theInventoryView slotOffset:nil];
  284. IJInventoryItem *item = [items objectAtIndex:itemIndex];
  285. //NSLog(@"%s index=%d item=%@", _cmd, itemIndex, item);
  286. if (item.itemId == 0 || lastItem == item)
  287. {
  288. // Perhaps caused by a bug, but it seems to be possible for the window to not be invisible at this point,
  289. // so we will set the alpha value here to be sure.
  290. [propertiesWindow setAlphaValue:0.0];
  291. propertiesViewController.item = nil;
  292. return; // can't show info on nothing
  293. }
  294. if (!propertiesViewController)
  295. {
  296. propertiesViewController = [[IJItemPropertiesViewController alloc] initWithNibName:@"ItemPropertiesView" bundle:nil];
  297. propertiesWindow = [[MAAttachedWindow alloc] initWithView:propertiesViewController.view
  298. attachedToPoint:point
  299. inWindow:self.window
  300. onSide:MAPositionRight
  301. atDistance:0];
  302. [propertiesWindow setBackgroundColor:[NSColor controlBackgroundColor]];
  303. [propertiesWindow setViewMargin:4.0];
  304. [propertiesWindow setAlphaValue:1.0];
  305. [[self window] addChildWindow:propertiesWindow ordered:NSWindowAbove];
  306. }
  307. if (observerObject)
  308. [[NSNotificationCenter defaultCenter] removeObserver:observerObject];
  309. observerObject = [[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidResignKeyNotification
  310. object:propertiesWindow
  311. queue:[NSOperationQueue mainQueue]
  312. usingBlock:^(NSNotification *notification) {
  313. [propertiesViewController commitEditing];
  314. if (item.count == 0)
  315. item.itemId = 0;
  316. [theInventoryView reloadItemAtIndex:itemIndex];
  317. [propertiesWindow setAlphaValue:0.0];
  318. }];
  319. propertiesViewController.item = item;
  320. [propertiesWindow setPoint:point side:MAPositionRight];
  321. [propertiesWindow makeKeyAndOrderFront:nil];
  322. [propertiesWindow setAlphaValue:1.0];
  323. }
  324. #pragma mark -
  325. #pragma mark Item Picker
  326. - (IBAction)updateItemSearchFilter:(id)sender
  327. {
  328. NSString *filterString = [sender stringValue];
  329. if (filterString.length == 0)
  330. {
  331. [filteredItemIds autorelease];
  332. filteredItemIds = [allItemIds retain];
  333. [itemTableView reloadData];
  334. return;
  335. }
  336. NSMutableArray *results = [NSMutableArray array];
  337. for (NSNumber *itemId in allItemIds)
  338. {
  339. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  340. NSRange range = [name rangeOfString:filterString options:NSCaseInsensitiveSearch];
  341. if (range.location != NSNotFound)
  342. {
  343. [results addObject:itemId];
  344. continue;
  345. }
  346. // Also search the item id:
  347. range = [[itemId stringValue] rangeOfString:filterString];
  348. if (range.location != NSNotFound)
  349. {
  350. [results addObject:itemId];
  351. continue;
  352. }
  353. }
  354. [filteredItemIds autorelease];
  355. filteredItemIds = [results retain];
  356. [itemTableView reloadData];
  357. }
  358. - (NSInteger)numberOfRowsInTableView:(NSTableView *)theTableView
  359. {
  360. return filteredItemIds.count;
  361. }
  362. - (id)tableView:(NSTableView *)theTableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row
  363. {
  364. NSNumber *itemId = [filteredItemIds objectAtIndex:row];
  365. if ([tableColumn.identifier isEqual:@"itemId"])
  366. {
  367. return itemId;
  368. }
  369. else if ([tableColumn.identifier isEqual:@"image"])
  370. {
  371. return [IJInventoryItem imageForItemId:[itemId shortValue]];
  372. }
  373. else
  374. {
  375. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  376. return name;
  377. }
  378. }
  379. - (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard
  380. {
  381. [pboard declareTypes:[NSArray arrayWithObjects:IJPasteboardTypeInventoryItem, nil] owner:nil];
  382. NSNumber *itemId = [filteredItemIds objectAtIndex:[rowIndexes firstIndex]];
  383. IJInventoryItem *item = [[IJInventoryItem alloc] init];
  384. item.itemId = [itemId shortValue];
  385. item.count = 1;
  386. item.damage = 0;
  387. item.slot = 0;
  388. [pboard setData:[NSKeyedArchiver archivedDataWithRootObject:item]
  389. forType:IJPasteboardTypeInventoryItem];
  390. [item release];
  391. return YES;
  392. }
  393. #pragma mark -
  394. #pragma mark NSWindowDelegate
  395. - (void)windowWillClose:(NSNotification *)notification
  396. {
  397. [NSApp terminate:nil];
  398. }
  399. @end