/**
 * Copyright 2014 Netflix, Inc.
 * 
 * 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 rx.operators;

import rx.Observable;
import rx.Observable.OnSubscribeFunc;
import rx.Observer;
import rx.Subscription;
import rx.functions.Func0;
import rx.functions.Func1;
import rx.observers.Subscribers;
import rx.subscriptions.CompositeSubscription;
import rx.subscriptions.Subscriptions;

/**
 * Constructs an observable sequence that depends on a resource object.
 */
public class OperationUsing {

    public static <T, RESOURCE extends Subscription> OnSubscribeFunc<T> using(
            final Func0<RESOURCE> resourceFactory,
            final Func1<RESOURCE, Observable<T>> observableFactory) {
        return new OnSubscribeFunc<T>() {
            @Override
            public Subscription onSubscribe(Observer<? super T> observer) {
                Subscription resourceSubscription = Subscriptions.empty();
                try {
                    RESOURCE resource = resourceFactory.call();
                    if (resource != null) {
                        resourceSubscription = resource;
                    }
                    Observable<T> observable = observableFactory.call(resource);
                    SafeObservableSubscription subscription = new SafeObservableSubscription();
                    // Use SafeObserver to guarantee resourceSubscription will
                    // be unsubscribed.
                    return subscription.wrap(new CompositeSubscription(
                            observable.unsafeSubscribe(new SafeObserver<T>(
                                    subscription, observer)),
                            resourceSubscription));
                } catch (Throwable e) {
                    resourceSubscription.unsubscribe();
                    return Observable.<T> error(e).unsafeSubscribe(Subscribers.from(observer));
                }
            }
        };
    }
}
