ratpack.sse.internal.DefaultEvent Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package ratpack.sse.internal;

import com.google.common.base.Strings;
import io.netty.buffer.Unpooled;
import org.reactivestreams.Publisher;
import ratpack.func.Action;
import ratpack.func.Function;
import ratpack.sse.ServerSentEvent;
import ratpack.stream.Streams;
import ratpack.stream.TransformablePublisher;

import java.nio.charset.StandardCharsets;

@SuppressWarnings("deprecation")
public class DefaultEvent implements ratpack.sse.Event {

  private final T item;

  private String id;
  private String event;
  private String data;
  private String comment;

  public DefaultEvent(T item) {
    this.item = item;
  }

  public static  TransformablePublisher toEvents(Publisher publisher, Action> action) {
    return Streams.map(publisher, item -> {
      ratpack.sse.Event event = action.with(new DefaultEvent<>(item));
      if (event.getData() == null && event.getId() == null && event.getEvent() == null && event.getComment() == null) {
        throw new IllegalArgumentException("You must supply at least one of data, event, id or comment");
      }
      return ServerSentEvent.builder()
          .id(event.getId() == null ? Unpooled.EMPTY_BUFFER : Unpooled.wrappedBuffer(event.getId().getBytes(StandardCharsets.UTF_8)))
          .event(event.getEvent() == null ? Unpooled.EMPTY_BUFFER : Unpooled.wrappedBuffer(event.getEvent().getBytes(StandardCharsets.UTF_8)))
          .comment(event.getComment() == null ? Unpooled.EMPTY_BUFFER : Unpooled.wrappedBuffer(event.getComment().getBytes(StandardCharsets.UTF_8)))
          .data(event.getData() == null ? Unpooled.EMPTY_BUFFER : Unpooled.wrappedBuffer(event.getData().getBytes(StandardCharsets.UTF_8)))
          .build();
    });
  }

  public static  ratpack.sse.Event fromServerSentEvent(ServerSentEvent e) {
    try {
      return new DefaultEvent(null)
          .id(Strings.emptyToNull(e.getIdAsString()))
          .event(Strings.emptyToNull(e.getEventAsString()))
          .data(Strings.emptyToNull(e.getDataAsString()))
          .comment(Strings.emptyToNull(e.getCommentAsString()));
    } finally {
      e.close();
    }
  }

  @Override
  public T getItem() {
    return item;
  }

  @Override
  public ratpack.sse.Event id(Function id) throws Exception {
    id(id.apply(item));
    return this;
  }

  @Override
  public ratpack.sse.Event id(String id) {
    if (id.contains("\n")) {
      throw new IllegalArgumentException("id must not contain \\n - '" + id + "'");
    }
    this.id = id;
    return this;
  }

  @Override
  public ratpack.sse.Event event(Function id) throws Exception {
    event(id.apply(item));
    return this;
  }

  @Override
  public ratpack.sse.Event event(String event) {
    if (event.contains("\n")) {
      throw new IllegalArgumentException("event must not contain \\n - '" + event + "'");
    }
    this.event = event;
    return this;
  }

  @Override
  public ratpack.sse.Event data(Function id) throws Exception {
    data(id.apply(item));
    return this;
  }

  @Override
  public ratpack.sse.Event data(String data) {
    this.data = data;
    return this;
  }

  @Override
  public ratpack.sse.Event comment(String comment) {
    this.comment = comment;
    return this;
  }

  @Override
  public String getId() {
    return id;
  }

  @Override
  public String getEvent() {
    return event;
  }

  @Override
  public String getData() {
    return data;
  }

  @Override
  public String getComment() {
    return comment;
  }

  @Override
  public String toString() {
    return "Event{id='" + id + '\'' + ", event='" + event + '\'' + ", data='" + data + '\'' + '}';
  }
}