dicklesworthstone/pi_agent_rust/firecrawl-enterprise-rbac

enum FireCrawlRole {
  Admin = 'admin',
  Developer = 'developer',
  Viewer = 'viewer',
  Service = 'service',
}
interface FireCrawlPermissions {
  read: boolean;
  write: boolean;
  delete: boolean;
  admin: boolean;
}
const ROLE_PERMISSIONS: Record<FireCrawlRole, FireCrawlPermissions> = {
  admin: { read: true, write: true, delete: true, admin: true },
  developer: { read: true, write: true, delete: false, admin: false },
  viewer: { read: true, write: false, delete: false, admin: false },
  service: { read: true, write: true, delete: false, admin: false },
};
function checkPermission(
  role: FireCrawlRole,
  action: keyof FireCrawlPermissions
): boolean {
  return ROLE_PERMISSIONS[role][action];
}

// FireCrawl SAML setup
const samlConfig = {
  entryPoint: 'https://idp.company.com/saml/sso',
  issuer: 'https://firecrawl.com/saml/metadata',
  cert: process.env.SAML_CERT,
  callbackUrl: 'https://app.yourcompany.com/auth/firecrawl/callback',
};
// Map IdP groups to FireCrawl roles
const groupRoleMapping: Record<string, FireCrawlRole> = {
  'Engineering': FireCrawlRole.Developer,
  'Platform-Admins': FireCrawlRole.Admin,
  'Data-Team': FireCrawlRole.Viewer,
};

import { OAuth2Client } from '@firecrawl/sdk';
const oauthClient = new OAuth2Client({
  clientId: process.env.FIRECRAWL_OAUTH_CLIENT_ID!,
  clientSecret: process.env.FIRECRAWL_OAUTH_CLIENT_SECRET!,
  redirectUri: 'https://app.yourcompany.com/auth/firecrawl/callback',
  scopes: ['read', 'write'],
});

interface FireCrawlOrganization {
  id: string;
  name: string;
  ssoEnabled: boolean;
  enforceSso: boolean;
  allowedDomains: string[];
  defaultRole: FireCrawlRole;
}
async function createOrganization(
  config: FireCrawlOrganization
): Promise<void> {
  await firecrawlClient.organizations.create({
    ...config,
    settings: {
      sso: {
        enabled: config.ssoEnabled,
        enforced: config.enforceSso,
        domains: config.allowedDomains,
      },
    },
  });
}

function requireFireCrawlPermission(
  requiredPermission: keyof FireCrawlPermissions
) {
  return async (req: Request, res: Response, next: NextFunction) => {
    const user = req.user as { firecrawlRole: FireCrawlRole };
    if (!checkPermission(user.firecrawlRole, requiredPermission)) {
      return res.status(403).json({
        error: 'Forbidden',
        message: `Missing permission: ${requiredPermission}`,
      });
    }
    next();
  };
}
// Usage
app.delete('/firecrawl/resource/:id',
  requireFireCrawlPermission('delete'),
  deleteResourceHandler
);

interface FireCrawlAuditEntry {
  timestamp: Date;
  userId: string;
  role: FireCrawlRole;
  action: string;
  resource: string;
  success: boolean;
  ipAddress: string;
}
async function logFireCrawlAccess(entry: FireCrawlAuditEntry): Promise<void> {
  await auditDb.insert(entry);
  // Alert on suspicious activity
  if (entry.action === 'delete' && !entry.success) {
    await alertOnSuspiciousActivity(entry);
  }
}

if (!checkPermission(user.role, 'write')) {
  throw new ForbiddenError('Write permission required');
}