alpaca-trade 0.24.2

use std::fmt;
use std::sync::Arc;

use alpaca_http::{NoContent, RequestParts};
use reqwest::Method;

use crate::client::ClientInner;
use crate::positions::{
    CloseAllRequest, ClosePositionBody, ClosePositionRequest, ClosePositionResult,
    DoNotExerciseAccepted, ExercisePositionBody, Position,
};
use crate::{Error, positions::request};

#[derive(Clone)]
pub struct PositionsClient {
    inner: Arc<ClientInner>,
}

impl PositionsClient {
    pub(crate) fn new(inner: Arc<ClientInner>) -> Self {
        Self { inner }
    }

    pub async fn list(&self) -> Result<Vec<Position>, Error> {
        let request =
            RequestParts::new(Method::GET, "/v2/positions").with_operation("positions.list");

        self.inner
            .send_json::<Vec<Position>>(request)
            .await
            .map(|response| response.into_body())
    }

    pub async fn get(&self, symbol_or_asset_id: &str) -> Result<Position, Error> {
        let request = RequestParts::new(
            Method::GET,
            format!(
                "/v2/positions/{}",
                request::validate_symbol_or_asset_id(symbol_or_asset_id)?
            ),
        )
        .with_operation("positions.get");

        self.inner
            .send_json::<Position>(request)
            .await
            .map(|response| response.into_body())
    }

    pub async fn close_all(
        &self,
        request: CloseAllRequest,
    ) -> Result<Vec<ClosePositionResult>, Error> {
        let request = RequestParts::new(Method::DELETE, "/v2/positions")
            .with_operation("positions.close_all")
            .with_query(request.into_query());

        self.inner
            .send_json::<Vec<ClosePositionResult>>(request)
            .await
            .map(|response| response.into_body())
    }

    pub async fn close(
        &self,
        symbol_or_asset_id: &str,
        request: ClosePositionRequest,
    ) -> Result<ClosePositionBody, Error> {
        let request = RequestParts::new(
            Method::DELETE,
            format!(
                "/v2/positions/{}",
                request::validate_symbol_or_asset_id(symbol_or_asset_id)?
            ),
        )
        .with_operation("positions.close")
        .with_query(request.into_query());

        self.inner
            .send_json::<ClosePositionBody>(request)
            .await
            .map(|response| response.into_body())
    }

    pub async fn exercise(
        &self,
        symbol_or_contract_id: &str,
    ) -> Result<ExercisePositionBody, Error> {
        let request = RequestParts::new(
            Method::POST,
            format!(
                "/v2/positions/{}/exercise",
                request::validate_symbol_or_contract_id(symbol_or_contract_id)?
            ),
        )
        .with_operation("positions.exercise");

        self.inner
            .send_json::<ExercisePositionBody>(request)
            .await
            .map(|response| response.into_body())
    }

    pub async fn do_not_exercise(
        &self,
        symbol_or_contract_id: &str,
    ) -> Result<DoNotExerciseAccepted, Error> {
        let request = RequestParts::new(
            Method::POST,
            format!(
                "/v2/positions/{}/do-not-exercise",
                request::validate_symbol_or_contract_id(symbol_or_contract_id)?
            ),
        )
        .with_operation("positions.do_not_exercise");

        self.inner
            .send_no_content(request)
            .await
            .map(|_response: alpaca_http::HttpResponse<NoContent>| DoNotExerciseAccepted)
    }

    #[allow(dead_code)]
    #[must_use]
    pub(crate) fn inner(&self) -> &Arc<ClientInner> {
        &self.inner
    }
}

impl fmt::Debug for PositionsClient {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("PositionsClient")
            .field("base_url", self.inner.base_url())
            .finish()
    }
}