@codesys/mcp-toolkit 1.5.7 → 2.0.0-alpha.0

package/dist/scripts/stop_session.py

@@ -1,324 +0,0 @@
-import sys, os, time, traceback
-import tempfile
-
-# Import JSON handling for IronPython 2.7 compatibility
-try:
-    import json
-except ImportError:
-    # For IronPython 2.7, we need to use simplejson instead
-    try:
-        import simplejson as json
-    except ImportError:
-        print("Neither json nor simplejson could be imported. Check IronPython installation.")
-        # Provide a minimal fallback implementation for basic needs
-        class DummyJson:
-            @staticmethod
-            def load(file_obj):
-                content = file_obj.read()
-                # Very basic parsing for our limited needs
-                if content.strip().startswith("{") and content.strip().endswith("}"):
-                    result = {}
-                    parts = content.strip()[1:-1].split(",")
-                    for part in parts:
-                        if ":" in part:
-                            k, v = part.split(":", 1)
-                            k = k.strip().strip('"\'')
-                            v = v.strip().strip('"\'')
-                            result[k] = v
-                    return result
-                return {}
-                
-            @staticmethod
-            def loads(s):
-                if s.strip().startswith("{") and s.strip().endswith("}"):
-                    result = {}
-                    parts = s.strip()[1:-1].split(",")
-                    for part in parts:
-                        if ":" in part:
-                            k, v = part.split(":", 1)
-                            k = k.strip().strip('"\'')
-                            v = v.strip().strip('"\'')
-                            result[k] = v
-                    return result
-                return {}
-                
-            @staticmethod
-            def dump(obj, file_obj, indent=None):
-                if isinstance(obj, dict):
-                    pairs = []
-                    for k, v in obj.items():
-                        if isinstance(v, str):
-                            pairs.append('"{0}": "{1}"'.format(k, v))
-                        else:
-                            pairs.append('"{0}": {1}'.format(k, v))
-                    file_obj.write("{" + ", ".join(pairs) + "}")
-                else:
-                    file_obj.write(str(obj))
-        
-        json = DummyJson()
-
-# Print Python and platform info
-print("Python version: {0}.{1}.{2}".format(sys.version_info[0], sys.version_info[1], sys.version_info[2]))
-print("Platform: {0}".format(sys.platform))
-print("Executable: {0}".format(sys.executable))
-
-# Define markers for script results
-SCRIPT_SUCCESS_MARKER = "SCRIPT_SUCCESS"
-SCRIPT_ERROR_MARKER = "SCRIPT_ERROR"
-
-# Session state and command files
-SESSION_DIR = os.path.join(tempfile.gettempdir(), "codesys_session")
-SESSION_STATE_FILE = os.path.join(SESSION_DIR, "session_state.json")
-COMMAND_FILE = os.path.join(SESSION_DIR, "command.json")
-RESULT_FILE = os.path.join(SESSION_DIR, "result.json")
-SESSION_LOCK_FILE = os.path.join(SESSION_DIR, "session.lock")
-
-# Print key information at start
-print("Session stop script starting")
-print("tempdir: {0}".format(tempfile.gettempdir()))
-print("SESSION_DIR: {0}".format(SESSION_DIR))
-print("SESSION_STATE_FILE: {0}".format(SESSION_STATE_FILE))
-print("COMMAND_FILE: {0}".format(COMMAND_FILE))
-print("RESULT_FILE: {0}".format(RESULT_FILE))
-print("SESSION_LOCK_FILE: {0}".format(SESSION_LOCK_FILE))
-
-def stop_session():
-    """Stop a running CODESYS session"""
-    try:
-        print("Attempting to stop CODESYS session")
-        
-        # Check if session directory exists
-        if not os.path.exists(SESSION_DIR):
-            print("{0}: No session directory found at {1}".format(SCRIPT_SUCCESS_MARKER, SESSION_DIR))
-            # Consider this a success since there's no session to stop
-            return True
-
-        # Check if session state file exists
-        if not os.path.exists(SESSION_STATE_FILE):
-            print("{0}: No session state file found at {1}".format(SCRIPT_SUCCESS_MARKER, SESSION_STATE_FILE))
-            # No state file means no active session - clear any other potential session files
-            try:
-                for filename in os.listdir(SESSION_DIR):
-                    filepath = os.path.join(SESSION_DIR, filename)
-                    try:
-                        if os.path.isfile(filepath):
-                            os.unlink(filepath)
-                            print("Removed orphaned session file: {0}".format(filepath))
-                    except Exception as e:
-                        print("Error removing file {0}: {1}".format(filepath, e))
-            except Exception as e:
-                print("Error cleaning session directory: {0}".format(e))
-            
-            return True
-
-        # Read session state
-        try:
-            with open(SESSION_STATE_FILE, 'r') as f:
-                session_state = json.load(f)
-        except Exception as e:
-            print("Error reading session state file: {0}".format(e))
-            # Force cleanup regardless of the error
-            try:
-                for filename in os.listdir(SESSION_DIR):
-                    filepath = os.path.join(SESSION_DIR, filename)
-                    try:
-                        if os.path.isfile(filepath):
-                            os.unlink(filepath)
-                            print("Removed session file due to corrupt state: {0}".format(filepath))
-                    except Exception as rm_err:
-                        print("Error removing file {0}: {1}".format(filepath, rm_err))
-            except Exception as dir_err:
-                print("Error cleaning session directory: {0}".format(dir_err))
-            
-            print("{0}: Session files cleaned up forcefully due to corrupt state".format(SCRIPT_SUCCESS_MARKER))
-            return True
-
-        # Check if session is already marked as inactive
-        if not session_state.get('active', False) or session_state.get('status') in ['stopped', 'error', 'timeout']:
-            print("{0}: Session is already marked as {1}".format(
-                SCRIPT_SUCCESS_MARKER, session_state.get('status', 'inactive')))
-            
-            # Remove session files to ensure a clean state
-            try:
-                for filename in os.listdir(SESSION_DIR):
-                    filepath = os.path.join(SESSION_DIR, filename)
-                    try:
-                        if os.path.isfile(filepath):
-                            os.unlink(filepath)
-                            print("Removed session file for cleanup: {0}".format(filepath))
-                    except Exception as e:
-                        print("Error removing file {0}: {1}".format(filepath, e))
-            except Exception as e:
-                print("Error cleaning session directory: {0}".format(e))
-            
-            return True
-
-        # Get lock file age to check if session is responsive
-        lock_is_stale = False
-        current_time = time.time()
-        
-        try:
-            if os.path.exists(SESSION_LOCK_FILE):
-                with open(SESSION_LOCK_FILE, 'r') as f:
-                    lock_time = float(f.read().strip())
-                    lock_age = current_time - lock_time
-                    
-                if lock_age > 30:
-                    print("WARNING: Session lock file is stale ({0:.1f} seconds old). Session may have crashed.".format(lock_age))
-                    lock_is_stale = True
-            else:
-                print("WARNING: Session lock file not found. Session may have crashed.")
-                lock_is_stale = True
-        except Exception as e:
-            print("Warning: Error reading lock file: {0}".format(e))
-            lock_is_stale = True
-
-        # If the command file doesn't exist, we have a zombie session
-        if not os.path.exists(COMMAND_FILE):
-            print("WARNING: Command file not found. Session is in a zombie state.")
-            lock_is_stale = True
-        
-        # If the session appears to be a zombie (stale lock), don't try to send a command
-        # Just force cleanup immediately
-        if lock_is_stale:
-            print("Session appears to be unresponsive. Skipping command and forcing cleanup.")
-            try:
-                # Update state file first
-                try:
-                    session_state['active'] = False
-                    session_state['status'] = "error"
-                    session_state['end_time'] = current_time
-                    session_state['error_reason'] = "Unresponsive session, forced cleanup"
-                    
-                    with open(SESSION_STATE_FILE, 'w') as f:
-                        json.dump(session_state, f, indent=2)
-                        print("Updated session state to error before cleanup")
-                except Exception as state_err:
-                    print("Warning: Failed to update session state file: {0}".format(state_err))
-                
-                # Remove all session files
-                for filename in os.listdir(SESSION_DIR):
-                    filepath = os.path.join(SESSION_DIR, filename)
-                    try:
-                        if os.path.isfile(filepath):
-                            os.unlink(filepath)
-                            print("Removed zombie session file: {0}".format(filepath))
-                    except Exception as e:
-                        print("Error removing file {0}: {1}".format(filepath, e))
-                
-                print("{0}: Zombie session files cleaned up forcefully".format(SCRIPT_SUCCESS_MARKER))
-                return True
-            except Exception as e:
-                print("{0}: Failed to force cleanup - {1}".format(SCRIPT_ERROR_MARKER, str(e)))
-                return False
-        
-        # For responsive sessions, try to send a stop command first
-        try:
-            # Create stop command
-            stop_command = {
-                "id": "stop_{0}".format(int(time.time())),
-                "type": "stop",
-                "timestamp": current_time
-            }
-
-            # Write command to file with pretty-printing
-            command_data = {
-                "status": "pending",
-                "command": stop_command,
-                "timestamp": current_time
-            }
-
-            print("Writing stop command to command file: {0}".format(COMMAND_FILE))
-            with open(COMMAND_FILE, 'w') as f:
-                json.dump(command_data, f, indent=2)
-
-            print("Stop command sent to session")
-        except Exception as cmd_err:
-            print("Error sending stop command: {0}".format(cmd_err))
-            # Continue to forced cleanup
-
-        # Wait for result (with timeout)
-        timeout = 15  # seconds (increased timeout for more reliability)
-        start_time = time.time()
-        result_received = False
-
-        while not result_received and (time.time() - start_time) < timeout:
-            # Check if result file exists
-            if not os.path.exists(RESULT_FILE):
-                time.sleep(0.5)
-                continue
-
-            try:
-                with open(RESULT_FILE, 'r') as f:
-                    result_data = json.load(f)
-                    
-                    if result_data.get("status") == "complete" and result_data.get("result"):
-                        result = result_data.get("result")
-                        if result.get('id') == stop_command['id']:
-                            result_received = True
-                            print("Session stop result: {0}".format(result.get('output', 'No output')))
-                            break
-            except Exception as e:
-                print("Error reading result: {0}".format(e))
-                time.sleep(0.5)
-
-        # Update session state to indicate it's stopped regardless of result
-        try:
-            with open(SESSION_STATE_FILE, 'r') as f:
-                session_state = json.load(f)
-                
-            session_state['active'] = False
-            session_state['status'] = "stopped"
-            session_state['end_time'] = time.time()
-
-            with open(SESSION_STATE_FILE, 'w') as f:
-                json.dump(session_state, f, indent=2)
-                print("Updated session state to stopped")
-        except Exception as e:
-            print("Warning: Failed to update session state file: {0}".format(e))
-
-        # Remove all session files for a clean state
-        print("Cleaning up all session files")
-        try:
-            for filename in os.listdir(SESSION_DIR):
-                filepath = os.path.join(SESSION_DIR, filename)
-                try:
-                    if os.path.isfile(filepath):
-                        os.unlink(filepath)
-                        print("Removed session file: {0}".format(filepath))
-                except Exception as e:
-                    print("Error removing file {0}: {1}".format(filepath, e))
-            
-            print("{0}: Session stopped and files cleaned up successfully".format(SCRIPT_SUCCESS_MARKER))
-            return True
-        except Exception as e:
-            print("{0}: Error during final cleanup - {1}".format(SCRIPT_ERROR_MARKER, str(e)))
-            return False
-
-    except Exception as e:
-        print("{0}: Error stopping session - {1}".format(SCRIPT_ERROR_MARKER, str(e)))
-        traceback.print_exc()
-        
-        # Try one final desperate cleanup even if everything else failed
-        try:
-            if os.path.exists(SESSION_DIR):
-                for filename in os.listdir(SESSION_DIR):
-                    filepath = os.path.join(SESSION_DIR, filename)
-                    try:
-                        if os.path.isfile(filepath):
-                            os.unlink(filepath)
-                            print("Emergency cleanup: Removed session file: {0}".format(filepath))
-                    except Exception:
-                        pass
-                print("Emergency session cleanup attempted after error")
-        except Exception:
-            pass
-            
-        return False
-
-# Main script
-if __name__ == "__main__":
-    if stop_session():
-        sys.exit(0)
-    else:
-        sys.exit(1)