diff options
| -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}." |