IJInventoryWindowController.m 17 KB

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