import sublime
import sublime_plugin
import os
import re

class IncrementalSaveCommand(sublime_plugin.TextCommand):
    def run(self, edit):
        current_file = self.view.file_name()
        
        if not current_file:
            # If file hasn't been saved yet, prompt for save
            self.view.window().run_command("save_as")
            return
        
        # Get directory and filename
        directory = os.path.dirname(current_file)
        filename = os.path.basename(current_file)
        name, ext = os.path.splitext(filename)
        
        # Check if filename ends with a number
        match = re.search(r'-(\d+)$', name)
        
        if match:
            # Extract the number and increment it
            current_num = int(match.group(1))
            new_num = current_num + 1
            # Replace the old number with the new one
            new_name = re.sub(r'-\d+$', f'-{new_num:02d}', name)
        else:
            # No number found, add -01
            new_name = name + '-01'
        
        new_filename = new_name + ext
        new_path = os.path.join(directory, new_filename)
        
        # Save the file with the new name
        self.view.set_scratch(False)
        self.view.retarget(new_path)
        self.view.run_command("save")
        
        sublime.status_message(f"Saved as: {new_filename}")