/*
 * Copyright 2015 Open Networking Laboratory
 *
 * 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 org.onosproject.net.intent.impl;

import org.onosproject.net.intent.IntentData;

import java.util.Optional;

import static com.google.common.base.Preconditions.checkNotNull;
import static org.onosproject.net.intent.IntentState.WITHDRAWN;

/**
 * Represents a phase of requesting a withdraw of an intent.
 */
class WithdrawRequest implements IntentUpdate {

    // TODO: define an interface and use it, instead of IntentManager
    private final IntentManager intentManager;
    private final IntentData pending;

    WithdrawRequest(IntentManager intentManager, IntentData intentData) {
        this.intentManager = checkNotNull(intentManager);
        this.pending = checkNotNull(intentData);
    }

    @Override
    public Optional<IntentUpdate> execute() {
        //FIXME need store interface
        IntentData current = intentManager.store.getIntentData(pending.key());
        if (current == null) {
            return Optional.of(new Withdrawn(pending, WITHDRAWN));
        }
        //TODO perhaps we want to validate that the pending and current are the
        // same version i.e. they are the same
        // Note: this call is not just the symmetric version of submit
        return Optional.of(new WithdrawCoordinating(intentManager, pending, current));
    }
}
