from flask import Flask, request, jsonify import requests app = Flask(__name__) # Joplin API token and base URL JOPLIN_API_TOKEN = 'your_joplin_api_token' JOPLIN_API_URL = 'http://localhost:41184' class Tools: def __init__(self): pass def get_notebook_id(self, notebook_title): """ Fetches the ID of the notebook with the given title. If the notebook does not exist, returns None. """ response = requests.get(f'{JOPLIN_API_URL}/folders?token={JOPLIN_API_TOKEN}') if response.status_code == 200: folders = response.json()['items'] for folder in folders: if folder['title'] == notebook_title: return folder['id'] return None else: raise Exception(f"Error fetching folders: {response.status_code}") def create_notebook(self, notebook_title): """ Creates a new notebook with the given title and returns its ID. """ payload = { 'title': notebook_title } response = requests.post(f'{JOPLIN_API_URL}/folders?token={JOPLIN_API_TOKEN}', json=payload) if response.status_code == 200: return response.json()['id'] else: raise Exception(f"Error creating folder: {response.status_code}") def create_or_update_note(self, data): """ Creates or updates a note in Joplin based on the provided data. """ notebook_title = data.get('notebookTitle') note_title = data.get('title') note_body = data.get('body') tags = data.get('tags', []) resources = data.get('resources', []) notebook_id = self.get_notebook_id(notebook_title) if not notebook_id: notebook_id = self.create_notebook(notebook_title) # Check if note already exists response = requests.get(f'{JOPLIN_API_URL}/notes?token={JOPLIN_API_TOKEN}') if response.status_code == 200: notes = response.json()['items'] for note in notes: if note['title'] == note_title: note_id = note['id'] payload = { 'body': note_body, 'parent_id': notebook_id } response = requests.put(f'{JOPLIN_API_URL}/notes/{note_id}?token={JOPLIN_API_TOKEN}', json=payload) if response.status_code == 200: return "note updated successfully" else: raise Exception(f"Error updating note: {response.status_code}") # If note does not exist, create a new one payload = { 'title': note_title, 'body': note_body, 'parent_id': notebook_id } response = requests.post(f'{JOPLIN_API_URL}/notes?token={JOPLIN_API_TOKEN}', json=payload) if response.status_code == 200: note_id = response.json()['id'] for tag in tags: tag_id = self.create_or_get_tag_id(tag) requests.post(f'{JOPLIN_API_URL}/notes/{note_id}/tags/{tag_id}?token={JOPLIN_API_TOKEN}') for resource in resources: self.create_resource(note_id, resource) return "note saved successfully" else: raise Exception(f"Error creating note: {response.status_code}") def create_or_get_tag_id(self, tag_name): """ Fetches the ID of the tag with the given title. If the tag does not exist, creates a new tag and returns its ID. """ response = requests.get(f'{JOPLIN_API_URL}/tags?token={JOPLIN_API_TOKEN}') if response.status_code == 200: tags = response.json()['items'] for tag in tags: if tag['title'] == tag_name: return tag['id'] payload = { 'title': tag_name } response = requests.post(f'{JOPLIN_API_URL}/tags?token={JOPLIN_API_TOKEN}', json=payload) if response.status_code == 200: return response.json()['id'] else: raise Exception(f"Error creating tag: {response.status_code}") def create_resource(self, note_id, resource): """ Creates a new resource (e.g., file, image) and associates it with the given note. """ file_name = resource.get('fileName') content_type = resource.get('contentType') data = resource.get('data') # Base64 encoded string or path to file payload = { 'title': file_name, 'mime_type': content_type } response = requests.post(f'{JOPLIN_API_URL}/resources?token={JOPLIN_API_TOKEN}', json=payload) if response.status_code == 200: resource_id = response.json()['id'] response = requests.put(f'{JOPLIN_API_URL}/resources/{resource_id}?token={JOPLIN_API_TOKEN}', data=data) if response.status_code == 200: requests.post(f'{JOPLIN_API_URL}/notes/{note_id}/resources/{resource_id}?token={JOPLIN_API_TOKEN}') else: raise Exception(f"Error updating resource: {response.status_code}") else: raise Exception(f"Error creating resource: {response.status_code}") tools = Tools() @app.route('/save_to_joplin', methods=['POST']) def save_to_joplin(): """ Handles POST requests from OpenChat to save notes to Joplin. """ try: data = request.get_json() result = tools.create_or_update_note(data) return jsonify({'message': result}), 200 except Exception as e: return jsonify({'error': str(e)}), 400 if __name__ == '__main__': app.run(host='0.0.0.0', port=5000)