import dcc.config
import os

class List(dcc.config.Base):
	def get_parser(self, prog_name):
		parser = super().get_parser(prog_name)
		parser.add_argument("target")
		parser.add_argument("--wad")
		return parser

	def take_action(self, parsed_args):
		match parsed_args.target:
			case "pwads":
				self.list(x.name for x in os.scandir(self.pwads) if x.is_dir())
			case "iwads":
				self.list(x.name for x in os.scandir(self.iwads) if x.is_file())
			case "demos":
				self.list(x.name for x in os.scandir(self.demos.joinpath(parsed_args.wad)) if x.name.endswith(".lmp"))
			case "videos":
				self.list(x.name for x in os.scandir(self.fabricate.joinpath(parsed_args.wad)) if x.name.endswith(".mp4"))
			case _:
				raise Exception(f"unknown target {parsed_args.target}")

	def list(self, gen):
		# TODO: fancy text?
		for i in sorted(gen):
			print(i)