import { z } from 'zod';
import {
  defineContext,
  defineRule,
  definePolicy,
  evaluatePolicy,
  allow,
  deny,
} from '@bantai-dev/core';

// Define context
const context = defineContext(
  z.object({
    userId: z.string(),
    age: z.number(),
  })
);

// Define rules
const ageRule = defineRule(context, 'age-check', async (input) => {
  if (input.age >= 18) {
    return allow({ reason: 'User is of legal age' });
  }
  return deny({ reason: 'User must be 18 or older' });
});

const userRule = defineRule(context, 'user-check', async (input) => {
  if (input.userId.startsWith('admin')) {
    return allow({ reason: 'User is admin' });
  }
  return deny({ reason: 'User is not admin' });
});

// Define policy
const policy = definePolicy(context, 'access-policy', [ageRule, userRule], {
  defaultStrategy: 'preemptive',
});

// Evaluate policy
const result = await evaluatePolicy(policy, {
  userId: 'admin123',
  age: 25,
});

console.log(result.decision); // 'allow' or 'deny'
console.log(result.isAllowed); // true or false
console.log(result.violatedRules); // Array of violations
console.log(result.evaluatedRules); // Array of all evaluated rules