add seed data generation (#13)

* merge upstream

* add seed data generation

src/api/user_info.rs

@@ -1,15 +1,23 @@
 use std::collections::HashMap;
-
+use std::io::SeekFrom;
+use std::ptr::null;
 use actix_identity::Identity;
 use actix_web::{HttpResponse, post, web};
+use chrono::{FixedOffset, Utc};
+use sea_orm::ActiveValue::NotSet;
 use serde::{Deserialize, Serialize};
 use crate::api::JsonResponse;
 use crate::AppState;
+use crate::entity::{doc_info, tag_info};
 use crate::entity::user_info::Model;
 use crate::errors::{AppError, UserError};
 use crate::service::user_info::Mutation;
 use crate::service::user_info::Query;
 
+fn now()->chrono::DateTime<FixedOffset>{
+    Utc::now().with_timezone(&FixedOffset::east_opt(3600*8).unwrap())
+}
+
 pub(crate) fn create_auth_token(user: &Model) -> u64 {
     use std::{
         collections::hash_map::DefaultHasher,
@@ -58,8 +66,51 @@ async fn login(
 #[post("/v1.0/register")]
 async fn register(model: web::Json<Model>, data: web::Data<AppState>) -> Result<HttpResponse, AppError> {
     let mut result = HashMap::new();
+    let u = Query::find_user_infos(&data.conn, &model.email).await?;
+    if let Some(_) = u {
+        let json_response = JsonResponse {
+            code: 500,
+            err: "Email registered!".to_owned(),
+            data: (),
+        };
+        return Ok(HttpResponse::Ok()
+            .content_type("application/json")
+            .body(serde_json::to_string(&json_response)?));
+    }
+
     let usr = Mutation::create_user(&data.conn, &model).await?;
-    result.insert("uid", usr.uid.unwrap());
+    result.insert("uid", usr.uid.clone().unwrap());
+    crate::service::doc_info::Mutation::create_doc_info(&data.conn, doc_info::Model{
+        did:Default::default(),
+        uid:  usr.uid.clone().unwrap(),
+        doc_name: "/".into(),
+        size: 0,
+        location: "".into(),
+        r#type: "folder".to_string(),
+        created_at: now(),
+        updated_at: now(),
+        is_deleted:Default::default(),
+    }).await?;
+    let tnm = vec!["视频","图片","音乐","文档"];
+    let tregx = vec![
+        ".*\\.(mpg|mpeg|avi|rm|rmvb|mov|wmv|asf|dat|asx|wvx|mpe|mpa)",
+        ".*\\.(png|tif|gif|pcx|tga|exif|fpx|svg|psd|cdr|pcd|dxf|ufo|eps|ai|raw|WMF|webp|avif|apng)",
+        ".*\\.(WAV|FLAC|APE|ALAC|WavPack|WV|MP3|AAC|Ogg|Vorbis|Opus)",
+        ".*\\.(pdf|doc|ppt|yml|xml|htm|json|csv|txt|ini|xsl|wps|rtf|hlp)"
+    ];
+    for i in 0..4 {
+        crate::service::tag_info::Mutation::create_tag(&data.conn, tag_info::Model{
+            tid: Default::default(),
+            uid: usr.uid.clone().unwrap(),
+            tag_name: tnm[i].to_owned(),
+            regx: tregx[i].to_owned(),
+            color: (i+1).to_owned() as i16,
+            icon: (i+1).to_owned() as i16,
+            folder_id: 0,
+            created_at: Default::default(),
+            updated_at: Default::default(),
+        }).await?;
+    }
     let json_response = JsonResponse {
         code: 200,
         err: "".to_owned(),

src/entity/kb2_doc.rs

@@ -11,7 +11,6 @@ pub struct Model {
     pub kb_id: i64,
     #[sea_orm(index)]
     pub did: i64,
-<<<<<<< HEAD
     #[serde(skip_deserializing)]
     pub kb_progress: f32,
     #[serde(skip_deserializing)]
@@ -20,8 +19,6 @@ pub struct Model {
     pub updated_at: DateTime<FixedOffset>,
     #[serde(skip_deserializing)]
     pub is_deleted: bool,
-=======
->>>>>>> upstream/main
 }
 
 #[derive(Debug, Clone, Copy, EnumIter)]

src/entity/tag_info.rs

@@ -11,19 +11,12 @@ pub struct Model {
     #[sea_orm(index)]
     pub uid: i64,
     pub tag_name: String,
-<<<<<<< HEAD
     #[serde(skip_deserializing)]
     pub regx: String,
     pub color: i16,
     pub icon: i16,
     #[serde(skip_deserializing)]
     pub folder_id: i64,
-=======
-    pub regx: Option<String>,
-    pub color: u16,
-    pub icon: u16,
-    pub dir: Option<String>,
->>>>>>> upstream/main
 
     #[serde(skip_deserializing)]
     pub created_at: DateTime<FixedOffset>,

src/main.rs

@@ -98,11 +98,8 @@ fn init(cfg: &mut web::ServiceConfig) {
     cfg.service(api::kb_info::delete);
     cfg.service(api::kb_info::list);
     cfg.service(api::kb_info::add_docs_to_kb);
-<<<<<<< HEAD
     cfg.service(api::kb_info::anti_kb_docs);
     cfg.service(api::kb_info::all_relevents);
-=======
->>>>>>> upstream/main
 
     cfg.service(api::doc_info::list);
     cfg.service(api::doc_info::delete);

src/service/tag_info.rs

@@ -1,6 +1,6 @@
 use chrono::{FixedOffset, Utc};
 use sea_orm::{ActiveModelTrait, DbConn, DbErr, DeleteResult, EntityTrait, PaginatorTrait, QueryOrder, ColumnTrait, QueryFilter};
-use sea_orm::ActiveValue::Set;
+use sea_orm::ActiveValue::{Set, NotSet};
 use crate::entity::tag_info;
 use crate::entity::tag_info::Entity;
 
@@ -51,7 +51,10 @@ impl Mutation {
             regx: Set(form_data.regx.to_owned()),
             color: Set(form_data.color.to_owned()),
             icon: Set(form_data.icon.to_owned()),
-            folder_id: Set(form_data.folder_id.to_owned()),
+            folder_id: match form_data.folder_id {
+                0 => NotSet,
+                _ => Set(form_data.folder_id.to_owned())
+            },
             created_at: Set(now()),
             updated_at: Set(now()),
         }

src/service/user_info.rs

@@ -23,8 +23,11 @@ impl Query {
             .await
     }
 
-    pub async fn find_user_infos(db: &DbConn) -> Result<Vec<user_info::Model>, DbErr> {
-        Entity::find().all(db).await
+    pub async fn find_user_infos(db: &DbConn, email:&String) -> Result<Option<user_info::Model>, DbErr> {
+        Entity::find()
+            .filter(user_info::Column::Email.eq(email))
+            .one(db)
+            .await
     }
 
     pub async fn find_user_infos_in_page(