package main

import (
	"fmt"

	"github.com/jinzhu/gorm"
	_ "github.com/lib/pq"
)

type Company struct {
	gorm.Model
	Name  string
	Users []User
}

type User struct {
	gorm.Model
	Company   Company
	CompanyID uint   `gorm:"column:company_id"`
	Email     string `gorm:"column:email"`
}

func main() {
	db, err := gorm.Open("postgres", "user=gorm password=123 dbname=gorm sslmode=disable")
	if err != nil {
		panic(err)
	}

	db.CreateTable(&User{}, &Company{})

	// db.Create(&Company{
	// 	Users: []User{
	// 		{Email: "admin@google.com"},
	// 	},
	// })

	db.LogMode(true)
	user := User{}
	db.Model(&user).Where("email = ?", "admin@google.com").Find(&user)
	db.Model(&user).Related(&user.Company)
	fmt.Println(user.ID)
	fmt.Println(user.Company.ID)
}