From e9ba117fa975fcbe449aefe6b19f6ceb5c28c9b5 Mon Sep 17 00:00:00 2001
From: Colin Basnett <cmbasnett@gmail.com>
Date: Sat, 20 Jan 2024 14:48:18 -0800
Subject: [PATCH] Added file handler for PSK/PSKX files

---
 io_scene_psk_psa/psk/import_/operators.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/io_scene_psk_psa/psk/import_/operators.py b/io_scene_psk_psa/psk/import_/operators.py
index 1cabb22..75141d7 100644
--- a/io_scene_psk_psa/psk/import_/operators.py
+++ b/io_scene_psk_psa/psk/import_/operators.py
@@ -2,7 +2,7 @@ import os
 import sys
 
 from bpy.props import StringProperty, BoolProperty, EnumProperty, FloatProperty
-from bpy.types import Operator
+from bpy.types import Operator, FileHandler, Context
 from bpy_extras.io_utils import ImportHelper
 
 from ..importer import PskImportOptions, import_psk
@@ -11,6 +11,17 @@ from ..reader import read_psk
 empty_set = set()
 
 
+class PSX_FH_psk(FileHandler):
+    bl_idname = 'PSX_FH_psk'
+    bl_label = 'Unreal PSK/PSKX'
+    bl_import_operator = 'import_scene.psk'
+    bl_file_extensions = '.psk;.pskx'
+
+    @classmethod
+    def poll_drop(cls, context: Context):
+        return context.area.type == 'VIEW_3D'
+
+
 class PSK_OT_import(Operator, ImportHelper):
     bl_idname = 'import_scene.psk'
     bl_label = 'Import'
@@ -141,4 +152,5 @@ class PSK_OT_import(Operator, ImportHelper):
 
 classes = (
     PSK_OT_import,
+    PSX_FH_psk,
 )