hotime/example/app/user.go

package app

import (
	. "../../../hotime"
	. "../../../hotime/common"
	"../../../hotime/dri/download"
	"encoding/base64"
	"io/ioutil"
	"os"
	"strings"
	"time"
	//"strings"
)

var User = Ctr{
	"test": func(this *Context) {
		this.Session("uid", 1)
		this.Display(0, "开始测试")
	},
	//获取个人信息
	"info": func(this *Context) {

		if this.Session("user_id").Data == nil {
			this.Display(2, "还没有登录")
			return
		}
		if this.Session("user_id").Data == nil {
			this.Display(2, "还没有登录")
			return
		}

		user := this.Db.Get("user", "*", Map{"id": this.Session("user_id").Data})
		if user == nil {
			this.Display(4, "找不到该用户")
			return
		}
		company := this.Db.Get("company", "id,name,status", Map{"id": user.GetCeilInt("company_id")})
		user["company"] = company

		t := time.Now().Unix()

		questions := this.Db.Select("question", "*", Map{"AND": Map{"org_id": user.GetCeilInt("org_id"), "start_time[<=]": t, "end_time[>=]": t, "state": 0}})
		re := []Map{}

		for _, v := range questions {
			questionCompany := this.Db.Get("question_company", "*", Map{"AND": Map{"question_id": v.GetCeilInt("id"), "company_id": user.GetCeilInt("company_id")}})
			if questionCompany == nil {
				questionCompany = Map{
					"name":          v.GetString("name"),
					"question_data": v.GetString("question_data"),
					"org_id":        v.GetString("org_id"),
					"question_id":   v.GetString("id"),
					"company_id":    user.GetCeilInt("company_id"),
					"user_id":       user.GetString("id"),
					"create_time":   t,
					"modify_time":   t,
					"status":        0,
					"state":         0,
				}
				questionCompany["id"] = this.Db.Insert("question_company", questionCompany)
				if questionCompany.GetCeilInt("id") == 0 {
					this.Display(4, "无法创建调查数据")
					return
				}
			}

			delete(v, "question_data")
			questionCompany["question"] = v
			questionCompany["question_data"] = questionCompany.GetSlice("question_data")
			re = append(re, questionCompany)
		}
		user["questions"] = re

		this.Display(0, user)

	},
	//身份认证
	"auth": func(this *Context) {

		//if this.Session("wechatInfo").Data == nil {
		//	this.Display(2, "尚未登录")
		//	return
		//}

		//if this.Session("code").Data == nil {
		//	this.Display(8, "验证码没有获取")
		//	return
		//}
		orgId := ObjToInt(this.Req.FormValue("org_id"))
		//questionCompanyId:=ObjToInt(this.Req.FormValue("id"))
		if orgId == 0 {
			this.Display(3, "没有选择组织")
			return
		}
		name := this.Req.FormValue("name")
		phone := this.Req.FormValue("phone")
		idcard := this.Req.FormValue("idcard")
		idcardFrontImg := this.Req.FormValue("idcard_front_img")
		idcardBackImg := this.Req.FormValue("idcard_back_img")
		companyImg := this.Req.FormValue("company_img")
		companyName := this.Req.FormValue("company_name")
		companySn := this.Req.FormValue("company_sn")

		if name == "" || phone == "" || idcard == "" || idcardFrontImg == "" || idcardBackImg == "" || companyImg == "" || companyName == "" || companySn == "" {
			this.Display(3, "参数异常")
			return
		}
		t := time.Now().Unix()

		company := this.Db.Get("company", "id", Map{"sn": companySn})
		if company == nil {
			company = Map{
				"name":        companyName,
				"sn":          companySn,
				"create_time": t,
				"org_id":      orgId,
				"modify_time": t,
				"status":      1,
				"state":       0,
				"img":         companyImg,
			}

			company["id"] = this.Db.Insert("company", company)

			if company.GetCeilInt("id") == 0 {
				this.Display(4, "无法创建企业")
				return
			}
		} else {
			company["status"] = 1
			company["modify_time"] = t
			company["img"] = companyImg
			company["org_id"] = orgId
			company["name"] = companyName
			//company["img"] = companyImg
			re := this.Db.Update("company", company, Map{"id": company.GetCeilInt("id")})
			if re == 0 {
				this.Display(4, "无法更新企业")
				return
			}
		}

		userInfo := this.Session("wechatInfo").ToMap()
		user := this.Db.Get("user", "*", Map{"phone": phone})
		data := Map{"name": name,
			"phone":            phone,
			"org_id":           orgId,
			"idcard":           idcard,
			"idcard_front_img": idcardFrontImg,
			"company_id":       company.GetCeilInt("id"),
			"idcard_back_img":  idcardBackImg,
			"modify_time":      t,
			"status":           1,
			"company_img":      companyImg}

		//用户已经注册则更新
		if user != nil {
			re := this.Db.Update("user", data, Map{"id": user.GetCeilInt("id")})
			if re == 0 {
				this.Display(4, "无法更新用户")
				return
			}
		} else {
			data["create_time"] = t
			if userInfo != nil {
				data["nickname"] = userInfo.GetString("nickname")
				path := time.Now().Format(this.Config.GetString("imgPath"))
				filename := Md5(ObjToStr(t)) + ".jpg"
				down := download.Down(userInfo.GetString("avatar"), this.Config.GetString("tpt")+"/"+path, filename)
				if down {
					data["avatar"] = path + filename
				}
			}
			user = data
			user["id"] = this.Db.Insert("user", data)
			if user.GetCeilInt("id") == 0 {
				this.Display(4, "无法创建用户")
				return
			}
		}
		if userInfo != nil {
			delete(userInfo, "avatar")
			delete(userInfo, "nickname")
			userInfo["modify_time"] = t
			userInfo["user_id"] = user.GetCeilInt("id")

			wechatDb := this.Db.Get("wechat", "*", Map{"openid": userInfo.GetString("openid")})

			if wechatDb != nil {

				this.Db.Update("wechat", userInfo, Map{"id": wechatDb.GetCeilInt("id")})
				//userInfo["wid"]=wechatDb.GetCeilInt("wid")
			} else {
				userInfo["create_time"] = t
				userInfo["id"] = this.Db.Insert("wechat", userInfo)
			}
		}
		this.Session("user_id", user.GetCeilInt("id"))
		this.Display(0, this.SessionId)

	},

	//修改密码g
	"forget": func(this *Context) {
		if this.Session("code").Data == nil {
			this.Display(8, "验证码没有获取")
			return
		}

		if this.Req.FormValue("code") == "" || this.Req.FormValue("umobile") == "" {
			this.Display(3, "验证码或者手机号没有输入")
			return
		}

		if this.Req.FormValue("umobile") != this.Session("umobile").ToStr() ||
			this.Req.FormValue("code") != this.Session("code").ToStr() {
			this.Display(14, "验证码验证失败")
			return
		}

		if len(this.Req.FormValue("upwd")) < 6 {
			this.Display(3, "密码没有输入")
			return

		}

		res := this.Db.Update("user", Map{"upwd": Md5(this.Req.FormValue("upwd"))}, Map{"umobile": this.Req.FormValue("umobile")})
		if res == 0 {
			this.Display(5, "找不到该用户")
			return
		}

		this.Display(0, "成功")
	},

	"upload": func(this *Context) {
		file := this.Req.FormValue("file")
		if len(file) < 100 {
			this.Display(3, "图片上传错误")
			return
		}
		//fmt.Println(uimg)
		btes, e := base64.StdEncoding.DecodeString(file[strings.Index(file, ",")+1:]) //成图片文件并把文件写入到buffer
		//btes, e := base64.StdEncoding.DecodeString(file) //成图片文件并把文件写入到buffer
		if e != nil {
			this.Display(3, "无法解析图片")
			return
		}

		//uimgPath:=time.Now().Format(this.Config.GetString("uimgPath"))
		path := time.Now().Format(this.Config.GetString("wxFilePath"))
		os.MkdirAll(this.Config.GetString("tpt")+"/"+path, os.ModeDir)
		filePath := path + Md5(ObjToStr(time.Now().Unix())) + ".jpg"

		err2 := ioutil.WriteFile(this.Config.GetString("tpt")+"/"+filePath, btes, 0666) //buffer输出到jpg文件中（不做处理，直接写到文件）

		if err2 != nil {
			this.Display(3, "图片保存失败")
			return
		}
		this.Display(0, filePath)
	},
	//"upload": func(this *Context) {
	//
	//	//读取网络文件
	//	fi, fheader, err := this.Req.FormFile("file")
	//	if err != nil {
	//		this.Display(3, err)
	//		return
	//
	//	}
	//	filePath := this.Config.GetString("wxFilePath")
	//	if filePath == "" {
	//		filePath = "file/2006/01/02/"
	//	}
	//
	//	path := time.Now().Format(filePath)
	//	e := os.MkdirAll(this.Config.GetString("tpt")+"/"+path, os.ModeDir)
	//
	//	if e != nil {
	//		this.Display(3, e)
	//		return
	//	}
	//
	//	filePath = path + Md5(ObjToStr(RandX(100000, 9999999))) + fheader.Filename[strings.LastIndex(fheader.Filename, "."):]
	//	newFile, e := os.Create(this.Config.GetString("tpt") + "/" + filePath)
	//
	//	if e != nil {
	//		this.Display(3, e)
	//		return
	//	}
	//
	//	_, e = io.Copy(newFile, fi)
	//
	//	if e != nil {
	//		this.Display(3, e)
	//		return
	//	}
	//
	//	//this.Display(0, filePath)
	//
	//	//读取excel
	//	//fmt.Println(Org[orgId],OrgId)
	//
	//
	//	this.Display(0, filePath)
	//
	//},
	//更新个人资料
	//"update": func(this *Context) {
	//	if this.Req.Form["unickname"] == nil ||
	//		this.Req.Form["uemail"] == nil ||
	//		this.Req.Form["usex"] == nil ||
	//		this.Req.Form["ubirthday"] == nil ||
	//		this.Req.Form["ubelieve"] == nil ||
	//		this.Req.Form["utaste"] == nil ||
	//		this.Req.Form["uname"] == nil ||
	//		this.Req.Form["uaddreman"] == nil ||
	//		this.Req.Form["uaddremobile"] == nil ||
	//		this.Req.Form["uaddre"] == nil ||
	//		this.Req.Form["uidcard"] == nil {
	//		this.Display(3, "参数不足")
	//	}
	//	//更新代码
	//	update := Map{
	//		"unickname":    this.Req.FormValue("unickname"),
	//		"uemail":       this.Req.FormValue("uemail"),
	//		"usex":         this.Req.FormValue("usex"),
	//		"ubelieve":     this.Req.FormValue("ubelieve"),
	//		"utaste":       this.Req.FormValue("utaste"),
	//		"uaddreman":    this.Req.FormValue("uaddreman"),
	//		"uidcard":      this.Req.FormValue("uidcard"),
	//		"ubirthday":    this.Req.FormValue("ubirthday"),
	//		"uaddremobile": this.Req.FormValue("uaddremobile"),
	//		"uaddre":       this.Req.FormValue("uaddre"),
	//		"uname":        this.Req.FormValue("uname"),
	//	}
	//
	//	this.Db.Update("user", update, Map{"uid": this.Session("user_id").Data})
	//	this.Display(0, "ok")
	//
	//},
}