use actix_web::{
    dev::Service,
    http::header::{self, HeaderName, HeaderValue},
    web, App, HttpResponse, HttpServer, Responder,
};
use futures::FutureExt;
use serde_json::json;

//index
async fn index() -> impl Responder {
    HttpResponse::Ok().json(json!({
        "code": 1,
        "message":"请求完成!",
        "data": "index"
    }))
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    HttpServer::new(|| {
        App::new()
            .route("/", web::get().to(index))
            .wrap_fn(|req, srv| {
                srv.call(req).map(move |mut res| {
                    // 修改头部信息
                    res.as_mut()
                        .unwrap()
                        .headers_mut()
                        .insert(header::SERVER, HeaderValue::from_static("wcharServer"));
                    //cus-header 自定义头
                    res.as_mut()
                        .unwrap()
                        .headers_mut()
                        .insert(HeaderName::from_static("cus-header"), HeaderValue::from_static("are you ok?"));
                
                    res
                })
            })
    })
    .bind("127.0.0.1:8080")?
    .run()
    .await
}

[package]
name = "grpc"
version = "0.1.0"
edition = "2021"

[dependencies]
tonic = "0.7"
prost = "0.10"
tokio = { version = "1", features = ["full"] }

[build-dependencies]
tonic-build = "0.7"

use tokio::fs::{create_dir_all, File};
use tokio::io::{AsyncReadExt as _, AsyncWriteExt as _};
use futures_util::StreamExt as _; // for file payload.next()

//上传文件，可以上传多个文件,都是file字段
#[post("/upload")]
async fn upload(mut payload: Multipart) -> impl Responder {
    //文件上传根目录
    let base_path = "./assets/upload";

    //创建根目录
    if let Err(e) = create_dir_all(base_path).await {
        return  HttpResponse::Ok().body(format!("创建文件夹失败,请检查是否有权限! {}".to_string(),e));
    }
    
    //多个文件数组
    let mut file_paths = Vec::new();

    //其他参数 user_id
    let mut user_id = String::new();
    
    //其他参数 dept_id
    let mut dept_id = String::new();

    //遍历 payload 需要引入 use futures_util::StreamExt as _;
    while let Some(Ok(mut field)) = payload.next().await {
        // 获取字段信息
        let content_disposition = field.content_disposition();
        let field_name = content_disposition.get_name().unwrap();
        
        //file为上传文件字段
        if field_name == "file" {
            // Handle file upload
            let file_name = content_disposition.get_filename().unwrap_or("default_file");
            
            //完整路径 base_path(根目录) + 要上传的文件
            let file_path_local: String = format!("{}/{}", base_path, file_name);
            file_paths.push(file_path_local.clone());

            let mut file = File::create(file_path_local).await.unwrap();

            while let Some(chunk) = field.next().await {
                let data = chunk.unwrap();
                file.write_all(&data).await.unwrap();
            }
            
        //处理其他参数 userId
        } else if field_name == "userId" {
            while let Some(chunk) = field.next().await {
                let data = chunk.unwrap();
                user_id.push_str(&String::from_utf8_lossy(&data));
            }
         //处理其他参数 deptId
        } else if field_name == "deptId" {
            while let Some(chunk) = field.next().await {
                let data = chunk.unwrap();
                dept_id.push_str(&String::from_utf8_lossy(&data));
            }
        }
    }
    
    //file_paths 为路径数组
    
    return  HttpResponse::Ok().body("文件上传成功!".to_string());
 
}