gistfile1.groovy

def updateCompanySubscription(companyId,duration) {
    if(duration <= 0) {
      return deactivateCompany(companyId)
    } else if(duration > 0) {
      def company = Company.get(companyId)
      company.isActive = true
      def today = new Date()
      if(company.endDate < today ) {
        company.startDate = today
        company.endDate = today
      }
      use(TimeCategory) {
        company.endDate += duration.months
      }
      company.isActive = true
      company.save()
      return company
    }
  }