Cloudpods/backend/pkg/compute/models/guestcdrom.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 (
	"fmt"
	"time"

	"yunion.io/x/log"
	"yunion.io/x/pkg/errors"

	api "yunion.io/x/onecloud/pkg/apis/compute"
	"yunion.io/x/onecloud/pkg/cloudcommon/db"
)

// +onecloud:swagger-gen-ignore
type SGuestcdromManager struct {
	db.SModelBaseManager
}

var GuestcdromManager *SGuestcdromManager

func init() {
	GuestcdromManager = &SGuestcdromManager{
		SModelBaseManager: db.NewModelBaseManager(
			SGuestcdrom{},
			"guestcdrom_tbl",
			"guestcdrom",
			"guestcdroms",
		),
	}
	GuestcdromManager.SetVirtualObject(GuestcdromManager)
}

// +onecloud:swagger-gen-ignore
type SGuestcdrom struct {
	db.SModelBase

	RowId         int64     `primary:"true" auto_increment:"true" list:"user"`
	Ordinal       int       `nullable:"false" default:"0"`                // = Column(Integer, nullable=False, default=0)
	Id            string    `width:"36" charset:"ascii"`                  // = Column(VARCHAR(36, charset='ascii'), primary_key=True)
	ImageId       string    `width:"36" charset:"ascii" nullable:"true"`  // Column(VARCHAR(36, charset='ascii'), nullable=True)
	Name          string    `width:"64" charset:"ascii" nullable:"true"`  // Column(VARCHAR(64, charset='ascii'), nullable=True)
	Path          string    `width:"256" charset:"ascii" nullable:"true"` // Column(VARCHAR(256, charset='ascii'), nullable=True)
	Size          int64     `nullable:"false" default:"0"`                // = Column(Integer, nullable=False, default=0)
	BootIndex     int8      `nullable:"false" default:"-1" list:"user" update:"user"`
	UpdatedAt     time.Time `nullable:"false" updated_at:"true" nullable:"false"`
	UpdateVersion int       `default:"0" nullable:"false" auto_version:"true"`
}

func (self *SGuestcdrom) insertIso(imageId string) bool {
	if len(self.ImageId) == 0 {
		_, err := db.Update(self, func() error {
			self.ImageId = imageId
			self.Name = ""
			self.Path = ""
			self.Size = 0
			return nil
		})
		if err != nil {
			log.Errorf("insertISO saveupdate fail: %s", err)
			return false
		}
		return true
	} else {
		return false
	}
}

func (self *SGuestcdrom) insertIsoSucc(imageId string, path string, size int64, name string, bootIndex *int8) bool {
	if self.ImageId == imageId {
		_, err := db.Update(self, func() error {
			self.Name = name
			self.Path = path
			self.Size = size
			if bootIndex != nil {
				self.BootIndex = *bootIndex
			} else {
				self.BootIndex = -1
			}
			return nil
		})
		if err != nil {
			log.Errorf("insertIsoSucc saveUpdate fail %s", err)
			return false
		}
		return true
	} else {
		return false
	}
}

func (self *SGuestcdrom) ejectIso() bool {
	if len(self.ImageId) > 0 {
		_, err := db.Update(self, func() error {
			self.ImageId = ""
			self.Name = ""
			self.Path = ""
			self.Size = 0
			self.BootIndex = -1
			return nil
		})
		if err != nil {
			log.Errorf("ejectIso saveUpdate fail %s", err)
			return false
		}
		return true
	} else {
		return false
	}
}

func (self *SGuestcdrom) GetImage() (*SCachedimage, error) {
	if len(self.ImageId) == 0 {
		return nil, fmt.Errorf("empty image_id")
	}
	image, err := CachedimageManager.FetchById(self.ImageId)
	if err != nil {
		return nil, errors.Wrapf(err, "CachedimageManager.FetchById(%s)", self.ImageId)
	}
	return image.(*SCachedimage), nil
}

func (self *SGuestcdrom) GetDetails() string {
	if len(self.ImageId) > 0 {
		if self.Size > 0 {
			// Size is stored in bytes, convert to MB for display
			sizeMB := self.Size / 1024 / 1024
			return fmt.Sprintf("%s(%s/%dMB)", self.Name, self.ImageId, sizeMB)
		} else {
			return fmt.Sprintf("%s(inserting)", self.ImageId)
		}
	} else {
		return ""
	}
}

func (self *SGuestcdrom) SetBootIndex(bootIndex int8) error {
	_, err := db.Update(self, func() error {
		self.BootIndex = bootIndex
		return nil
	})
	return err
}

func (self *SGuestcdrom) getJsonDesc() *api.GuestcdromJsonDesc {
	if len(self.ImageId) > 0 && len(self.Path) > 0 {
		return &api.GuestcdromJsonDesc{
			Ordinal:   self.Ordinal,
			ImageId:   self.ImageId,
			Path:      self.Path,
			Name:      self.Name,
			Size:      self.Size,
			BootIndex: self.BootIndex,
		}
	}
	return nil
}