Cloudpods/backend/pkg/webconsole/models/commandlog.go

// Copyright 2019 Yunion
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package models

import (
	"context"
	"time"

	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/errors"
	"yunion.io/x/sqlchemy"

	api "yunion.io/x/onecloud/pkg/apis/webconsole"
	"yunion.io/x/onecloud/pkg/cloudcommon/consts"
	"yunion.io/x/onecloud/pkg/cloudcommon/db"
	"yunion.io/x/onecloud/pkg/mcclient"
)

var commandLogManager *SCommandLogManager

type CommandType string

const (
	CommandTypeSSH = "ssh"
)

func InitCommandLog() {
	commandLogManager = GetCommandLogManager()
}

func GetCommandLogManager() *SCommandLogManager {
	if commandLogManager != nil {
		return commandLogManager
	}
	commandLogManager = &SCommandLogManager{
		SOpsLogManager: db.NewOpsLogManager(SCommandLog{}, "command_log_tbl", "commandlog", "commandlogs", "start_time", consts.OpsLogWithClickhouse),
	}
	commandLogManager.SetVirtualObject(commandLogManager)
	return commandLogManager
}

type SCommandLogManager struct {
	db.SOpsLogManager
}

type SCommandLog struct {
	db.SOpsLog

	SessionId  string      `width:"128" charset:"ascii" list:"user"`
	AccessedAt time.Time   `nullable:"false" list:"user" create:"required"`
	Type       CommandType `width:"32" charset:"utf8" nullable:"true" list:"user" create:"required"`
	LoginUser  string      `charset:"utf8" list:"user" create:"required"`
	StartTime  time.Time   `list:"user" create:"required"`
	Ps1        string      `charset:"utf8" list:"user" create:"optional" json:"ps1"`
	Command    string      `charset:"utf8" list:"user" create:"required"`
}

type CommandLogCreateInput struct {
	ObjId           string
	ObjName         string
	ObjType         string
	Action          string
	UserId          string
	User            string
	TenantId        string
	Tenant          string
	DomainId        string
	Domain          string
	ProjectDomainId string
	ProjectDomain   string
	Roles           string
	SessionId       string
	AccessedAt      time.Time
	Type            CommandType
	LoginUser       string
	StartTime       time.Time
	Ps1             string `json:"ps1"`
	Command         string
	Notes           jsonutils.JSONObject
}

func (m *SCommandLogManager) Create(ctx context.Context, userCred mcclient.TokenCredential, input *CommandLogCreateInput) (*SCommandLog, error) {
	record := &SCommandLog{}
	record.ObjId = input.ObjId
	record.ObjName = input.ObjName
	record.Action = input.Action
	record.UserId = input.UserId
	record.User = input.User
	record.ProjectId = input.TenantId
	record.Project = input.Tenant
	record.DomainId = input.DomainId
	record.Domain = input.Domain
	record.ProjectDomainId = input.ProjectDomainId
	record.ProjectDomain = input.ProjectDomain
	record.Roles = input.Roles
	record.SessionId = input.SessionId
	record.AccessedAt = input.AccessedAt
	record.Type = input.Type
	record.LoginUser = input.LoginUser
	record.StartTime = input.StartTime
	record.Ps1 = input.Ps1
	record.Command = input.Command
	record.Notes = input.Notes.String()
	record.OpsTime = input.StartTime

	record.SetModelManager(m, record)
	err := m.TableSpec().Insert(ctx, record)
	if err != nil {
		return nil, errors.Wrap(err, "Insert")
	}
	return record, nil
}

func (m *SCommandLogManager) ListItemFilter(
	ctx context.Context,
	q *sqlchemy.SQuery,
	userCred mcclient.TokenCredential,
	input api.CommandLogListInput,
) (*sqlchemy.SQuery, error) {
	q, err := m.SOpsLogManager.ListItemFilter(ctx, q, userCred, input.OpsLogListInput)
	if err != nil {
		return nil, errors.Wrap(err, "SOpsLogManager.ListItemFilter")
	}

	return q, nil
}