import os
import base64


class OnshapeBase:
    def __init__(self):
        self.access_key = os.environ.get("ONSHAPE_ACCESS_KEY")
        self.secret_key = os.environ.get("ONSHAPE_SECRET_KEY")
        self.base_url = "https://cad.onshape.com/api/v10"

        if not self.access_key or not self.secret_key:
            raise ValueError(
                "ONSHAPE_ACCESS_KEY and ONSHAPE_SECRET_KEY must be set in environment variables."
            )

    def get_auth_header(self):
        credentials_raw = f"{self.access_key}:{self.secret_key}"
        credentials_base64 = base64.b64encode(credentials_raw.encode("utf-8")).decode(
            "utf-8"
        )
        return {"Authorization": f"Basic {credentials_base64}"}