Add Filters!
This commit is contained in:
parent
5fede6d6da
commit
6697358960
1
go.mod
1
go.mod
@ -10,6 +10,7 @@ require (
|
||||
github.com/dsoprea/go-exif/v3 v3.0.0-20201216222538-db167117f483
|
||||
github.com/gabriel-vasile/mimetype v1.1.2
|
||||
github.com/google/uuid v1.1.5
|
||||
github.com/iancoleman/strcase v0.1.3
|
||||
github.com/lestrrat-go/jwx v1.0.8
|
||||
github.com/mattn/go-sqlite3 v1.14.6
|
||||
github.com/sirupsen/logrus v1.7.0
|
||||
|
2
go.sum
2
go.sum
@ -96,6 +96,8 @@ github.com/h2non/parth v0.0.0-20190131123155-b4df798d6542/go.mod h1:Ow0tF8D4Kplb
|
||||
github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||
github.com/hashicorp/golang-lru v0.5.1 h1:0hERBMJE1eitiLkihrMvRVBYAkpHzc/J3QdDN+dAcgU=
|
||||
github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||
github.com/iancoleman/strcase v0.1.3 h1:dJBk1m2/qjL1twPLf68JND55vvivMupZ4wIzE8CTdBw=
|
||||
github.com/iancoleman/strcase v0.1.3/go.mod h1:SK73tn/9oHe+/Y0h39VT4UCxmurVJkR5NA7kMEAOgSE=
|
||||
github.com/jessevdk/go-flags v1.4.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI=
|
||||
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
|
||||
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
|
||||
|
File diff suppressed because it is too large
Load Diff
@ -16,15 +16,14 @@ type Album struct {
|
||||
CreatedAt *time.Time `json:"createdAt" `
|
||||
UpdatedAt *time.Time `json:"updatedAt" `
|
||||
Name string `json:"name" gorm:"unique;not null"`
|
||||
UserID string `json:"userID" gorm:"not null"`
|
||||
}
|
||||
|
||||
type AlbumFilter struct {
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
Name *StringFilter `json:"name" `
|
||||
And *MediaItemFilter `json:"and" `
|
||||
Or *MediaItemFilter `json:"or" `
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
Name *StringFilter `json:"name" `
|
||||
}
|
||||
|
||||
type AlbumResponse struct {
|
||||
@ -39,10 +38,8 @@ type AuthResponse struct {
|
||||
}
|
||||
|
||||
type AuthTypeFilter struct {
|
||||
EqualTo *AuthType `json:"equalTo" `
|
||||
NotEqualTo *AuthType `json:"notEqualTo" `
|
||||
In []AuthType `json:"in" `
|
||||
NotIn []AuthType `json:"notIn" `
|
||||
EqualTo *AuthType `json:"equalTo" `
|
||||
NotEqualTo *AuthType `json:"notEqualTo" `
|
||||
}
|
||||
|
||||
type BooleanFilter struct {
|
||||
@ -56,8 +53,8 @@ type Device struct {
|
||||
UpdatedAt *time.Time `json:"updatedAt" `
|
||||
Name string `json:"name" gorm:"not null"`
|
||||
Type DeviceType `json:"type" gorm:"default:Unknown;not null"`
|
||||
UserID string `json:"userID" gorm:"not null"`
|
||||
RefreshKey *string `json:"refreshKey" `
|
||||
UserID string `json:"userID" gorm:"not null"`
|
||||
}
|
||||
|
||||
type DeviceFilter struct {
|
||||
@ -66,8 +63,6 @@ type DeviceFilter struct {
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
Name *StringFilter `json:"name" `
|
||||
Type *DeviceTypeFilter `json:"type" `
|
||||
And *MediaItemFilter `json:"and" `
|
||||
Or *MediaItemFilter `json:"or" `
|
||||
}
|
||||
|
||||
type DeviceResponse struct {
|
||||
@ -76,39 +71,31 @@ type DeviceResponse struct {
|
||||
}
|
||||
|
||||
type DeviceTypeFilter struct {
|
||||
EqualTo *DeviceType `json:"equalTo" `
|
||||
NotEqualTo *DeviceType `json:"notEqualTo" `
|
||||
In []DeviceType `json:"in" `
|
||||
NotIn []DeviceType `json:"notIn" `
|
||||
EqualTo *DeviceType `json:"equalTo" `
|
||||
NotEqualTo *DeviceType `json:"notEqualTo" `
|
||||
}
|
||||
|
||||
type FloatFilter struct {
|
||||
EqualTo *float64 `json:"equalTo" `
|
||||
NotEqualTo *float64 `json:"notEqualTo" `
|
||||
LessThan *float64 `json:"lessThan" `
|
||||
LessThanOrEqualTo *float64 `json:"lessThanOrEqualTo" `
|
||||
MoreThan *float64 `json:"moreThan" `
|
||||
MoreThanOrEqualTo *float64 `json:"moreThanOrEqualTo" `
|
||||
In []float64 `json:"in" `
|
||||
NotIn []float64 `json:"notIn" `
|
||||
EqualTo *float64 `json:"equalTo" `
|
||||
NotEqualTo *float64 `json:"notEqualTo" `
|
||||
LessThan *float64 `json:"lessThan" `
|
||||
LessThanOrEqualTo *float64 `json:"lessThanOrEqualTo" `
|
||||
GreaterThan *float64 `json:"greaterThan" `
|
||||
GreaterThanOrEqualTo *float64 `json:"greaterThanOrEqualTo" `
|
||||
}
|
||||
|
||||
type IDFilter struct {
|
||||
EqualTo *string `json:"equalTo" `
|
||||
NotEqualTo *string `json:"notEqualTo" `
|
||||
In []string `json:"in" `
|
||||
NotIn []string `json:"notIn" `
|
||||
EqualTo *string `json:"equalTo" `
|
||||
NotEqualTo *string `json:"notEqualTo" `
|
||||
}
|
||||
|
||||
type IntFilter struct {
|
||||
EqualTo *int `json:"equalTo" `
|
||||
NotEqualTo *int `json:"notEqualTo" `
|
||||
LessThan *int `json:"lessThan" `
|
||||
LessThanOrEqualTo *int `json:"lessThanOrEqualTo" `
|
||||
MoreThan *int `json:"moreThan" `
|
||||
MoreThanOrEqualTo *int `json:"moreThanOrEqualTo" `
|
||||
In []int `json:"in" `
|
||||
NotIn []int `json:"notIn" `
|
||||
EqualTo *int `json:"equalTo" `
|
||||
NotEqualTo *int `json:"notEqualTo" `
|
||||
LessThan *int `json:"lessThan" `
|
||||
LessThanOrEqualTo *int `json:"lessThanOrEqualTo" `
|
||||
GreaterThan *int `json:"greaterThan" `
|
||||
GreaterThanOrEqualTo *int `json:"greaterThanOrEqualTo" `
|
||||
}
|
||||
|
||||
type MediaItem struct {
|
||||
@ -116,29 +103,27 @@ type MediaItem struct {
|
||||
CreatedAt *time.Time `json:"createdAt" `
|
||||
UpdatedAt *time.Time `json:"updatedAt" `
|
||||
ExifDate *time.Time `json:"exifDate" `
|
||||
Latitude *float64 `json:"latitude" `
|
||||
Longitude *float64 `json:"longitude" `
|
||||
Latitude *float64 `json:"latitude" gorm:"precision:5"`
|
||||
Longitude *float64 `json:"longitude" gorm:"precision:5"`
|
||||
IsVideo bool `json:"isVideo" gorm:"default:false;not null"`
|
||||
FileName string `json:"fileName" gorm:"not null"`
|
||||
OrigName string `json:"origName" gorm:"not null"`
|
||||
Tags []*Tag `json:"tags" gorm:"many2many:media_tags"`
|
||||
Albums []*Album `json:"albums" gorm:"many2many:media_albums"`
|
||||
Tags []*Tag `json:"tags" gorm:"many2many:media_tags;foreignKey:ID,UserID;References:ID"`
|
||||
Albums []*Album `json:"albums" gorm:"many2many:media_albums;foreignKey:ID,UserID;Refrences:ID"`
|
||||
UserID string `json:"userID" gorm:"not null"`
|
||||
}
|
||||
|
||||
type MediaItemFilter struct {
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
ExifDate *TimeFilter `json:"exifDate" `
|
||||
Latitude *FloatFilter `json:"latitude" `
|
||||
Longitude *FloatFilter `json:"longitude" `
|
||||
IsVideo *BooleanFilter `json:"isVideo" `
|
||||
OrigName *StringFilter `json:"origName" `
|
||||
Tags *TagFilter `json:"tags" `
|
||||
Albums *AlbumFilter `json:"albums" `
|
||||
And *MediaItemFilter `json:"and" `
|
||||
Or *MediaItemFilter `json:"or" `
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
ExifDate *TimeFilter `json:"exifDate" `
|
||||
Latitude *FloatFilter `json:"latitude" `
|
||||
Longitude *FloatFilter `json:"longitude" `
|
||||
IsVideo *BooleanFilter `json:"isVideo" `
|
||||
OrigName *StringFilter `json:"origName" `
|
||||
Tags *TagFilter `json:"tags" `
|
||||
Albums *AlbumFilter `json:"albums" `
|
||||
}
|
||||
|
||||
type MediaItemResponse struct {
|
||||
@ -177,29 +162,19 @@ type PageInfo struct {
|
||||
}
|
||||
|
||||
type RoleFilter struct {
|
||||
EqualTo *Role `json:"equalTo" `
|
||||
NotEqualTo *Role `json:"notEqualTo" `
|
||||
In []Role `json:"in" `
|
||||
NotIn []Role `json:"notIn" `
|
||||
EqualTo *Role `json:"equalTo" `
|
||||
NotEqualTo *Role `json:"notEqualTo" `
|
||||
}
|
||||
|
||||
type StringFilter struct {
|
||||
EqualTo *string `json:"equalTo" `
|
||||
NotEqualTo *string `json:"notEqualTo" `
|
||||
StartWith *string `json:"startWith" `
|
||||
NotStartWith *string `json:"notStartWith" `
|
||||
EndWith *string `json:"endWith" `
|
||||
NotEndWith *string `json:"notEndWith" `
|
||||
Contain *string `json:"contain" `
|
||||
NotContain *string `json:"notContain" `
|
||||
In []string `json:"in" `
|
||||
NotIn []string `json:"notIn" `
|
||||
StartWithStrict *string `json:"startWithStrict" `
|
||||
NotStartWithStrict *string `json:"notStartWithStrict" `
|
||||
EndWithStrict *string `json:"endWithStrict" `
|
||||
NotEndWithStrict *string `json:"notEndWithStrict" `
|
||||
ContainStrict *string `json:"containStrict" `
|
||||
NotContainStrict *string `json:"notContainStrict" `
|
||||
EqualTo *string `json:"equalTo" `
|
||||
NotEqualTo *string `json:"notEqualTo" `
|
||||
StartsWith *string `json:"startsWith" `
|
||||
NotStartsWith *string `json:"notStartsWith" `
|
||||
EndsWith *string `json:"endsWith" `
|
||||
NotEndsWith *string `json:"notEndsWith" `
|
||||
Contains *string `json:"contains" `
|
||||
NotContains *string `json:"notContains" `
|
||||
}
|
||||
|
||||
type Tag struct {
|
||||
@ -207,15 +182,14 @@ type Tag struct {
|
||||
CreatedAt *time.Time `json:"createdAt" `
|
||||
UpdatedAt *time.Time `json:"updatedAt" `
|
||||
Name string `json:"name" gorm:"unique;not null"`
|
||||
UserID string `json:"userID" gorm:"not null"`
|
||||
}
|
||||
|
||||
type TagFilter struct {
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
Name *StringFilter `json:"name" `
|
||||
And *MediaItemFilter `json:"and" `
|
||||
Or *MediaItemFilter `json:"or" `
|
||||
ID *IDFilter `json:"id" `
|
||||
CreatedAt *TimeFilter `json:"createdAt" `
|
||||
UpdatedAt *TimeFilter `json:"updatedAt" `
|
||||
Name *StringFilter `json:"name" `
|
||||
}
|
||||
|
||||
type TagResponse struct {
|
||||
@ -224,12 +198,12 @@ type TagResponse struct {
|
||||
}
|
||||
|
||||
type TimeFilter struct {
|
||||
EqualTo *time.Time `json:"equalTo" `
|
||||
NotEqualTo *time.Time `json:"notEqualTo" `
|
||||
LessThan *time.Time `json:"lessThan" `
|
||||
LessThanOrEqualTo *time.Time `json:"lessThanOrEqualTo" `
|
||||
MoreThan *time.Time `json:"moreThan" `
|
||||
MoreThanOrEqualTo *time.Time `json:"moreThanOrEqualTo" `
|
||||
EqualTo *time.Time `json:"equalTo" `
|
||||
NotEqualTo *time.Time `json:"notEqualTo" `
|
||||
LessThan *time.Time `json:"lessThan" `
|
||||
LessThanOrEqualTo *time.Time `json:"lessThanOrEqualTo" `
|
||||
GreaterThan *time.Time `json:"greaterThan" `
|
||||
GreaterThanOrEqualTo *time.Time `json:"greaterThanOrEqualTo" `
|
||||
}
|
||||
|
||||
type User struct {
|
||||
@ -256,8 +230,6 @@ type UserFilter struct {
|
||||
LastName *StringFilter `json:"lastName" `
|
||||
Role *RoleFilter `json:"role" `
|
||||
AuthType *AuthTypeFilter `json:"authType" `
|
||||
And *UserFilter `json:"and" `
|
||||
Or *UserFilter `json:"or" `
|
||||
}
|
||||
|
||||
type UserResponse struct {
|
||||
|
@ -56,8 +56,8 @@ input TimeFilter {
|
||||
notEqualTo: Time
|
||||
lessThan: Time
|
||||
lessThanOrEqualTo: Time
|
||||
moreThan: Time
|
||||
moreThanOrEqualTo: Time
|
||||
greaterThan: Time
|
||||
greaterThanOrEqualTo: Time
|
||||
}
|
||||
|
||||
input IntFilter {
|
||||
@ -65,10 +65,8 @@ input IntFilter {
|
||||
notEqualTo: Int
|
||||
lessThan: Int
|
||||
lessThanOrEqualTo: Int
|
||||
moreThan: Int
|
||||
moreThanOrEqualTo: Int
|
||||
in: [Int!]
|
||||
notIn: [Int!]
|
||||
greaterThan: Int
|
||||
greaterThanOrEqualTo: Int
|
||||
}
|
||||
|
||||
input FloatFilter {
|
||||
@ -76,10 +74,8 @@ input FloatFilter {
|
||||
notEqualTo: Float
|
||||
lessThan: Float
|
||||
lessThanOrEqualTo: Float
|
||||
moreThan: Float
|
||||
moreThanOrEqualTo: Float
|
||||
in: [Float!]
|
||||
notIn: [Float!]
|
||||
greaterThan: Float
|
||||
greaterThanOrEqualTo: Float
|
||||
}
|
||||
|
||||
input BooleanFilter {
|
||||
@ -90,49 +86,32 @@ input BooleanFilter {
|
||||
input IDFilter {
|
||||
equalTo: ID
|
||||
notEqualTo: ID
|
||||
in: [ID!]
|
||||
notIn: [ID!]
|
||||
}
|
||||
|
||||
input StringFilter {
|
||||
equalTo: String
|
||||
notEqualTo: String
|
||||
startWith: String
|
||||
notStartWith: String
|
||||
endWith: String
|
||||
notEndWith: String
|
||||
contain: String
|
||||
notContain: String
|
||||
in: [String!]
|
||||
notIn: [String!]
|
||||
|
||||
startWithStrict: String
|
||||
notStartWithStrict: String
|
||||
endWithStrict: String
|
||||
notEndWithStrict: String
|
||||
containStrict: String
|
||||
notContainStrict: String
|
||||
startsWith: String
|
||||
notStartsWith: String
|
||||
endsWith: String
|
||||
notEndsWith: String
|
||||
contains: String
|
||||
notContains: String
|
||||
}
|
||||
|
||||
input RoleFilter {
|
||||
equalTo: Role
|
||||
notEqualTo: Role
|
||||
in: [Role!]
|
||||
notIn: [Role!]
|
||||
}
|
||||
|
||||
input DeviceTypeFilter {
|
||||
equalTo: DeviceType
|
||||
notEqualTo: DeviceType
|
||||
in: [DeviceType!]
|
||||
notIn: [DeviceType!]
|
||||
}
|
||||
|
||||
input AuthTypeFilter {
|
||||
equalTo: AuthType
|
||||
notEqualTo: AuthType
|
||||
in: [AuthType!]
|
||||
notIn: [AuthType!]
|
||||
}
|
||||
|
||||
# ------------------------------------------------------------
|
||||
@ -160,8 +139,8 @@ type Device {
|
||||
updatedAt: Time
|
||||
name: String! @meta(gorm: "not null")
|
||||
type: DeviceType! @meta(gorm: "default:Unknown;not null")
|
||||
userID: ID! @meta(gorm: "not null")
|
||||
refreshKey: String @isPrivate
|
||||
userID: ID! @meta(gorm: "not null")
|
||||
}
|
||||
|
||||
type MediaItem {
|
||||
@ -169,21 +148,22 @@ type MediaItem {
|
||||
createdAt: Time
|
||||
updatedAt: Time
|
||||
exifDate: Time
|
||||
latitude: Float
|
||||
longitude: Float
|
||||
latitude: Float @meta(gorm: "precision:5")
|
||||
longitude: Float @meta(gorm: "precision:5")
|
||||
isVideo: Boolean! @meta(gorm: "default:false;not null")
|
||||
fileName: String! @meta(gorm: "not null")
|
||||
origName: String! @meta(gorm: "not null")
|
||||
tags: [Tag] @meta(gorm: "many2many:media_tags")
|
||||
albums: [Album] @meta(gorm: "many2many:media_albums")
|
||||
tags: [Tag] @meta(gorm: "many2many:media_tags;foreignKey:ID,UserID;References:ID")
|
||||
albums: [Album] @meta(gorm: "many2many:media_albums;foreignKey:ID,UserID;Refrences:ID")
|
||||
userID: ID! @meta(gorm: "not null")
|
||||
}
|
||||
|
||||
type Tag {
|
||||
id: ID! @meta(gorm: "primaryKey;not null")
|
||||
id: ID! @meta(gorm: "primaryKey;not null")
|
||||
createdAt: Time
|
||||
updatedAt: Time
|
||||
name: String! @meta(gorm: "unique;not null")
|
||||
name: String! @meta(gorm: "unique;not null")
|
||||
userID: ID! @meta(gorm: "not null")
|
||||
}
|
||||
|
||||
type Album {
|
||||
@ -191,7 +171,7 @@ type Album {
|
||||
createdAt: Time
|
||||
updatedAt: Time
|
||||
name: String! @meta(gorm: "unique;not null")
|
||||
# userID: ID! @meta(gorm: "not null")
|
||||
userID: ID! @meta(gorm: "not null")
|
||||
}
|
||||
|
||||
# ------------------------------------------------------------
|
||||
@ -208,8 +188,8 @@ input UserFilter {
|
||||
role: RoleFilter
|
||||
authType: AuthTypeFilter
|
||||
|
||||
and: UserFilter
|
||||
or: UserFilter
|
||||
# and: UserFilter
|
||||
# or: UserFilter
|
||||
}
|
||||
|
||||
input MediaItemFilter {
|
||||
@ -224,8 +204,8 @@ input MediaItemFilter {
|
||||
tags: TagFilter
|
||||
albums: AlbumFilter
|
||||
|
||||
and: MediaItemFilter
|
||||
or: MediaItemFilter
|
||||
# and: MediaItemFilter
|
||||
# or: MediaItemFilter
|
||||
}
|
||||
|
||||
input DeviceFilter {
|
||||
@ -235,8 +215,8 @@ input DeviceFilter {
|
||||
name: StringFilter
|
||||
type: DeviceTypeFilter
|
||||
|
||||
and: MediaItemFilter
|
||||
or: MediaItemFilter
|
||||
# and: MediaItemFilter
|
||||
# or: MediaItemFilter
|
||||
}
|
||||
|
||||
input TagFilter {
|
||||
@ -245,8 +225,8 @@ input TagFilter {
|
||||
updatedAt: TimeFilter
|
||||
name: StringFilter
|
||||
|
||||
and: MediaItemFilter
|
||||
or: MediaItemFilter
|
||||
# and: MediaItemFilter
|
||||
# or: MediaItemFilter
|
||||
}
|
||||
|
||||
input AlbumFilter {
|
||||
@ -255,8 +235,8 @@ input AlbumFilter {
|
||||
updatedAt: TimeFilter
|
||||
name: StringFilter
|
||||
|
||||
and: MediaItemFilter
|
||||
or: MediaItemFilter
|
||||
# and: MediaItemFilter
|
||||
# or: MediaItemFilter
|
||||
}
|
||||
|
||||
# ------------------------------------------------------------
|
||||
|
@ -102,13 +102,14 @@ func (r *mutationResolver) CreateAlbum(ctx context.Context, input model.NewAlbum
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
_, ok := accessToken.Get("sub")
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Upload Failed")
|
||||
}
|
||||
|
||||
album := &model.Album{
|
||||
Name: input.Name,
|
||||
Name: input.Name,
|
||||
UserID: userID.(string),
|
||||
}
|
||||
|
||||
err := r.DB.CreateAlbum(album)
|
||||
@ -120,8 +121,17 @@ func (r *mutationResolver) CreateAlbum(ctx context.Context, input model.NewAlbum
|
||||
}
|
||||
|
||||
func (r *mutationResolver) CreateTag(ctx context.Context, input model.NewTag) (*model.Tag, error) {
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Upload Failed")
|
||||
}
|
||||
|
||||
tag := &model.Tag{
|
||||
Name: input.Name,
|
||||
Name: input.Name,
|
||||
UserID: userID.(string),
|
||||
}
|
||||
|
||||
err := r.DB.CreateTag(tag)
|
||||
@ -223,13 +233,20 @@ func (r *queryResolver) Logout(ctx context.Context) (*model.AuthResponse, error)
|
||||
}
|
||||
|
||||
func (r *queryResolver) MediaItem(ctx context.Context, id string, delete *bool) (*model.MediaItem, error) {
|
||||
// TODO: User Specific
|
||||
deviceID, err := uuid.Parse(id)
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Context Error")
|
||||
}
|
||||
|
||||
mediaItemID, err := uuid.Parse(id)
|
||||
if err != nil {
|
||||
return nil, errors.New("Invalid ID Format")
|
||||
}
|
||||
|
||||
foundMediaItem := &model.MediaItem{ID: deviceID.String()}
|
||||
foundMediaItem := &model.MediaItem{ID: mediaItemID.String(), UserID: userID.(string)}
|
||||
count, err := r.DB.MediaItem(foundMediaItem)
|
||||
if err != nil {
|
||||
return nil, errors.New("DB Error")
|
||||
@ -240,13 +257,20 @@ func (r *queryResolver) MediaItem(ctx context.Context, id string, delete *bool)
|
||||
}
|
||||
|
||||
func (r *queryResolver) Device(ctx context.Context, id string, delete *bool) (*model.Device, error) {
|
||||
// TODO: User Specific
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Context Error")
|
||||
}
|
||||
|
||||
deviceID, err := uuid.Parse(id)
|
||||
if err != nil {
|
||||
return nil, errors.New("Invalid ID Format")
|
||||
}
|
||||
|
||||
foundDevice := &model.Device{ID: deviceID.String()}
|
||||
foundDevice := &model.Device{ID: deviceID.String(), UserID: userID.(string)}
|
||||
count, err := r.DB.Device(foundDevice)
|
||||
if err != nil {
|
||||
return nil, errors.New("DB Error")
|
||||
@ -257,13 +281,20 @@ func (r *queryResolver) Device(ctx context.Context, id string, delete *bool) (*m
|
||||
}
|
||||
|
||||
func (r *queryResolver) Album(ctx context.Context, id string, delete *bool) (*model.Album, error) {
|
||||
// TODO: User Specific
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Context Error")
|
||||
}
|
||||
|
||||
albumID, err := uuid.Parse(id)
|
||||
if err != nil {
|
||||
return nil, errors.New("Invalid ID Format")
|
||||
}
|
||||
|
||||
foundAlbum := &model.Album{ID: albumID.String()}
|
||||
foundAlbum := &model.Album{ID: albumID.String(), UserID: userID.(string)}
|
||||
count, err := r.DB.Album(foundAlbum)
|
||||
if err != nil {
|
||||
return nil, errors.New("DB Error")
|
||||
@ -323,7 +354,7 @@ func (r *queryResolver) Me(ctx context.Context, delete *bool) (*model.User, erro
|
||||
}
|
||||
|
||||
func (r *queryResolver) MediaItems(ctx context.Context, delete *bool, filter *model.MediaItemFilter, count *int, page *int) (*model.MediaItemResponse, error) {
|
||||
resp, totalCount, err := r.DB.MediaItems()
|
||||
resp, totalCount, err := r.DB.MediaItems(filter)
|
||||
if err != nil {
|
||||
return nil, errors.New("Context Error")
|
||||
}
|
||||
@ -338,10 +369,18 @@ func (r *queryResolver) MediaItems(ctx context.Context, delete *bool, filter *mo
|
||||
}
|
||||
|
||||
func (r *queryResolver) Devices(ctx context.Context, delete *bool, filter *model.DeviceFilter, count *int, page *int) (*model.DeviceResponse, error) {
|
||||
// TODO: User Specific
|
||||
// Get Context
|
||||
authContext := ctx.Value("auth").(*model.AuthContext)
|
||||
accessToken := *authContext.AccessToken
|
||||
userID, ok := accessToken.Get("sub")
|
||||
if !ok {
|
||||
return nil, errors.New("Context Error")
|
||||
}
|
||||
_ = userID
|
||||
|
||||
resp, totalCount, err := r.DB.Devices()
|
||||
if err != nil {
|
||||
return nil, errors.New("Context Error")
|
||||
return nil, errors.New("DB Error")
|
||||
}
|
||||
return &model.DeviceResponse{
|
||||
Data: resp,
|
||||
|
193
internal/db/filters.go
Normal file
193
internal/db/filters.go
Normal file
@ -0,0 +1,193 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
|
||||
"github.com/iancoleman/strcase"
|
||||
"gorm.io/gorm"
|
||||
"reichard.io/imagini/graph/model"
|
||||
)
|
||||
|
||||
// Generic function used to generate filters for the DB
|
||||
func (dbm *DBManager) generateFilters(filter interface{}) (*gorm.DB, error) {
|
||||
tx := dbm.db.Session(&gorm.Session{}).Debug()
|
||||
|
||||
v := reflect.ValueOf(filter)
|
||||
|
||||
for i := 0; i < v.NumField(); i++ {
|
||||
fieldName := strcase.ToSnake(v.Type().Field(i).Name)
|
||||
fieldVal := v.Field(i)
|
||||
|
||||
if fieldVal.IsNil() {
|
||||
continue
|
||||
}
|
||||
|
||||
switch valType := fieldVal.Type(); valType {
|
||||
case reflect.TypeOf(&model.StringFilter{}):
|
||||
tx = generateStringFilter(tx, fieldName, fieldVal.Interface().(*model.StringFilter))
|
||||
case reflect.TypeOf(&model.BooleanFilter{}):
|
||||
tx = generateBooleanFilter(tx, fieldName, fieldVal.Interface().(*model.BooleanFilter))
|
||||
case reflect.TypeOf(&model.FloatFilter{}):
|
||||
tx = generateFloatFilter(tx, fieldName, fieldVal.Interface().(*model.FloatFilter))
|
||||
case reflect.TypeOf(&model.IntFilter{}):
|
||||
tx = generateIntFilter(tx, fieldName, fieldVal.Interface().(*model.IntFilter))
|
||||
case reflect.TypeOf(&model.IDFilter{}):
|
||||
tx = generateIDFilter(tx, fieldName, fieldVal.Interface().(*model.IDFilter))
|
||||
case reflect.TypeOf(&model.TimeFilter{}):
|
||||
tx = generateTimeFilter(tx, fieldName, fieldVal.Interface().(*model.TimeFilter))
|
||||
case reflect.TypeOf(&model.RoleFilter{}):
|
||||
tx = generateRoleFilter(tx, fieldName, fieldVal.Interface().(*model.RoleFilter))
|
||||
case reflect.TypeOf(&model.DeviceTypeFilter{}):
|
||||
tx = generateDeviceTypeFilter(tx, fieldName, fieldVal.Interface().(*model.DeviceTypeFilter))
|
||||
case reflect.TypeOf(&model.AuthTypeFilter{}):
|
||||
tx = generateAuthTypeFilter(tx, fieldName, fieldVal.Interface().(*model.AuthTypeFilter))
|
||||
}
|
||||
}
|
||||
|
||||
return tx, nil
|
||||
}
|
||||
|
||||
func generateStringFilter(tx *gorm.DB, fieldName string, filter *model.StringFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
if filter.StartsWith != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s LIKE ?", fieldName), fmt.Sprintf("%s%%", *filter.StartsWith))
|
||||
}
|
||||
if filter.NotStartsWith != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s NOT LIKE ?", fieldName), fmt.Sprintf("%s%%", *filter.NotStartsWith))
|
||||
}
|
||||
if filter.EndsWith != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s LIKE ?", fieldName), fmt.Sprintf("%%%s", *filter.EndsWith))
|
||||
}
|
||||
if filter.NotEndsWith != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s NOT LIKE ?", fieldName), fmt.Sprintf("%%%s", *filter.NotEndsWith))
|
||||
}
|
||||
if filter.Contains != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s LIKE ?", fieldName), fmt.Sprintf("%%%s%%", *filter.Contains))
|
||||
}
|
||||
if filter.NotContains != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s NOT LIKE ?", fieldName), fmt.Sprintf("%%%s%%", *filter.NotContains))
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateBooleanFilter(tx *gorm.DB, fieldName string, filter *model.BooleanFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateFloatFilter(tx *gorm.DB, fieldName string, filter *model.FloatFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
if filter.GreaterThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s > ?", fieldName), *filter.GreaterThan)
|
||||
}
|
||||
if filter.GreaterThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s >= ?", fieldName), *filter.GreaterThanOrEqualTo)
|
||||
}
|
||||
if filter.LessThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s < ?", fieldName), *filter.LessThan)
|
||||
}
|
||||
if filter.LessThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s <= ?", fieldName), *filter.LessThanOrEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateIntFilter(tx *gorm.DB, fieldName string, filter *model.IntFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
if filter.GreaterThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s > ?", fieldName), *filter.GreaterThan)
|
||||
}
|
||||
if filter.GreaterThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s >= ?", fieldName), *filter.GreaterThanOrEqualTo)
|
||||
}
|
||||
if filter.LessThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s < ?", fieldName), *filter.LessThan)
|
||||
}
|
||||
if filter.LessThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s <= ?", fieldName), *filter.LessThanOrEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateIDFilter(tx *gorm.DB, fieldName string, filter *model.IDFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateTimeFilter(tx *gorm.DB, fieldName string, filter *model.TimeFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
if filter.GreaterThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s > ?", fieldName), *filter.GreaterThan)
|
||||
}
|
||||
if filter.GreaterThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s >= ?", fieldName), *filter.GreaterThanOrEqualTo)
|
||||
}
|
||||
if filter.LessThan != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s < ?", fieldName), *filter.LessThan)
|
||||
}
|
||||
if filter.LessThanOrEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s <= ?", fieldName), *filter.LessThanOrEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateRoleFilter(tx *gorm.DB, fieldName string, filter *model.RoleFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateDeviceTypeFilter(tx *gorm.DB, fieldName string, filter *model.DeviceTypeFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
||||
|
||||
func generateAuthTypeFilter(tx *gorm.DB, fieldName string, filter *model.AuthTypeFilter) *gorm.DB {
|
||||
if filter.EqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s == ?", fieldName), *filter.EqualTo)
|
||||
}
|
||||
if filter.NotEqualTo != nil {
|
||||
tx = tx.Where(fmt.Sprintf("%s != ?", fieldName), *filter.NotEqualTo)
|
||||
}
|
||||
return tx
|
||||
}
|
@ -18,9 +18,14 @@ func (dbm *DBManager) MediaItem(mediaItem *model.MediaItem) (int64, error) {
|
||||
return count, err
|
||||
}
|
||||
|
||||
func (dbm *DBManager) MediaItems() ([]*model.MediaItem, int64, error) {
|
||||
func (dbm *DBManager) MediaItems(filters *model.MediaItemFilter) ([]*model.MediaItem, int64, error) {
|
||||
// Perform Filters
|
||||
tx, err := dbm.generateFilters(*filters)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
var mediaItems []*model.MediaItem
|
||||
var count int64
|
||||
err := dbm.db.Find(&mediaItems).Count(&count).Error
|
||||
err = tx.Find(&mediaItems).Count(&count).Error
|
||||
return mediaItems, count, err
|
||||
}
|
||||
|
Reference in New Issue
Block a user