import { EventApi } from '@fullcalendar/core'
import ResourceApi from './ResourceApi'
import '../ambient'


EventApi.prototype.getResources = function(this: EventApi): ResourceApi[] {
  let calendar = this._calendar

  return this._def.resourceIds.map(function(resourceId) {
    return calendar.getResourceById(resourceId)
  })
}

EventApi.prototype.setResources = function(this: EventApi, resources: (string | ResourceApi)[]) {
  let resourceIds = []

  // massage resources -> resourceIds
  for (let resource of resources) {
    let resourceId = null

    if (typeof resource === 'string') {
      resourceId = resource
    } else if (typeof resource === 'number') {
      resourceId = String(resource)
    } else if (resource instanceof ResourceApi) {
      resourceId = resource.id // guaranteed to always have an ID. hmmm
    } else {
      console.warn('unknown resource type: ' + resource)
    }

    if (resourceId) {
      resourceIds.push(resourceId)
    }
  }

  this.mutate({
    standardProps: {
      resourceIds
    }
  })
}