package repositories

import (
    "xorm.io/xorm"
    "xps/datamodels"
)

type CompanyRepo struct {
    engine *xorm.Engine
}

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

func (d *CompanyRepo) GetById(ocId int64) (*datamodels.Company, error){
    data := &datamodels.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 *datamodels.Company) error{
    _, err := d.engine.Where("oc_id = ?", ocId).Update(data)
    return err
}