import bpy
import os
import struct
from . import common

class export_cm3d2_tex(bpy.types.Operator):
	bl_idname = 'image.export_cm3d2_tex'
	bl_label = "Save As .tex"
	bl_description = "Current image will be saved as a (.tex)."
	bl_options = {'REGISTER'}
	
	filepath = bpy.props.StringProperty(subtype='FILE_PATH')
	filename_ext = ".tex"
	filter_glob = bpy.props.StringProperty(default="*.tex", options={'HIDDEN'})
	
	is_backup = bpy.props.BoolProperty(name="Backup", default=True, description="Overwritten files will be backed up.")
	
	version = bpy.props.IntProperty(name="Version", default=1000, min=1000, max=1111, soft_min=1000, soft_max=1111, step=1)
	path = bpy.props.StringProperty(name="Path", default="assets/texture/texture/*.png")
	
	@classmethod
	def poll(cls, context):
		img = context.edit_image
		if img:
			if len(img.pixels) or img.source == 'VIEWER':
				return True
		return False
	
	def invoke(self, context, event):
		img = context.edit_image
		if img.filepath:
			common.preferences().tex_export_path = bpy.path.abspath(img.filepath)
		if common.preferences().tex_default_path:
			self.filepath = common.default_cm3d2_dir(common.preferences().tex_default_path, common.remove_serial_number(img.name), "tex")
		else:
			self.filepath = common.default_cm3d2_dir(common.preferences().tex_export_path, common.remove_serial_number(img.name), "tex")
		self.is_backup = bool(common.preferences().backup_ext)
		if 'cm3d2_path' in img:
			self.path = img['cm3d2_path']
		else:
			self.path = "assets/texture/texture/" + os.path.basename(self.filepath)
		if 'tex Name' in img:
			self.filepath = os.path.join(os.path.dirname(self.filepath), img['tex Name'])
		context.window_manager.fileselect_add(self)
		return {'RUNNING_MODAL'}
	
	def draw(self, context):
		row = self.layout.row()
		row.prop(self, 'is_backup', icon='FILE_BACKUP')
		if not common.preferences().backup_ext:
			row.enabled = False
		self.layout.prop(self, 'version', icon='LINENUMBERS_ON')
		self.layout.prop(self, 'path', icon='ANIM')
	
	def execute(self, context):
		common.preferences().tex_export_path = self.filepath
		
		try:
			file = common.open_temporary(self.filepath, 'wb', is_backup=self.is_backup)
		except:
			self.report(type={'ERROR'}, message="Failed to save file, possibly inaccessible.")
			return {'CANCELLED'}
		
		try:
			with file:
				self.write_texture(context, file)
		except common.CM3D2ExportException as e:
			self.report(type={'ERROR'}, message=str(e))
			return {'CANCELLED'}
		
		return {'FINISHED'}

	def write_texture(self, context, file):
		# とりあえずpngで保存
		img = context.edit_image
		if img.source != 'VIEWER':
			temp_path = self.filepath + ".temp.png"
		else:
			temp_path = os.path.splitext(self.filepath)[0] + ".png"
		pre_filepath = bpy.path.abspath(img.filepath)
		pre_source = img.source
		override = context.copy()
		override['edit_image'] = img
		try:
			save_as_render = True if pre_source == 'VIEWER' else False
			copy = True if pre_source == 'VIEWER' else False
			bpy.ops.image.save_as(override, save_as_render=save_as_render, copy=copy, filepath=temp_path, relative_path=True, show_multiview=False, use_multiview=False)
			is_remove = True
		except:
			if os.path.exists( bpy.path.abspath(img.filepath) ):
				temp_path = bpy.path.abspath(img.filepath)
				is_remove = False
			else:
				raise common.CM3D2ExportException("Could not export .tex file.")
		if pre_source != 'VIEWER':
			img.filepath = pre_filepath
			img.source = pre_source
		
		# pngバイナリを全て読み込み
		with open(temp_path, 'rb') as temp_file:
			temp_data = temp_file.read()
		# 一時ファイルを削除
		if is_remove:
			os.remove(temp_path)
		
		# 本命ファイルに書き込み
		common.write_str(file, 'CM3D2_TEX')
		file.write(struct.pack('<i', self.version))
		common.write_str(file, self.path)
		file.write(struct.pack('<i', len(temp_data)))
		file.write(temp_data)

# メニューを登録する関数
def menu_func(self, context):
	self.layout.operator(export_cm3d2_tex.bl_idname, icon_value=common.preview_collections['main']['KISS'].icon_id)