|
1 | 1 | package services |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "bytes" |
4 | 5 | "context" |
5 | 6 | "errors" |
| 7 | + "fmt" |
| 8 | + "io" |
| 9 | + "mime" |
| 10 | + "mime/multipart" |
| 11 | + "path/filepath" |
| 12 | + "strings" |
6 | 13 | "time" |
7 | 14 |
|
8 | 15 | "github.com/google/uuid" |
9 | 16 | "github.com/rs/zerolog" |
| 17 | + "github.com/swamphacks/core/apps/api/internal/config" |
10 | 18 | ctxu "github.com/swamphacks/core/apps/api/internal/ctxutils" |
11 | 19 | "github.com/swamphacks/core/apps/api/internal/db/repository" |
12 | 20 | "github.com/swamphacks/core/apps/api/internal/db/sqlc" |
| 21 | + "github.com/swamphacks/core/apps/api/internal/storage" |
13 | 22 | ) |
14 | 23 |
|
15 | 24 | var ( |
16 | | - ErrFailedToCreateEvent = errors.New("failed to create event") |
17 | | - ErrFailedToGetEvent = errors.New("failed to get event") |
18 | | - ErrFailedToUpdateEvent = errors.New("failed to update event") |
19 | | - ErrFailedToDeleteEvent = errors.New("failed to delete event") |
20 | | - ErrFailedToParseUUID = errors.New("failed to parse uuid") |
21 | | - ErrMissingFields = errors.New("missing fields") |
22 | | - ErrMissingPerms = errors.New("missing perms") |
| 25 | + ErrFailedToCreateEvent = errors.New("failed to create event") |
| 26 | + ErrFailedToGetEvent = errors.New("failed to get event") |
| 27 | + ErrFailedToUpdateEvent = errors.New("failed to update event") |
| 28 | + ErrFailedToDeleteEvent = errors.New("failed to delete event") |
| 29 | + ErrFailedToParseUUID = errors.New("failed to parse uuid") |
| 30 | + ErrMissingFields = errors.New("missing fields") |
| 31 | + ErrMissingPerms = errors.New("missing perms") |
| 32 | + ErrFailedToUploadBanner = errors.New("failed to upload banner") |
| 33 | + ErrUnexpectedFileType = errors.New("did not expect this file type") |
23 | 34 |
|
24 | 35 | ErrFailedToSubmitApplication = errors.New("failed to submit application") |
25 | 36 | ) |
26 | 37 |
|
27 | 38 | type EventService struct { |
28 | 39 | eventRepo *repository.EventRepository |
29 | 40 | userRepo *repository.UserRepository |
| 41 | + storage storage.Storage |
| 42 | + buckets *config.CoreBuckets |
30 | 43 | logger zerolog.Logger |
31 | 44 | } |
32 | 45 |
|
33 | | -func NewEventService(eventRepo *repository.EventRepository, userRepo *repository.UserRepository, logger zerolog.Logger) *EventService { |
| 46 | +func NewEventService(eventRepo *repository.EventRepository, userRepo *repository.UserRepository, storage storage.Storage, buckets *config.CoreBuckets, logger zerolog.Logger) *EventService { |
34 | 47 | return &EventService{ |
35 | 48 | eventRepo: eventRepo, |
36 | 49 | userRepo: userRepo, |
| 50 | + storage: storage, |
| 51 | + buckets: buckets, |
37 | 52 | logger: logger.With().Str("service", "EventService").Str("component", "events").Logger(), |
38 | 53 | } |
39 | 54 | } |
@@ -197,3 +212,49 @@ func (s *EventService) IsApplicationsOpen(ctx context.Context, eventId uuid.UUID |
197 | 212 | open := now.After(event.ApplicationOpen) && now.Before(event.ApplicationClose) |
198 | 213 | return open, nil |
199 | 214 | } |
| 215 | + |
| 216 | +func (s *EventService) UploadBanner(ctx context.Context, eventId uuid.UUID, banner multipart.File, header *multipart.FileHeader) (*string, error) { |
| 217 | + bannerFileBuffer := bytes.NewBuffer(nil) |
| 218 | + |
| 219 | + fileName := header.Filename |
| 220 | + fileExt := strings.ToLower(filepath.Ext(fileName)) |
| 221 | + |
| 222 | + switch fileExt { |
| 223 | + case ".jpg", ".png", ".jpeg": |
| 224 | + // Do nothing |
| 225 | + default: |
| 226 | + return nil, ErrUnexpectedFileType |
| 227 | + } |
| 228 | + |
| 229 | + fileType := mime.TypeByExtension(fileExt) |
| 230 | + |
| 231 | + if fileType == "" { |
| 232 | + return nil, ErrFailedToUploadBanner |
| 233 | + } |
| 234 | + |
| 235 | + if _, err := io.Copy(bannerFileBuffer, banner); err != nil { |
| 236 | + return nil, ErrFailedToUploadBanner |
| 237 | + } |
| 238 | + |
| 239 | + uploadKey := fmt.Sprintf("%s/banner%s", eventId, fileExt) |
| 240 | + |
| 241 | + err := s.storage.Store(ctx, s.buckets.EventAssets, uploadKey, bannerFileBuffer.Bytes(), &fileType) |
| 242 | + if err != nil { |
| 243 | + return nil, ErrFailedToUploadBanner |
| 244 | + } |
| 245 | + |
| 246 | + // Reconstrust URL |
| 247 | + url := fmt.Sprintf("%s/%s", s.buckets.EventAssetsBaseUrl, uploadKey) |
| 248 | + |
| 249 | + err = s.eventRepo.UpdateEventById(ctx, sqlc.UpdateEventByIdParams{ |
| 250 | + ID: eventId, |
| 251 | + BannerDoUpdate: true, |
| 252 | + Banner: &url, |
| 253 | + }) |
| 254 | + if err != nil { |
| 255 | + return nil, ErrFailedToUpdateEvent |
| 256 | + } |
| 257 | + |
| 258 | + return &url, nil |
| 259 | + |
| 260 | +} |
0 commit comments