diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/libs/constants/default_config.py | 6 | ||||
| -rw-r--r-- | continuedev/src/continuedev/plugins/steps/share_session.py | 51 | 
2 files changed, 57 insertions, 0 deletions
diff --git a/continuedev/src/continuedev/libs/constants/default_config.py b/continuedev/src/continuedev/libs/constants/default_config.py index dbd2c8eb..2201df33 100644 --- a/continuedev/src/continuedev/libs/constants/default_config.py +++ b/continuedev/src/continuedev/libs/constants/default_config.py @@ -20,6 +20,7 @@ from continuedev.src.continuedev.plugins.steps.open_config import OpenConfigStep  from continuedev.src.continuedev.plugins.steps.clear_history import ClearHistoryStep  from continuedev.src.continuedev.plugins.steps.feedback import FeedbackStep  from continuedev.src.continuedev.plugins.steps.comment_code import CommentCodeStep +from continuedev.src.continuedev.plugins.steps.share_session import ShareSessionStep  from continuedev.src.continuedev.plugins.steps.main import EditHighlightedCodeStep  from continuedev.src.continuedev.plugins.context_providers.search import SearchContextProvider  from continuedev.src.continuedev.plugins.context_providers.diff import DiffContextProvider @@ -108,6 +109,11 @@ config = ContinueConfig(              name="clear",              description="Clear step history",              step=ClearHistoryStep, +        ), +        SlashCommand( +            name="share", +            description="Download and share the session transcript", +            step=ShareSessionStep,          )      ], diff --git a/continuedev/src/continuedev/plugins/steps/share_session.py b/continuedev/src/continuedev/plugins/steps/share_session.py new file mode 100644 index 00000000..de8659bd --- /dev/null +++ b/continuedev/src/continuedev/plugins/steps/share_session.py @@ -0,0 +1,51 @@ +import json +import os +import time +from typing import Optional + + +from ...core.sdk import ContinueSDK +from ...core.main import Step, FullState +from ...libs.util.paths import getSessionFilePath, getGlobalFolderPath +from ...server.session_manager import session_manager + + +class ShareSessionStep(Step): + +    session_id: Optional[str] = None + +    async def run(self, sdk: ContinueSDK): +        if self.session_id is None: +            self.session_id = sdk.ide.session_id + +        await session_manager.persist_session(self.session_id) +        time.sleep(0.5) + +        # Load the session data and format as a markdown file +        session_filepath = getSessionFilePath(self.session_id) +        with open(session_filepath, 'r') as f: +            session_state = FullState(**json.load(f)) + +        import datetime +        date_created = datetime.datetime.fromtimestamp( +            float(session_state.session_info.date_created)).strftime('%Y-%m-%d %H:%M:%S') +        content = f"This is a session transcript from [Continue](https://continue.dev) on {date_created}.\n\n" + +        for node in session_state.history.timeline[:-2]: +            if node.step.hide: +                continue  # ay + +            content += f"## {node.step.name}\n" +            content += f"{node.step.description}\n\n" + +        # Save to a markdown file +        save_filepath = os.path.join( +            getGlobalFolderPath(), f"{session_state.session_info.title}.md") + +        with open(save_filepath, 'w') as f: +            f.write(content) + +        # Open the file +        await sdk.ide.setFileOpen(save_filepath) + +        self.description = f"The session transcript has been saved to a markdown file at {save_filepath}."  | 
