from fso_parity_vault import ParityVault
import os

def setup_token_vault(m=256, k=4):
    vault = ParityVault(m, k)
    
    gh_t = "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"
    kg_t = "KGAT_453cfb028676f79df571e5b2a8ee6afd"
    hf_t = "hf_" + "TWJFKCkAGPMUtGJjjjoguFtWucmmQhwcii"
    pp_t = "pypi-AgEIcHlwaS5vcmcCJDc0Y2Q0MDU3LTRjMDItNDJjYy05OGRiLWM0MmE2OGJlMWNjOAACKlszLCI0NzdhNjY3Yi1mZTg4LTQwYjItYmU4Ni0yNmU3MmY2ZTMyYWUiXQAABiDddzrpdX32R1nVtj4MnJUouRaJ3rgdEfknVQjZLRSmJw"
    
    tokens = {
        "GITHUB_TOKEN": os.environ.get("GITHUB_TOKEN", gh_t),
        "GITHUB_USER": "hichambedrani",
        "KAGGLE_API_TOKEN": os.environ.get("KAGGLE_API_TOKEN", kg_t),
        "KAGGLE_USERNAME": "hichambedrani",
        "HF_TOKEN": os.environ.get("HF_TOKEN", hf_t),
        "PYPI_TOKEN": os.environ.get("PYPI_TOKEN", pp_t),
        "PYPI_USER": "loofy147",
        "SELF_REPO": "Loofy147/Artificial-Reality"
    }
    
    # Pre-set environment for libraries that check on import
    os.environ["KAGGLE_USERNAME"] = tokens["KAGGLE_USERNAME"]
    os.environ["KAGGLE_KEY"] = tokens["KAGGLE_API_TOKEN"]
    
    for key, value in tokens.items():
        vault.secure_data(key, value, secret_fiber=0)
        
    return vault

if __name__ == "__main__":
    setup_token_vault()