dfhack/scripts/create-items.rb

# create first necessity items under cursor

category = $script_args[0] || 'help'
mat_raw  = $script_args[1] || 'list'
count    = $script_args[2]


category = df.match_rawname(category, ['help', 'bars', 'boulders', 'plants', 'logs', 'webs', 'anvils']) || 'help'

if category == 'help'
	puts <<EOS
Create first necessity items under the cursor.
Usage:
 create-items [category] [raws token] [number]

Item categories:
 bars, boulders, plants, logs, webs, anvils

Raw token:
 Either a full token (PLANT_MAT:ADLER:WOOD) or the middle part only
 (the missing part is autocompleted depending on the item category)
 Use 'list' to show all possibilities

Exemples:
 create-items boulders hematite 30
 create-items bars CREATURE_MAT:CAT:SOAP 10
 create-items web cave_giant
 create-items plants list

EOS
	throw :script_finished

elsif mat_raw == 'list'
	# allowed with no cursor

elsif df.cursor.x == -30000
	puts "Please place the game cursor somewhere"
	throw :script_finished

elsif !(maptile = df.map_tile_at(df.cursor))
	puts "Error: unallocated map block !"
	throw :script_finished

elsif !maptile.shape_passablehigh
	puts "Error: impassible tile !"
	throw :script_finished
end


def match_list(tok, list)
	if tok != 'list'
		tok = df.match_rawname(tok, list)
		if not tok
			puts "Invalid raws token, use one in:"
			tok = 'list'
		end
	end
	if tok == 'list'
		puts list.map { |w| w =~ /[^\w]/ ? w.inspect : w }.join(' ')
		throw :script_finished
	end
	tok
end


case category
when 'bars'
	# create metal bar, eg createbar INORGANIC:IRON
	cls = DFHack::ItemBarst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.inorganics.find_all { |ino|
			ino.material.flags[:IS_METAL]
		}.map { |ino| ino.id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "INORGANIC:#{mat_raw}"
		puts mat_raw
	end
	customize = lambda { |item|
		item.dimension = 150
		item.subtype = -1
	}

when 'boulders'
	cls = DFHack::ItemBoulderst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.inorganics.find_all { |ino|
			ino.material.flags[:IS_STONE]
		}.map { |ino| ino.id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "INORGANIC:#{mat_raw}"
		puts mat_raw
	end

when 'plants'
	cls = DFHack::ItemPlantst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.plants.all.find_all { |plt|
			plt.material.find { |mat| mat.id == 'STRUCTURAL' }
		}.map { |plt| plt.id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "PLANT_MAT:#{mat_raw}:STRUCTURAL"
		puts mat_raw
	end

when 'logs'
	cls = DFHack::ItemWoodst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.plants.all.find_all { |plt|
			plt.material.find { |mat| mat.id == 'WOOD' }
		}.map { |plt| plt.id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "PLANT_MAT:#{mat_raw}:WOOD"
		puts mat_raw
	end

when 'webs'
	cls = DFHack::ItemThreadst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.creatures.all.find_all { |cre|
			cre.material.find { |mat| mat.id == 'SILK' }
		}.map { |cre| cre.creature_id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "CREATURE_MAT:#{mat_raw}:SILK"
		puts mat_raw
	end
	count ||= 1
	customize = lambda { |item|
		item.flags.spider_web = true
		item.dimension = 15000	# XXX may depend on creature (this is for GCS)
	}

when 'anvils'
	cls = DFHack::ItemAnvilst
	if mat_raw !~ /:/ and !(df.decode_mat(mat_raw) rescue nil)
		list = df.world.raws.inorganics.find_all { |ino|
			ino.material.flags[:IS_METAL]
		}.map { |ino| ino.id }
		mat_raw = match_list(mat_raw, list)
		mat_raw = "INORGANIC:#{mat_raw}"
		puts mat_raw
	end
	count ||= 1

end


mat = df.decode_mat mat_raw

count ||= 20
count.to_i.times {
	item = cls.cpp_new
	item.id = df.item_next_id
	item.stack_size = 1
	item.mat_type = mat.mat_type
	item.mat_index = mat.mat_index

	customize[item] if customize

	df.item_next_id += 1
	item.categorize(true)
	df.world.items.all << item

	item.pos = df.cursor
	item.flags.on_ground = true
	df.map_tile_at.mapblock.items << item.id
	df.map_tile_at.occupancy.item = true
}


# move game view, so that the ui menu updates
if df.cursor.z > 5
	df.curview.feed_keys(:CURSOR_DOWN_Z)
	df.curview.feed_keys(:CURSOR_UP_Z)
else
	df.curview.feed_keys(:CURSOR_UP_Z)
	df.curview.feed_keys(:CURSOR_DOWN_Z)
end