use {
    crate::{assertions::assert_user, errors::ErrorCode, state::*},
    anchor_lang::prelude::*,
    hpl_events::HplEvents,
    hpl_utils::{reallocate, traits::Default},
};

#[derive(Accounts)]
#[instruction(args: CreateProfileArgs)]
pub struct CreateProfile<'info> {
    /// The user state account
    #[account()]
    pub user: Box<Account<'info, User>>,

    /// The project state account
    #[account()]
    pub project: Box<Account<'info, Project>>,

    /// User's profile for the provided project
    #[account(
        init, payer = wallet,
        space = Profile::get_initial_len(&args),
        seeds = [
            b"profile".as_ref(),
            project.key().as_ref(),
            user.key().as_ref(),
            &args.identity.to_bytes()[..]
        ],
        bump
      )]
    pub profile: Account<'info, Profile>,

    /// One of the user's wallet
    #[account(mut)]
    pub wallet: Signer<'info>,

    /// NATIVE RENT SYSVAR
    pub rent_sysvar: Sysvar<'info, Rent>,

    /// System Program.
    pub system_program: Program<'info, System>,

    /// NO OP program
    pub hpl_events: Program<'info, HplEvents>,

    /// NATIVE SYSVAR CLOCK
    pub clock: Sysvar<'info, Clock>,

    /// NATIVE INSTRUCTIONS SYSVAR
    /// CHECK: This is not dangerous because we don't read or write from this account
    #[account(address = anchor_lang::solana_program::sysvar::instructions::ID)]
    pub instructions_sysvar: AccountInfo<'info>,

    /// The vault that collects the fees.
    /// CHECK: This is not dangerous
    #[account(mut)]
    pub vault: AccountInfo<'info>,
}

/// Structure representing the arguments for creating a new profile on the chain.
#[derive(AnchorSerialize, AnchorDeserialize, Clone, Debug, PartialEq)]
pub struct CreateProfileArgs {
    /// Identity of the profile being created. It can be one of the following variants:
    /// - `ProfileIdentity::Main`: Represents the main identity of the profile.
    /// - `ProfileIdentity::Value`: Represents an identity with a custom string value.
    pub identity: ProfileIdentity,

    /// Profile specific name.
    pub name: Option<String>,

    /// Profile specific description.
    pub bio: Option<String>,

    /// Profile specific pfp url.
    pub pfp: Option<String>,
}

pub fn create_profile(ctx: Context<CreateProfile>, args: CreateProfileArgs) -> Result<()> {
    assert_user(&ctx.accounts.user, ctx.accounts.wallet.key())?;

    let profile = &mut ctx.accounts.profile;
    profile.set_defaults();
    profile.bump = ctx.bumps["profile"];
    profile.project = ctx.accounts.project.key();
    profile.user = ctx.accounts.user.key();
    profile.identity = args.identity.clone();
    profile.name = args.name.clone();
    profile.bio = args.bio.clone();
    profile.pfp = args.pfp.clone();

    Event::new_profile(
        profile.key(),
        profile.try_to_vec().unwrap(),
        &ctx.accounts.clock,
    )
    .emit(ctx.accounts.hpl_events.to_account_info())?;

    Ok(())
}

#[derive(Accounts)]
pub struct DeleteProfile<'info> {
    /// The user state account
    #[account()]
    pub user: Box<Account<'info, User>>,

    /// The project state account
    #[account()]
    pub project: Box<Account<'info, Project>>,

    /// User's profile for the provided project
    #[account(mut, has_one = user, has_one = project, close = wallet)]
    pub profile: Account<'info, Profile>,

    /// One of the user's wallet
    #[account(mut)]
    pub wallet: Signer<'info>,

    /// System Program.
    pub system_program: Program<'info, System>,

    /// NO OP program
    pub hpl_events: Program<'info, HplEvents>,

    /// NATIVE SYSVAR CLOCK
    pub clock: Sysvar<'info, Clock>,

    /// NATIVE RENT SYSVAR
    pub rent_sysvar: Sysvar<'info, Rent>,

    /// NATIVE INSTRUCTIONS SYSVAR
    /// CHECK: This is not dangerous because we don't read or write from this account
    #[account(address = anchor_lang::solana_program::sysvar::instructions::ID)]
    pub instructions_sysvar: AccountInfo<'info>,

    /// The vault that collects the fees.
    /// CHECK: This is not dangerous
    #[account(mut)]
    pub vault: AccountInfo<'info>,
}

pub fn delete_profile(ctx: Context<DeleteProfile>) -> Result<()> {
    assert_user(&ctx.accounts.user, ctx.accounts.wallet.key())?;
    Ok(())
}

#[derive(Accounts)]
pub struct ManageProfileData<'info> {
    /// The project state account
    #[account()]
    pub project: Box<Account<'info, Project>>,

    /// User's profile for the provided project
    #[account(mut, has_one = project)]
    pub profile: Box<Account<'info, Profile>>,

    /// [Option] delegate authority account
    #[account()]
    pub delegate_authority: Option<Account<'info, DelegateAuthority>>,

    /// The authority (or delegate) of the project.
    pub authority: Signer<'info>,

    /// The wallet that pays for the rent.
    #[account(mut)]
    pub payer: Signer<'info>,

    /// The system program.
    pub system_program: Program<'info, System>,

    /// NO OP program
    pub hpl_events: Program<'info, HplEvents>,

    /// NATIVE SYSVAR CLOCK
    pub clock: Sysvar<'info, Clock>,

    /// NATIVE RENT SYSVAR
    pub rent_sysvar: Sysvar<'info, Rent>,

    /// NATIVE Instructions SYSVAR
    /// CHECK: This is not dangerous
    #[account(address = anchor_lang::solana_program::sysvar::instructions::ID)]
    pub instructions_sysvar: AccountInfo<'info>,

    /// The vault that collects the fees.
    /// CHECK: This is not dangerous
    #[account(mut)]
    pub vault: AccountInfo<'info>,
}
/// Structure representing the arguments for adding profile data to a profile.
///
/// # Fields
///
/// - `label`: Label or name of the profile data being added.
/// - `value`: Value of the profile data being added. It can be one of the following variants:
///     - `None`: If no value is provided.
///     - `Some(ManageProfileDataArgsValue::SingleValue)`: Represents a single string value.
///     - `Some(ManageProfileDataArgsValue::MultiValue)`: Represents a list of string values.
///     - `Some(ManageProfileDataArgsValue::EntityData)`: Represents an entity data node (32-byte array).
#[derive(AnchorSerialize, AnchorDeserialize)]
pub struct ManageProfileDataArgs {
    pub label: String,
    pub value: Option<ProfileData>,
    pub is_app_context: bool,
}

pub fn manage_profile_data(
    ctx: Context<ManageProfileData>,
    args: ManageProfileDataArgs,
) -> Result<()> {
    assert_eq!(
        ctx.accounts
            .project
            .profile_data_config
            .contains_key(&args.label),
        true
    );

    let profile_info = ctx.accounts.profile.to_account_info();
    let profile = &mut ctx.accounts.profile;

    let to_update = if args.is_app_context {
        &mut profile.app_context
    } else {
        &mut profile.data
    };

    let mut len = if to_update.contains_key(&args.label) {
        if args.value.is_none() {
            args.label.len() as isize * -1
        } else {
            0
        }
    } else {
        args.label.len() as isize
    };

    if let Some(v) = to_update.get(&args.label) {
        len -= v.size() as isize;
    }

    if let Some(value) = &args.value {
        len += value.size() as isize;
    }

    if len > 0 {
        reallocate(
            len,
            profile_info.clone(),
            ctx.accounts.payer.to_account_info(),
            &ctx.accounts.rent_sysvar,
            &ctx.accounts.system_program,
        )?;
    }

    if let Some(profile_data) = args.value {
        if !ctx
            .accounts
            .project
            .validate_profile_data(&args.label, &profile_data)
        {
            return Err(ErrorCode::InvalidValueType.into());
        }
        profile.data.insert(args.label, profile_data);
    } else {
        profile.data.remove(&args.label);
    }

    if len < 0 {
        reallocate(
            len,
            profile_info,
            ctx.accounts.payer.to_account_info(),
            &ctx.accounts.rent_sysvar,
            &ctx.accounts.system_program,
        )?;
    }

    Event::update_profile(
        profile.key(),
        profile.try_to_vec().unwrap(),
        &ctx.accounts.clock,
    )
    .emit(ctx.accounts.hpl_events.to_account_info())?;

    Ok(())
}