package repository

import (
    "xorm.io/xorm"
    "xps/pkg/system/datamodel"
)

type CompanyRepo struct {
    engine *xorm.Engine
}

func NewCompanyRepo(engine *xorm.Engine) *CompanyRepo {
    return &CompanyRepo{
        engine: engine,
    }
}

func (d *CompanyRepo) GetById(ocId int64) (*datamodel.Company, error) {
    data := &datamodel.Company{}

    has, err := d.engine.Where("oc_id = ?", ocId).Get(data)

    if !has || err != nil {
        return nil, err
    } else {
        return data, nil
    }
}

func (d *CompanyRepo) Update(ocId int64, data *datamodel.Company) error {
    _, err := d.engine.Where("oc_id = ?", ocId).Update(data)
    return err
}