Targets working?

This commit is contained in:
Owen Schwartz 2024-10-19 20:47:05 -04:00
parent 0fa3382cda
commit bf106c221e
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
3 changed files with 208 additions and 188 deletions

View file

@ -33,7 +33,7 @@ export async function verifyTargetAccess(req: Request, res: Response, next: Next
const resourceId = target[0].resourceId; const resourceId = target[0].resourceId;
if (resourceId) { if (!resourceId) {
return next( return next(
createHttpError( createHttpError(
HttpCode.INTERNAL_SERVER_ERROR, HttpCode.INTERNAL_SERVER_ERROR,

View file

@ -13,9 +13,15 @@ import api from "@app/api"
import { AxiosResponse } from "axios" import { AxiosResponse } from "axios"
import { ListTargetsResponse } from "@server/routers/target/listTargets" import { ListTargetsResponse } from "@server/routers/target/listTargets"
const isValidIPAddress = (ip: string) => {
const ipv4Regex = /^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/;
return ipv4Regex.test(ip);
}
export default function ReverseProxyTargets({ params }: { params: { resourceId: string } }) { export default function ReverseProxyTargets({ params }: { params: { resourceId: string } }) {
const [targets, setTargets] = useState<ListTargetsResponse["targets"]>([]) const [targets, setTargets] = useState<ListTargetsResponse["targets"]>([])
const [nextId, setNextId] = useState(1) const [nextId, setNextId] = useState(1)
const [ipError, setIpError] = useState("")
useEffect(() => { useEffect(() => {
if (typeof window !== "undefined") { if (typeof window !== "undefined") {
@ -30,21 +36,27 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
const [newTarget, setNewTarget] = useState({ const [newTarget, setNewTarget] = useState({
resourceId: params.resourceId, resourceId: params.resourceId,
ip: "", ip: "",
method: "GET", method: "http",
port: 80, port: 80,
protocol: "http", protocol: "TCP",
}) })
const addTarget = async () => { const addTarget = () => {
const res = await api.put(`/resource/${params.resourceId}/target`, { if (!isValidIPAddress(newTarget.ip)) {
setIpError("Invalid IP address format");
return;
}
setIpError("");
api.put(`/resource/${params.resourceId}/target`, {
...newTarget, ...newTarget,
resourceId: undefined resourceId: undefined
}) })
.catch((err) => { .catch((err) => {
console.error(err) console.error(err)
}); }).then((res) => {
// console.log(res)
setTargets([...targets, { ...newTarget, targetId: nextId, enabled: true }]) setTargets([...targets, { ...newTarget, targetId: nextId, enabled: true }])
setNextId(nextId + 1) setNextId(nextId + 1)
setNewTarget({ setNewTarget({
@ -54,11 +66,17 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
port: 80, port: 80,
protocol: "http", protocol: "http",
}) })
});
} }
const removeTarget = async (targetId: number) => { const removeTarget = (targetId: number) => {
setTargets(targets.filter((target) => target.targetId !== targetId)) api.delete(`/target/${targetId}`)
const res = await api.delete(`/target/${targetId}`) .catch((err) => {
console.error(err)
}).then((res) => {
setTargets(targets.filter((target) => target.targetId !== targetId));
});
} }
const toggleTarget = (targetId: number) => { const toggleTarget = (targetId: number) => {
@ -67,7 +85,12 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
target.targetId === targetId ? { ...target, enabled: !target.enabled } : target target.targetId === targetId ? { ...target, enabled: !target.enabled } : target
) )
) )
const res = api.post(`/target/${targetId}`, { enabled: !targets.find((target) => target.targetId === targetId)?.enabled }) api.post(`/target/${targetId}`, { enabled: !targets.find((target) => target.targetId === targetId)?.enabled })
.catch((err) => {
console.error(err)
}).then((res) => {
console.log(res)
});
// Add a visual feedback // Add a visual feedback
const targetElement = document.getElementById(`target-${targetId}`) const targetElement = document.getElementById(`target-${targetId}`)
@ -81,11 +104,6 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
return ( return (
<div className="space-y-6"> <div className="space-y-6">
{/* <Card>
<CardHeader> */}
{/* <CardTitle>Add New Target</CardTitle>
</CardHeader>
<CardContent> */}
<form <form
onSubmit={(e) => { onSubmit={(e) => {
e.preventDefault() e.preventDefault()
@ -99,9 +117,13 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
<Input <Input
id="ip" id="ip"
value={newTarget.ip} value={newTarget.ip}
onChange={(e) => setNewTarget({ ...newTarget, ip: e.target.value })} onChange={(e) => {
setNewTarget({ ...newTarget, ip: e.target.value })
setIpError("")
}}
required required
/> />
{ipError && <p className="text-red-500 text-sm">{ipError}</p>}
</div> </div>
<div className="space-y-2"> <div className="space-y-2">
<Label htmlFor="method">Method</Label> <Label htmlFor="method">Method</Label>
@ -113,10 +135,8 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
<SelectValue placeholder="Select method" /> <SelectValue placeholder="Select method" />
</SelectTrigger> </SelectTrigger>
<SelectContent> <SelectContent>
<SelectItem value="GET">GET</SelectItem> <SelectItem value="http">HTTP</SelectItem>
<SelectItem value="POST">POST</SelectItem> <SelectItem value="https">HTTPS</SelectItem>
<SelectItem value="PUT">PUT</SelectItem>
<SelectItem value="DELETE">DELETE</SelectItem>
</SelectContent> </SelectContent>
</Select> </Select>
</div> </div>
@ -140,8 +160,8 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
<SelectValue placeholder="Select protocol" /> <SelectValue placeholder="Select protocol" />
</SelectTrigger> </SelectTrigger>
<SelectContent> <SelectContent>
<SelectItem value="http">HTTP</SelectItem> <SelectItem value="UDP">UDP</SelectItem>
<SelectItem value="https">HTTPS</SelectItem> <SelectItem value="TCP">TCP</SelectItem>
</SelectContent> </SelectContent>
</Select> </Select>
</div> </div>
@ -150,24 +170,32 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
<PlusCircle className="mr-2 h-4 w-4" /> Add Target <PlusCircle className="mr-2 h-4 w-4" /> Add Target
</Button> </Button>
</form> </form>
{/* </CardContent>
</Card> */}
<div className="grid gap-4 md:grid-cols-2 lg:grid-cols-3"> <div className="space-y-4">
{targets.map((target) => ( {targets.map((target) => (
<Card key={target.targetId} id={`target-${target.targetId}`} > <Card key={target.targetId} id={`target-${target.targetId}`} className="w-full p-4">
<CardHeader className="flex flex-row items-center justify-between space-y-0 pb-2"> <CardHeader className="flex flex-row items-start justify-between space-y-0 pb-2 px-0 pt-0">
<CardTitle className="text-sm font-medium flex items-center"> <CardTitle className="text-lg font-medium flex items-center">
<Server className="mr-2 h-4 w-4" /> <Server className="mr-2 h-5 w-5" />
Target {target.targetId} Target {target.targetId}
</CardTitle> </CardTitle>
<div className="flex flex-col items-end space-y-2">
<Switch <Switch
checked={target.enabled} checked={target.enabled}
onCheckedChange={() => toggleTarget(target.targetId)} onCheckedChange={() => toggleTarget(target.targetId)}
/> />
<Button
variant="ghost"
size="sm"
className="text-destructive hover:text-destructive hover:bg-destructive/10"
onClick={() => removeTarget(target.targetId)}
>
<Trash2 className="h-4 w-4" />
</Button>
</div>
</CardHeader> </CardHeader>
<CardContent> <CardContent className="px-0 py-2">
<div className="grid gap-2"> <div className="grid gap-2 md:grid-cols-2 lg:grid-cols-3">
<div className="flex items-center"> <div className="flex items-center">
<Globe className="mr-2 h-4 w-4 text-muted-foreground" /> <Globe className="mr-2 h-4 w-4 text-muted-foreground" />
<span className="text-sm">{target.ip}:{target.port}</span> <span className="text-sm">{target.ip}:{target.port}</span>
@ -181,14 +209,6 @@ export default function ReverseProxyTargets({ params }: { params: { resourceId:
<Badge variant={target.enabled ? "default" : "secondary"}>{target.protocol?.toUpperCase()}</Badge> <Badge variant={target.enabled ? "default" : "secondary"}>{target.protocol?.toUpperCase()}</Badge>
</div> </div>
</div> </div>
<Button
variant="destructive"
size="sm"
className="mt-4 w-full"
onClick={() => removeTarget(target.targetId)}
>
<Trash2 className="mr-2 h-4 w-4" /> Remove
</Button>
</CardContent> </CardContent>
</Card> </Card>
))} ))}