models

type Attribute struct {
	Name   string  `json:"name" bson:"name"`
	Values []Value `json:"values" bson:"values"`
}

type CategoryGroupWiseCount struct {
	CategoryGroup []string `json:"_id" bson:"_id"`
	Count         int      `json:"category_count" bson:"category_count"`
}

type CategoryWiseCount struct {
	Category string `json:"_id" bson:"_id"`
	Count    int    `json:"category_count" bson:"category_count"`
}

type Entity struct {
	Id              bson.ObjectId        `json:"id" bson:"_id,omitempty"`
	Title           string               `json:"title" bson:"title"`
	ImageURL        string               `json:"image_url" bson:"image_url"`
	Source          string               `json:"source" bson:"source"`
	SourceSignature string               `json:"source_signature" bson:"source_signature"`
	SourceDate      time.Time            `json:"source_date" bson:"source_date"`
	Attributes      map[string]Attribute `json:"attributes" bson:"attributes"`
	Links           []Link               `json:"links" bson:"links"`
	Categories      []string             `json:"categories" bson:"categories"`
	CreatedAt       time.Time            `json:"created_at" bson:"created_at"`
	UpdatedAt       time.Time            `json:"updated_at" bson:"updated_at"`
	Snippet         string               `json:"snippet" bson:"snippet"`
	SearchText      string               `json:"search_text" bson:"search_text"`
}

type EntityStats struct {
	Id                     bson.ObjectId            `json:"-" bson:"_id,omitempty"`
	EntityCount            int                      `json:"entity_count" bson:"entity_count"`
	RelationCount          int                      `json:"relation_count" bson:"relation_count"`
	CategoryWiseCount      []CategoryWiseCount      `json:"category_wise_count" bson:"category_wise_count"`
	CategoryGroupWiseCount []CategoryGroupWiseCount `json:"category_group_wise_count" bson:"category_group_wise_count"`
	CreatedAt              time.Time                `json:"created_at" bson:"created_at"`
}

type Link struct {
	Title string      `json:"title" bson:"title"`
	Dates []time.Time `json:"dates" bson:"dates"`
}

type Login struct {
	Username string `json:"username" bson:"username"`
	Password string `json:"password" bson:"password"`
}

type NERResult struct {
	Category   string
	EntityName string
}

type NewReader struct {
	Email    string `json:"email" bson:"email"`
	Password string `json:"password" bson:"password"`
}

type NewUser struct {
	Name     string `json:"name" bson:"name"`
	Email    string `json:"email" bson:"email"`
	Role     string `json:"role" bson:"role"`
	Password string `json:"password" bson:"password"`
}

type NormalizedName struct {
	Id             bson.ObjectId `json:"id" bson:"_id"`
	SearchText     string        `json:"search_text" bson:"search_text"`
	NormalizedText string        `json:"normalized_text" bson:"normalized_text"`
	CreatedAt      time.Time     `json:"created_at" bson:"created_at"`
}

type SearchResult struct {
	Title      string               `json:"title" bson:"title"`
	Snippet    string               `json:"snippet" bson:"snippet"`
	Categories []string             `json:"categories" bson:"categories"`
	Attributes map[string]Attribute `json:"attributes" bson:"attributes"`
	Links      []Link               `json:"links" bson:"links"`
	SourceDate time.Time            `json:"source_date" bson:"source_date"`
	Source     string               `json:"source" bson:"source"`
	CreatedAt  time.Time            `json:"created_at" bson:"created_at"`
	UpdatedAt  time.Time            `json:"updated_at" bson:"updated_at"`
	ImageURL   string               `json:"image_url" bson:"image_url"`
}

type Upload struct {
	Source string `json:"source" bson:"source"`
	Title  string `json:"title" bson:"title"`
}

type User struct {
	Id       bson.ObjectId `json:"id" bson:"_id,omitempty"`
	Name     string        `json:"name" bson:"name"`
	Email    string        `json:"email" bson:"email"`
	Role     string        `json:"role" bson:"role"`
	Password []byte        `json:"-"`
	ApiKey   string        `json:"apikey" bson:"apikey"`
}

type UserToken struct {
	Name  string `json:"name" bson:"name"`
	Email string `json:"email" bson:"email"`
	Role  string `json:"role" bson:"role"`
	Token string `json:"token" bson:"token"`
}

type Value struct {
	ValueType   string    `json:"value_type" bson:"value_type"`
	ValueString string    `json:"value_string" bson:"value_string"`
	Source      string    `json:"source" bson:"source"`
	Date        time.Time `json:"date" bson:"date"`
	UpdatedAt   time.Time `json:"updated_at" bson:"updated_at"`
}