1 files changed, 36 insertions, 4 deletions

diff --git a/client/src/app/shared/shared-main/video-channel/video-channel.model.ts b/client/src/app/shared/shared-main/video-channel/video-channel.model.ts
index b4c3365a9..d8be42eef 100644
--- a/client/src/app/shared/shared-main/video-channel/video-channel.model.ts
+++ b/client/src/app/shared/shared-main/video-channel/video-channel.model.ts
@@ -1,3 +1,4 @@
+import { getAbsoluteAPIUrl } from '@app/helpers'
 import { Account as ServerAccount, ActorImage, VideoChannel as ServerVideoChannel, ViewsPerDate } from '@shared/models'
 import { Account } from '../account/account.model'
 import { Actor } from '../account/actor.model'
@@ -6,10 +7,15 @@ export class VideoChannel extends Actor implements ServerVideoChannel {
   displayName: string
   description: string
   support: string
+
   isLocal: boolean
+
   nameWithHost: string
   nameWithHostForced: string
 
+  banner: ActorImage
+  bannerUrl: string
+
   ownerAccount?: ServerAccount
   ownerBy?: string
   ownerAvatarUrl?: string
@@ -22,6 +28,18 @@ export class VideoChannel extends Actor implements ServerVideoChannel {
     return Actor.GET_ACTOR_AVATAR_URL(actor) || this.GET_DEFAULT_AVATAR_URL()
   }
 
+  static GET_ACTOR_BANNER_URL (channel: ServerVideoChannel) {
+    if (channel?.banner?.url) return channel.banner.url
+
+    if (channel && channel.banner) {
+      const absoluteAPIUrl = getAbsoluteAPIUrl()
+
+      return absoluteAPIUrl + channel.banner.path
+    }
+
+    return ''
+  }
+
   static GET_DEFAULT_AVATAR_URL () {
     return `${window.location.origin}/client/assets/images/default-avatar-videochannel.png`
   }
@@ -29,12 +47,14 @@ export class VideoChannel extends Actor implements ServerVideoChannel {
   constructor (hash: ServerVideoChannel) {
     super(hash)
 
-    this.updateComputedAttributes()
-
     this.displayName = hash.displayName
     this.description = hash.description
     this.support = hash.support
+
+    this.banner = hash.banner
+
     this.isLocal = hash.isLocal
+
     this.nameWithHost = Actor.CREATE_BY_STRING(this.name, this.host)
     this.nameWithHostForced = Actor.CREATE_BY_STRING(this.name, this.host, true)
 
@@ -49,6 +69,8 @@ export class VideoChannel extends Actor implements ServerVideoChannel {
       this.ownerBy = Actor.CREATE_BY_STRING(hash.ownerAccount.name, hash.ownerAccount.host)
       this.ownerAvatarUrl = Account.GET_ACTOR_AVATAR_URL(this.ownerAccount)
     }
+
+    this.updateComputedAttributes()
   }
 
   updateAvatar (newAvatar: ActorImage) {
@@ -58,11 +80,21 @@ export class VideoChannel extends Actor implements ServerVideoChannel {
   }
 
   resetAvatar () {
-    this.avatar = null
-    this.avatarUrl = VideoChannel.GET_DEFAULT_AVATAR_URL()
+    this.updateAvatar(null)
+  }
+
+  updateBanner (newBanner: ActorImage) {
+    this.banner = newBanner
+
+    this.updateComputedAttributes()
+  }
+
+  resetBanner () {
+    this.updateBanner(null)
   }
 
   private updateComputedAttributes () {
     this.avatarUrl = VideoChannel.GET_ACTOR_AVATAR_URL(this)
+    this.bannerUrl = VideoChannel.GET_ACTOR_BANNER_URL(this)
   }
 }