import os
from huggingface_hub import SpaceStorage

# Initialize Spaces storage
storage = SpaceStorage()

# Custom storage adapter for Mastodon (simplified)
class MastodonStorage:
    def __init__(self):
        self.storage = storage

    def set(self, key, value):
        self.storage.set(key, value)

    def get(self, key):
        return self.storage.get(key, default=None)

    def delete(self, key):
        self.storage.delete(key)

# Example functions to hook into Mastodon
def save_user(username, data):
    storage.set(f"user:{username}", data)

def get_user(username):
    return storage.get(f"user:{username}")

def save_post(post_id, content):
    storage.set(f"post:{post_id}", content)

def get_post(post_id):
    return storage.get(f"post:{post_id}")

# Called from Ruby via system call (simplified integration)
if __name__ == "__main__":
    import sys
    action = sys.argv[1] if len(sys.argv) > 1 else None
    if action == "save_user":
        save_user(sys.argv[2], sys.argv[3])
    elif action == "get_user":
        print(get_user(sys.argv[2]) or "")
    elif action == "save_post":
        save_post(sys.argv[2], sys.argv[3])
    elif action == "get_post":
        print(get_post(sys.argv[2]) or "")