Package UDF - Создание собственных PKR архивов

#Include "Package.au3"

Global Const $sPackage = @ScriptDir & '\MyFile.pkr'

$hPackage = _PK_Create($sPackage, 'Demonstration Package', 1)
_PK_AddPacketFromFile($hPackage, @ScriptFullPath, @ScriptName, 1)
_PK_DisplayPackage($hPackage)
_PK_Close($hPackage)

#Include "Package.au3"

Global Const $sPackage = @ScriptDir & '\MyFile.pkr'

$hPackage = _PK_Open($sPackage)
_PK_ExtractPacketDataToFile($hPackage, -1, StringRegExpReplace(_PK_GetPacketDescription($hPackage, -1), '\.[^.]*\Z', '.bak'), 1)
_PK_Close($hPackage)

#Include "Package.au3"

Global Const $sPackage = @ScriptDir & '\MyData.pkr'

$hPackage = _PK_Create($sPackage, 'Demonstration Package', 1)
$tData = DllStructCreate('byte[16]')
DllStructSetData($tData, 1, '0x00112233445566778899AABBCCDDEEFF')
_PK_AddPacket($hPackage, DllStructGetPtr($tData), DllStructGetSize($tData), 'Binary data', 1)
_PK_DisplayPackage($hPackage)
_PK_Close($hPackage)

#Include "Package.au3"

Global Const $sPackage = @ScriptDir & '\MyData.pkr'

$hPackage = _PK_Open($sPackage)
$tData = DllStructCreate('byte[' & _PK_GetPacketDataLength($hPackage, -1) & ']')
_PK_ExtractPacketData($hPackage, -1, DllStructGetPtr($tData), DllStructGetSize($tData))
ConsoleWrite(DllStructGetData($tData, 1) & @CR)
_PK_Close($hPackage)

#Include <WinAPIEx.au3>

#Include "Package.au3"

Opt('MustDeclareVars', 1)

Global Const $sPackage = @ScriptDir & '\MyPackage.pkr'

Global $hPackage, $ID, $File

If Not FileExists($sPackage) Then
	$hPackage = _PK_Create($sPackage, 'Demonstration Package')
Else
	$hPackage = _PK_Open($sPackage)
EndIf

Do
	$File = FileOpenDialog('Add File', @WorkingDir, 'All Files (*.*)', 3)
	If @Error Then
		ExitLoop
	EndIf
	$ID = _PK_AddPacketFromFile($hPackage, $File, _WinAPI_PathStripPath($File))
	If @Error Then
		$ID = MsgBox(20, 'Package', 'An error occurred while adding file to the package.' & @CR & @CR & 'Error: ' & @Error & @CR & @CR & 'Do you want to add another file?')
	Else
		$ID = MsgBox(68, 'Package', 'File added successfully.' & @CR & @CR & 'ID: ' & $ID & @CR & @CR & 'Do you want to add another file?')
	EndIf
Until $ID <> 6

_PK_DisplayPackage($hPackage)

_PK_Close($hPackage)

#Include <Array.au3>
#Include <WinAPIEx.au3>

#Include "Package.au3"

Opt('MustDeclareVars', 1)

Global Const $sPackage = @ScriptDir & '\MyPackage.pkr'

Global $hPackage, $ID, $Data, $File, $Path

$hPackage = _PK_Open($sPackage)
If @Error Then
	MsgBox(16, 'Package', _WinAPI_PathStripPath($sPackage) & ' not found.')
	Exit
EndIf

$Data = _PK_EnumPackets($hPackage)
If @Error Then
	MsgBox(16, 'Package', 'Unable to enumerate packets in the package.' & @CR & @CR & 'Error: ' & @Error)
Else
	For $i = 1 To $Data[0][0]
		ConsoleWrite($i & ' - ' & $Data[$i][1] & @CR)
	Next
	While 1
		$ID = InputBox('Package', 'Type a one based index of the packet to extract.', '1', '', 274, 142)
		If @Error Then
			ExitLoop
		EndIf
		$File = _PK_GetPacketDescription($hPackage, -$ID)
		If @Error Then
			MsgBox(16, 'Package', 'Packet not found.')
		Else
			$Path = _WinAPI_PathYetAnotherMakeUniqueName(@ScriptDir & '\' & $File)
			If Not _PK_ExtractPacketDataToFile($hPackage, -$ID, $Path) Then
				$ID = MsgBox(20, 'Package', 'An error occurred while extracting file from the package.' & @CR & @CR & 'Error: ' & @Error & @CR & @CR & 'Do you want to extract another file?')
			Else
				$ID = MsgBox(68, 'Package', 'File has been saved to ' & _WinAPI_PathStripPath($Path) & '.' & @CR & @CR & 'Do you want to extract another file?')
			EndIf
			If $ID <> 6 Then
				ExitLoop
			EndIf
		EndIf
	WEnd
EndIf

_PK_Close($hPackage)

#NoTrayIcon

#Include <APIConstants.au3>
#Include <GUIListView.au3>
#Include <GUIConstantsEx.au3>
#Include <ListViewConstants.au3>
#Include <StaticConstants.au3>
#Include <WinAPIEx.au3>

#Include "Package.au3"

Opt('MustDeclareVars', 1)
Opt('WinWaitDelay', 0)

#Region Initialization

Global $hForm, $hDrop, $hPackage, $hLV, $LV, $Button[3], $Dummy[2], $Label, $Area, $Item

#EndRegion Initialization

#Region Body

_CreateForm()

While 1
	Switch GUIGetMsg()
		Case 0
			ContinueLoop
		Case $GUI_EVENT_CLOSE
			ExitLoop
		Case $GUI_EVENT_DROPPED
			_AddPacket(@GUI_DRAGFILE)
		Case $Button[0]
			_AddPacket()
		Case $Button[1], $Dummy[1]
			_RemovePacket()
		Case $Button[2], $Dummy[0]
			_ExtractPacket()
		Case Else

	EndSwitch
WEnd

_DeleteForm()

#EndRegion Body

#Region Additional Functions

Func _AddPacket($sFile = '')

	Local $tInfo, $ID, $Data, $Flag, $Error

	If Not $sFile Then
		$sFile = FileOpenDialog('Add File', @WorkingDir, 'All Files (*.*)', 3, '', $hDrop)
		If @Error Then
			Return
		EndIf
	EndIf
	If FileGetSize($sFile) > 104857600 Then
		If MsgBox(52, 'Package', _WinAPI_PathStripPath($sFile) & ' is too large (over 100 MB) and can not be compressed on the fly.' & @CR & @CR & 'Do you want to add this file without compression?', 0, $hDrop) <> 6 Then
			Return
		EndIf
		$Flag = 0
	Else
		$Flag = 1
	EndIf
	Opt('GUIOnEventMode', 1)
	GUISetState(@SW_DISABLE, $hForm)
	GUISetCursor(15, 1, $hForm)
	$tInfo = _GetFileInfo($sFile)
	$ID = _PK_AddPacketFromFile($hPackage, $sFile, _WinAPI_PathStripPath($sFile), $Flag, DllStructGetPtr($tInfo), DllStructGetSize($tInfo), '_Progress')
	$Error = @Error
	GUISetCursor(2, 0, $hForm)
	GUISetState(@SW_ENABLE, $hForm)
	Opt('GUIOnEventMode', 0)
	If $Error Then
		MsgBox(16, 'Package', 'An error occurred while adding file to the package.' & @CR & @CR & 'Error: ' & $Error, 0, $hDrop)
		Return
	EndIf
	$Data = _PK_QueryPacket($hPackage, $ID)
	If @Error Then
		Return
	EndIf
	_GUICtrlListView_BeginUpdate($hLV)
	$ID = _GUICtrlListView_AddItem($hLV, $Data[0])
	For $i = 1 To 4
		_GUICtrlListView_AddSubItem($hLV, $ID, $Data[$i], $i)
	Next
	_GUICtrlListView_SetItemParam($hLV, $ID, $Data[5])
	_GUICtrlListView_SetItemSelected($hLV, $ID, 1, 1)
	_GUICtrlListView_EnsureVisible($hLV, $ID)
	_GUICtrlListView_EndUpdate($hLV)
	GUICtrlSetState($LV, $GUI_FOCUS)
	WinActivate($hDrop)
EndFunc   ;==>_AddPacket

Func _CreateForm()

	Local $ID, $Path, $File, $Data

	$Path = FileOpenDialog('Open Package', @ScriptDir, 'Package Files (*.pkr)|All Files (*.*)', 10, 'MyPackage.pkr')
	If @Error Then
		Exit
	EndIf
	$File = _WinAPI_PathStripPath($Path)
	If Not FileExists($Path) Then
		$hPackage = _PK_Create($Path, 'Demonstration Package')
	Else
		$hPackage = _PK_Open($Path)
	EndIf
	If @Error Then
		MsgBox(16, 'Package', 'Unable to open ' & $File & '.' & @CR & @CR & 'Error: ' & @Error)
		Exit
	EndIf
	OnAutoItExitRegister('_Quit')
	If _PK_GetCount($hPackage) Then
		$Data = _PK_EnumPackets($hPackage)
		If @Error Then
			MsgBox(16, 'Package', 'Unexpected error.' & @CR & @CR & 'Error: ' & @Error)
			Exit
		EndIf
	Else
		$Data = 0
	EndIf
	$hForm = GUICreate($File & ' - ' & _PK_GetComment($hPackage), 660, 440, -1, -1, BitOR($WS_CAPTION, $WS_MINIMIZEBOX, $WS_MAXIMIZEBOX, $WS_POPUP, $WS_SIZEBOX, $WS_SYSMENU))
	$hDrop = GUICreate('', 660, 398, 0, 0, BitOR($WS_CHILD, $WS_TABSTOP, $WS_VISIBLE), $WS_EX_ACCEPTFILES, $hForm)
	$LV = GUICtrlCreateListView('', 0, 0, 660, 398, BitOR($LVS_DEFAULT, $LVS_NOSORTHEADER), BitOR($LVS_EX_DOUBLEBUFFER, $LVS_EX_FULLROWSELECT, $LVS_EX_INFOTIP))
	GUICtrlSetFont(-1, 8.5, 400, 0, 'Tahoma')
	GUICtrlSetState(-1, $GUI_DROPACCEPTED)
	$hLV = GUICtrlGetHandle(-1)
	_GUICtrlListView_AddColumn($hLV, '#', 30)
	_GUICtrlListView_AddColumn($hLV, 'ID', 140)
	_GUICtrlListView_AddColumn($hLV, 'Description', 313)
	_GUICtrlListView_AddColumn($hLV, 'Info Length', 80, 1)
	_GUICtrlListView_AddColumn($hLV, 'Data Length', 80, 1)
	If _WinAPI_GetVersion() >= '6.0' Then
		_WinAPI_SetWindowTheme($hLV, 'Explorer')
	EndIf
	If IsArray($Data) Then
		For $i = 1 To $Data[0][0]
			$ID = _GUICtrlListView_AddItem($hLV, $i)
			For $j = 0 To 3
				_GUICtrlListView_AddSubItem($hLV, $ID, $Data[$i][$j], $j + 1)
			Next
			_GUICtrlListView_SetItemParam($hLV, $ID, $Data[$i][4])
		Next
		_GUICtrlListView_SetItemSelected($hLV, 0, 1, 1)
		$Item = 0
	Else
		$Item =-1
	EndIf
	GUISwitch($hForm)
	GUISetIcon(@ScriptDir & '\Package.ico')
	$Label = GUICtrlCreateLabel('', 0, 398, 663, 2, $SS_ETCHEDHORZ)
	$Button[0] = GUICtrlCreateButton('Add...', 5, 405, 214, 30)
	$Button[1] = GUICtrlCreateButton('Remove', 223, 405, 214, 30)
	$Button[2] = GUICtrlCreateButton('Extract...', 441, 405, 214, 30)
	For $i = 0 To 1
		$Dummy[$i] = GUICtrlCreateDummy()
	Next
	$Area = WinGetPos($hForm)
	GUIRegisterMsg($WM_GETMINMAXINFO, 'WM_GETMINMAXINFO')
	GUIRegisterMsg($WM_NOTIFY, 'WM_NOTIFY')
	GUIRegisterMsg($WM_SIZE, 'WM_SIZE')
	GUICtrlSetState($LV, $GUI_FOCUS)
	GUISetState()
EndFunc   ;==>_CreateForm

Func _DeleteForm()
	GUIDelete($hForm)
	GUIDelete($hDrop)
EndFunc   ;==>_DeleteForm

Func _ExtractPacket()

	Local $tInfo, $File, $Error

	$File = FileSaveDialog('Extract File', @WorkingDir, 'All Files (*.*)', 18, _PK_GetPacketDescription($hPackage, -($Item + 1)), $hDrop)
	If @Error Then
		Return
	EndIf
	Opt('GUIOnEventMode', 1)
	GUISetState(@SW_DISABLE, $hForm)
	GUISetCursor(15, 1, $hForm)
	_PK_ExtractPacketDataToFile($hPackage, -($Item + 1), $File, 1, '_Progress')
	$Error = @Error
	If $Error Then

	Else
		$tInfo = _PK_GetPacketInfo($hPackage, -($Item + 1))
		If Not _SetFileInfo($File, $tInfo) Then
			; Nothing
		EndIf
	EndIf
	GUISetCursor(2, 0, $hForm)
	GUISetState(@SW_ENABLE, $hForm)
	Opt('GUIOnEventMode', 0)
	If $Error Then
		MsgBox(16, 'Package', 'An error occurred while adding file to the package.' & @CR & @CR & 'Error: ' & $Error, 0, $hDrop)
	EndIf
EndFunc   ;==>_ExtractPacket

Func _GetFileInfo($sFile)

	Local $hFile, $tInfo

	$hFile = _WinAPI_CreateFileEx($sFile, $OPEN_EXISTING, $GENERIC_READ, BitOR($FILE_SHARE_READ, $FILE_SHARE_WRITE))
	If @Error Then
		Return 0
	EndIf
	$tInfo = _WinAPI_GetFileInformationByHandleEx($hFile)
	_WinAPI_CloseHandle($hFile)
	Return $tInfo
EndFunc   ;==>_GetFileInfo

Func _Quit()
	_PK_Close($hPackage)
EndFunc   ;==>_Quit

Func _RemovePacket()

	Local $Count, $Error, $Sel = $Item

	If MsgBox(36, 'Package', 'Are you sure you want to remove ' & _PK_GetPacketDescription($hPackage, -($Item + 1)) & ' from the package?', 0, $hDrop) <> 6 Then
		Return
	EndIf
	Opt('GUIOnEventMode', 1)
	GUISetState(@SW_DISABLE, $hForm)
	GUISetCursor(15, 1, $hForm)
	_PK_RemovePacket($hPackage, -($Item + 1))
	$Error = @Error
	GUISetCursor(2, 0, $hForm)
	GUISetState(@SW_ENABLE, $hForm)
	Opt('GUIOnEventMode', 0)
	If $Error Then
		MsgBox(16, 'Package', 'An error occurred while remove file from the package.' & @CR & @CR & 'Error: ' & $Error, 0, $hDrop)
		Return
	EndIf
	_GUICtrlListView_BeginUpdate($hLV)
	_GUICtrlListView_DeleteItem($hLV, $Item)
	$Count = _GUICtrlListView_GetItemCount($hLV)
	If $Count Then
		If $Sel > $Count - 1 Then
			$Sel -= 1
		Else
			For $i = $Sel To $Count - 1
				_GUICtrlListView_SetItemText($hLV, $i, $i + 1)
			Next
		EndIf
		_GUICtrlListView_SetItemSelected($hLV, $Sel, 1, 1)
	EndIf
	_GUICtrlListView_EndUpdate($hLV)
	GUICtrlSetState($LV, $GUI_FOCUS)
	WinActivate($hDrop)
EndFunc   ;==>_RemovePacket

Func _SetFileInfo($sFile, $tInfo)

	Local $hFile

	If DllStructGetSize($tInfo) < 40 Then
		Return 0
	EndIf
	$hFile = _WinAPI_CreateFileEx($sFile, $OPEN_EXISTING, $GENERIC_WRITE, BitOR($FILE_SHARE_READ, $FILE_SHARE_WRITE))
	If @Error Then
		Return 0
	EndIf
	_WinAPI_SetFileInformationByHandleEx($hFile, $tInfo)
	_WinAPI_CloseHandle($hFile)
	Return 1
EndFunc   ;==>_SetFileInfo

#EndRegion Additional Functions

#Region Callback Functions

Func _Progress($hPackage, $ID, $iTotalSize, $iTotalBytesTransferred, $vData)

	#forceref $hPackage, $ID, $vData

	Static $hDlg = Ptr(0), $Progress

	Local $Pos, $Percent

	If $iTotalSize <> $iTotalBytesTransferred Then
		$Percent = Round($iTotalBytesTransferred / $iTotalSize * 100)
	Else
		$Percent = 100
	EndIf
	If Not $hDlg Then
		If $Percent < 100 Then
			$Pos = WinGetPos($hDrop)
			$hDlg = GUICreate('', 300, 44, $Pos[0] + ($Pos[2] - 300) / 2, $Pos[1] + ($Pos[3] - 44) / 2, BitOR($WS_BORDER, $WS_DISABLED, $WS_POPUP), 0, $hForm)
			$Progress = GUICtrlCreateProgress(15, 15, 270, 14)
			GUICtrlSetData(-1, $Percent)
			GUISetCursor(15, 1)
			GUISetState()
		EndIf
	Else
		GUICtrlSetData($Progress, $Percent)
		If $Percent = 100 Then
			GUISetState(@SW_ENABLE, $hForm)
			GUIDelete($hDlg)
			$hDlg = 0
		EndIf
	EndIf
	Return $PK_PROGRESS_CONTINUE
EndFunc   ;==>_Progress

#EndRegion Callback Functions

#Region Windows Message Functions

Func WM_GETMINMAXINFO($hWnd, $iMsg, $wParam, $lParam)

	Local $tMMI = DllStructCreate('long Reserved[2];long MaxSize[2];long MaxPosition[2];long MinTrackSize[2];long MaxTrackSize[2]', $lParam)

	Switch $hWnd
		Case $hForm
			If IsArray($Area) Then
				DllStructSetData($tMMI, 'MinTrackSize', $Area[2], 1)
				DllStructSetData($tMMI, 'MinTrackSize', $Area[3], 2)
			EndIf
			Return 0
	EndSwitch
	Return $GUI_RUNDEFMSG
EndFunc   ;==>WM_GETMINMAXINFO

Func WM_NOTIFY($hWnd, $iMsg, $wParam, $lParam)

	Static $Flag = False

	If @AutoItX64 Then
		Local $tNMLV = DllStructCreate($tagNMHDR & ';uint Aligment;int Item;int SubItem;uint NewState;uint OldState;uint Changed;long ActionX;long ActionY;lparam Param', $lParam)
	Else
		Local $tNMLV = DllStructCreate($tagNMHDR & ';int Item;int SubItem;uint NewState;uint OldState;uint Changed;long ActionX;long ActionY;lparam Param', $lParam)
	EndIf

	Local $hTarget = DllStructGetData($tNMLV, 'hWndFrom')
	Local $ID = DllStructGetData($tNMLV, 'Code')

	Switch $hTarget
		Case $hLV
			Switch $ID
				Case $LVN_BEGINDRAG
					Return 0
				Case $LVN_ITEMACTIVATE
					If $Item <> -1 Then
						GUICtrlSendToDummy($Dummy[0])
					EndIf
				Case $LVN_ITEMCHANGED
					If (BitAND(DllStructGetData($tNMLV, 'Changed'), $LVIF_STATE)) And (BitXOR(DllStructGetData($tNMLV, 'NewState'), DllStructGetData($tNMLV, 'OldState'))) Then
						If BitAND(DllStructGetData($tNMLV, 'NewState'), $LVIS_SELECTED) Then
							$Item = DllStructGetData($tNMLV, 'Item')
							For $i = 1 To 2
								GUICtrlSetState($Button[$i], $GUI_ENABLE)
							Next
							$Flag = 0
						Else
							If BitAND(DllStructGetData($tNMLV, 'OldState'), $LVIS_FOCUSED) Then
								$Flag = 1
							Else
								If Not $Flag Then
									For $i = 1 To 2
										GUICtrlSetState($Button[$i], $GUI_DISABLE)
									Next
									$Item = -1
								EndIf
								$Flag = 0
							EndIf
						EndIf
					EndIf
				Case $NM_CUSTOMDRAW

					If @AutoItX64 Then
						Local $tNMLVCD = DllStructCreate($tagNMHDR & ';uint Aligment1;dword DrawStage;hwnd hDC;long Left;long Top;long Right;long Bottom;dword_ptr ItemSpec;uint ItemState;lparam ItemParam;uint Aligment2[5];dword clrText;dword clrTextBk;int SubItem;dword ItemType;dword clrFace;int IconEffect;int IconPhase;int PartId;int StateId;long TextLeft;long TextTop;long TextRight;long TextBottom;uint Align', $lParam)
					Else
						Local $tNMLVCD = DllStructCreate($tagNMHDR & ';dword DrawStage;hwnd hDC;long Left;long Top;long Right;long Bottom;dword_ptr ItemSpec;uint ItemState;lparam ItemParam;dword clrText;dword clrTextBk;int SubItem;dword ItemType;dword clrFace;int IconEffect;int IconPhase;int PartId;int StateId;long TextLeft;long TextTop;long TextRight;long TextBottom;uint Align', $lParam)
					EndIf

					Local $Stage = DllStructGetData($tNMLVCD, 'DrawStage')

					Switch $Stage
						Case $CDDS_ITEMPREPAINT
							Return $CDRF_NOTIFYSUBITEMDRAW
						Case BitOR($CDDS_ITEMPREPAINT, $CDDS_SUBITEM)
							If _GUICtrlListView_GetItemParam($hLV, DllStructGetData($tNMLVCD, 'ItemSpec')) Then
								DllStructSetData($tNMLVCD, 'clrText', 0xFF0000)
							EndIf
							Return $CDRF_DODEFAULT
						Case Else

					EndSwitch
				Case $LVN_KEYDOWN

					If @AutoItX64 Then
						Local $tNMLVKD = DllStructCreate($tagNMHDR & ';uint Aligment;ushort VKey;uint Flags', $lParam)
					Else
						Local $tNMLVKD = DllStructCreate($tagNMHDR & ';ushort VKey;uint Flags', $lParam)
					EndIf

					Local $Key = DllStructGetData($tNMLVKD, 'VKey')

					Switch BitAND($Key, 0xFF)
						Case 0x2E ; VK_DELETE
							If $Item <> -1 Then
								GUICtrlSendToDummy($Dummy[1])
							EndIf
						Case Else

					EndSwitch
				Case $NM_CLICK, $NM_DBLCLK, $NM_RCLICK, $NM_RDBLCLK
					Return 0
			EndSwitch
	EndSwitch
	Return $GUI_RUNDEFMSG
EndFunc   ;==>WM_NOTIFY

Func WM_SIZE($hWnd, $iMsg, $wParam, $lParam)

	Local $WC, $HC, $WT

	Switch $hWnd
		Case $hForm
			$WC = _WinAPI_LoWord($lParam)
			$HC = _WinAPI_HiWord($lParam)
			$WT = Floor(($WC - 18) / 3)
			WinMove($hDrop, '', 0, 0, $WC, $HC - 42)
			GUICtrlSetPos($LV, 0, 0, $WC, $HC - 42)
			GUICtrlSetPos($Label, 0, $HC - 42, $WC + 3)
			GUICtrlSetPos($Button[0], 5, $HC - 35, $WT)
			GUICtrlSetPos($Button[1], $WT + 9, $HC - 35, $WC - 2 * $WT - 18)
			GUICtrlSetPos($Button[2], $WC - $WT - 5, $HC - 35, $WT)
			Return 0
	EndSwitch
	Return $GUI_RUNDEFMSG
EndFunc   ;==>WM_SIZE

#EndRegion Windows Message Functions