feat: inplement auth for cases

This commit is contained in:
Takeshi Kimata
2024-06-06 11:29:33 +09:00
parent d9cc30c839
commit 145f9fed68
10 changed files with 168 additions and 81 deletions

View File

@@ -51,67 +51,6 @@ function authMiddleware(sequelize) {
next(); next();
} }
/**
* Verify user can access project
* (have to be called after verifySignedIn() middleware)
*/
async function verifyProjectVisible(req, res, next) {
const Project = defineProject(sequelize, DataTypes);
const Folder = defineFolder(sequelize, DataTypes);
const Member = defineMember(sequelize, DataTypes);
Project.hasMany(Member, { foreignKey: 'projectId' });
let projectId = req.params.projectId || req.query.projectId;
const folderId = req.params.folderId || req.query.folderId;
if (!projectId && !folderId) {
return res.status(400).json({ error: 'projectId or folderId is required' });
}
if (!projectId) {
// find project id from folderId
const folder = await Folder.findByPk(folderId);
if (folder && folder.projectId) {
projectId = folder.projectId;
} else {
return res.status(404).send('failed to find project from folderId');
}
}
const project = await Project.findOne({
where: { id: projectId },
include: [
{
model: Member,
where: { userId: req.userId },
required: false,
},
],
});
if (!project) {
return res.status(404).send('Project not found');
}
// if project is public, everyone can see
if (project.isPublic) {
next();
return;
}
// if project is private, owner and project member can see
if (project.userId === req.userId) {
next();
return;
}
const member = project.Members && project.Members[0];
if (member) {
next();
return;
}
return res.status(403).json({ error: 'Forbidden' });
}
/** /**
* Verify user has project * Verify user has project
* (have to be called after verifySignedIn() middleware) * (have to be called after verifySignedIn() middleware)
@@ -188,8 +127,9 @@ function authMiddleware(sequelize) {
const Folder = defineFolder(sequelize, DataTypes); const Folder = defineFolder(sequelize, DataTypes);
const Case = defineCase(sequelize, DataTypes); const Case = defineCase(sequelize, DataTypes);
const Member = defineMember(sequelize, DataTypes); const Member = defineMember(sequelize, DataTypes);
Project.hasMany(Member, { foreignKey: 'projectId' }); Project.hasMany(Folder, { foreignKey: 'projectId' });
Folder.hasMany(Case, { foreignKey: 'folderId' }); Folder.hasMany(Case, { foreignKey: 'folderId' });
Project.hasMany(Member, { foreignKey: 'projectId' });
let projectId = req.params.projectId || req.query.projectId; let projectId = req.params.projectId || req.query.projectId;
const folderId = req.params.folderId || req.query.folderId; const folderId = req.params.folderId || req.query.folderId;
@@ -258,7 +198,6 @@ function authMiddleware(sequelize) {
return { return {
verifySignedIn, verifySignedIn,
verifyAdmin, verifyAdmin,
verifyProjectVisible,
verifyProjectOwner, verifyProjectOwner,
verifyProjectManager, verifyProjectManager,
verifyProjectDeveloper, verifyProjectDeveloper,

View File

@@ -0,0 +1,134 @@
const { DataTypes } = require('sequelize');
const defineMember = require('../models/members');
const defineProject = require('../models/projects');
const defineFolder = require('../models/folders');
const defineCase = require('../models/cases');
function verifyVisibleMiddleware(sequelize) {
/**
* Verify user can read project by projectId
* (have to be called after verifySignedIn() middleware)
*/
async function verifyProjectVisibleFromProjectId(req, res, next) {
let projectId = req.params.projectId || req.query.projectId;
if (!projectId) {
return res.status(400).json({ error: 'projectId is required' });
}
const isVisble = await isVisible(projectId, req.userId);
if (isVisble) {
next();
return;
}
return res.status(403).json({ error: 'Forbidden' });
}
/**
* Verify user can read project by folderId
* (have to be called after verifySignedIn() middleware)
*/
async function verifyProjectVisibleFromFolderId(req, res, next) {
const Folder = defineFolder(sequelize, DataTypes);
const folderId = req.params.folderId || req.query.folderId;
if (!folderId) {
return res.status(400).json({ error: 'folderId is required' });
}
// find project id from folderId
const folder = await Folder.findByPk(folderId);
const projectId = folder && folder.id;
if (!projectId) {
return res.status(404).send('failed to find projectId');
}
const isVisble = await isVisible(projectId, req.userId);
if (isVisble) {
next();
return;
}
return res.status(403).json({ error: 'Forbidden' });
}
async function verifyProjectVisibleFromCaseId(req, res, next) {
const Project = defineProject(sequelize, DataTypes);
const Folder = defineFolder(sequelize, DataTypes);
const Case = defineCase(sequelize, DataTypes);
Project.hasMany(Folder, { foreignKey: 'projectId' });
Folder.hasMany(Case, { foreignKey: 'folderId' });
Folder.belongsTo(Project, { foreignKey: 'projectId' });
Case.belongsTo(Folder, { foreignKey: 'folderId' });
const caseId = req.params.caseId || req.query.caseId;
if (!caseId) {
return res.status(400).json({ error: 'caseId is required' });
}
// find project id from caseId
const testCase = await Case.findByPk(caseId, {
include: {
model: Folder,
include: Project,
},
});
const projectId = testCase && testCase.Folder && testCase.Folder.Project && testCase.Folder.Project.id;
if (!projectId) {
return res.status(404).send('failed to find projectId');
}
const isVisble = await isVisible(projectId, req.userId);
if (isVisble) {
next();
return;
}
return res.status(403).json({ error: 'Forbidden' });
}
async function isVisible(projectId, userId) {
const Project = defineProject(sequelize, DataTypes);
const Member = defineMember(sequelize, DataTypes);
Project.hasMany(Member, { foreignKey: 'projectId' });
const project = await Project.findOne({
where: { id: projectId },
include: [
{
model: Member,
where: { userId: userId },
required: false,
},
],
});
if (!project) {
return res.status(404).send('Project not found');
}
// if project is public, everyone can see
if (project.isPublic) {
return true;
}
// if project is private, owner and project member can see
if (project.userId === userId) {
return true;
}
const member = project.Members && project.Members[0];
if (member) {
return true;
}
return false;
}
return {
verifyProjectVisibleFromProjectId,
verifyProjectVisibleFromFolderId,
verifyProjectVisibleFromCaseId,
};
}
module.exports = verifyVisibleMiddleware;

View File

@@ -5,9 +5,10 @@ const { DataTypes } = require('sequelize');
module.exports = function (sequelize) { module.exports = function (sequelize) {
const Case = defineCase(sequelize, DataTypes); const Case = defineCase(sequelize, DataTypes);
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromFolderId } = require('../../middleware/verifyVisible')(sequelize);
router.get('/', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/', verifySignedIn, verifyProjectVisibleFromFolderId, async (req, res) => {
const { folderId } = req.query; const { folderId } = req.query;
if (!folderId) { if (!folderId) {

View File

@@ -13,9 +13,10 @@ module.exports = function (sequelize) {
Step.belongsToMany(Case, { through: 'caseSteps' }); Step.belongsToMany(Case, { through: 'caseSteps' });
Case.belongsToMany(Attachment, { through: 'caseAttachments' }); Case.belongsToMany(Attachment, { through: 'caseAttachments' });
Attachment.belongsToMany(Case, { through: 'caseAttachments' }); Attachment.belongsToMany(Case, { through: 'caseAttachments' });
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromCaseId } = require('../../middleware/verifyVisible')(sequelize);
router.get('/:caseId', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/:caseId', verifySignedIn, verifyProjectVisibleFromCaseId, async (req, res) => {
const caseId = req.params.caseId; const caseId = req.params.caseId;
if (!caseId) { if (!caseId) {

View File

@@ -4,10 +4,11 @@ const defineFolder = require('../../models/folders');
const { DataTypes } = require('sequelize'); const { DataTypes } = require('sequelize');
module.exports = function (sequelize) { module.exports = function (sequelize) {
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromProjectId } = require('../../middleware/verifyVisible')(sequelize);
const Folder = defineFolder(sequelize, DataTypes); const Folder = defineFolder(sequelize, DataTypes);
router.get('/', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/', verifySignedIn, verifyProjectVisibleFromProjectId, async (req, res) => {
const { projectId } = req.query; const { projectId } = req.query;
if (!projectId) { if (!projectId) {

View File

@@ -8,7 +8,8 @@ const defineRunCase = require('../../models/runCases');
const { DataTypes } = require('sequelize'); const { DataTypes } = require('sequelize');
module.exports = function (sequelize) { module.exports = function (sequelize) {
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromProjectId } = require('../../middleware/verifyVisible')(sequelize);
const Project = defineProject(sequelize, DataTypes); const Project = defineProject(sequelize, DataTypes);
const Folder = defineFolder(sequelize, DataTypes); const Folder = defineFolder(sequelize, DataTypes);
@@ -20,7 +21,7 @@ module.exports = function (sequelize) {
Project.hasMany(Run, { foreignKey: 'projectId' }); Project.hasMany(Run, { foreignKey: 'projectId' });
Run.hasMany(RunCase, { foreignKey: 'runId' }); Run.hasMany(RunCase, { foreignKey: 'runId' });
router.get('/:projectId', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/:projectId', verifySignedIn, verifyProjectVisibleFromProjectId, async (req, res) => {
const projectId = req.params.projectId; const projectId = req.params.projectId;
if (!projectId) { if (!projectId) {

View File

@@ -5,12 +5,13 @@ const defineMember = require('../../models/members');
const { DataTypes } = require('sequelize'); const { DataTypes } = require('sequelize');
module.exports = function (sequelize) { module.exports = function (sequelize) {
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromProjectId } = require('../../middleware/verifyVisible')(sequelize);
const User = defineUser(sequelize, DataTypes); const User = defineUser(sequelize, DataTypes);
const Member = defineMember(sequelize, DataTypes); const Member = defineMember(sequelize, DataTypes);
Member.belongsTo(User, { foreignKey: 'userId' }); Member.belongsTo(User, { foreignKey: 'userId' });
router.get('/', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/', verifySignedIn, verifyProjectVisibleFromProjectId, async (req, res) => {
const { projectId } = req.query; const { projectId } = req.query;
if (!projectId) { if (!projectId) {

View File

@@ -5,12 +5,13 @@ const defineFolder = require('../../models/folders');
const { DataTypes } = require('sequelize'); const { DataTypes } = require('sequelize');
module.exports = function (sequelize) { module.exports = function (sequelize) {
const { verifySignedIn, verifyProjectVisible } = require('../../middleware/auth')(sequelize); const { verifySignedIn } = require('../../middleware/auth')(sequelize);
const { verifyProjectVisibleFromProjectId } = require('../../middleware/verifyVisible')(sequelize);
const Project = defineProject(sequelize, DataTypes); const Project = defineProject(sequelize, DataTypes);
const Folder = defineFolder(sequelize, DataTypes); const Folder = defineFolder(sequelize, DataTypes);
Project.hasMany(Folder, { foreignKey: 'projectId' }); Project.hasMany(Folder, { foreignKey: 'projectId' });
router.get('/:projectId', verifySignedIn, verifyProjectVisible, async (req, res) => { router.get('/:projectId', verifySignedIn, verifyProjectVisibleFromProjectId, async (req, res) => {
const projectId = req.params.projectId; const projectId = req.params.projectId;
if (!projectId) { if (!projectId) {
return res.status(400).json({ error: 'projectId is required' }); return res.status(400).json({ error: 'projectId is required' });

View File

@@ -1,5 +1,5 @@
'use client'; 'use client';
import { useEffect, useState } from 'react'; import { useState, useEffect, useContext } from 'react';
import { Input, Textarea, Select, SelectItem, Button, Divider, Tooltip } from '@nextui-org/react'; import { Input, Textarea, Select, SelectItem, Button, Divider, Tooltip } from '@nextui-org/react';
import { useRouter } from '@/src/navigation'; import { useRouter } from '@/src/navigation';
import { Save, Plus, ArrowLeft, ArrowUpFromLine, Circle } from 'lucide-react'; import { Save, Plus, ArrowLeft, ArrowUpFromLine, Circle } from 'lucide-react';
@@ -10,6 +10,7 @@ import { CaseType, AttachmentType, CaseMessages } from '@/types/case';
import { fetchCase, updateCase } from '../caseControl'; import { fetchCase, updateCase } from '../caseControl';
import { fetchCreateStep, fetchDeleteStep } from './stepControl'; import { fetchCreateStep, fetchDeleteStep } from './stepControl';
import { fetchCreateAttachments, fetchDownloadAttachment, fetchDeleteAttachment } from './attachmentControl'; import { fetchCreateAttachments, fetchDownloadAttachment, fetchDeleteAttachment } from './attachmentControl';
import { TokenContext } from '@/utils/TokenProvider';
const defaultTestCase = { const defaultTestCase = {
id: 0, id: 0,
@@ -23,6 +24,10 @@ const defaultTestCase = {
preConditions: '', preConditions: '',
expectedResults: '', expectedResults: '',
folderId: 0, folderId: 0,
Steps: [],
Attachments: [],
isIncluded: false,
runStatus: 0,
}; };
type Props = { type Props = {
@@ -34,13 +39,14 @@ type Props = {
}; };
export default function CaseEditor({ projectId, folderId, caseId, messages, locale }: Props) { export default function CaseEditor({ projectId, folderId, caseId, messages, locale }: Props) {
const context = useContext(TokenContext);
const [testCase, setTestCase] = useState<CaseType>(defaultTestCase); const [testCase, setTestCase] = useState<CaseType>(defaultTestCase);
const [isTitleInvalid, setIsTitleInvalid] = useState<boolean>(false); const [isTitleInvalid, setIsTitleInvalid] = useState<boolean>(false);
const [isUpdating, setIsUpdating] = useState<boolean>(false); const [isUpdating, setIsUpdating] = useState<boolean>(false);
const router = useRouter(); const router = useRouter();
const onPlusClick = async (newStepNo: number) => { const onPlusClick = async (newStepNo: number) => {
const newStep = await fetchCreateStep(newStepNo, caseId); const newStep = await fetchCreateStep(newStepNo, Number(caseId));
if (newStep) { if (newStep) {
newStep.caseSteps = { stepNo: newStepNo }; newStep.caseSteps = { stepNo: newStepNo };
const updatedSteps = testCase.Steps.map((step) => { const updatedSteps = testCase.Steps.map((step) => {
@@ -136,8 +142,11 @@ export default function CaseEditor({ projectId, folderId, caseId, messages, loca
useEffect(() => { useEffect(() => {
async function fetchDataEffect() { async function fetchDataEffect() {
if (!context.isSignedIn()) {
return;
}
try { try {
const data = await fetchCase(caseId); const data = await fetchCase(context.token.access_token, Number(caseId));
setTestCase(data); setTestCase(data);
} catch (error: any) { } catch (error: any) {
console.error('Error in effect:', error.message); console.error('Error in effect:', error.message);
@@ -145,7 +154,7 @@ export default function CaseEditor({ projectId, folderId, caseId, messages, loca
} }
fetchDataEffect(); fetchDataEffect();
}, []); }, [context]);
return ( return (
<> <>
@@ -170,7 +179,7 @@ export default function CaseEditor({ projectId, folderId, caseId, messages, loca
isLoading={isUpdating} isLoading={isUpdating}
onPress={async () => { onPress={async () => {
setIsUpdating(true); setIsUpdating(true);
await updateCase(testCase); await updateCase(context.token.access_token, testCase);
setIsUpdating(false); setIsUpdating(false);
}} }}
> >

View File

@@ -96,7 +96,6 @@ async function updateCase(jwt: string, updateCaseData: CaseType) {
}; };
const url = `${apiServer}/cases/${updateCaseData.id}`; const url = `${apiServer}/cases/${updateCaseData.id}`;
try { try {
const response = await fetch(url, fetchOptions); const response = await fetch(url, fetchOptions);
if (!response.ok) { if (!response.ok) {