/*
 * Copyright (C) 2014 Apple Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#import <Foundation/Foundation.h>

#if USE(APPLE_INTERNAL_SDK)

#import <AssertionServices/BKSApplicationStateMonitor.h>
#import <AssertionServices/BKSProcess.h>
#import <AssertionServices/BKSProcessAssertion.h>

#else

enum {
    BKSApplicationStateUnknown                      = 0,
    BKSApplicationStateTerminated                   = (1 << 0),
    BKSApplicationStateBackgroundTaskSuspended      = (1 << 1),
    BKSApplicationStateBackgroundRunning            = (1 << 2),
    BKSApplicationStateForegroundRunning            = (1 << 3),
    BKSApplicationStateProcessServer                = (1 << 4),
    BKSApplicationStateForegroundRunningObscured    = (1 << 5),
};
typedef uint32_t BKSApplicationState;

extern NSString * const BKSApplicationStateMostElevatedStateForProcessIDKey;
extern NSString * const BKSApplicationStateProcessIDKey;

typedef void(^BKSApplicationStateChangedHandler)(NSDictionary *userInfo);

@interface BKSApplicationStateMonitor : NSObject

@property (nonatomic, copy) BKSApplicationStateChangedHandler handler;

- (BKSApplicationState)mostElevatedApplicationStateForPID:(pid_t)pid;
- (void)invalidate;

@end

enum {
    BKSProcessAssertionNone = 0,
    BKSProcessAssertionPreventTaskSuspend  = 1 << 0,
    BKSProcessAssertionPreventTaskThrottleDown = 1 << 1,
    BKSProcessAssertionAllowIdleSleep = 1 << 2,
    BKSProcessAssertionWantsForegroundResourcePriority = 1 << 3,
    BKSProcessAssertionAllowSuspendOnSleep = 1 << 4,
};
typedef uint32_t BKSProcessAssertionFlags;

enum {
    BKSProcessAssertionReasonMediaPlayback = 1,
    BKSProcessAssertionReasonFinishTask = 4,
    BKSProcessAssertionReasonExtension = 13,
    BKSProcessAssertionReasonFinishTaskUnbounded = 10004,
};
typedef uint32_t BKSProcessAssertionReason;

typedef void (^BKSProcessAssertionInvalidationHandler)(void);
typedef void (^BKSProcessAssertionAcquisitionHandler)(BOOL acquired);

@interface BKSProcessAssertion : NSObject
@end

@interface BKSProcessAssertion ()
@property (nonatomic, assign) BKSProcessAssertionFlags flags;
- (id)initWithPID:(pid_t)pid flags:(BKSProcessAssertionFlags)flags reason:(BKSProcessAssertionReason)reason name:(NSString *)name withHandler:(BKSProcessAssertionAcquisitionHandler)handler;

@property (nonatomic, readonly) BOOL valid;
@property (nonatomic, copy) BKSProcessAssertionInvalidationHandler invalidationHandler;
- (void)invalidate;
@end

enum {
    BKSProcessTaskStateNone,
    BKSProcessTaskStateRunning,
    BKSProcessTaskStateSuspended,
};
typedef uint32_t BKSProcessTaskState;

@class BKSProcess;

@protocol BKSProcessDelegate <NSObject>
@optional
- (void)process:(BKSProcess *)process taskStateDidChange:(BKSProcessTaskState)newState;
@end

@interface BKSProcess : NSObject
@end

@interface BKSProcess ()
+ (BKSProcess *)currentProcess;
@property (nonatomic, readwrite, weak) id <BKSProcessDelegate> delegate;
@property (nonatomic, readonly, assign) BKSProcessTaskState taskState;
@end

#endif
