From bc16aae3b4ea47ed3ab261abc087049f7801a471 Mon Sep 17 00:00:00 2001
From: Lilfade <lilfade.lor@gmail.com>
Date: Wed, 24 Apr 2024 01:30:00 -0500
Subject: [PATCH] update files

---
 error.py    |   6 --
 header.py   | 120 --------------------------------------
 main.py     |  18 +++++-
 tools.py    |  36 ------------
 unpacker.py | 163 +++++++++++++++++++++++++++++++++++++++++++++++++---
 5 files changed, 170 insertions(+), 173 deletions(-)
 delete mode 100644 error.py
 delete mode 100644 header.py
 delete mode 100644 tools.py

diff --git a/error.py b/error.py
deleted file mode 100644
index 0cc3cfe..0000000
--- a/error.py
+++ /dev/null
@@ -1,6 +0,0 @@
-""" Package specific exceptions. """
-
-
-class Ai1wmError(Exception):
-    """ Exceptions raised from this package. """
-    pass
diff --git a/header.py b/header.py
deleted file mode 100644
index 6153e64..0000000
--- a/header.py
+++ /dev/null
@@ -1,120 +0,0 @@
-""" Parses ai1wm file header. """
-
-import collections
-import struct
-from error import Ai1wmError
-from tools import b__, s__
-
-
-class Ai1wmHeader(tuple):
-    """ Parses an `All-in-One WP Migration` header. """
-
-    SIZE = 4377
-    EOF = b'\x00' * SIZE
-
-    _Location = collections.namedtuple('_Location', ['offset', 'size'])
-    _LOC_NAME = _Location(0, 255)       # File name
-    _LOC_SIZE = _Location(255, 14)      # File size
-    _LOC_TIME = _Location(269, 12)      # Last modified time
-    _LOC_PATH = _Location(281, 4096)    # File path
-
-    def __new__(cls, path=None, name=None, size=None, time=None):
-        """ Returns a new instance of the object. """
-
-        if path or name or size or time:
-            if not isinstance(path, str) or path == '':
-                raise ValueError('<path> must be a nonempty string')
-            if not isinstance(name, str) or name == '':
-                raise ValueError('<name> must be a nonempty string')
-            if not isinstance(size, int) or size < 0:
-                raise ValueError('<size> must be a non-negative integer')
-            if not isinstance(time, int) or time < 0:
-                raise ValueError('<time> must be a non-negative integer')
-
-        return super(Ai1wmHeader, cls).__new__(cls, [path, name, size, time])
-
-    @classmethod
-    def unpack(cls, header):
-        """ Unpacks a binary header. """
-
-        if len(header) != cls.SIZE:
-            raise Ai1wmError('invalid header size')
-
-        if header == cls.EOF:
-            return cls()
-
-        return cls(
-            path=s__(cls.__extract_field(header, cls._LOC_PATH)),
-            name=s__(cls.__extract_field(header, cls._LOC_NAME)),
-            size=cls.__extract_int(header, cls._LOC_SIZE),
-            time=cls.__extract_int(header, cls._LOC_TIME),
-        )
-
-    def pack(self):
-        """ Packs to a binary header. """
-
-        attributes, formats, locations = [], '', [
-            ('name', self._LOC_NAME),
-            ('size', self._LOC_SIZE),
-            ('time', self._LOC_TIME),
-            ('path', self._LOC_PATH),
-        ]
-
-        for name, location in locations:
-            attribute = b__(getattr(self, name))
-            if len(attribute) > location.size:
-                raise Ai1wmError('{} is too long to pack: {}'.format(name, getattr(self, name)))
-            attributes.append(attribute)
-            formats += '{}s'.format(location.size)
-
-        return struct.pack(formats, *attributes)
-
-    @property
-    def path(self):
-        """ Path of the file. """
-
-        return self[0]
-
-    @property
-    def name(self):
-        """ Name of the file. """
-
-        return self[1]
-
-    @property
-    def size(self):
-        """ Size of the file. """
-
-        return self[2]
-
-    @property
-    def time(self):
-        """ Time of the file. """
-
-        return self[3]
-
-    @property
-    def eof(self):
-        """ Indicates if this is an EOF header. """
-
-        return not any(self)
-
-    @classmethod
-    def __extract_field(cls, header, location):
-        """ Extracts a header field. """
-
-        try:
-            field = struct.unpack_from('{}s'.format(location.size), header, offset=location.offset)[0]
-        except struct.error as e:
-            raise Ai1wmError('error extracting a header field, error: {}'.format(e))
-
-        return field.rstrip(b'\x00')
-
-    @classmethod
-    def __extract_int(cls, header, location):
-        """ Extracts an integral header field. """
-
-        try:
-            return int(cls.__extract_field(header, location))
-        except ValueError:
-            raise Ai1wmError('invalid header field')
diff --git a/main.py b/main.py
index a3074f6..f0dc3db 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,6 @@ import argparse
 import os
 import sys
 import unpacker
-from error import Ai1wmError
 
 '''
 obtain new info for sites database
@@ -10,6 +9,21 @@ install core files
 move wp-content to correct place
 import db to website
 check for any issues
+
+package.json
+
+function ai1wm_activate_plugins( $plugins ) {
+	$current = get_option( AI1WM_ACTIVE_PLUGINS, array() );
+
+	// Add plugins
+	foreach ( $plugins as $plugin ) {
+		if ( ! in_array( $plugin, $current ) && ! is_wp_error( validate_plugin( $plugin ) ) ) {
+			$current[] = $plugin;
+		}
+	}
+
+	return update_option( AI1WM_ACTIVE_PLUGINS, $current );
+}
 '''
 
 if __name__ == '__main__':
@@ -23,7 +37,7 @@ if __name__ == '__main__':
     try:
         if os.path.isfile(args.source):
             unpacker.Ai1wmUnpacker().unpack(args.source, args.target)
-    except Ai1wmError as e:
+    except Exception as e:
         print(e)
         sys.exit(-1)
 
diff --git a/tools.py b/tools.py
deleted file mode 100644
index 067dfca..0000000
--- a/tools.py
+++ /dev/null
@@ -1,36 +0,0 @@
-import sys
-
-
-def s__(obj):
-    """
-    Converts an object to str format.
-    :rtype: str
-    """
-
-    if isinstance(obj, str):
-        return obj
-    v = sys.version_info[0]
-    if v == 2:
-        if isinstance(obj, unicode):
-            return obj.encode('utf-8')
-    elif v == 3:
-        if isinstance(obj, bytes):
-            return obj.decode('utf-8')
-    return str(obj)
-
-
-def b__(obj):
-    """
-    Converts an object to bytes format.
-    :rtype: bytes
-    """
-
-    v = sys.version_info[0]
-    if v == 2:
-        return s__(obj)
-
-    if isinstance(obj, bytes):
-        return obj
-    if not isinstance(obj, str):
-        obj = str(obj)
-    return obj.encode('utf-8')
diff --git a/unpacker.py b/unpacker.py
index 47e65c0..c29e1e6 100644
--- a/unpacker.py
+++ b/unpacker.py
@@ -1,14 +1,159 @@
 """ Unpacks an `All-in-One WP Migration` package. """
 
+import collections
 import errno
 import os
-from error import Ai1wmError
-from header import Ai1wmHeader
-from tools import s__
+import struct
+import sys
 
 
-class Ai1wmUnpacker(object):
+class Ai1wmUnpacker(tuple):
     """ Unpacks an `All-in-One WP Migration` package. """
+    SIZE = 4377
+    EOF = b'\x00' * SIZE
+
+    _Location = collections.namedtuple('_Location', ['offset', 'size'])
+    _LOC_NAME = _Location(0, 255)       # File name
+    _LOC_SIZE = _Location(255, 14)      # File size
+    _LOC_TIME = _Location(269, 12)      # Last modified time
+    _LOC_PATH = _Location(281, 4096)    # File path
+
+    def __new__(cls, path=None, name=None, size=None, time=None):
+        """ Returns a new instance of the object. """
+
+        if path or name or size or time:
+            if not isinstance(path, str) or path == '':
+                raise ValueError('<path> must be a nonempty string')
+            if not isinstance(name, str) or name == '':
+                raise ValueError('<name> must be a nonempty string')
+            if not isinstance(size, int) or size < 0:
+                raise ValueError('<size> must be a non-negative integer')
+            if not isinstance(time, int) or time < 0:
+                raise ValueError('<time> must be a non-negative integer')
+
+        return super(Ai1wmUnpacker, cls).__new__(cls, [path, name, size, time])
+
+    @classmethod
+    def unpack_header(cls, header):
+        """ Unpacks a binary header. """
+
+        if len(header) != cls.SIZE:
+            raise Exception('invalid header size')
+
+        if header == cls.EOF:
+            return cls()
+
+        return cls(
+            path=cls.s__(cls.__extract_field(header, cls._LOC_PATH)),
+            name=cls.s__(cls.__extract_field(header, cls._LOC_NAME)),
+            size=cls.__extract_int(header, cls._LOC_SIZE),
+            time=cls.__extract_int(header, cls._LOC_TIME),
+        )
+    
+    @classmethod
+    def s__(cls, obj):
+        """
+        Converts an object to str format.
+        :rtype: str
+        """
+
+        if isinstance(obj, str):
+            return obj
+        v = sys.version_info[0]
+        if v == 2:
+            if isinstance(obj, unicode):
+                return obj.encode('utf-8')
+        elif v == 3:
+            if isinstance(obj, bytes):
+                return obj.decode('utf-8')
+        return str(obj)
+
+    @classmethod
+    def b__(cls, obj):
+        """
+        Converts an object to bytes format.
+        :rtype: bytes
+        """
+
+        v = sys.version_info[0]
+        if v == 2:
+            return cls.s__(obj)
+
+        if isinstance(obj, bytes):
+            return obj
+        if not isinstance(obj, str):
+            obj = str(obj)
+        return obj.encode('utf-8')
+
+    def pack(self):
+        """ Packs to a binary header. """
+
+        attributes, formats, locations = [], '', [
+            ('name', self._LOC_NAME),
+            ('size', self._LOC_SIZE),
+            ('time', self._LOC_TIME),
+            ('path', self._LOC_PATH),
+        ]
+
+        for name, location in locations:
+            attribute = self.b__(getattr(self, name))
+            if len(attribute) > location.size:
+                raise Exception('{} is too long to pack: {}'.format(name, getattr(self, name)))
+            attributes.append(attribute)
+            formats += '{}s'.format(location.size)
+
+        return struct.pack(formats, *attributes)
+
+    @property
+    def path(self):
+        """ Path of the file. """
+
+        return self[0]
+
+    @property
+    def name(self):
+        """ Name of the file. """
+
+        return self[1]
+
+    @property
+    def size(self):
+        """ Size of the file. """
+
+        return self[2]
+
+    @property
+    def time(self):
+        """ Time of the file. """
+
+        return self[3]
+
+    @property
+    def eof(self):
+        """ Indicates if this is an EOF header. """
+
+        return not any(self)
+
+    @classmethod
+    def __extract_field(cls, header, location):
+        """ Extracts a header field. """
+
+        try:
+            field = struct.unpack_from('{}s'.format(location.size), header, offset=location.offset)[0]
+        except struct.error as e:
+            raise Exception('error extracting a header field, error: {}'.format(e))
+
+        return field.rstrip(b'\x00')
+
+    @classmethod
+    def __extract_int(cls, header, location):
+        """ Extracts an integral header field. """
+
+        try:
+            return int(cls.__extract_field(header, location))
+        except ValueError:
+            raise Exception('invalid header field')
+
 
     @staticmethod
     def __make_dirs(path, mode=0o777):
@@ -18,7 +163,7 @@ class Ai1wmUnpacker(object):
             os.makedirs(path, mode=mode)
         except OSError as e:
             if e.errno != errno.EEXIST:
-                raise Ai1wmError('error creating a directory: {}, error: {}'.format(path, e))
+                raise Exception('error creating a directory: {}, error: {}'.format(path, e))
         return path
 
     @classmethod
@@ -33,7 +178,7 @@ class Ai1wmUnpacker(object):
                     block_size = size
                 block = stream.read(block_size)
                 if len(block) != block_size:
-                    raise Ai1wmError('error extracting a file: {}, error: bad file size'.format(path))
+                    raise Exception('error extracting a file: {}, error: bad file size'.format(path))
                 f.write(block)
                 size -= len(block)
 
@@ -42,7 +187,7 @@ class Ai1wmUnpacker(object):
         """ Unpacks a package. """
 
         while True:
-            header = Ai1wmHeader.unpack(stream.read(Ai1wmHeader.SIZE))
+            header = cls.unpack_header(stream.read(Ai1wmUnpacker.SIZE))
             if header.eof:
                 break
 
@@ -56,10 +201,10 @@ class Ai1wmUnpacker(object):
     def unpack(cls, source, target):
         """ Unpacks a package. """
 
-        source, target = s__(os.path.realpath(source)), s__(os.path.realpath(target))
+        source, target = cls.s__(os.path.realpath(source)), cls.s__(os.path.realpath(target))
         try:
             with open(source, 'rb') as f:
                 cls.__unpack(f, target)
         except Exception as ex:
-            raise Ai1wmError('error unpacking a file: {}, error: {}'.format(source, ex))
+            raise Exception('error unpacking a file: {}, error: {}'.format(source, ex))
         return target