From 74ff43b76ac70bc086ee309b1409958974396f52 Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Tue, 26 Jul 2022 17:10:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=9F=A5=E8=AF=A2=E7=99=BB?=
 =?UTF-8?q?=E5=BD=95=E4=BF=A1=E6=81=AF=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 models/user_analysis_for_activity.go | 14 +++++++
 routers/api/v1/api.go                |  1 +
 routers/repo/user_data_analysis.go   | 60 ++++++++++++++++++++++++++++
 3 files changed, 75 insertions(+)

diff --git a/models/user_analysis_for_activity.go b/models/user_analysis_for_activity.go
index d8e4a5500..2066697d2 100644
--- a/models/user_analysis_for_activity.go
+++ b/models/user_analysis_for_activity.go
@@ -6,6 +6,7 @@ import (
 
 	"code.gitea.io/gitea/modules/log"
 	"code.gitea.io/gitea/modules/timeutil"
+	"xorm.io/builder"
 )
 
 type UserBusinessAnalysisForActivity struct {
@@ -435,3 +436,16 @@ func queryUserModelPublic(start_unix int64, end_unix int64, publicAllRepo map[in
 	}
 	return resultMap
 }
+
+func QueryUserLoginInfo(userIds []int64) []*UserLoginLog {
+	statictisSess := xStatistic.NewSession()
+	defer statictisSess.Close()
+	var cond = builder.NewCond()
+	cond = cond.And(builder.In("u_id", userIds))
+	statictisSess.Select("*").Table(new(UserLoginLog)).Where(cond)
+	loginList := make([]*UserLoginLog, 0)
+
+	statictisSess.Find(&loginList)
+
+	return loginList
+}
diff --git a/routers/api/v1/api.go b/routers/api/v1/api.go
index 3d9452f93..aa51c6e1a 100755
--- a/routers/api/v1/api.go
+++ b/routers/api/v1/api.go
@@ -571,6 +571,7 @@ func RegisterRoutes(m *macaron.Macaron) {
 		m.Get("/query_user_yesterday", operationReq, repo_ext.QueryUserStaticYesterday)
 		m.Get("/query_user_all", operationReq, repo_ext.QueryUserStaticAll)
 		m.Get("/query_user_activity", operationReq, repo_ext.QueryUserActivity)
+		m.Get("/query_user_login", operationReq, repo_ext.QueryUserLoginInfo)
 		//cloudbrain board
 		m.Group("/cloudbrainboard", func() {
 			m.Get("/downloadAll", repo.DownloadCloudBrainBoard)
diff --git a/routers/repo/user_data_analysis.go b/routers/repo/user_data_analysis.go
index bafe46878..be64fddf9 100755
--- a/routers/repo/user_data_analysis.go
+++ b/routers/repo/user_data_analysis.go
@@ -5,6 +5,8 @@ import (
 	"net/http"
 	"net/url"
 	"os"
+	"strconv"
+	"strings"
 	"time"
 
 	"code.gitea.io/gitea/models"
@@ -847,3 +849,61 @@ func writeUserActivityToExcel(startTime time.Time, endTime time.Time, filePath s
 		log.Info("write to file succeed, filepath=" + filePath)
 	}
 }
+
+// URL:  /api/v1/query_user_login?userId=1,2,3,4
+func QueryUserLoginInfo(ctx *context.Context) {
+	userId := ctx.Query("userId")
+	userIds := strings.Split(userId, ",")
+	userIdInt := make([]int64, 0)
+	for _, id := range userIds {
+		idInt, err := strconv.ParseInt(id, 10, 64)
+		if err == nil {
+			userIdInt = append(userIdInt, idInt)
+		}
+	}
+	result := models.QueryUserLoginInfo(userIdInt)
+
+	xlsx := excelize.NewFile()
+	sheetName := ctx.Tr("用户登录信息")
+	index := xlsx.NewSheet(sheetName)
+	xlsx.DeleteSheet("Sheet1")
+
+	excelHeader := make([]string, 0)
+	excelHeader = append(excelHeader, "用户ID")
+	excelHeader = append(excelHeader, "登录IP")
+	excelHeader = append(excelHeader, "登录时间")
+
+	excelHeaderMap := make(map[string]string, 0)
+	var j byte
+	j = 0
+	for _, value := range excelHeader {
+		excelColumn := getColumn(j) + fmt.Sprint(1)
+		log.Info("excelColumn=" + excelColumn)
+		excelHeaderMap[excelColumn] = value
+		j++
+	}
+	for k, v := range excelHeaderMap {
+		//设置单元格的值
+		xlsx.SetCellValue(sheetName, k, v)
+	}
+	for i, userLogin := range result {
+		row := i + 2
+		rows := fmt.Sprint(row)
+		var tmp byte
+		tmp = 0
+		xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userLogin.UId)
+		tmp = tmp + 1
+		xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userLogin.IpAddr)
+		tmp = tmp + 1
+		formatTime := userLogin.CreatedUnix.Format("2006-01-02 15:04:05")
+		xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime[0:len(formatTime)-3])
+	}
+	//设置默认打开的表单
+	xlsx.SetActiveSheet(index)
+	filename := sheetName + "_" + time.Now().Format("2006-01-02 15:04:05") + ".xlsx"
+	ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
+	ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
+	if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
+		log.Info("writer exel error." + err.Error())
+	}
+}