admin_workflow.rs

// SPDX-License-Identifier: Apache-2.0

//! Admin Workflow Example: Setting up storage infrastructure
//!
//! This example demonstrates the admin flow in the bucket-based model.
//! Admins are responsible for:
//! - Creating buckets in Layer 0
//! - Establishing storage agreements with providers
//! - Assigning users to buckets (Reader+Writer roles)
//!
//! Users then create drives on their assigned buckets without managing
//! any infrastructure details.

use sp_keyring::AccountKeyring;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    tracing_subscriber::fmt::init();

    println!("==================================================");
    println!("  ADMIN WORKFLOW: Setting up Storage Infrastructure");
    println!("==================================================\n");

    // Admin account (e.g., Alice in dev)
    let admin = AccountKeyring::Alice;
    let alice_user = AccountKeyring::Bob; // Alice will assign Bob as user

    println!("Admin: {:?}", admin.to_account_id());
    println!("User to assign: {:?}\n", alice_user.to_account_id());

    // ============================================================
    // Step 1: Create Bucket in Layer 0
    // ============================================================

    println!("Step 1: Creating bucket in Layer 0...");

    // NOTE: In a real implementation, this would call Layer 0 pallet:
    // let bucket_id = storage_provider.create_bucket(
    //     origin: admin,
    //     min_providers: 3,
    // );

    let bucket_id = 1u64; // Placeholder
    println!("✓ Bucket {} created by admin\n", bucket_id);

    // ============================================================
    // Step 2: Request Storage Agreements with Providers
    // ============================================================

    println!("Step 2: Requesting storage agreements with providers...");

    // In production, admin would:
    // 1. Find available providers
    // 2. Request agreements with 1 primary + N replicas
    // 3. Wait for providers to accept

    // NOTE: This would call Layer 0 pallet:
    // let agreement_1 = storage_provider.request_agreement(
    //     origin: admin,
    //     bucket_id: bucket_id,
    //     provider: provider_1_account,
    //     max_bytes: 100_000_000_000, // 100 GB
    //     duration: 30 * DAYS,
    //     max_payment: 100 * TOKENS,
    //     replica_params: None, // Primary
    // );

    let agreement_1 = 101u64; // Primary
    let agreement_2 = 102u64; // Replica 1
    let agreement_3 = 103u64; // Replica 2

    println!("✓ Primary agreement: {}", agreement_1);
    println!("✓ Replica 1 agreement: {}", agreement_2);
    println!("✓ Replica 2 agreement: {}", agreement_3);
    println!();

    // ============================================================
    // Step 3: Assign User to Bucket (Reader + Writer)
    // ============================================================

    println!("Step 3: Assigning user to bucket...");

    // The user needs both Reader and Writer roles to:
    // - Reader: Download files from the bucket
    // - Writer: Upload files to the bucket

    // NOTE: This would call Layer 0 pallet:
    // storage_provider.add_bucket_member(
    //     origin: admin,
    //     bucket_id: bucket_id,
    //     member: alice_user,
    //     role: Role::Reader | Role::Writer,
    // );

    println!("✓ User assigned as Reader+Writer to bucket {}", bucket_id);
    println!();

    // ============================================================
    // Step 4: Monitor and Maintain (Ongoing)
    // ============================================================

    println!("Step 4: Ongoing admin responsibilities:");
    println!("  - Monitor storage challenges");
    println!("  - Replace failed providers if needed");
    println!("  - Adjust bucket capacity as needed");
    println!("  - Manage user access (grant/revoke)");
    println!();

    // Example: Replace failed provider
    println!("Example: If provider fails, admin would:");
    println!("  1. Create new agreement with replacement provider");
    println!("  2. System automatically replicates data");
    println!("  3. Old agreement terminated");
    println!();

    // ============================================================
    // Done!
    // ============================================================

    println!("==================================================");
    println!("✓ Infrastructure setup complete!");
    println!("==================================================");
    println!();
    println!("User can now:");
    println!("  1. Query available buckets: list_my_buckets()");
    println!("  2. Create drive: create_drive_on_bucket(bucket_id)");
    println!("  3. Upload/download files normally");
    println!();
    println!("User NEVER needs to:");
    println!("  ✗ Create buckets");
    println!("  ✗ Manage agreements");
    println!("  ✗ Handle challenges");
    println!("  ✗ Replace providers");
    println!();

    Ok(())
}