From 1b861a41f9bbc7262affae671e11903e32b54cdf Mon Sep 17 00:00:00 2001
From: Qiu Jian <qiujian@yunionyun.com>
Date: Wed, 22 Apr 2020 22:36:15 +0800
Subject: [PATCH] fix: opslog create fail

---
 pkg/apis/input.go            | 25 +++++++++++++++
 pkg/apis/output.go           | 35 +++++++++++++++++++++
 pkg/cloudcommon/db/opslog.go | 61 ++++++++++++++++++++++++++++++++++++
 3 files changed, 121 insertions(+)

diff --git a/pkg/apis/input.go b/pkg/apis/input.go
index 819a20921c..729252ecc7 100644
--- a/pkg/apis/input.go
+++ b/pkg/apis/input.go
@@ -224,3 +224,28 @@ type EnabledStatusInfrasResourceBaseCreateInput struct {
 	StatusInfrasResourceBaseCreateInput
 	EnabledBaseResourceCreateInput
 }
+
+type OpsLogCreateInput struct {
+	ModelBaseCreateInput
+
+	ObjType string `json:"obj_type"`
+	ObjId   string `json:"obj_id"`
+	ObjName string `json:"obj_name"`
+	Action  string `json:"action"`
+	Notes   string `json:"notes"`
+
+	ProjectId string `json:"tenant_id"`
+	Project   string `json:"tenant"`
+
+	ProjectDomainId string `json:"project_domain_id"`
+	ProjectDomain   string `json:"project_domain"`
+
+	UserId   string `json:"user_id"`
+	User     string `json:"user"`
+	DomainId string `json:"domain_id"`
+	Domain   string `json:"domain"`
+	Roles    string `json:"roles"`
+
+	OwnerDomainId  string `json:"owner_domain_id"`
+	OwnerProjectId string `json:"owner_tenant_id"`
+}
diff --git a/pkg/apis/output.go b/pkg/apis/output.go
index d0c6f78b31..71abf6bfc3 100644
--- a/pkg/apis/output.go
+++ b/pkg/apis/output.go
@@ -14,6 +14,10 @@
 
 package apis
 
+import (
+	"time"
+)
+
 type ModelBaseDetails struct {
 	Meta
 
@@ -160,3 +164,34 @@ type EnabledStatusInfrasResourceBaseDetails struct {
 type ChangeOwnerCandidateDomainsOutput struct {
 	Candidates []SharedDomain `json:"candidates"`
 }
+
+type OpsLogDetails struct {
+	ModelBaseDetails
+
+	Id      int64  `json:"id"`
+	ObjType string `json:"obj_type"`
+	ObjId   string `json:"obj_id"`
+	ObjName string `json:"obj_name"`
+	Action  string `json:"action"`
+	Notes   string `json:"notes"`
+
+	ProjectId string `json:"tenant_id"`
+	Project   string `json:"tenant"`
+
+	ProjectDomainId string `json:"project_domain_id"`
+	ProjectDomain   string `json:"project_domain"`
+
+	UserId   string `json:"user_id"`
+	User     string `json:"user"`
+	DomainId string `json:"domain_id"`
+	Domain   string `json:"domain"`
+	Roles    string `json:"roles"`
+
+	OpsTime time.Time `json:"ops_time"`
+
+	OwnerDomainId  string `json:"owner_domain_id"`
+	OwnerProjectId string `json:"owner_project_id"`
+
+	OwnerDomain  string `json:"owner_domain"`
+	OwnerProject string `json:"owner_tenant"`
+}
diff --git a/pkg/cloudcommon/db/opslog.go b/pkg/cloudcommon/db/opslog.go
index 139f25fa87..4af9998100 100644
--- a/pkg/cloudcommon/db/opslog.go
+++ b/pkg/cloudcommon/db/opslog.go
@@ -24,14 +24,17 @@ import (
 
 	"yunion.io/x/jsonutils"
 	"yunion.io/x/log"
+	"yunion.io/x/pkg/util/reflectutils"
 	"yunion.io/x/pkg/util/stringutils"
 	"yunion.io/x/sqlchemy"
 
+	"yunion.io/x/onecloud/pkg/apis"
 	"yunion.io/x/onecloud/pkg/appsrv"
 	"yunion.io/x/onecloud/pkg/cloudcommon/consts"
 	"yunion.io/x/onecloud/pkg/httperrors"
 	"yunion.io/x/onecloud/pkg/mcclient"
 	"yunion.io/x/onecloud/pkg/util/rbacutils"
+	"yunion.io/x/onecloud/pkg/util/stringutils2"
 )
 
 const (
@@ -661,3 +664,61 @@ func (manager *SOpsLogManager) GetPagingConfig() *SPagingConfig {
 		DefaultLimit: 20,
 	}
 }
+
+func (manager *SOpsLogManager) FetchOwnerId(ctx context.Context, data jsonutils.JSONObject) (mcclient.IIdentityProvider, error) {
+	return FetchProjectInfo(ctx, data)
+}
+
+func (manager *SOpsLogManager) ValidateCreateData(ctx context.Context,
+	userCred mcclient.TokenCredential,
+	ownerId mcclient.IIdentityProvider,
+	query jsonutils.JSONObject,
+	data apis.OpsLogCreateInput,
+) (apis.OpsLogCreateInput, error) {
+	data.Project = ownerId.GetProjectName()
+	data.ProjectId = ownerId.GetProjectId()
+	data.ProjectDomainId = ownerId.GetProjectDomainId()
+	data.ProjectDomain = ownerId.GetProjectDomain()
+	return data, nil
+}
+
+func (manager *SOpsLogManager) FetchCustomizeColumns(
+	ctx context.Context,
+	userCred mcclient.TokenCredential,
+	query jsonutils.JSONObject,
+	objs []interface{},
+	fields stringutils2.SSortedStrings,
+	isList bool,
+) []apis.OpsLogDetails {
+	rows := make([]apis.OpsLogDetails, len(objs))
+
+	projectIds := make([]string, len(rows))
+	domainIds := make([]string, len(rows))
+	for i := range rows {
+		var base *SOpsLog
+		err := reflectutils.FindAnonymouStructPointer(objs[i], &base)
+		if err != nil {
+			log.Errorf("Cannot find OpsLog in %#v: %s", objs[i], err)
+		} else {
+			if len(base.OwnerProjectId) > 0 {
+				projectIds[i] = base.OwnerProjectId
+			} else if len(base.OwnerDomainId) > 0 {
+				domainIds[i] = base.OwnerDomainId
+			}
+		}
+	}
+
+	projects := DefaultProjectsFetcher(ctx, projectIds, false)
+	domains := DefaultProjectsFetcher(ctx, domainIds, true)
+
+	for i := range rows {
+		if project, ok := projects[projectIds[i]]; ok {
+			rows[i].OwnerProject = project.Name
+			rows[i].OwnerDomain = project.Domain
+		} else if domain, ok := domains[domainIds[i]]; ok {
+			rows[i].OwnerDomain = domain.Name
+		}
+	}
+
+	return rows
+}