IJInventoryWindowController.m 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719
  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. - (void)loadWorldAtFolder:(NSString *)worldFolder;
  18. - (BOOL)isDocumentEdited;
  19. @end
  20. @implementation IJInventoryWindowController
  21. @synthesize worldSelectionControl;
  22. @synthesize statusTextField;
  23. @synthesize inventoryView, armorView, quickView;
  24. @synthesize itemSearchField, itemTableView;
  25. - (void)awakeFromNib
  26. {
  27. armorInventory = [[NSMutableArray alloc] init];
  28. quickInventory = [[NSMutableArray alloc] init];
  29. normalInventory = [[NSMutableArray alloc] init];
  30. statusTextField.stringValue = @"";
  31. [inventoryView setRows:3 columns:9 invert:NO];
  32. [quickView setRows:1 columns:9 invert:NO];
  33. [armorView setRows:4 columns:1 invert:YES];
  34. inventoryView.delegate = self;
  35. quickView.delegate = self;
  36. armorView.delegate = self;
  37. // Item Table View setup
  38. NSArray *keys = [[IJInventoryItem itemIdLookup] allKeys];
  39. keys = [keys sortedArrayUsingSelector:@selector(compare:)];
  40. allItemIds = [[NSArray alloc] initWithArray:keys];
  41. filteredItemIds = [allItemIds retain];
  42. [itemTableView setTarget:self];
  43. [itemTableView setDoubleAction:@selector(itemTableViewDoubleClicked:)];
  44. }
  45. - (void)dealloc
  46. {
  47. [propertiesViewController release];
  48. [armorInventory release];
  49. [quickInventory release];
  50. [normalInventory release];
  51. [inventory release];
  52. [level release];
  53. [player release];
  54. [super dealloc];
  55. }
  56. #pragma mark -
  57. #pragma mark World Selection
  58. - (void)dirtyLoadSheetDidEnd:(NSWindow *)sheet returnCode:(int)returnCode contextInfo:(void *)contextInfo
  59. {
  60. if (returnCode == NSAlertOtherReturn) // Cancel
  61. {
  62. [worldSelectionControl setSelectedSegment:loadedWorldIndex-1];
  63. return;
  64. }
  65. if (returnCode == NSAlertDefaultReturn) // Save
  66. {
  67. [self saveWorld];
  68. [self loadWorldAtFolder:attemptedLoadWorldFolder];
  69. }
  70. else if (returnCode == NSAlertAlternateReturn) // Don't save
  71. {
  72. [self setDocumentEdited:NO]; // Slightly hacky -- prevent the alert from being put up again.
  73. [self loadWorldAtFolder:attemptedLoadWorldFolder];
  74. }
  75. }
  76. - (void)loadWorldPlayerInventory:(NSString *)PlayerName
  77. {
  78. /*
  79. * If passing NULL to PlayerName, we will use level.dat instead of
  80. * Players/PlayerName.dat file
  81. */
  82. }
  83. - (void)loadWorldAtFolder:(NSString *)worldPath
  84. {
  85. if ([self isDocumentEdited])
  86. {
  87. attemptedLoadWorldFolder = worldPath;
  88. 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.");
  89. return;
  90. }
  91. [armorInventory removeAllObjects];
  92. [quickInventory removeAllObjects];
  93. [normalInventory removeAllObjects];
  94. [inventoryView setItems:normalInventory];
  95. [quickView setItems:quickInventory];
  96. [armorView setItems:armorInventory];
  97. [self willChangeValueForKey:@"worldTime"];
  98. [level release];
  99. level = nil;
  100. [player release];
  101. player = nil;
  102. [inventory release];
  103. inventory = nil;
  104. [self didChangeValueForKey:@"worldTime"];
  105. statusTextField.stringValue = @"No world loaded.";
  106. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtFolder:worldPath];
  107. NSData *fileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:levelPath]];
  108. if (!fileData)
  109. {
  110. // Error loading
  111. NSBeginCriticalAlertSheet(@"Error loading world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", levelPath);
  112. return;
  113. }
  114. [self willChangeValueForKey:@"worldTime"];
  115. /* Now search for first player .dat file (but by default try to load from level.dat */
  116. #if 1
  117. loadedPlayer = nil;
  118. NSString *playerPath = [IJMinecraftLevel pathForPlayer:loadedPlayer withWorld:worldPath];
  119. #else
  120. NSString *playerPath = [worldPath stringByAppendingString:@"/players/Godzil.dat"];
  121. #endif
  122. NSData *playerFileData = [NSData dataWithContentsOfURL:[NSURL fileURLWithPath:playerPath]];
  123. if (!playerFileData)
  124. {
  125. // Error loading
  126. NSBeginCriticalAlertSheet(@"Error loading player.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"InsideJob was unable to load the level at %@.", playerPath);
  127. return;
  128. }
  129. level = [[IJMinecraftLevel nbtContainerWithData:fileData] retain];
  130. player = [[IJMinecraftLevel nbtContainerWithData:playerFileData] retain];
  131. inventory = [[player inventory] retain];
  132. [self didChangeValueForKey:@"worldTime"];
  133. // Add placeholder inventory items:
  134. for (int i = 0; i < IJInventorySlotQuickLast + 1 - IJInventorySlotQuickFirst; i++)
  135. [quickInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotQuickFirst + i]];
  136. for (int i = 0; i < IJInventorySlotNormalLast + 1 - IJInventorySlotNormalFirst; i++)
  137. [normalInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotNormalFirst + i]];
  138. for (int i = 0; i < IJInventorySlotArmorLast + 1 - IJInventorySlotArmorFirst; i++)
  139. [armorInventory addObject:[IJInventoryItem emptyItemWithSlot:IJInventorySlotArmorFirst + i]];
  140. // Overwrite the placeholders with actual inventory:
  141. for (IJInventoryItem *item in inventory)
  142. {
  143. if (IJInventorySlotQuickFirst <= item.slot && item.slot <= IJInventorySlotQuickLast)
  144. {
  145. [quickInventory replaceObjectAtIndex:item.slot - IJInventorySlotQuickFirst withObject:item];
  146. }
  147. else if (IJInventorySlotNormalFirst <= item.slot && item.slot <= IJInventorySlotNormalLast)
  148. {
  149. [normalInventory replaceObjectAtIndex:item.slot - IJInventorySlotNormalFirst withObject:item];
  150. }
  151. else if (IJInventorySlotArmorFirst <= item.slot && item.slot <= IJInventorySlotArmorLast)
  152. {
  153. [armorInventory replaceObjectAtIndex:item.slot - IJInventorySlotArmorFirst withObject:item];
  154. }
  155. }
  156. // NSLog(@"normal: %@", normalInventory);
  157. // NSLog(@"quick: %@", quickInventory);
  158. [inventoryView setItems:normalInventory];
  159. [quickView setItems:quickInventory];
  160. [armorView setItems:armorInventory];
  161. [self setDocumentEdited:NO];
  162. statusTextField.stringValue = @"";
  163. loadedWorldFolder = [worldPath retain];
  164. }
  165. - (void)loadWorldAtIndex:(int)worldIndex
  166. {
  167. NSString *worldPath;
  168. worldPath = [IJMinecraftLevel pathForWorldAtIndex:worldIndex];
  169. [self loadWorldAtFolder: worldPath];
  170. }
  171. - (void)saveWorld
  172. {
  173. NSString *worldPath = loadedWorldFolder;
  174. if (inventory == nil)
  175. return; // no world loaded, nothing to save
  176. if (![IJMinecraftLevel checkSessionLockAtFolder:worldPath value:sessionLockValue])
  177. {
  178. NSBeginCriticalAlertSheet(@"Another application has modified this world.", @"Dismiss", nil, nil, self.window, nil, nil, nil, nil, @"The session lock was changed by another application.");
  179. return;
  180. }
  181. NSString *levelPath = [IJMinecraftLevel pathForLevelDatAtFolder:worldPath];
  182. NSString *playerPath = [IJMinecraftLevel pathForPlayer:loadedPlayer withWorld:worldPath];
  183. NSMutableArray *newInventory = [NSMutableArray array];
  184. for (NSArray *items in [NSArray arrayWithObjects:armorInventory, quickInventory, normalInventory, nil])
  185. {
  186. for (IJInventoryItem *item in items)
  187. {
  188. if (item.count > 0 && item.itemId > 0)
  189. [newInventory addObject:item];
  190. }
  191. }
  192. [player setInventory:newInventory];
  193. NSString *backupLevelPath = [levelPath stringByAppendingPathExtension:@"insidejobbackup"];
  194. NSString *backupPlayerPath = [playerPath stringByAppendingPathExtension:@"insidejobbackup"];
  195. BOOL success = NO;
  196. NSError *error = nil;
  197. // Remove a previously-created .insidejobbackup, if it exists:
  198. if ([[NSFileManager defaultManager] fileExistsAtPath:backupLevelPath])
  199. {
  200. success = [[NSFileManager defaultManager] removeItemAtPath:backupLevelPath error:&error];
  201. if (!success)
  202. {
  203. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  204. 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]);
  205. return;
  206. }
  207. }
  208. // Remove a previously-created .insidejobbackup, if it exists:
  209. if ([[NSFileManager defaultManager] fileExistsAtPath:backupLevelPath])
  210. {
  211. success = [[NSFileManager defaultManager] removeItemAtPath:backupPlayerPath error:&error];
  212. if (!success)
  213. {
  214. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  215. 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]);
  216. return;
  217. }
  218. }
  219. // Create the backup:
  220. success = [[NSFileManager defaultManager] copyItemAtPath:levelPath toPath:backupLevelPath error:&error];
  221. if (!success)
  222. {
  223. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  224. 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]);
  225. return;
  226. }
  227. success = [[NSFileManager defaultManager] copyItemAtPath:playerPath toPath:backupPlayerPath error:&error];
  228. if (!success)
  229. {
  230. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  231. 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]);
  232. return;
  233. }
  234. // Write the new level.dat out:
  235. success = [[player writeData] writeToURL:[NSURL fileURLWithPath:levelPath] options:0 error:&error];
  236. if (!success)
  237. {
  238. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [error localizedDescription]);
  239. NSError *restoreError = nil;
  240. success = [[NSFileManager defaultManager] copyItemAtPath:backupLevelPath toPath:levelPath error:&restoreError];
  241. if (!success)
  242. {
  243. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [restoreError localizedDescription]);
  244. 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]);
  245. }
  246. success = [[NSFileManager defaultManager] copyItemAtPath:backupPlayerPath toPath:playerPath error:&restoreError];
  247. if (!success)
  248. {
  249. NSLog(@"%s:%d %@", __PRETTY_FUNCTION__, __LINE__, [restoreError localizedDescription]);
  250. 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]);
  251. }
  252. else
  253. {
  254. 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]);
  255. }
  256. return;
  257. }
  258. [self setDocumentEdited:NO];
  259. statusTextField.stringValue = @"Saved.";
  260. }
  261. - (void)setDocumentEdited:(BOOL)edited
  262. {
  263. [super setDocumentEdited:edited];
  264. if (edited)
  265. statusTextField.stringValue = @"World has unsaved changes.";
  266. }
  267. - (BOOL)isDocumentEdited
  268. {
  269. return [self.window isDocumentEdited];
  270. }
  271. #pragma mark -
  272. #pragma mark Actions
  273. - (IBAction)menuSelectWorld:(id)sender
  274. {
  275. int worldIndex = [sender tag];
  276. [self loadWorldAtIndex:worldIndex];
  277. [worldSelectionControl setSelectedSegment:worldIndex - 1];
  278. }
  279. - (IBAction)menuSelectWorldFromPath:(id)sender
  280. {
  281. NSInteger openResult;
  282. /* Ask user for world folder path */
  283. NSOpenPanel *panel = [NSOpenPanel openPanel];
  284. NSString *worldPath;
  285. /* Only allow to choose a folder */
  286. [panel setCanChooseDirectories:YES];
  287. [panel setCanChooseFiles:NO];
  288. openResult = [panel runModal];
  289. if (openResult == NSOKButton)
  290. {
  291. worldPath = [[panel directoryURL] path];
  292. /* Verify for level.dat */
  293. if (![IJMinecraftLevel worldExistsAtFolder: worldPath])
  294. {
  295. 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.");
  296. return;
  297. }
  298. /* Now try to open the world... */
  299. [self loadWorldAtFolder:[[panel directoryURL] path]];
  300. }
  301. }
  302. - (IBAction)worldSelectionChanged:(id)sender
  303. {
  304. int worldIndex = [worldSelectionControl selectedSegment] + 1;
  305. [self loadWorldAtIndex:worldIndex];
  306. }
  307. - (void)saveDocument:(id)sender
  308. {
  309. [self saveWorld];
  310. }
  311. - (void)delete:(id)sender
  312. {
  313. // IJInventoryItem *item = [outlineView itemAtRow:[outlineView selectedRow]];
  314. // item.count = 0;
  315. // item.itemId = 0;
  316. // item.damage = 0;
  317. // [self setDocumentEdited:YES];
  318. // [outlineView reloadItem:item];
  319. }
  320. - (IBAction)makeSearchFieldFirstResponder:(id)sender
  321. {
  322. [itemSearchField becomeFirstResponder];
  323. }
  324. - (BOOL)validateUserInterfaceItem:(id <NSValidatedUserInterfaceItem>)anItem
  325. {
  326. if (anItem.action == @selector(saveDocument:))
  327. return inventory != nil;
  328. return YES;
  329. }
  330. - (NSNumber *)worldTime
  331. {
  332. return [level worldTimeContainer].numberValue;
  333. }
  334. - (void)setWorldTime:(NSNumber *)number
  335. {
  336. [self willChangeValueForKey:@"worldTime"];
  337. [level worldTimeContainer].numberValue = number;
  338. [self didChangeValueForKey:@"worldTime"];
  339. [self setDocumentEdited:YES];
  340. }
  341. - (NSString *)playerName
  342. {
  343. return [level ].numberValue;
  344. }
  345. - (void)setPlayerName:(NSString *)playerName
  346. {
  347. }
  348. #pragma mark -
  349. #pragma mark IJInventoryViewDelegate
  350. - (IJInventoryView *)inventoryViewForItemArray:(NSMutableArray *)theItemArray
  351. {
  352. if (theItemArray == normalInventory)
  353. return inventoryView;
  354. if (theItemArray == quickInventory)
  355. return quickView;
  356. if (theItemArray == armorInventory)
  357. return armorView;
  358. return nil;
  359. }
  360. - (NSMutableArray *)itemArrayForInventoryView:(IJInventoryView *)theInventoryView slotOffset:(int*)slotOffset
  361. {
  362. if (theInventoryView == inventoryView)
  363. {
  364. if (slotOffset) *slotOffset = IJInventorySlotNormalFirst;
  365. return normalInventory;
  366. }
  367. else if (theInventoryView == quickView)
  368. {
  369. if (slotOffset) *slotOffset = IJInventorySlotQuickFirst;
  370. return quickInventory;
  371. }
  372. else if (theInventoryView == armorView)
  373. {
  374. if (slotOffset) *slotOffset = IJInventorySlotArmorFirst;
  375. return armorInventory;
  376. }
  377. return nil;
  378. }
  379. - (void)inventoryView:(IJInventoryView *)theInventoryView removeItemAtIndex:(int)itemIndex
  380. {
  381. int slotOffset = 0;
  382. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  383. if (itemArray)
  384. {
  385. IJInventoryItem *item = [IJInventoryItem emptyItemWithSlot:slotOffset + itemIndex];
  386. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  387. [theInventoryView setItems:itemArray];
  388. }
  389. [self setDocumentEdited:YES];
  390. }
  391. - (void)inventoryView:(IJInventoryView *)theInventoryView setItem:(IJInventoryItem *)item atIndex:(int)itemIndex
  392. {
  393. int slotOffset = 0;
  394. NSMutableArray *itemArray = [self itemArrayForInventoryView:theInventoryView slotOffset:&slotOffset];
  395. if (itemArray)
  396. {
  397. [itemArray replaceObjectAtIndex:itemIndex withObject:item];
  398. item.slot = slotOffset + itemIndex;
  399. [theInventoryView setItems:itemArray];
  400. }
  401. [self setDocumentEdited:YES];
  402. }
  403. - (void)inventoryView:(IJInventoryView *)theInventoryView selectedItemAtIndex:(int)itemIndex
  404. {
  405. // Show the properties window for this item.
  406. IJInventoryItem *lastItem = propertiesViewController.item;
  407. NSPoint itemLocationInView = [theInventoryView pointForItemAtIndex:itemIndex];
  408. NSPoint point = [theInventoryView convertPoint:itemLocationInView toView:nil];
  409. point.x += 16 + 8;
  410. point.y -= 16;
  411. NSArray *items = [self itemArrayForInventoryView:theInventoryView slotOffset:nil];
  412. IJInventoryItem *item = [items objectAtIndex:itemIndex];
  413. //NSLog(@"%s index=%d item=%@", _cmd, itemIndex, item);
  414. if (item.itemId == 0 || lastItem == item)
  415. {
  416. // Perhaps caused by a bug, but it seems to be possible for the window to not be invisible at this point,
  417. // so we will set the alpha value here to be sure.
  418. [propertiesWindow setAlphaValue:0.0];
  419. propertiesViewController.item = nil;
  420. return; // can't show info on nothing
  421. }
  422. if (!propertiesViewController)
  423. {
  424. propertiesViewController = [[IJItemPropertiesViewController alloc] initWithNibName:@"ItemPropertiesView" bundle:nil];
  425. propertiesWindow = [[MAAttachedWindow alloc] initWithView:propertiesViewController.view
  426. attachedToPoint:point
  427. inWindow:self.window
  428. onSide:MAPositionRight
  429. atDistance:0];
  430. [propertiesWindow setBackgroundColor:[NSColor controlBackgroundColor]];
  431. [propertiesWindow setViewMargin:4.0];
  432. [propertiesWindow setAlphaValue:1.0];
  433. [[self window] addChildWindow:propertiesWindow ordered:NSWindowAbove];
  434. }
  435. if (observerObject)
  436. [[NSNotificationCenter defaultCenter] removeObserver:observerObject];
  437. observerObject = [[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidResignKeyNotification
  438. object:propertiesWindow
  439. queue:[NSOperationQueue mainQueue]
  440. usingBlock:^(NSNotification *notification) {
  441. [propertiesViewController commitEditing];
  442. if (item.count == 0)
  443. item.itemId = 0;
  444. [theInventoryView reloadItemAtIndex:itemIndex];
  445. [propertiesWindow setAlphaValue:0.0];
  446. }];
  447. propertiesViewController.item = item;
  448. [propertiesWindow setPoint:point side:MAPositionRight];
  449. [propertiesWindow makeKeyAndOrderFront:nil];
  450. [propertiesWindow setAlphaValue:1.0];
  451. }
  452. #pragma mark -
  453. #pragma mark Item Picker
  454. - (IBAction)updateItemSearchFilter:(id)sender
  455. {
  456. NSString *filterString = [sender stringValue];
  457. if (filterString.length == 0)
  458. {
  459. [filteredItemIds autorelease];
  460. filteredItemIds = [allItemIds retain];
  461. [itemTableView reloadData];
  462. return;
  463. }
  464. NSMutableArray *results = [NSMutableArray array];
  465. for (NSNumber *itemId in allItemIds)
  466. {
  467. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  468. NSRange range = [name rangeOfString:filterString options:NSCaseInsensitiveSearch];
  469. if (range.location != NSNotFound)
  470. {
  471. [results addObject:itemId];
  472. continue;
  473. }
  474. // Also search the item id:
  475. range = [[itemId stringValue] rangeOfString:filterString];
  476. if (range.location != NSNotFound)
  477. {
  478. [results addObject:itemId];
  479. continue;
  480. }
  481. }
  482. [filteredItemIds autorelease];
  483. filteredItemIds = [results retain];
  484. [itemTableView reloadData];
  485. }
  486. - (NSInteger)numberOfRowsInTableView:(NSTableView *)theTableView
  487. {
  488. return filteredItemIds.count;
  489. }
  490. - (id)tableView:(NSTableView *)theTableView objectValueForTableColumn:(NSTableColumn *)tableColumn row:(NSInteger)row
  491. {
  492. NSNumber *itemId = [filteredItemIds objectAtIndex:row];
  493. if ([tableColumn.identifier isEqual:@"itemId"])
  494. {
  495. return itemId;
  496. }
  497. else if ([tableColumn.identifier isEqual:@"image"])
  498. {
  499. return [IJInventoryItem imageForItemId:[itemId shortValue]];
  500. }
  501. else
  502. {
  503. NSString *name = [[IJInventoryItem itemIdLookup] objectForKey:itemId];
  504. return name;
  505. }
  506. }
  507. - (BOOL)tableView:(NSTableView *)tableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard
  508. {
  509. [pboard declareTypes:[NSArray arrayWithObjects:IJPasteboardTypeInventoryItem, nil] owner:nil];
  510. NSNumber *itemId = [filteredItemIds objectAtIndex:[rowIndexes firstIndex]];
  511. IJInventoryItem *item = [[IJInventoryItem alloc] init];
  512. item.itemId = [itemId shortValue];
  513. item.count = 1;
  514. item.damage = 0;
  515. item.slot = 0;
  516. [pboard setData:[NSKeyedArchiver archivedDataWithRootObject:item]
  517. forType:IJPasteboardTypeInventoryItem];
  518. [item release];
  519. return YES;
  520. }
  521. - (NSMutableArray *)inventoryArrayWithEmptySlot:(NSUInteger *)slot
  522. {
  523. for (NSMutableArray *inventoryArray in [NSArray arrayWithObjects:quickInventory, normalInventory, nil])
  524. {
  525. __block BOOL found = NO;
  526. [inventoryArray enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL *stop) {
  527. IJInventoryItem *item = obj;
  528. if (item.count == 0)
  529. {
  530. *slot = index;
  531. *stop = YES;
  532. found = YES;
  533. }
  534. }];
  535. if (found)
  536. return inventoryArray;
  537. }
  538. return nil;
  539. }
  540. - (IBAction)itemTableViewDoubleClicked:(id)sender
  541. {
  542. NSUInteger slot;
  543. NSMutableArray *inventoryArray = [self inventoryArrayWithEmptySlot:&slot];
  544. if (!inventoryArray)
  545. return;
  546. IJInventoryItem *item = [inventoryArray objectAtIndex:slot];
  547. item.itemId = [[filteredItemIds objectAtIndex:[itemTableView selectedRow]] shortValue];
  548. item.count = 1;
  549. [self setDocumentEdited:YES];
  550. IJInventoryView *invView = [self inventoryViewForItemArray:inventoryArray];
  551. [invView reloadItemAtIndex:slot];
  552. [self inventoryView:invView selectedItemAtIndex:slot];
  553. }
  554. #pragma mark -
  555. #pragma mark NSWindowDelegate
  556. - (void)dirtyCloseSheetDidDismiss:(NSWindow *)sheet returnCode:(int)returnCode contextInfo:(void *)contextInfo
  557. {
  558. if (returnCode == NSAlertOtherReturn) // Cancel
  559. return;
  560. if (returnCode == NSAlertDefaultReturn) // Save
  561. {
  562. [self saveWorld];
  563. [self.window performClose:nil];
  564. }
  565. else if (returnCode == NSAlertAlternateReturn) // Don't save
  566. {
  567. [self setDocumentEdited:NO]; // Slightly hacky -- prevent the alert from being put up again.
  568. [self.window performClose:nil];
  569. }
  570. }
  571. - (BOOL)windowShouldClose:(id)sender
  572. {
  573. if ([self isDocumentEdited])
  574. {
  575. // Note: We use the didDismiss selector becuase the sheet needs to be closed in order for performClose: to work.
  576. 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.");
  577. return NO;
  578. }
  579. return YES;
  580. }
  581. - (void)windowWillClose:(NSNotification *)notification
  582. {
  583. [NSApp terminate:nil];
  584. }
  585. #pragma mark -
  586. #pragma mark NSControlTextEditingDelegate
  587. - (BOOL)control:(NSControl *)control textView:(NSTextView *)textView doCommandBySelector:(SEL)command
  588. {
  589. if (command == @selector(moveDown:))
  590. {
  591. if ([itemTableView numberOfRows] > 0)
  592. {
  593. [self.window makeFirstResponder:itemTableView];
  594. [itemTableView selectRowIndexes:[NSIndexSet indexSetWithIndex:0] byExtendingSelection:NO];
  595. }
  596. return YES;
  597. }
  598. return YES;
  599. }
  600. @end