package act.test;

/*-
 * #%L
 * ACT Framework
 * %%
 * Copyright (C) 2014 - 2019 ActFramework
 * %%
 * 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.
 * #L%
 */

import act.Act;
import act.job.JobManager;
import act.sys.Env;
import act.util.ProgressGauge;
import act.ws.WebSocketConnectionListener;
import act.ws.WebSocketContext;
import act.ws.WsEndpoint;
import com.alibaba.fastjson.JSON;
import org.osgl.util.C;
import org.osgl.util.E;

import javax.inject.Inject;

import static java.util.concurrent.TimeUnit.MILLISECONDS;

@Env.RequireMode(Act.Mode.DEV)
@WsEndpoint("/~/ws/test/progress")
public class TestWsProgress implements WebSocketConnectionListener {

    @Inject
    private Test test;

    @Inject
    private JobManager jobManager;

    @Override
    public void onConnect(final WebSocketContext context) {
        if (null == test.gauge) {
            if (test.result().isEmpty()) {
                context.sendJsonToSelf(C.Map("error", true));
            } else {
                context.sendJsonToSelf(C.Map("done", true));
            }
            return;
        }
        E.illegalStateIfNot(Test.inProgress() || test.gauge.isDone());
        test.gauge.addListener(new ProgressGauge.Listener() {
            @Override
            public void onUpdate(ProgressGauge progressGauge) {
                context.send(JSON.toJSONString(progressGauge));
                if (progressGauge.isDone()) {
                    jobManager.delay(new Runnable() {
                        @Override
                        public void run() {
                            context.connection().close();
                        }
                    }, 100, MILLISECONDS);
                }
            }
        });
    }

    @Override
    public void onClose(WebSocketContext context) {
    }
}
