split storage into its own file

Author: alan412

backend/main.py

@@ -1,20 +1,20 @@
 # Standard library imports
 import argparse
-import json
 import os
 
 # Third-party imports
 from flask import Flask, jsonify, request, send_from_directory
-from flask_restful import Api, Resource
+from flask_restful import Api
 from flask_sqlalchemy import SQLAlchemy
-from werkzeug.utils import secure_filename
 
 # Our imports
 from deploy import DeployResource
+from storage import StorageEntryResource, StorageFileRenameResource, StorageRootResource, StorageResource
 
 app = Flask(__name__, static_folder='../dist', static_url_path='')
 app.url_map.merge_slashes = False  # Don't merge consecutive slashes
 api = Api(app)
+db = SQLAlchemy(app)
 
 # Add CORS headers
 @app.after_request
@@ -38,201 +38,6 @@ def handle_preflight():
 app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{os.path.join(basedir, "projects.db")}'
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 
-db = SQLAlchemy(app)
-
-# Storage models for key-value pairs and files
-class StorageEntry(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
-    entry_key = db.Column(db.String(255), nullable=False, unique=True)
-    entry_value = db.Column(db.Text, nullable=False)
-    
-    def to_dict(self):
-        return {
-            'key': self.entry_key,
-            'value': self.entry_value
-        }
-
-class StorageFile(db.Model):
-    id = db.Column(db.Integer, primary_key=True)
-    file_path = db.Column(db.String(500), nullable=False, unique=True)
-    file_content = db.Column(db.Text, nullable=False)
-    
-    def to_dict(self):
-        return {
-            'path': self.file_path,
-            'content': self.file_content
-        }
-
-# Storage Resources for key-value and file operations
-class StorageEntryResource(Resource):
-    def get(self, entry_key):
-        """Fetch entry value by key"""
-        entry = StorageEntry.query.filter_by(entry_key=entry_key).first()
-        if entry:
-            return {'value': entry.entry_value}
-        else:
-            # Return default value if provided in query params
-            default_value = request.args.get('default', '')
-            return {'value': default_value}
-    
-    def post(self, entry_key):
-        """Save entry value"""
-        data = request.get_json()
-        if not data or 'value' not in data:
-            return {'error': 'Entry value is required'}, 400
-        
-        entry = StorageEntry.query.filter_by(entry_key=entry_key).first()
-        if entry:
-            entry.entry_value = data['value']
-        else:
-            entry = StorageEntry(entry_key=entry_key, entry_value=data['value'])
-            db.session.add(entry)
-        
-        try:
-            db.session.commit()
-            return {'message': 'Entry saved successfully'}
-        except Exception as e:
-            db.session.rollback()
-            return {'error': 'Failed to save entry'}, 500
-
-class StorageResource(Resource):
-    def get(self, path):
-        """Get file content or list directory based on path"""
-        # Handle empty path as root directory
-        if not path:
-            path = ""
-        
-        if path.endswith('/') or path == "":
-            # List directory
-            # For root directory, find all files
-            if path == "":
-                files = StorageFile.query.all()
-                # Extract top-level files and directories
-                children = set()
-                for file in files:
-                    if '/' in file.file_path:
-                        # It's in a subdirectory, add the directory name
-                        dir_name = file.file_path.split('/')[0] + '/'
-                        children.add(dir_name)
-                    else:
-                        # It's a top-level file
-                        children.add(file.file_path)
-            else:
-                # Find all files that start with this path
-                files = StorageFile.query.filter(StorageFile.file_path.like(f'{path}%')).all()
-                
-                # Extract immediate children (not nested)
-                children = set()
-                for file in files:
-                    relative_path = file.file_path[len(path):]
-                    if '/' in relative_path:
-                        # It's in a subdirectory, add the directory name
-                        dir_name = relative_path.split('/')[0] + '/'
-                        children.add(dir_name)
-                    else:
-                        # It's a direct child file
-                        children.add(relative_path)
-            
-            return {'files': sorted(list(children))}
-        else:
-            # Get file content
-            file_obj = StorageFile.query.filter_by(file_path=path).first()
-            if file_obj:
-                return {'content': file_obj.file_content}
-            else:
-                return {'error': 'File not found'}, 404
-    
-    def post(self, path):
-        """Save file content (only for files, not directories)"""
-        if path.endswith('/'):
-            return {'error': 'Cannot save content to a directory path'}, 400
-        
-        data = request.get_json()
-        if not data or 'content' not in data:
-            return {'error': 'File content is required'}, 400
-        
-        file_obj = StorageFile.query.filter_by(file_path=path).first()
-        if file_obj:
-            file_obj.file_content = data['content']
-        else:
-            file_obj = StorageFile(file_path=path, file_content=data['content'])
-            db.session.add(file_obj)
-        
-        try:
-            db.session.commit()
-            return {'message': 'File saved successfully'}
-        except Exception as e:
-            db.session.rollback()
-            return {'error': 'Failed to save file'}, 500
-    
-    def delete(self, path):
-        """Delete file or directory"""
-        if path.endswith('/'):
-            # Delete directory (all files starting with this path)
-            files = StorageFile.query.filter(StorageFile.file_path.like(f'{path}%')).all()
-            for file_obj in files:
-                db.session.delete(file_obj)
-        else:
-            # Delete specific file
-            file_obj = StorageFile.query.filter_by(file_path=path).first()
-            if not file_obj:
-                return {'error': 'File not found'}, 404
-            db.session.delete(file_obj)
-        
-        try:
-            db.session.commit()
-            return {'message': 'Deleted successfully'}
-        except Exception as e:
-            db.session.rollback()
-            return {'error': 'Failed to delete'}, 500
-
-class StorageFileRenameResource(Resource):
-    def post(self):
-        """Rename file or directory"""
-        data = request.get_json()
-        if not data or 'old_path' not in data or 'new_path' not in data:
-            return {'error': 'Both old_path and new_path are required'}, 400
-        
-        old_path = data['old_path']
-        new_path = data['new_path']
-        
-        if old_path.endswith('/'):
-            # Rename directory (all files starting with old_path)
-            files = StorageFile.query.filter(StorageFile.file_path.like(f'{old_path}%')).all()
-            for file_obj in files:
-                new_file_path = file_obj.file_path.replace(old_path, new_path, 1)
-                file_obj.file_path = new_file_path
-        else:
-            # Rename specific file
-            file_obj = StorageFile.query.filter_by(file_path=old_path).first()
-            if not file_obj:
-                return {'error': 'File not found'}, 404
-            file_obj.file_path = new_path
-        
-        try:
-            db.session.commit()
-            return {'message': 'Renamed successfully'}
-        except Exception as e:
-            db.session.rollback()
-            return {'error': 'Failed to rename'}, 500
-
-class StorageRootResource(Resource):
-    def get(self):
-        """List all top-level files and directories"""
-        files = StorageFile.query.all()
-        # Extract top-level files and directories
-        children = set()
-        for file in files:
-            if '/' in file.file_path:
-                # It's in a subdirectory, add the directory name
-                dir_name = file.file_path.split('/')[0] + '/'
-                children.add(dir_name)
-            else:
-                # It's a top-level file
-                children.add(file.file_path)
-        
-        return {'files': sorted(list(children))}
-
 # Register API routes
 # Storage API routes (more specific routes first)
 api.add_resource(StorageEntryResource, '/entries/<path:entry_key>')