mirror of
https://github.com/dstotijn/hetty.git
synced 2025-07-01 18:47:29 -04:00
Replace SQLite with BadgerDB
This commit is contained in:
@ -8,7 +8,6 @@ linters:
|
|||||||
- metalinter
|
- metalinter
|
||||||
- module
|
- module
|
||||||
- performance
|
- performance
|
||||||
- sql
|
|
||||||
- style
|
- style
|
||||||
- test
|
- test
|
||||||
- unused
|
- unused
|
||||||
@ -28,8 +27,6 @@ linters:
|
|||||||
- wrapcheck
|
- wrapcheck
|
||||||
|
|
||||||
linters-settings:
|
linters-settings:
|
||||||
errcheck:
|
|
||||||
ignore: "database/sql:Rollback"
|
|
||||||
gci:
|
gci:
|
||||||
local-prefixes: github.com/dstotijn/hetty
|
local-prefixes: github.com/dstotijn/hetty
|
||||||
godot:
|
godot:
|
||||||
|
@ -2,7 +2,7 @@ ARG GO_VERSION=1.16
|
|||||||
ARG NODE_VERSION=14.11
|
ARG NODE_VERSION=14.11
|
||||||
ARG ALPINE_VERSION=3.12
|
ARG ALPINE_VERSION=3.12
|
||||||
|
|
||||||
ARG CGO_ENABLED=1
|
ARG CGO_ENABLED=0
|
||||||
|
|
||||||
FROM node:${NODE_VERSION}-alpine AS node-builder
|
FROM node:${NODE_VERSION}-alpine AS node-builder
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
|
2
Makefile
2
Makefile
@ -7,7 +7,7 @@ build-admin:
|
|||||||
|
|
||||||
.PHONY: build
|
.PHONY: build
|
||||||
build: build-admin
|
build: build-admin
|
||||||
CGO_ENABLED=1 mv admin/dist cmd/hetty/admin && go build ./cmd/hetty
|
CGO_ENABLED=0 mv admin/dist cmd/hetty/admin && go build ./cmd/hetty
|
||||||
|
|
||||||
.PHONY: release-dry-run
|
.PHONY: release-dry-run
|
||||||
release-dry-run: build-admin
|
release-dry-run: build-admin
|
||||||
|
17
README.md
17
README.md
@ -33,7 +33,7 @@ for details.
|
|||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
Hetty compiles to a self-contained binary, with an embedded SQLite database
|
Hetty compiles to a self-contained binary, with an embedded BadgerDB database
|
||||||
and web based admin interface.
|
and web based admin interface.
|
||||||
|
|
||||||
### Install pre-built release (recommended)
|
### Install pre-built release (recommended)
|
||||||
@ -47,11 +47,10 @@ and web based admin interface.
|
|||||||
- [Go 1.16](https://golang.org/)
|
- [Go 1.16](https://golang.org/)
|
||||||
- [Yarn](https://yarnpkg.com/)
|
- [Yarn](https://yarnpkg.com/)
|
||||||
|
|
||||||
Hetty depends on SQLite (via [mattn/go-sqlite3](https://github.com/mattn/go-sqlite3))
|
When building from source, the static resources for the admin interface
|
||||||
and needs `cgo` to compile. Additionally, the static resources for the admin interface
|
|
||||||
(Next.js) need to be generated via [Yarn](https://yarnpkg.com/). The generated
|
(Next.js) need to be generated via [Yarn](https://yarnpkg.com/). The generated
|
||||||
files will be embedded (using the [embed](https://golang.org/pkg/embed/) package)
|
files will be embedded (using the [embed](https://golang.org/pkg/embed/)
|
||||||
when you use the `build` Makefile target.
|
package) when you use the `build` Makefile target.
|
||||||
|
|
||||||
Clone the repository and use the `build` make target to create a binary:
|
Clone the repository and use the `build` make target to create a binary:
|
||||||
|
|
||||||
@ -64,7 +63,7 @@ $ make build
|
|||||||
### Docker
|
### Docker
|
||||||
|
|
||||||
A Docker image is available on Docker Hub: [`dstotijn/hetty`](https://hub.docker.com/r/dstotijn/hetty).
|
A Docker image is available on Docker Hub: [`dstotijn/hetty`](https://hub.docker.com/r/dstotijn/hetty).
|
||||||
For persistent storage of CA certificates and project databases, mount a volume:
|
For persistent storage of CA certificates and projects database, mount a volume:
|
||||||
|
|
||||||
```
|
```
|
||||||
$ mkdir -p $HOME/.hetty
|
$ mkdir -p $HOME/.hetty
|
||||||
@ -77,7 +76,7 @@ When Hetty is run, by default it listens on `:8080` and is accessible via
|
|||||||
http://localhost:8080. Depending on incoming HTTP requests, it either acts as a
|
http://localhost:8080. Depending on incoming HTTP requests, it either acts as a
|
||||||
MITM proxy, or it serves the API and web interface.
|
MITM proxy, or it serves the API and web interface.
|
||||||
|
|
||||||
By default, project database files and CA certificates are stored in a `.hetty`
|
By default, the projects database files and CA certificates are stored in a `.hetty`
|
||||||
directory under the user's home directory (`$HOME` on Linux/macOS, `%USERPROFILE%`
|
directory under the user's home directory (`$HOME` on Linux/macOS, `%USERPROFILE%`
|
||||||
on Windows).
|
on Windows).
|
||||||
|
|
||||||
@ -101,8 +100,8 @@ Usage of ./hetty:
|
|||||||
CA certificate filepath. Creates a new CA certificate if file doesn't exist (default "~/.hetty/hetty_cert.pem")
|
CA certificate filepath. Creates a new CA certificate if file doesn't exist (default "~/.hetty/hetty_cert.pem")
|
||||||
-key string
|
-key string
|
||||||
CA private key filepath. Creates a new CA private key if file doesn't exist (default "~/.hetty/hetty_key.pem")
|
CA private key filepath. Creates a new CA private key if file doesn't exist (default "~/.hetty/hetty_key.pem")
|
||||||
-projects string
|
-db string
|
||||||
Projects directory path (default "~/.hetty/projects")
|
Database directory path (default "~/.hetty/db")
|
||||||
```
|
```
|
||||||
|
|
||||||
You should see:
|
You should see:
|
||||||
|
@ -24,9 +24,19 @@ const useStyles = makeStyles((theme: Theme) =>
|
|||||||
})
|
})
|
||||||
);
|
);
|
||||||
|
|
||||||
|
const CREATE_PROJECT = gql`
|
||||||
|
mutation CreateProject($name: String!) {
|
||||||
|
createProject(name: $name) {
|
||||||
|
id
|
||||||
|
name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
`;
|
||||||
|
|
||||||
const OPEN_PROJECT = gql`
|
const OPEN_PROJECT = gql`
|
||||||
mutation OpenProject($name: String!) {
|
mutation OpenProject($id: ID!) {
|
||||||
openProject(name: $name) {
|
openProject(id: $id) {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
}
|
}
|
||||||
@ -37,8 +47,15 @@ function NewProject(): JSX.Element {
|
|||||||
const classes = useStyles();
|
const classes = useStyles();
|
||||||
const [input, setInput] = useState(null);
|
const [input, setInput] = useState(null);
|
||||||
|
|
||||||
const [openProject, { error, loading }] = useMutation(OPEN_PROJECT, {
|
const [createProject, { error: createProjErr, loading: createProjLoading }] = useMutation(CREATE_PROJECT, {
|
||||||
onError: () => {},
|
onError: () => { },
|
||||||
|
onCompleted(data) {
|
||||||
|
input.value = "";
|
||||||
|
openProject({ variables: { id: data.createProject.id } });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const [openProject, { error: openProjErr, loading: openProjLoading }] = useMutation(OPEN_PROJECT, {
|
||||||
|
onError: () => { },
|
||||||
onCompleted() {
|
onCompleted() {
|
||||||
input.value = "";
|
input.value = "";
|
||||||
},
|
},
|
||||||
@ -47,10 +64,11 @@ function NewProject(): JSX.Element {
|
|||||||
fields: {
|
fields: {
|
||||||
activeProject() {
|
activeProject() {
|
||||||
const activeProjRef = cache.writeFragment({
|
const activeProjRef = cache.writeFragment({
|
||||||
id: openProject.name,
|
id: openProject.id,
|
||||||
data: openProject,
|
data: openProject,
|
||||||
fragment: gql`
|
fragment: gql`
|
||||||
fragment ActiveProject on Project {
|
fragment ActiveProject on Project {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
type
|
type
|
||||||
@ -61,10 +79,11 @@ function NewProject(): JSX.Element {
|
|||||||
},
|
},
|
||||||
projects(_, { DELETE }) {
|
projects(_, { DELETE }) {
|
||||||
cache.writeFragment({
|
cache.writeFragment({
|
||||||
id: openProject.name,
|
id: openProject.id,
|
||||||
data: openProject,
|
data: openProject,
|
||||||
fragment: gql`
|
fragment: gql`
|
||||||
fragment OpenProject on Project {
|
fragment OpenProject on Project {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
type
|
type
|
||||||
@ -78,9 +97,9 @@ function NewProject(): JSX.Element {
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
const handleNewProjectForm = (e: React.SyntheticEvent) => {
|
const handleCreateAndOpenProjectForm = (e: React.SyntheticEvent) => {
|
||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
openProject({ variables: { name: input.value } });
|
createProject({ variables: { name: input.value } });
|
||||||
};
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@ -88,7 +107,7 @@ function NewProject(): JSX.Element {
|
|||||||
<Box mb={3}>
|
<Box mb={3}>
|
||||||
<Typography variant="h6">New project</Typography>
|
<Typography variant="h6">New project</Typography>
|
||||||
</Box>
|
</Box>
|
||||||
<form onSubmit={handleNewProjectForm} autoComplete="off">
|
<form onSubmit={handleCreateAndOpenProjectForm} autoComplete="off">
|
||||||
<TextField
|
<TextField
|
||||||
className={classes.projectName}
|
className={classes.projectName}
|
||||||
color="secondary"
|
color="secondary"
|
||||||
@ -100,8 +119,8 @@ function NewProject(): JSX.Element {
|
|||||||
}}
|
}}
|
||||||
label="Project name"
|
label="Project name"
|
||||||
placeholder="Project name…"
|
placeholder="Project name…"
|
||||||
error={Boolean(error)}
|
error={Boolean(createProjErr || openProjErr)}
|
||||||
helperText={error && error.message}
|
helperText={createProjErr && createProjErr.message || openProjErr && openProjErr.message}
|
||||||
/>
|
/>
|
||||||
<Button
|
<Button
|
||||||
className={classes.button}
|
className={classes.button}
|
||||||
@ -109,8 +128,8 @@ function NewProject(): JSX.Element {
|
|||||||
variant="contained"
|
variant="contained"
|
||||||
color="secondary"
|
color="secondary"
|
||||||
size="large"
|
size="large"
|
||||||
disabled={loading}
|
disabled={createProjLoading || openProjLoading}
|
||||||
startIcon={loading ? <CircularProgress size={22} /> : <AddIcon />}
|
startIcon={createProjLoading || openProjLoading ? <CircularProgress size={22} /> : <AddIcon />}
|
||||||
>
|
>
|
||||||
Create & open project
|
Create & open project
|
||||||
</Button>
|
</Button>
|
||||||
|
@ -48,6 +48,7 @@ const useStyles = makeStyles((theme: Theme) =>
|
|||||||
const PROJECTS = gql`
|
const PROJECTS = gql`
|
||||||
query Projects {
|
query Projects {
|
||||||
projects {
|
projects {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
}
|
}
|
||||||
@ -55,8 +56,9 @@ const PROJECTS = gql`
|
|||||||
`;
|
`;
|
||||||
|
|
||||||
const OPEN_PROJECT = gql`
|
const OPEN_PROJECT = gql`
|
||||||
mutation OpenProject($name: String!) {
|
mutation OpenProject($id: ID!) {
|
||||||
openProject(name: $name) {
|
openProject(id: $id) {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
}
|
}
|
||||||
@ -72,8 +74,8 @@ const CLOSE_PROJECT = gql`
|
|||||||
`;
|
`;
|
||||||
|
|
||||||
const DELETE_PROJECT = gql`
|
const DELETE_PROJECT = gql`
|
||||||
mutation DeleteProject($name: String!) {
|
mutation DeleteProject($id: ID!) {
|
||||||
deleteProject(name: $name) {
|
deleteProject(id: $id) {
|
||||||
success
|
success
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -89,7 +91,7 @@ function ProjectList(): JSX.Element {
|
|||||||
{ error: openProjErr, loading: openProjLoading },
|
{ error: openProjErr, loading: openProjLoading },
|
||||||
] = useMutation(OPEN_PROJECT, {
|
] = useMutation(OPEN_PROJECT, {
|
||||||
errorPolicy: "all",
|
errorPolicy: "all",
|
||||||
onError: () => {},
|
onError: () => { },
|
||||||
update(cache, { data: { openProject } }) {
|
update(cache, { data: { openProject } }) {
|
||||||
cache.modify({
|
cache.modify({
|
||||||
fields: {
|
fields: {
|
||||||
@ -98,6 +100,7 @@ function ProjectList(): JSX.Element {
|
|||||||
data: openProject,
|
data: openProject,
|
||||||
fragment: gql`
|
fragment: gql`
|
||||||
fragment ActiveProject on Project {
|
fragment ActiveProject on Project {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
type
|
type
|
||||||
@ -108,10 +111,11 @@ function ProjectList(): JSX.Element {
|
|||||||
},
|
},
|
||||||
projects(_, { DELETE }) {
|
projects(_, { DELETE }) {
|
||||||
cache.writeFragment({
|
cache.writeFragment({
|
||||||
id: openProject.name,
|
id: openProject.id,
|
||||||
data: openProject,
|
data: openProject,
|
||||||
fragment: gql`
|
fragment: gql`
|
||||||
fragment OpenProject on Project {
|
fragment OpenProject on Project {
|
||||||
|
id
|
||||||
name
|
name
|
||||||
isActive
|
isActive
|
||||||
type
|
type
|
||||||
@ -129,7 +133,7 @@ function ProjectList(): JSX.Element {
|
|||||||
});
|
});
|
||||||
const [closeProject, { error: closeProjErr }] = useMutation(CLOSE_PROJECT, {
|
const [closeProject, { error: closeProjErr }] = useMutation(CLOSE_PROJECT, {
|
||||||
errorPolicy: "all",
|
errorPolicy: "all",
|
||||||
onError: () => {},
|
onError: () => { },
|
||||||
update(cache) {
|
update(cache) {
|
||||||
cache.modify({
|
cache.modify({
|
||||||
fields: {
|
fields: {
|
||||||
@ -151,7 +155,7 @@ function ProjectList(): JSX.Element {
|
|||||||
{ loading: deleteProjLoading, error: deleteProjErr },
|
{ loading: deleteProjLoading, error: deleteProjErr },
|
||||||
] = useMutation(DELETE_PROJECT, {
|
] = useMutation(DELETE_PROJECT, {
|
||||||
errorPolicy: "all",
|
errorPolicy: "all",
|
||||||
onError: () => {},
|
onError: () => { },
|
||||||
update(cache) {
|
update(cache) {
|
||||||
cache.modify({
|
cache.modify({
|
||||||
fields: {
|
fields: {
|
||||||
@ -165,14 +169,14 @@ function ProjectList(): JSX.Element {
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
const [deleteProjName, setDeleteProjName] = useState(null);
|
const [deleteProj, setDeleteProj] = useState(null);
|
||||||
const [deleteDiagOpen, setDeleteDiagOpen] = useState(false);
|
const [deleteDiagOpen, setDeleteDiagOpen] = useState(false);
|
||||||
const handleDeleteButtonClick = (name: string) => {
|
const handleDeleteButtonClick = (project: any) => {
|
||||||
setDeleteProjName(name);
|
setDeleteProj(project);
|
||||||
setDeleteDiagOpen(true);
|
setDeleteDiagOpen(true);
|
||||||
};
|
};
|
||||||
const handleDeleteConfirm = () => {
|
const handleDeleteConfirm = () => {
|
||||||
deleteProject({ variables: { name: deleteProjName } });
|
deleteProject({ variables: { id: deleteProj.id } });
|
||||||
};
|
};
|
||||||
const handleDeleteCancel = () => {
|
const handleDeleteCancel = () => {
|
||||||
setDeleteDiagOpen(false);
|
setDeleteDiagOpen(false);
|
||||||
@ -190,7 +194,7 @@ function ProjectList(): JSX.Element {
|
|||||||
<div>
|
<div>
|
||||||
<Dialog open={deleteDiagOpen} onClose={handleDeleteCancel}>
|
<Dialog open={deleteDiagOpen} onClose={handleDeleteCancel}>
|
||||||
<DialogTitle>
|
<DialogTitle>
|
||||||
Delete project “<strong>{deleteProjName}</strong>”?
|
Delete project “<strong>{deleteProj?.name}</strong>”?
|
||||||
</DialogTitle>
|
</DialogTitle>
|
||||||
<DialogContent>
|
<DialogContent>
|
||||||
<DialogContentText>
|
<DialogContentText>
|
||||||
@ -223,7 +227,7 @@ function ProjectList(): JSX.Element {
|
|||||||
onClose={handleCloseDeleteNotif}
|
onClose={handleCloseDeleteNotif}
|
||||||
>
|
>
|
||||||
<Alert onClose={handleCloseDeleteNotif} severity="info">
|
<Alert onClose={handleCloseDeleteNotif} severity="info">
|
||||||
Project <strong>{deleteProjName}</strong> was deleted.
|
Project <strong>{deleteProj?.name}</strong> was deleted.
|
||||||
</Alert>
|
</Alert>
|
||||||
</Snackbar>
|
</Snackbar>
|
||||||
|
|
||||||
@ -253,7 +257,7 @@ function ProjectList(): JSX.Element {
|
|||||||
{projData?.projects.length > 0 && (
|
{projData?.projects.length > 0 && (
|
||||||
<List className={classes.projectsList}>
|
<List className={classes.projectsList}>
|
||||||
{projData.projects.map((project) => (
|
{projData.projects.map((project) => (
|
||||||
<ListItem key={project.name}>
|
<ListItem key={project.id}>
|
||||||
<ListItemAvatar>
|
<ListItemAvatar>
|
||||||
<Avatar
|
<Avatar
|
||||||
className={
|
className={
|
||||||
@ -281,7 +285,7 @@ function ProjectList(): JSX.Element {
|
|||||||
disabled={openProjLoading || projLoading}
|
disabled={openProjLoading || projLoading}
|
||||||
onClick={() =>
|
onClick={() =>
|
||||||
openProject({
|
openProject({
|
||||||
variables: { name: project.name },
|
variables: { id: project.id },
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
>
|
>
|
||||||
@ -293,7 +297,7 @@ function ProjectList(): JSX.Element {
|
|||||||
<Tooltip title="Delete project">
|
<Tooltip title="Delete project">
|
||||||
<span>
|
<span>
|
||||||
<IconButton
|
<IconButton
|
||||||
onClick={() => handleDeleteButtonClick(project.name)}
|
onClick={() => handleDeleteButtonClick(project)}
|
||||||
disabled={project.isActive}
|
disabled={project.isActive}
|
||||||
>
|
>
|
||||||
<DeleteIcon />
|
<DeleteIcon />
|
||||||
|
@ -32,7 +32,7 @@ const HTTP_REQUEST_LOG = gql`
|
|||||||
`;
|
`;
|
||||||
|
|
||||||
interface Props {
|
interface Props {
|
||||||
requestId: number;
|
requestId: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
function LogDetail({ requestId: id }: Props): JSX.Element {
|
function LogDetail({ requestId: id }: Props): JSX.Element {
|
||||||
|
@ -15,12 +15,10 @@ import { useHttpRequestLogs } from "./hooks/useHttpRequestLogs";
|
|||||||
|
|
||||||
function LogsOverview(): JSX.Element {
|
function LogsOverview(): JSX.Element {
|
||||||
const router = useRouter();
|
const router = useRouter();
|
||||||
const detailReqLogId =
|
const detailReqLogId = router.query.id as string | undefined;
|
||||||
router.query.id && parseInt(router.query.id as string, 10);
|
|
||||||
|
|
||||||
const { loading, error, data } = useHttpRequestLogs();
|
const { loading, error, data } = useHttpRequestLogs();
|
||||||
|
|
||||||
const handleLogClick = (reqId: number) => {
|
const handleLogClick = (reqId: string) => {
|
||||||
router.push("/proxy/logs?id=" + reqId, undefined, {
|
router.push("/proxy/logs?id=" + reqId, undefined, {
|
||||||
shallow: false,
|
shallow: false,
|
||||||
});
|
});
|
||||||
|
@ -31,8 +31,8 @@ const useStyles = makeStyles((theme: Theme) =>
|
|||||||
|
|
||||||
interface Props {
|
interface Props {
|
||||||
logs: Array<any>;
|
logs: Array<any>;
|
||||||
selectedReqLogId?: number;
|
selectedReqLogId?: string;
|
||||||
onLogClick(requestId: number): void;
|
onLogClick(requestId: string): void;
|
||||||
theme: Theme;
|
theme: Theme;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -63,8 +63,8 @@ function RequestList({
|
|||||||
|
|
||||||
interface RequestListTableProps {
|
interface RequestListTableProps {
|
||||||
logs?: any;
|
logs?: any;
|
||||||
selectedReqLogId?: number;
|
selectedReqLogId?: string;
|
||||||
onLogClick(requestId: number): void;
|
onLogClick(requestId: string): void;
|
||||||
theme: Theme;
|
theme: Theme;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -9,13 +9,7 @@ function createApolloClient() {
|
|||||||
link: new HttpLink({
|
link: new HttpLink({
|
||||||
uri: "/api/graphql/",
|
uri: "/api/graphql/",
|
||||||
}),
|
}),
|
||||||
cache: new InMemoryCache({
|
cache: new InMemoryCache(),
|
||||||
typePolicies: {
|
|
||||||
Project: {
|
|
||||||
keyFields: ["name"],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}),
|
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,30 +1,17 @@
|
|||||||
import {
|
import {
|
||||||
Avatar,
|
|
||||||
Box,
|
Box,
|
||||||
Button,
|
Button,
|
||||||
CircularProgress,
|
|
||||||
createStyles,
|
createStyles,
|
||||||
List,
|
|
||||||
ListItem,
|
|
||||||
ListItemAvatar,
|
|
||||||
ListItemText,
|
|
||||||
makeStyles,
|
makeStyles,
|
||||||
TextField,
|
|
||||||
Theme,
|
Theme,
|
||||||
Typography,
|
Typography,
|
||||||
} from "@material-ui/core";
|
} from "@material-ui/core";
|
||||||
import AddIcon from "@material-ui/icons/Add";
|
|
||||||
import FolderIcon from "@material-ui/icons/Folder";
|
import FolderIcon from "@material-ui/icons/Folder";
|
||||||
import DescriptionIcon from "@material-ui/icons/Description";
|
|
||||||
import PlayArrowIcon from "@material-ui/icons/PlayArrow";
|
|
||||||
import Link from "next/link";
|
import Link from "next/link";
|
||||||
|
|
||||||
import { useState } from "react";
|
|
||||||
import { gql, useMutation, useQuery } from "@apollo/client";
|
|
||||||
import { useRouter } from "next/router";
|
import { useRouter } from "next/router";
|
||||||
|
|
||||||
import Layout, { Page } from "../components/Layout";
|
import Layout, { Page } from "../components/Layout";
|
||||||
import { Alert } from "@material-ui/lab";
|
|
||||||
|
|
||||||
const useStyles = makeStyles((theme: Theme) =>
|
const useStyles = makeStyles((theme: Theme) =>
|
||||||
createStyles({
|
createStyles({
|
||||||
@ -37,107 +24,14 @@ const useStyles = makeStyles((theme: Theme) =>
|
|||||||
lineHeight: 2,
|
lineHeight: 2,
|
||||||
marginBottom: theme.spacing(5),
|
marginBottom: theme.spacing(5),
|
||||||
},
|
},
|
||||||
projectName: {
|
|
||||||
marginTop: -6,
|
|
||||||
marginRight: theme.spacing(2),
|
|
||||||
},
|
|
||||||
button: {
|
button: {
|
||||||
marginRight: theme.spacing(2),
|
marginRight: theme.spacing(2),
|
||||||
},
|
},
|
||||||
activeProject: {
|
|
||||||
color: theme.palette.getContrastText(theme.palette.secondary.main),
|
|
||||||
backgroundColor: theme.palette.secondary.main,
|
|
||||||
},
|
|
||||||
})
|
})
|
||||||
);
|
);
|
||||||
|
|
||||||
const ACTIVE_PROJECT = gql`
|
|
||||||
query ActiveProject {
|
|
||||||
activeProject {
|
|
||||||
name
|
|
||||||
}
|
|
||||||
}
|
|
||||||
`;
|
|
||||||
|
|
||||||
const OPEN_PROJECT = gql`
|
|
||||||
mutation OpenProject($name: String!) {
|
|
||||||
openProject(name: $name) {
|
|
||||||
name
|
|
||||||
isActive
|
|
||||||
}
|
|
||||||
}
|
|
||||||
`;
|
|
||||||
|
|
||||||
function Index(): JSX.Element {
|
function Index(): JSX.Element {
|
||||||
const classes = useStyles();
|
const classes = useStyles();
|
||||||
const router = useRouter();
|
|
||||||
const [input, setInput] = useState(null);
|
|
||||||
const { error: activeProjErr, data: activeProjData } = useQuery(
|
|
||||||
ACTIVE_PROJECT,
|
|
||||||
{
|
|
||||||
pollInterval: 1000,
|
|
||||||
}
|
|
||||||
);
|
|
||||||
const [
|
|
||||||
openProject,
|
|
||||||
{ error: openProjErr, data: openProjData, loading: openProjLoading },
|
|
||||||
] = useMutation(OPEN_PROJECT, {
|
|
||||||
onError: () => {},
|
|
||||||
onCompleted({ openProject }) {
|
|
||||||
if (openProject) {
|
|
||||||
router.push("/get-started");
|
|
||||||
}
|
|
||||||
},
|
|
||||||
update(cache, { data: { openProject } }) {
|
|
||||||
cache.modify({
|
|
||||||
fields: {
|
|
||||||
activeProject() {
|
|
||||||
const activeProjRef = cache.writeFragment({
|
|
||||||
id: openProject.name,
|
|
||||||
data: openProject,
|
|
||||||
fragment: gql`
|
|
||||||
fragment ActiveProject on Project {
|
|
||||||
name
|
|
||||||
isActive
|
|
||||||
type
|
|
||||||
}
|
|
||||||
`,
|
|
||||||
});
|
|
||||||
return activeProjRef;
|
|
||||||
},
|
|
||||||
projects(_, { DELETE }) {
|
|
||||||
cache.writeFragment({
|
|
||||||
id: openProject.name,
|
|
||||||
data: openProject,
|
|
||||||
fragment: gql`
|
|
||||||
fragment OpenProject on Project {
|
|
||||||
name
|
|
||||||
isActive
|
|
||||||
type
|
|
||||||
}
|
|
||||||
`,
|
|
||||||
});
|
|
||||||
return DELETE;
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const handleForm = (e: React.SyntheticEvent) => {
|
|
||||||
e.preventDefault();
|
|
||||||
openProject({ variables: { name: input.value } });
|
|
||||||
};
|
|
||||||
|
|
||||||
if (activeProjErr) {
|
|
||||||
return (
|
|
||||||
<Layout page={Page.Home} title="">
|
|
||||||
<Alert severity="error">
|
|
||||||
Error fetching active project: {activeProjErr.message}
|
|
||||||
</Alert>
|
|
||||||
</Layout>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<Layout page={Page.Home} title="">
|
<Layout page={Page.Home} title="">
|
||||||
@ -157,97 +51,18 @@ function Index(): JSX.Element {
|
|||||||
<code>:8080</code>…
|
<code>:8080</code>…
|
||||||
</Typography>
|
</Typography>
|
||||||
|
|
||||||
{activeProjData?.activeProject?.name ? (
|
<Link href="/projects" passHref>
|
||||||
<div>
|
<Button
|
||||||
<Box mb={1}>
|
className={classes.button}
|
||||||
<Typography variant="h6">Active project:</Typography>
|
variant="contained"
|
||||||
</Box>
|
color="secondary"
|
||||||
<Box ml={-2} mb={2}>
|
component="a"
|
||||||
<List>
|
size="large"
|
||||||
<ListItem>
|
startIcon={<FolderIcon />}
|
||||||
<ListItemAvatar>
|
>
|
||||||
<Avatar className={classes.activeProject}>
|
Manage projects
|
||||||
<DescriptionIcon />
|
</Button>
|
||||||
</Avatar>
|
</Link>
|
||||||
</ListItemAvatar>
|
|
||||||
<ListItemText primary={activeProjData.activeProject.name} />
|
|
||||||
</ListItem>
|
|
||||||
</List>
|
|
||||||
</Box>
|
|
||||||
<div>
|
|
||||||
<Link href="/get-started" passHref>
|
|
||||||
<Button
|
|
||||||
className={classes.button}
|
|
||||||
variant="outlined"
|
|
||||||
component="a"
|
|
||||||
color="secondary"
|
|
||||||
size="large"
|
|
||||||
startIcon={<PlayArrowIcon />}
|
|
||||||
>
|
|
||||||
Get started
|
|
||||||
</Button>
|
|
||||||
</Link>
|
|
||||||
<Link href="/projects" passHref>
|
|
||||||
<Button
|
|
||||||
className={classes.button}
|
|
||||||
variant="outlined"
|
|
||||||
component="a"
|
|
||||||
size="large"
|
|
||||||
startIcon={<FolderIcon />}
|
|
||||||
>
|
|
||||||
Manage projects
|
|
||||||
</Button>
|
|
||||||
</Link>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<form onSubmit={handleForm} autoComplete="off">
|
|
||||||
<TextField
|
|
||||||
className={classes.projectName}
|
|
||||||
color="secondary"
|
|
||||||
inputProps={{
|
|
||||||
id: "projectName",
|
|
||||||
ref: (node) => {
|
|
||||||
setInput(node);
|
|
||||||
},
|
|
||||||
}}
|
|
||||||
label="Project name"
|
|
||||||
placeholder="Project name…"
|
|
||||||
error={Boolean(openProjErr)}
|
|
||||||
helperText={openProjErr && openProjErr.message}
|
|
||||||
/>
|
|
||||||
<Button
|
|
||||||
className={classes.button}
|
|
||||||
type="submit"
|
|
||||||
variant="contained"
|
|
||||||
color="secondary"
|
|
||||||
size="large"
|
|
||||||
disabled={
|
|
||||||
openProjLoading || Boolean(openProjData?.openProject?.name)
|
|
||||||
}
|
|
||||||
startIcon={
|
|
||||||
openProjLoading || openProjData?.openProject ? (
|
|
||||||
<CircularProgress size={22} />
|
|
||||||
) : (
|
|
||||||
<AddIcon />
|
|
||||||
)
|
|
||||||
}
|
|
||||||
>
|
|
||||||
Create project
|
|
||||||
</Button>
|
|
||||||
<Link href="/projects" passHref>
|
|
||||||
<Button
|
|
||||||
className={classes.button}
|
|
||||||
variant="outlined"
|
|
||||||
component="a"
|
|
||||||
size="large"
|
|
||||||
startIcon={<FolderIcon />}
|
|
||||||
>
|
|
||||||
Open project…
|
|
||||||
</Button>
|
|
||||||
</Link>
|
|
||||||
</form>
|
|
||||||
)}
|
|
||||||
</Box>
|
</Box>
|
||||||
</Layout>
|
</Layout>
|
||||||
);
|
);
|
||||||
|
@ -12,7 +12,7 @@ function Index(): JSX.Element {
|
|||||||
</Box>
|
</Box>
|
||||||
<Typography paragraph>
|
<Typography paragraph>
|
||||||
Projects contain settings and data generated/processed by Hetty. They
|
Projects contain settings and data generated/processed by Hetty. They
|
||||||
are stored as SQLite database files on disk.
|
are stored in a single database on disk.
|
||||||
</Typography>
|
</Typography>
|
||||||
<Box my={4}>
|
<Box my={4}>
|
||||||
<Divider />
|
<Divider />
|
||||||
|
@ -15,11 +15,12 @@ import (
|
|||||||
|
|
||||||
"github.com/99designs/gqlgen/graphql/handler"
|
"github.com/99designs/gqlgen/graphql/handler"
|
||||||
"github.com/99designs/gqlgen/graphql/playground"
|
"github.com/99designs/gqlgen/graphql/playground"
|
||||||
|
badgerdb "github.com/dgraph-io/badger/v3"
|
||||||
"github.com/gorilla/mux"
|
"github.com/gorilla/mux"
|
||||||
"github.com/mitchellh/go-homedir"
|
"github.com/mitchellh/go-homedir"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/api"
|
"github.com/dstotijn/hetty/pkg/api"
|
||||||
"github.com/dstotijn/hetty/pkg/db/sqlite"
|
"github.com/dstotijn/hetty/pkg/db/badger"
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
"github.com/dstotijn/hetty/pkg/proj"
|
||||||
"github.com/dstotijn/hetty/pkg/proxy"
|
"github.com/dstotijn/hetty/pkg/proxy"
|
||||||
"github.com/dstotijn/hetty/pkg/reqlog"
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
@ -29,7 +30,7 @@ import (
|
|||||||
var (
|
var (
|
||||||
caCertFile string
|
caCertFile string
|
||||||
caKeyFile string
|
caKeyFile string
|
||||||
projPath string
|
dbPath string
|
||||||
addr string
|
addr string
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -50,7 +51,7 @@ func run() error {
|
|||||||
"CA certificate filepath. Creates a new CA certificate if file doesn't exist")
|
"CA certificate filepath. Creates a new CA certificate if file doesn't exist")
|
||||||
flag.StringVar(&caKeyFile, "key", "~/.hetty/hetty_key.pem",
|
flag.StringVar(&caKeyFile, "key", "~/.hetty/hetty_key.pem",
|
||||||
"CA private key filepath. Creates a new CA private key if file doesn't exist")
|
"CA private key filepath. Creates a new CA private key if file doesn't exist")
|
||||||
flag.StringVar(&projPath, "projects", "~/.hetty/projects", "Projects directory path")
|
flag.StringVar(&dbPath, "db", "~/.hetty/db", "Database directory path")
|
||||||
flag.StringVar(&addr, "addr", ":8080", "TCP address to listen on, in the form \"host:port\"")
|
flag.StringVar(&addr, "addr", ":8080", "TCP address to listen on, in the form \"host:port\"")
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
@ -65,7 +66,7 @@ func run() error {
|
|||||||
return fmt.Errorf("could not parse CA private key filepath: %w", err)
|
return fmt.Errorf("could not parse CA private key filepath: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
projPath, err := homedir.Expand(projPath)
|
dbPath, err := homedir.Expand(dbPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("could not parse projects filepath: %w", err)
|
return fmt.Errorf("could not parse projects filepath: %w", err)
|
||||||
}
|
}
|
||||||
@ -77,24 +78,27 @@ func run() error {
|
|||||||
return fmt.Errorf("could not create/load CA key pair: %w", err)
|
return fmt.Errorf("could not create/load CA key pair: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
db, err := sqlite.New(projPath)
|
badger, err := badger.OpenDatabase(badgerdb.DefaultOptions(dbPath))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("could not initialize database client: %w", err)
|
return fmt.Errorf("could not open badger database: %w", err)
|
||||||
}
|
}
|
||||||
|
defer badger.Close()
|
||||||
|
|
||||||
projService, err := proj.NewService(db)
|
scope := &scope.Scope{}
|
||||||
|
|
||||||
|
reqLogService := reqlog.NewService(reqlog.Config{
|
||||||
|
Scope: scope,
|
||||||
|
Repository: badger,
|
||||||
|
})
|
||||||
|
|
||||||
|
projService, err := proj.NewService(proj.Config{
|
||||||
|
Repository: badger,
|
||||||
|
ReqLogService: reqLogService,
|
||||||
|
Scope: scope,
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("could not create new project service: %w", err)
|
return fmt.Errorf("could not create new project service: %w", err)
|
||||||
}
|
}
|
||||||
defer projService.Close()
|
|
||||||
|
|
||||||
scope := scope.New(db, projService)
|
|
||||||
|
|
||||||
reqLogService := reqlog.NewService(reqlog.Config{
|
|
||||||
Scope: scope,
|
|
||||||
ProjectService: projService,
|
|
||||||
Repository: db,
|
|
||||||
})
|
|
||||||
|
|
||||||
p, err := proxy.NewProxy(caCert, caKey)
|
p, err := proxy.NewProxy(caCert, caKey)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -123,7 +127,6 @@ func run() error {
|
|||||||
handler.NewDefaultServer(api.NewExecutableSchema(api.Config{Resolvers: &api.Resolver{
|
handler.NewDefaultServer(api.NewExecutableSchema(api.Config{Resolvers: &api.Resolver{
|
||||||
RequestLogService: reqLogService,
|
RequestLogService: reqLogService,
|
||||||
ProjectService: projService,
|
ProjectService: projService,
|
||||||
ScopeService: scope,
|
|
||||||
}})))
|
}})))
|
||||||
|
|
||||||
// Admin interface.
|
// Admin interface.
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
Hetty compiles to a static binary, with an embedded SQLite database and web
|
Hetty compiles to a static binary, with an embedded BadgerDB database and web
|
||||||
admin interface.
|
admin interface.
|
||||||
|
|
||||||
### Install pre-built release (recommended)
|
### Install pre-built release (recommended)
|
||||||
@ -16,11 +16,10 @@ admin interface.
|
|||||||
- [Go 1.16](https://golang.org/)
|
- [Go 1.16](https://golang.org/)
|
||||||
- [Yarn](https://yarnpkg.com/)
|
- [Yarn](https://yarnpkg.com/)
|
||||||
|
|
||||||
Hetty depends on SQLite (via [mattn/go-sqlite3](https://github.com/mattn/go-sqlite3))
|
When building from source, the static resources for the admin interface
|
||||||
and needs `cgo` to compile. Additionally, the static resources for the admin interface
|
|
||||||
(Next.js) need to be generated via [Yarn](https://yarnpkg.com/). The generated
|
(Next.js) need to be generated via [Yarn](https://yarnpkg.com/). The generated
|
||||||
files will be embedded (using the [embed](https://golang.org/pkg/embed/) package)
|
files will be embedded (using the [embed](https://golang.org/pkg/embed/)
|
||||||
when you use the `build` Makefile target..
|
package) when you use the `build` Makefile target.
|
||||||
|
|
||||||
Clone the repository and use the `build` make target to create a binary:
|
Clone the repository and use the `build` make target to create a binary:
|
||||||
|
|
||||||
@ -46,7 +45,7 @@ When Hetty is started, by default it listens on `:8080` and is accessible via
|
|||||||
[http://localhost:8080](http://localhost:8080). Depending on incoming HTTP
|
[http://localhost:8080](http://localhost:8080). Depending on incoming HTTP
|
||||||
requests, it either acts as a MITM proxy, or it serves the API and web interface.
|
requests, it either acts as a MITM proxy, or it serves the API and web interface.
|
||||||
|
|
||||||
By default, project database files and CA certificates are stored in a `.hetty`
|
By default, the projects database files and CA certificates are stored in a `.hetty`
|
||||||
directory under the user's home directory (`$HOME` on Linux/macOS, `%USERPROFILE%`
|
directory under the user's home directory (`$HOME` on Linux/macOS, `%USERPROFILE%`
|
||||||
on Windows).
|
on Windows).
|
||||||
|
|
||||||
@ -80,6 +79,6 @@ Usage of ./hetty:
|
|||||||
CA certificate filepath. Creates a new CA certificate if file doesn't exist (default "~/.hetty/hetty_cert.pem")
|
CA certificate filepath. Creates a new CA certificate if file doesn't exist (default "~/.hetty/hetty_cert.pem")
|
||||||
-key string
|
-key string
|
||||||
CA private key filepath. Creates a new CA private key if file doesn't exist (default "~/.hetty/hetty_key.pem")
|
CA private key filepath. Creates a new CA private key if file doesn't exist (default "~/.hetty/hetty_key.pem")
|
||||||
-projects string
|
-db string
|
||||||
Projects directory path (default "~/.hetty/projects")
|
Database directory path (default "~/.hetty/db")
|
||||||
```
|
```
|
||||||
|
@ -17,7 +17,7 @@ features tailored to the needs of the infosec and bug bounty community.
|
|||||||
## Features
|
## Features
|
||||||
|
|
||||||
- Machine-in-the-middle (MITM) HTTP/1.1 proxy with logs
|
- Machine-in-the-middle (MITM) HTTP/1.1 proxy with logs
|
||||||
- Project based database storage (SQLite)
|
- Project based database storage (BadgerDB)
|
||||||
- Scope support
|
- Scope support
|
||||||
- Headless management API using GraphQL
|
- Headless management API using GraphQL
|
||||||
- Embedded web admin interface (Next.js)
|
- Embedded web admin interface (Next.js)
|
||||||
|
@ -14,7 +14,7 @@ The available modules:
|
|||||||
|
|
||||||
## Projects
|
## Projects
|
||||||
|
|
||||||
Projects are self-contained (SQLite) database files that contain module data.
|
Projects are stored in a single BadgerDB database on disk.
|
||||||
They allow you organize your work, for example to split your work between research
|
They allow you organize your work, for example to split your work between research
|
||||||
targets.
|
targets.
|
||||||
|
|
||||||
@ -25,19 +25,15 @@ typically the first thing you do when you start using Hetty.
|
|||||||
### Creating a new project
|
### Creating a new project
|
||||||
|
|
||||||
When you open the Hetty admin interface after starting the program, you’ll be prompted
|
When you open the Hetty admin interface after starting the program, you’ll be prompted
|
||||||
on the homepage to create a new project. Give it a name (alphanumeric and space character)
|
on the homepage to “Manage projects”, which leads to the “Projects” page where
|
||||||
and click the create button:
|
you can open an existing project or create a new one:
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
The project name will become the base for the database file on disk. For example,
|
|
||||||
if you name your project `My first project`, the file on disk will be
|
|
||||||
`My first project.db`.
|
|
||||||
|
|
||||||
::: tip INFO
|
::: tip INFO
|
||||||
Project database files by default are stored in `$HOME/.hetty/projects` on Linux
|
Projects are stored in a single BadgerDB database, stored in `$HOME/.hetty/db` on Linux
|
||||||
and macOS, and `%USERPROFILE%/.hetty` on Windows. You can override this path with
|
and macOS, and `%USERPROFILE%/.hetty/db` on Windows. You can override this path with
|
||||||
the `-projects` flag. See: [Usage](/guide/getting-started.md#usage).
|
the `-db` flag. See: [Usage](/guide/getting-started.md#usage).
|
||||||
:::
|
:::
|
||||||
|
|
||||||
### Managing projects
|
### Managing projects
|
||||||
|
13
go.mod
13
go.mod
@ -3,14 +3,13 @@ module github.com/dstotijn/hetty
|
|||||||
go 1.16
|
go 1.16
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/99designs/gqlgen v0.13.0
|
github.com/99designs/gqlgen v0.14.0
|
||||||
github.com/Masterminds/squirrel v1.4.0
|
github.com/dgraph-io/badger/v3 v3.2103.2
|
||||||
|
github.com/google/go-cmp v0.5.6
|
||||||
github.com/gorilla/mux v1.7.4
|
github.com/gorilla/mux v1.7.4
|
||||||
github.com/hashicorp/golang-lru v0.5.1 // indirect
|
github.com/hashicorp/golang-lru v0.5.1 // indirect
|
||||||
github.com/jmoiron/sqlx v1.2.0
|
github.com/matryer/moq v0.0.0-20200106131100-75d0ddfc0007
|
||||||
github.com/mattn/go-sqlite3 v1.14.4
|
|
||||||
github.com/mitchellh/go-homedir v1.1.0
|
github.com/mitchellh/go-homedir v1.1.0
|
||||||
github.com/mitchellh/mapstructure v1.1.2 // indirect
|
github.com/oklog/ulid v1.3.1
|
||||||
github.com/vektah/gqlparser/v2 v2.1.0
|
github.com/vektah/gqlparser/v2 v2.2.0
|
||||||
google.golang.org/appengine v1.6.6 // indirect
|
|
||||||
)
|
)
|
||||||
|
165
go.sum
165
go.sum
@ -1,26 +1,61 @@
|
|||||||
github.com/99designs/gqlgen v0.13.0 h1:haLTcUp3Vwp80xMVEg5KRNwzfUrgFdRmtBY8fuB8scA=
|
cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=
|
||||||
github.com/99designs/gqlgen v0.13.0/go.mod h1:NV130r6f4tpRWuAI+zsrSdooO/eWUv+Gyyoi3rEfXIk=
|
github.com/99designs/gqlgen v0.14.0 h1:Wg8aNYQUjMR/4v+W3xD+7SizOy6lSvVeQ06AobNQAXI=
|
||||||
|
github.com/99designs/gqlgen v0.14.0/go.mod h1:S7z4boV+Nx4VvzMUpVrY/YuHjFX4n7rDyuTqvAkuoRE=
|
||||||
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||||
github.com/Masterminds/squirrel v1.4.0 h1:he5i/EXixZxrBUWcxzDYMiju9WZ3ld/l7QBNuo/eN3w=
|
github.com/OneOfOne/xxhash v1.2.2 h1:KMrpdQIwFcEqXDklaen+P1axHaj9BSKzvpUUfnHldSE=
|
||||||
github.com/Masterminds/squirrel v1.4.0/go.mod h1:yaPeOnPG5ZRwL9oKdTsO/prlkPbXWZlRVMQ/gGlzIuA=
|
github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=
|
||||||
github.com/agnivade/levenshtein v1.0.1/go.mod h1:CURSv5d9Uaml+FovSIICkLbAUZ9S4RqaHDIsdSBg7lM=
|
github.com/agnivade/levenshtein v1.0.1/go.mod h1:CURSv5d9Uaml+FovSIICkLbAUZ9S4RqaHDIsdSBg7lM=
|
||||||
github.com/agnivade/levenshtein v1.0.3 h1:M5ZnqLOoZR8ygVq0FfkXsNOKzMCk0xRiow0R5+5VkQ0=
|
github.com/agnivade/levenshtein v1.1.0 h1:n6qGwyHG61v3ABce1rPVZklEYRT8NFpCMrpZdBUbYGM=
|
||||||
github.com/agnivade/levenshtein v1.0.3/go.mod h1:4SFRZbbXWLF4MU1T9Qg0pGgH3Pjs+t6ie5efyrwRJXs=
|
github.com/agnivade/levenshtein v1.1.0/go.mod h1:veldBMzWxcCG2ZvUTKD2kJNRdCk5hVbJomOvKkmgYbo=
|
||||||
github.com/andreyvit/diff v0.0.0-20170406064948-c7f18ee00883 h1:bvNMNQO63//z+xNgfBlViaCIJKLlCJ6/fmUseuG0wVQ=
|
github.com/andreyvit/diff v0.0.0-20170406064948-c7f18ee00883 h1:bvNMNQO63//z+xNgfBlViaCIJKLlCJ6/fmUseuG0wVQ=
|
||||||
github.com/andreyvit/diff v0.0.0-20170406064948-c7f18ee00883/go.mod h1:rCTlJbsFo29Kk6CurOXKm700vrz8f0KW0JNfpkRJY/8=
|
github.com/andreyvit/diff v0.0.0-20170406064948-c7f18ee00883/go.mod h1:rCTlJbsFo29Kk6CurOXKm700vrz8f0KW0JNfpkRJY/8=
|
||||||
github.com/arbovm/levenshtein v0.0.0-20160628152529-48b4e1c0c4d0 h1:jfIu9sQUG6Ig+0+Ap1h4unLjW6YQJpKZVmUzxsD4E/Q=
|
github.com/arbovm/levenshtein v0.0.0-20160628152529-48b4e1c0c4d0 h1:jfIu9sQUG6Ig+0+Ap1h4unLjW6YQJpKZVmUzxsD4E/Q=
|
||||||
github.com/arbovm/levenshtein v0.0.0-20160628152529-48b4e1c0c4d0/go.mod h1:t2tdKJDJF9BV14lnkjHmOQgcvEKgtqs5a1N3LNdJhGE=
|
github.com/arbovm/levenshtein v0.0.0-20160628152529-48b4e1c0c4d0/go.mod h1:t2tdKJDJF9BV14lnkjHmOQgcvEKgtqs5a1N3LNdJhGE=
|
||||||
|
github.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6/go.mod h1:grANhF5doyWs3UAsr3K4I6qtAmlQcZDesFNEHPZAzj8=
|
||||||
|
github.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=
|
||||||
|
github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
|
||||||
|
github.com/cespare/xxhash/v2 v2.1.1 h1:6MnRN8NT7+YBpUIWxHtefFZOKTAPgGjpQSxqLNn0+qY=
|
||||||
|
github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||||
|
github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=
|
||||||
|
github.com/coreos/etcd v3.3.10+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=
|
||||||
|
github.com/coreos/go-etcd v2.0.0+incompatible/go.mod h1:Jez6KQU2B/sWsbdaef3ED8NzMklzPG4d5KIOhIy30Tk=
|
||||||
|
github.com/coreos/go-semver v0.2.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=
|
||||||
|
github.com/cpuguy83/go-md2man v1.0.10 h1:BSKMNlYxDvnunlTymqtgONjNnaRV1sTpcovwwjF22jk=
|
||||||
|
github.com/cpuguy83/go-md2man v1.0.10/go.mod h1:SmD6nW6nTyfqj6ABTjUi3V3JVMnlJmwcJI5acqYI6dE=
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d h1:U+s90UTSYgptZMwQh2aRr3LuazLJIa+Pg3Kc1ylSYVY=
|
||||||
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
|
github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
|
||||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/dgryski/trifles v0.0.0-20190318185328-a8d75aae118c h1:TUuUh0Xgj97tLMNtWtNvI9mIV6isjEb9lBMNv+77IGM=
|
github.com/dgraph-io/badger/v3 v3.2103.2 h1:dpyM5eCJAtQCBcMCZcT4UBZchuTJgCywerHHgmxfxM8=
|
||||||
github.com/dgryski/trifles v0.0.0-20190318185328-a8d75aae118c/go.mod h1:if7Fbed8SFyPtHLHbg49SI7NAdJiC5WIA09pe59rfAA=
|
github.com/dgraph-io/badger/v3 v3.2103.2/go.mod h1:RHo4/GmYcKKh5Lxu63wLEMHJ70Pac2JqZRYGhlyAo2M=
|
||||||
github.com/go-chi/chi v3.3.2+incompatible/go.mod h1:eB3wogJHnLi3x/kFX2A+IbTBlXxmMeXJVKy9tTv1XzQ=
|
github.com/dgraph-io/ristretto v0.1.0 h1:Jv3CGQHp9OjuMBSne1485aDpUkTKEcUqF+jm/LuerPI=
|
||||||
github.com/go-sql-driver/mysql v1.4.0 h1:7LxgVwFb2hIQtMm87NdgAVfXjnt4OePseqT1tKx+opk=
|
github.com/dgraph-io/ristretto v0.1.0/go.mod h1:fux0lOrBhrVCJd3lcTHsIJhq1T2rokOu6v9Vcb3Q9ug=
|
||||||
github.com/go-sql-driver/mysql v1.4.0/go.mod h1:zAC/RDZ24gD3HViQzih4MyKcchzm+sOG5ZlKdlhCg5w=
|
github.com/dgryski/go-farm v0.0.0-20190423205320-6a90982ecee2 h1:tdlZCpZ/P9DhczCTSixgIKmwPv6+wP5DGjqLYw5SUiA=
|
||||||
github.com/gogo/protobuf v1.0.0/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=
|
github.com/dgryski/go-farm v0.0.0-20190423205320-6a90982ecee2/go.mod h1:SqUrOPUnsFjfmXRMNPybcSiG0BgUW2AuFH8PAnS2iTw=
|
||||||
|
github.com/dgryski/trifles v0.0.0-20200323201526-dd97f9abfb48 h1:fRzb/w+pyskVMQ+UbP35JkH8yB7MYb4q/qhBarqZE6g=
|
||||||
|
github.com/dgryski/trifles v0.0.0-20200323201526-dd97f9abfb48/go.mod h1:if7Fbed8SFyPtHLHbg49SI7NAdJiC5WIA09pe59rfAA=
|
||||||
|
github.com/dustin/go-humanize v1.0.0 h1:VSnTsYCnlFHaM2/igO1h6X3HA71jcobQuxemgkq4zYo=
|
||||||
|
github.com/dustin/go-humanize v1.0.0/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
|
||||||
|
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
|
||||||
|
github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q=
|
||||||
|
github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q=
|
||||||
|
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b h1:VKtxabqXZkF25pY9ekfRL6a582T4P37/31XEstQ5p58=
|
||||||
|
github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=
|
||||||
|
github.com/golang/groupcache v0.0.0-20190702054246-869f871628b6 h1:ZgQEtGgCBiWRM39fZuwSd1LwSqqSW0hOdXCYYDX0R3I=
|
||||||
|
github.com/golang/groupcache v0.0.0-20190702054246-869f871628b6/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=
|
||||||
|
github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=
|
||||||
|
github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
|
github.com/golang/protobuf v1.3.1 h1:YF8+flBXS5eO826T4nzqPrxfhQThhXl0YzfuUPu4SBg=
|
||||||
github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
github.com/golang/protobuf v1.3.1/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
|
||||||
|
github.com/golang/snappy v0.0.3 h1:fHPg5GQYlCeLIPB9BZqMVR5nR9A+IM5zcgeTdjMYmLA=
|
||||||
|
github.com/golang/snappy v0.0.3/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
|
||||||
|
github.com/google/flatbuffers v1.12.1 h1:MVlul7pQNoDzWRLTw5imwYsl+usrS1TXG2H4jg6ImGw=
|
||||||
|
github.com/google/flatbuffers v1.12.1/go.mod h1:1AeVuKshWv4vARoZatz6mlQ0JxURH0Kv5+zNeJKJCa8=
|
||||||
|
github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
|
||||||
|
github.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
|
github.com/google/go-cmp v0.5.6 h1:BKbKCqvP6I+rmFHt06ZmyQtvB8xAkWdhFyr0ZUNZcxQ=
|
||||||
|
github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
|
||||||
github.com/gorilla/context v0.0.0-20160226214623-1ea25387ff6f/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
|
github.com/gorilla/context v0.0.0-20160226214623-1ea25387ff6f/go.mod h1:kBGZzfjB9CEq2AlWe17Uuf7NDRt0dE0s8S51q0aT7Yg=
|
||||||
github.com/gorilla/mux v1.6.1/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
|
github.com/gorilla/mux v1.6.1/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs=
|
||||||
github.com/gorilla/mux v1.7.4 h1:VuZ8uybHlWmqV03+zRzdwKL4tUnIp1MAQtp1mIFE1bc=
|
github.com/gorilla/mux v1.7.4 h1:VuZ8uybHlWmqV03+zRzdwKL4tUnIp1MAQtp1mIFE1bc=
|
||||||
@ -30,76 +65,142 @@ github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/ad
|
|||||||
github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||||
github.com/hashicorp/golang-lru v0.5.1 h1:0hERBMJE1eitiLkihrMvRVBYAkpHzc/J3QdDN+dAcgU=
|
github.com/hashicorp/golang-lru v0.5.1 h1:0hERBMJE1eitiLkihrMvRVBYAkpHzc/J3QdDN+dAcgU=
|
||||||
github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
github.com/hashicorp/golang-lru v0.5.1/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8=
|
||||||
github.com/jmoiron/sqlx v1.2.0 h1:41Ip0zITnmWNR/vHV+S4m+VoUivnWY5E4OJfLZjCJMA=
|
github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
|
||||||
github.com/jmoiron/sqlx v1.2.0/go.mod h1:1FEQNm3xlJgrMD+FBdI9+xvCksHtbpVBBw5dYhBSsks=
|
github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
|
||||||
|
github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8=
|
||||||
|
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
|
||||||
|
github.com/klauspost/compress v1.12.3 h1:G5AfA94pHPysR56qqrkO2pxEexdDzrpFJ6yt/VqWxVU=
|
||||||
|
github.com/klauspost/compress v1.12.3/go.mod h1:8dP1Hq4DHOhN9w426knH3Rhby4rFm6D8eO+e+Dq5Gzg=
|
||||||
|
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
|
||||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
|
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
|
||||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 h1:SOEGU9fKiNWd/HOJuq6+3iTQz8KNCLtVX6idSoTLdUw=
|
|
||||||
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0/go.mod h1:dXGbAdH5GtBTC4WfIxhKZfyBF/HBFgRZSWwZ9g/He9o=
|
|
||||||
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 h1:P6pPBnrTSX3DEVR4fDembhRWSsG5rVo6hYhAB/ADZrk=
|
|
||||||
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0/go.mod h1:vmVJ0l/dxyfGW6FmdpVm2joNMFikkuWg0EoCKLGUMNw=
|
|
||||||
github.com/lib/pq v1.0.0 h1:X5PMW56eZitiTeO7tKzZxFCSpbFZJtkMMooicw2us9A=
|
|
||||||
github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
|
|
||||||
github.com/logrusorgru/aurora v0.0.0-20200102142835-e9ef32dff381/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4=
|
github.com/logrusorgru/aurora v0.0.0-20200102142835-e9ef32dff381/go.mod h1:7rIyQOR62GCctdiQpZ/zOJlFyk6y+94wXzv6RNZgaR4=
|
||||||
|
github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
|
||||||
|
github.com/matryer/moq v0.0.0-20200106131100-75d0ddfc0007 h1:reVOUXwnhsYv/8UqjvhrMOu5CNT9UapHFLbQ2JcXsmg=
|
||||||
github.com/matryer/moq v0.0.0-20200106131100-75d0ddfc0007/go.mod h1:9ELz6aaclSIGnZBoaSLZ3NAl1VTufbOrXBPvtcy6WiQ=
|
github.com/matryer/moq v0.0.0-20200106131100-75d0ddfc0007/go.mod h1:9ELz6aaclSIGnZBoaSLZ3NAl1VTufbOrXBPvtcy6WiQ=
|
||||||
github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
|
||||||
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
|
||||||
github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
|
github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
|
||||||
github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc=
|
|
||||||
github.com/mattn/go-sqlite3 v1.14.4 h1:4rQjbDxdu9fSgI/r3KN72G3c2goxknAqHHgPWWs8UlI=
|
|
||||||
github.com/mattn/go-sqlite3 v1.14.4/go.mod h1:WVKg1VTActs4Qso6iwGbiFih2UIHo0ENGwNd0Lj+XmI=
|
|
||||||
github.com/mitchellh/go-homedir v1.1.0 h1:lukF9ziXFxDFPkA1vsr5zpc1XuPDn/wFntq5mG+4E0Y=
|
github.com/mitchellh/go-homedir v1.1.0 h1:lukF9ziXFxDFPkA1vsr5zpc1XuPDn/wFntq5mG+4E0Y=
|
||||||
github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
|
github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
|
||||||
github.com/mitchellh/mapstructure v0.0.0-20180203102830-a4e142e9c047/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
github.com/mitchellh/mapstructure v0.0.0-20180203102830-a4e142e9c047/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
||||||
github.com/mitchellh/mapstructure v1.1.2 h1:fmNYVwqnSfB9mZU6OS2O6GsXM+wcskZDuKQzvN1EDeE=
|
github.com/mitchellh/mapstructure v1.1.2 h1:fmNYVwqnSfB9mZU6OS2O6GsXM+wcskZDuKQzvN1EDeE=
|
||||||
github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
|
||||||
|
github.com/oklog/ulid v1.3.1 h1:EGfNDEx6MqHz8B3uNV6QAib1UR2Lm97sHi3ocA6ESJ4=
|
||||||
|
github.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U=
|
||||||
github.com/opentracing/basictracer-go v1.0.0/go.mod h1:QfBfYuafItcjQuMwinw9GhYKwFXS9KnPs5lxoYwgW74=
|
github.com/opentracing/basictracer-go v1.0.0/go.mod h1:QfBfYuafItcjQuMwinw9GhYKwFXS9KnPs5lxoYwgW74=
|
||||||
github.com/opentracing/opentracing-go v1.0.2/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
|
github.com/opentracing/opentracing-go v1.0.2/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
|
||||||
|
github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
|
||||||
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
github.com/rs/cors v1.6.0/go.mod h1:gFx+x8UowdsKA9AchylcLynDq+nNFfI8FkUZdN/jGCU=
|
github.com/rs/cors v1.6.0/go.mod h1:gFx+x8UowdsKA9AchylcLynDq+nNFfI8FkUZdN/jGCU=
|
||||||
|
github.com/russross/blackfriday v1.5.2 h1:HyvC0ARfnZBqnXwABFeSZHpKvJHJJfPz81GNueLj0oo=
|
||||||
|
github.com/russross/blackfriday v1.5.2/go.mod h1:JO/DiYxRf+HjHt06OyowR9PTA263kcR/rfWxYHBV53g=
|
||||||
|
github.com/russross/blackfriday/v2 v2.0.1 h1:lPqVAte+HuHNfhJ/0LC98ESWRz8afy9tM/0RK8m9o+Q=
|
||||||
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
github.com/sergi/go-diff v1.1.0 h1:we8PVUC3FE2uYfodKH/nBHMSetSfHDR6scGdBi+erh0=
|
github.com/sergi/go-diff v1.1.0 h1:we8PVUC3FE2uYfodKH/nBHMSetSfHDR6scGdBi+erh0=
|
||||||
github.com/sergi/go-diff v1.1.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=
|
github.com/sergi/go-diff v1.1.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=
|
||||||
github.com/shurcooL/httpfs v0.0.0-20171119174359-809beceb2371/go.mod h1:ZY1cvUeJuFPAdZ/B6v7RHavJWZn2YPVFQ1OSXhCGOkg=
|
github.com/shurcooL/httpfs v0.0.0-20171119174359-809beceb2371/go.mod h1:ZY1cvUeJuFPAdZ/B6v7RHavJWZn2YPVFQ1OSXhCGOkg=
|
||||||
|
github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo=
|
||||||
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
|
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
|
||||||
github.com/shurcooL/vfsgen v0.0.0-20180121065927-ffb13db8def0/go.mod h1:TrYk7fJVaAttu97ZZKrO9UbRa8izdowaMIZcxYMbVaw=
|
github.com/shurcooL/vfsgen v0.0.0-20180121065927-ffb13db8def0/go.mod h1:TrYk7fJVaAttu97ZZKrO9UbRa8izdowaMIZcxYMbVaw=
|
||||||
|
github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
|
||||||
|
github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI=
|
||||||
|
github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
|
||||||
|
github.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ=
|
||||||
|
github.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
|
||||||
|
github.com/spf13/cobra v0.0.5/go.mod h1:3K3wKZymM7VvHMDS9+Akkh4K60UwM26emMESw8tLCHU=
|
||||||
|
github.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo=
|
||||||
|
github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
|
||||||
|
github.com/spf13/viper v1.3.2/go.mod h1:ZiWeW+zYFKm7srdB9IoDzzZXaJaI5eL9QjNiN/DMA2s=
|
||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
github.com/stretchr/testify v1.2.1/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
github.com/stretchr/testify v1.2.1/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
|
||||||
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
|
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
|
||||||
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
|
||||||
|
github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=
|
||||||
|
github.com/urfave/cli/v2 v2.1.1 h1:Qt8FeAtxE/vfdrLmR3rxR6JRE0RoVmbXu8+6kZtYU4k=
|
||||||
github.com/urfave/cli/v2 v2.1.1/go.mod h1:SE9GqnLQmjVa0iPEY0f1w3ygNIYcIJ0OKPMoW2caLfQ=
|
github.com/urfave/cli/v2 v2.1.1/go.mod h1:SE9GqnLQmjVa0iPEY0f1w3ygNIYcIJ0OKPMoW2caLfQ=
|
||||||
|
github.com/vektah/dataloaden v0.2.1-0.20190515034641-a19b9a6e7c9e h1:+w0Zm/9gaWpEAyDlU1eKOuk5twTjAjuevXqcJJw8hrg=
|
||||||
github.com/vektah/dataloaden v0.2.1-0.20190515034641-a19b9a6e7c9e/go.mod h1:/HUdMve7rvxZma+2ZELQeNh88+003LL7Pf/CZ089j8U=
|
github.com/vektah/dataloaden v0.2.1-0.20190515034641-a19b9a6e7c9e/go.mod h1:/HUdMve7rvxZma+2ZELQeNh88+003LL7Pf/CZ089j8U=
|
||||||
github.com/vektah/gqlparser/v2 v2.1.0 h1:uiKJ+T5HMGGQM2kRKQ8Pxw8+Zq9qhhZhz/lieYvCMns=
|
github.com/vektah/gqlparser/v2 v2.2.0 h1:bAc3slekAAJW6sZTi07aGq0OrfaCjj4jxARAaC7g2EM=
|
||||||
github.com/vektah/gqlparser/v2 v2.1.0/go.mod h1:SyUiHgLATUR8BiYURfTirrTcGpcE+4XkV2se04Px1Ms=
|
github.com/vektah/gqlparser/v2 v2.2.0/go.mod h1:i3mQIGIrbK2PD1RrCeMTlVbkF2FJ6WkU1KJlJlC+3F4=
|
||||||
|
github.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77/go.mod h1:aYKd//L2LvnjZzWKhF00oedf4jCCReLcmhLdhm1A27Q=
|
||||||
|
github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
|
github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
|
||||||
|
go.opencensus.io v0.22.5 h1:dntmOdLpSpHlVqbW5Eay97DelsZHe+55D+xC6i0dDS0=
|
||||||
|
go.opencensus.io v0.22.5/go.mod h1:5pWMHQbX5EPX2/62yrJeAkowc+lfs/XD7Uxpq3pI6kk=
|
||||||
|
golang.org/x/crypto v0.0.0-20181203042331-505ab145d0a9/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
||||||
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
|
||||||
golang.org/x/mod v0.1.1-0.20191105210325-c90efee705ee/go.mod h1:QqPTAvyqsEbceGzBzNggFXnrqF1CaUcvgkdR5Ot7KZg=
|
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
|
||||||
|
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||||
|
golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=
|
||||||
|
golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=
|
||||||
|
golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=
|
||||||
|
golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
|
golang.org/x/mod v0.3.0 h1:RM4zey1++hCTbCVQfnWeKs9/IEsaBLA8vTkd0WVtmH4=
|
||||||
|
golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||||
|
golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
|
golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||||
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
||||||
golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks=
|
|
||||||
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
|
||||||
|
golang.org/x/net v0.0.0-20201021035429-f5854403a974 h1:IX6qOQeG5uLjB/hjjwjedwfjND0hgjPMMyO1RoIXQNI=
|
||||||
|
golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
|
||||||
|
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||||
|
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
|
golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
|
golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||||
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20190502145724-3ef323f4f1fd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
|
golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c h1:VwygUrnw9jn88c4u8GD3rZQbqrP/tgas88tPUbBxQrk=
|
||||||
|
golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
||||||
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
|
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
golang.org/x/tools v0.0.0-20190125232054-d66bd3c5d5a6/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20190125232054-d66bd3c5d5a6/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
|
golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=
|
||||||
|
golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
||||||
golang.org/x/tools v0.0.0-20190515012406-7d7faa4812bd/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
golang.org/x/tools v0.0.0-20190515012406-7d7faa4812bd/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=
|
||||||
golang.org/x/tools v0.0.0-20200114235610-7ae403b6b589/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
|
golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE=
|
||||||
|
golang.org/x/tools v0.0.0-20210106214847-113979e3529a h1:CB3a9Nez8M13wwlr/E2YtwoU+qYHKfC+JrDa45RXXoQ=
|
||||||
|
golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA=
|
||||||
|
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/appengine v1.6.6 h1:lMO5rYAqUxkmaj76jAkRUvt5JZgFymx/+Q5Mzfivuhc=
|
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
google.golang.org/appengine v1.6.6/go.mod h1:8WjMMxjGQR8xUklV/ARdw2HLXBOI7O7uCIDZVag1xfc=
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE=
|
||||||
|
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||||
|
google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=
|
||||||
|
google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=
|
||||||
|
google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=
|
||||||
|
google.golang.org/genproto v0.0.0-20190425155659-357c62f0e4bb/go.mod h1:VzzqZJRnGkLBvHegQrXjBqPurQTc5/KpmUdxsrq26oE=
|
||||||
|
google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=
|
||||||
|
google.golang.org/grpc v1.20.1/go.mod h1:10oTOabMzJvdu6/UiuZezV6QK5dSlG84ov/aaiqXj38=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
|
||||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
gopkg.in/yaml.v2 v2.2.4 h1:/eiJrUcujPVeJ3xlSWaiNi3uSVmDGBK1pDHUHAnao1I=
|
gopkg.in/yaml.v2 v2.2.4 h1:/eiJrUcujPVeJ3xlSWaiNi3uSVmDGBK1pDHUHAnao1I=
|
||||||
gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
|
||||||
sourcegraph.com/sourcegraph/appdash v0.0.0-20180110180208-2cc67fd64755/go.mod h1:hI742Nqp5OhwiqlzhgfbWU4mW4yO10fP+LoT9WOswdU=
|
honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
|
||||||
|
sourcegraph.com/sourcegraph/appdash v0.0.0-20190731080439-ebfcffb1b5c0/go.mod h1:hI742Nqp5OhwiqlzhgfbWU4mW4yO10fP+LoT9WOswdU=
|
||||||
sourcegraph.com/sourcegraph/appdash-data v0.0.0-20151005221446-73f23eafcf67/go.mod h1:L5q+DGLGOQFpo1snNEkLOJT2d1YTW66rWNzatr3He1k=
|
sourcegraph.com/sourcegraph/appdash-data v0.0.0-20151005221446-73f23eafcf67/go.mod h1:L5q+DGLGOQFpo1snNEkLOJT2d1YTW66rWNzatr3He1k=
|
||||||
|
@ -45,7 +45,7 @@ omit_slice_element_pointers: true
|
|||||||
models:
|
models:
|
||||||
ID:
|
ID:
|
||||||
model:
|
model:
|
||||||
- github.com/99designs/gqlgen/graphql.Int64
|
- github.com/dstotijn/hetty/pkg/api.ULID
|
||||||
# Int:
|
# Int:
|
||||||
# model:
|
# model:
|
||||||
# - github.com/99designs/gqlgen/graphql.Int
|
# - github.com/99designs/gqlgen/graphql.Int
|
||||||
|
@ -80,7 +80,6 @@ type ComplexityRoot struct {
|
|||||||
Body func(childComplexity int) int
|
Body func(childComplexity int) int
|
||||||
Headers func(childComplexity int) int
|
Headers func(childComplexity int) int
|
||||||
Proto func(childComplexity int) int
|
Proto func(childComplexity int) int
|
||||||
RequestID func(childComplexity int) int
|
|
||||||
StatusCode func(childComplexity int) int
|
StatusCode func(childComplexity int) int
|
||||||
StatusReason func(childComplexity int) int
|
StatusReason func(childComplexity int) int
|
||||||
}
|
}
|
||||||
@ -88,20 +87,22 @@ type ComplexityRoot struct {
|
|||||||
Mutation struct {
|
Mutation struct {
|
||||||
ClearHTTPRequestLog func(childComplexity int) int
|
ClearHTTPRequestLog func(childComplexity int) int
|
||||||
CloseProject func(childComplexity int) int
|
CloseProject func(childComplexity int) int
|
||||||
DeleteProject func(childComplexity int, name string) int
|
CreateProject func(childComplexity int, name string) int
|
||||||
OpenProject func(childComplexity int, name string) int
|
DeleteProject func(childComplexity int, id ULID) int
|
||||||
|
OpenProject func(childComplexity int, id ULID) int
|
||||||
SetHTTPRequestLogFilter func(childComplexity int, filter *HTTPRequestLogFilterInput) int
|
SetHTTPRequestLogFilter func(childComplexity int, filter *HTTPRequestLogFilterInput) int
|
||||||
SetScope func(childComplexity int, scope []ScopeRuleInput) int
|
SetScope func(childComplexity int, scope []ScopeRuleInput) int
|
||||||
}
|
}
|
||||||
|
|
||||||
Project struct {
|
Project struct {
|
||||||
|
ID func(childComplexity int) int
|
||||||
IsActive func(childComplexity int) int
|
IsActive func(childComplexity int) int
|
||||||
Name func(childComplexity int) int
|
Name func(childComplexity int) int
|
||||||
}
|
}
|
||||||
|
|
||||||
Query struct {
|
Query struct {
|
||||||
ActiveProject func(childComplexity int) int
|
ActiveProject func(childComplexity int) int
|
||||||
HTTPRequestLog func(childComplexity int, id int64) int
|
HTTPRequestLog func(childComplexity int, id ULID) int
|
||||||
HTTPRequestLogFilter func(childComplexity int) int
|
HTTPRequestLogFilter func(childComplexity int) int
|
||||||
HTTPRequestLogs func(childComplexity int) int
|
HTTPRequestLogs func(childComplexity int) int
|
||||||
Projects func(childComplexity int) int
|
Projects func(childComplexity int) int
|
||||||
@ -121,15 +122,16 @@ type ComplexityRoot struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type MutationResolver interface {
|
type MutationResolver interface {
|
||||||
OpenProject(ctx context.Context, name string) (*Project, error)
|
CreateProject(ctx context.Context, name string) (*Project, error)
|
||||||
|
OpenProject(ctx context.Context, id ULID) (*Project, error)
|
||||||
CloseProject(ctx context.Context) (*CloseProjectResult, error)
|
CloseProject(ctx context.Context) (*CloseProjectResult, error)
|
||||||
DeleteProject(ctx context.Context, name string) (*DeleteProjectResult, error)
|
DeleteProject(ctx context.Context, id ULID) (*DeleteProjectResult, error)
|
||||||
ClearHTTPRequestLog(ctx context.Context) (*ClearHTTPRequestLogResult, error)
|
ClearHTTPRequestLog(ctx context.Context) (*ClearHTTPRequestLogResult, error)
|
||||||
SetScope(ctx context.Context, scope []ScopeRuleInput) ([]ScopeRule, error)
|
SetScope(ctx context.Context, scope []ScopeRuleInput) ([]ScopeRule, error)
|
||||||
SetHTTPRequestLogFilter(ctx context.Context, filter *HTTPRequestLogFilterInput) (*HTTPRequestLogFilter, error)
|
SetHTTPRequestLogFilter(ctx context.Context, filter *HTTPRequestLogFilterInput) (*HTTPRequestLogFilter, error)
|
||||||
}
|
}
|
||||||
type QueryResolver interface {
|
type QueryResolver interface {
|
||||||
HTTPRequestLog(ctx context.Context, id int64) (*HTTPRequestLog, error)
|
HTTPRequestLog(ctx context.Context, id ULID) (*HTTPRequestLog, error)
|
||||||
HTTPRequestLogs(ctx context.Context) ([]HTTPRequestLog, error)
|
HTTPRequestLogs(ctx context.Context) ([]HTTPRequestLog, error)
|
||||||
HTTPRequestLogFilter(ctx context.Context) (*HTTPRequestLogFilter, error)
|
HTTPRequestLogFilter(ctx context.Context) (*HTTPRequestLogFilter, error)
|
||||||
ActiveProject(ctx context.Context) (*Project, error)
|
ActiveProject(ctx context.Context) (*Project, error)
|
||||||
@ -278,13 +280,6 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
|
|
||||||
return e.complexity.HTTPResponseLog.Proto(childComplexity), true
|
return e.complexity.HTTPResponseLog.Proto(childComplexity), true
|
||||||
|
|
||||||
case "HttpResponseLog.requestId":
|
|
||||||
if e.complexity.HTTPResponseLog.RequestID == nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
return e.complexity.HTTPResponseLog.RequestID(childComplexity), true
|
|
||||||
|
|
||||||
case "HttpResponseLog.statusCode":
|
case "HttpResponseLog.statusCode":
|
||||||
if e.complexity.HTTPResponseLog.StatusCode == nil {
|
if e.complexity.HTTPResponseLog.StatusCode == nil {
|
||||||
break
|
break
|
||||||
@ -313,6 +308,18 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
|
|
||||||
return e.complexity.Mutation.CloseProject(childComplexity), true
|
return e.complexity.Mutation.CloseProject(childComplexity), true
|
||||||
|
|
||||||
|
case "Mutation.createProject":
|
||||||
|
if e.complexity.Mutation.CreateProject == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
args, err := ec.field_Mutation_createProject_args(context.TODO(), rawArgs)
|
||||||
|
if err != nil {
|
||||||
|
return 0, false
|
||||||
|
}
|
||||||
|
|
||||||
|
return e.complexity.Mutation.CreateProject(childComplexity, args["name"].(string)), true
|
||||||
|
|
||||||
case "Mutation.deleteProject":
|
case "Mutation.deleteProject":
|
||||||
if e.complexity.Mutation.DeleteProject == nil {
|
if e.complexity.Mutation.DeleteProject == nil {
|
||||||
break
|
break
|
||||||
@ -323,7 +330,7 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
return 0, false
|
return 0, false
|
||||||
}
|
}
|
||||||
|
|
||||||
return e.complexity.Mutation.DeleteProject(childComplexity, args["name"].(string)), true
|
return e.complexity.Mutation.DeleteProject(childComplexity, args["id"].(ULID)), true
|
||||||
|
|
||||||
case "Mutation.openProject":
|
case "Mutation.openProject":
|
||||||
if e.complexity.Mutation.OpenProject == nil {
|
if e.complexity.Mutation.OpenProject == nil {
|
||||||
@ -335,7 +342,7 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
return 0, false
|
return 0, false
|
||||||
}
|
}
|
||||||
|
|
||||||
return e.complexity.Mutation.OpenProject(childComplexity, args["name"].(string)), true
|
return e.complexity.Mutation.OpenProject(childComplexity, args["id"].(ULID)), true
|
||||||
|
|
||||||
case "Mutation.setHttpRequestLogFilter":
|
case "Mutation.setHttpRequestLogFilter":
|
||||||
if e.complexity.Mutation.SetHTTPRequestLogFilter == nil {
|
if e.complexity.Mutation.SetHTTPRequestLogFilter == nil {
|
||||||
@ -361,6 +368,13 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
|
|
||||||
return e.complexity.Mutation.SetScope(childComplexity, args["scope"].([]ScopeRuleInput)), true
|
return e.complexity.Mutation.SetScope(childComplexity, args["scope"].([]ScopeRuleInput)), true
|
||||||
|
|
||||||
|
case "Project.id":
|
||||||
|
if e.complexity.Project.ID == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
return e.complexity.Project.ID(childComplexity), true
|
||||||
|
|
||||||
case "Project.isActive":
|
case "Project.isActive":
|
||||||
if e.complexity.Project.IsActive == nil {
|
if e.complexity.Project.IsActive == nil {
|
||||||
break
|
break
|
||||||
@ -392,7 +406,7 @@ func (e *executableSchema) Complexity(typeName, field string, childComplexity in
|
|||||||
return 0, false
|
return 0, false
|
||||||
}
|
}
|
||||||
|
|
||||||
return e.complexity.Query.HTTPRequestLog(childComplexity, args["id"].(int64)), true
|
return e.complexity.Query.HTTPRequestLog(childComplexity, args["id"].(ULID)), true
|
||||||
|
|
||||||
case "Query.httpRequestLogFilter":
|
case "Query.httpRequestLogFilter":
|
||||||
if e.complexity.Query.HTTPRequestLogFilter == nil {
|
if e.complexity.Query.HTTPRequestLogFilter == nil {
|
||||||
@ -533,7 +547,6 @@ var sources = []*ast.Source{
|
|||||||
}
|
}
|
||||||
|
|
||||||
type HttpResponseLog {
|
type HttpResponseLog {
|
||||||
requestId: ID!
|
|
||||||
proto: String!
|
proto: String!
|
||||||
statusCode: Int!
|
statusCode: Int!
|
||||||
statusReason: String!
|
statusReason: String!
|
||||||
@ -547,6 +560,7 @@ type HttpHeader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Project {
|
type Project {
|
||||||
|
id: ID!
|
||||||
name: String!
|
name: String!
|
||||||
isActive: Boolean!
|
isActive: Boolean!
|
||||||
}
|
}
|
||||||
@ -605,9 +619,10 @@ type Query {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Mutation {
|
type Mutation {
|
||||||
openProject(name: String!): Project
|
createProject(name: String!): Project
|
||||||
|
openProject(id: ID!): Project
|
||||||
closeProject: CloseProjectResult!
|
closeProject: CloseProjectResult!
|
||||||
deleteProject(name: String!): DeleteProjectResult!
|
deleteProject(id: ID!): DeleteProjectResult!
|
||||||
clearHTTPRequestLog: ClearHTTPRequestLogResult!
|
clearHTTPRequestLog: ClearHTTPRequestLogResult!
|
||||||
setScope(scope: [ScopeRuleInput!]!): [ScopeRule!]!
|
setScope(scope: [ScopeRuleInput!]!): [ScopeRule!]!
|
||||||
setHttpRequestLogFilter(
|
setHttpRequestLogFilter(
|
||||||
@ -637,7 +652,7 @@ var parsedSchema = gqlparser.MustLoadSchema(sources...)
|
|||||||
|
|
||||||
// region ***************************** args.gotpl *****************************
|
// region ***************************** args.gotpl *****************************
|
||||||
|
|
||||||
func (ec *executionContext) field_Mutation_deleteProject_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
func (ec *executionContext) field_Mutation_createProject_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
||||||
var err error
|
var err error
|
||||||
args := map[string]interface{}{}
|
args := map[string]interface{}{}
|
||||||
var arg0 string
|
var arg0 string
|
||||||
@ -652,18 +667,33 @@ func (ec *executionContext) field_Mutation_deleteProject_args(ctx context.Contex
|
|||||||
return args, nil
|
return args, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) field_Mutation_openProject_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
func (ec *executionContext) field_Mutation_deleteProject_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
||||||
var err error
|
var err error
|
||||||
args := map[string]interface{}{}
|
args := map[string]interface{}{}
|
||||||
var arg0 string
|
var arg0 ULID
|
||||||
if tmp, ok := rawArgs["name"]; ok {
|
if tmp, ok := rawArgs["id"]; ok {
|
||||||
ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("name"))
|
ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("id"))
|
||||||
arg0, err = ec.unmarshalNString2string(ctx, tmp)
|
arg0, err = ec.unmarshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx, tmp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
args["name"] = arg0
|
args["id"] = arg0
|
||||||
|
return args, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ec *executionContext) field_Mutation_openProject_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
||||||
|
var err error
|
||||||
|
args := map[string]interface{}{}
|
||||||
|
var arg0 ULID
|
||||||
|
if tmp, ok := rawArgs["id"]; ok {
|
||||||
|
ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("id"))
|
||||||
|
arg0, err = ec.unmarshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx, tmp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
args["id"] = arg0
|
||||||
return args, nil
|
return args, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -715,10 +745,10 @@ func (ec *executionContext) field_Query___type_args(ctx context.Context, rawArgs
|
|||||||
func (ec *executionContext) field_Query_httpRequestLog_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
func (ec *executionContext) field_Query_httpRequestLog_args(ctx context.Context, rawArgs map[string]interface{}) (map[string]interface{}, error) {
|
||||||
var err error
|
var err error
|
||||||
args := map[string]interface{}{}
|
args := map[string]interface{}{}
|
||||||
var arg0 int64
|
var arg0 ULID
|
||||||
if tmp, ok := rawArgs["id"]; ok {
|
if tmp, ok := rawArgs["id"]; ok {
|
||||||
ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("id"))
|
ctx := graphql.WithPathContext(ctx, graphql.NewPathWithField("id"))
|
||||||
arg0, err = ec.unmarshalNID2int64(ctx, tmp)
|
arg0, err = ec.unmarshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx, tmp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@ -970,9 +1000,9 @@ func (ec *executionContext) _HttpRequestLog_id(ctx context.Context, field graphq
|
|||||||
}
|
}
|
||||||
return graphql.Null
|
return graphql.Null
|
||||||
}
|
}
|
||||||
res := resTmp.(int64)
|
res := resTmp.(ULID)
|
||||||
fc.Result = res
|
fc.Result = res
|
||||||
return ec.marshalNID2int64(ctx, field.Selections, res)
|
return ec.marshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx, field.Selections, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) _HttpRequestLog_url(ctx context.Context, field graphql.CollectedField, obj *HTTPRequestLog) (ret graphql.Marshaler) {
|
func (ec *executionContext) _HttpRequestLog_url(ctx context.Context, field graphql.CollectedField, obj *HTTPRequestLog) (ret graphql.Marshaler) {
|
||||||
@ -1281,41 +1311,6 @@ func (ec *executionContext) _HttpRequestLogFilter_searchExpression(ctx context.C
|
|||||||
return ec.marshalOString2ᚖstring(ctx, field.Selections, res)
|
return ec.marshalOString2ᚖstring(ctx, field.Selections, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) _HttpResponseLog_requestId(ctx context.Context, field graphql.CollectedField, obj *HTTPResponseLog) (ret graphql.Marshaler) {
|
|
||||||
defer func() {
|
|
||||||
if r := recover(); r != nil {
|
|
||||||
ec.Error(ctx, ec.Recover(ctx, r))
|
|
||||||
ret = graphql.Null
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
fc := &graphql.FieldContext{
|
|
||||||
Object: "HttpResponseLog",
|
|
||||||
Field: field,
|
|
||||||
Args: nil,
|
|
||||||
IsMethod: false,
|
|
||||||
IsResolver: false,
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx = graphql.WithFieldContext(ctx, fc)
|
|
||||||
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
|
||||||
ctx = rctx // use context from middleware stack in children
|
|
||||||
return obj.RequestID, nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
ec.Error(ctx, err)
|
|
||||||
return graphql.Null
|
|
||||||
}
|
|
||||||
if resTmp == nil {
|
|
||||||
if !graphql.HasFieldError(ctx, fc) {
|
|
||||||
ec.Errorf(ctx, "must not be null")
|
|
||||||
}
|
|
||||||
return graphql.Null
|
|
||||||
}
|
|
||||||
res := resTmp.(int64)
|
|
||||||
fc.Result = res
|
|
||||||
return ec.marshalNID2int64(ctx, field.Selections, res)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ec *executionContext) _HttpResponseLog_proto(ctx context.Context, field graphql.CollectedField, obj *HTTPResponseLog) (ret graphql.Marshaler) {
|
func (ec *executionContext) _HttpResponseLog_proto(ctx context.Context, field graphql.CollectedField, obj *HTTPResponseLog) (ret graphql.Marshaler) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
@ -1488,6 +1483,45 @@ func (ec *executionContext) _HttpResponseLog_headers(ctx context.Context, field
|
|||||||
return ec.marshalNHttpHeader2ᚕgithubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐHTTPHeaderᚄ(ctx, field.Selections, res)
|
return ec.marshalNHttpHeader2ᚕgithubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐHTTPHeaderᚄ(ctx, field.Selections, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ec *executionContext) _Mutation_createProject(ctx context.Context, field graphql.CollectedField) (ret graphql.Marshaler) {
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
ec.Error(ctx, ec.Recover(ctx, r))
|
||||||
|
ret = graphql.Null
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
fc := &graphql.FieldContext{
|
||||||
|
Object: "Mutation",
|
||||||
|
Field: field,
|
||||||
|
Args: nil,
|
||||||
|
IsMethod: true,
|
||||||
|
IsResolver: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx = graphql.WithFieldContext(ctx, fc)
|
||||||
|
rawArgs := field.ArgumentMap(ec.Variables)
|
||||||
|
args, err := ec.field_Mutation_createProject_args(ctx, rawArgs)
|
||||||
|
if err != nil {
|
||||||
|
ec.Error(ctx, err)
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
fc.Args = args
|
||||||
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
|
ctx = rctx // use context from middleware stack in children
|
||||||
|
return ec.resolvers.Mutation().CreateProject(rctx, args["name"].(string))
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
ec.Error(ctx, err)
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
if resTmp == nil {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
res := resTmp.(*Project)
|
||||||
|
fc.Result = res
|
||||||
|
return ec.marshalOProject2ᚖgithubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐProject(ctx, field.Selections, res)
|
||||||
|
}
|
||||||
|
|
||||||
func (ec *executionContext) _Mutation_openProject(ctx context.Context, field graphql.CollectedField) (ret graphql.Marshaler) {
|
func (ec *executionContext) _Mutation_openProject(ctx context.Context, field graphql.CollectedField) (ret graphql.Marshaler) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
@ -1513,7 +1547,7 @@ func (ec *executionContext) _Mutation_openProject(ctx context.Context, field gra
|
|||||||
fc.Args = args
|
fc.Args = args
|
||||||
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
ctx = rctx // use context from middleware stack in children
|
ctx = rctx // use context from middleware stack in children
|
||||||
return ec.resolvers.Mutation().OpenProject(rctx, args["name"].(string))
|
return ec.resolvers.Mutation().OpenProject(rctx, args["id"].(ULID))
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ec.Error(ctx, err)
|
ec.Error(ctx, err)
|
||||||
@ -1587,7 +1621,7 @@ func (ec *executionContext) _Mutation_deleteProject(ctx context.Context, field g
|
|||||||
fc.Args = args
|
fc.Args = args
|
||||||
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
ctx = rctx // use context from middleware stack in children
|
ctx = rctx // use context from middleware stack in children
|
||||||
return ec.resolvers.Mutation().DeleteProject(rctx, args["name"].(string))
|
return ec.resolvers.Mutation().DeleteProject(rctx, args["id"].(ULID))
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ec.Error(ctx, err)
|
ec.Error(ctx, err)
|
||||||
@ -1720,6 +1754,41 @@ func (ec *executionContext) _Mutation_setHttpRequestLogFilter(ctx context.Contex
|
|||||||
return ec.marshalOHttpRequestLogFilter2ᚖgithubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐHTTPRequestLogFilter(ctx, field.Selections, res)
|
return ec.marshalOHttpRequestLogFilter2ᚖgithubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐHTTPRequestLogFilter(ctx, field.Selections, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ec *executionContext) _Project_id(ctx context.Context, field graphql.CollectedField, obj *Project) (ret graphql.Marshaler) {
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
ec.Error(ctx, ec.Recover(ctx, r))
|
||||||
|
ret = graphql.Null
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
fc := &graphql.FieldContext{
|
||||||
|
Object: "Project",
|
||||||
|
Field: field,
|
||||||
|
Args: nil,
|
||||||
|
IsMethod: false,
|
||||||
|
IsResolver: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx = graphql.WithFieldContext(ctx, fc)
|
||||||
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
|
ctx = rctx // use context from middleware stack in children
|
||||||
|
return obj.ID, nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
ec.Error(ctx, err)
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
if resTmp == nil {
|
||||||
|
if !graphql.HasFieldError(ctx, fc) {
|
||||||
|
ec.Errorf(ctx, "must not be null")
|
||||||
|
}
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
res := resTmp.(ULID)
|
||||||
|
fc.Result = res
|
||||||
|
return ec.marshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx, field.Selections, res)
|
||||||
|
}
|
||||||
|
|
||||||
func (ec *executionContext) _Project_name(ctx context.Context, field graphql.CollectedField, obj *Project) (ret graphql.Marshaler) {
|
func (ec *executionContext) _Project_name(ctx context.Context, field graphql.CollectedField, obj *Project) (ret graphql.Marshaler) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
@ -1815,7 +1884,7 @@ func (ec *executionContext) _Query_httpRequestLog(ctx context.Context, field gra
|
|||||||
fc.Args = args
|
fc.Args = args
|
||||||
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
ctx = rctx // use context from middleware stack in children
|
ctx = rctx // use context from middleware stack in children
|
||||||
return ec.resolvers.Query().HTTPRequestLog(rctx, args["id"].(int64))
|
return ec.resolvers.Query().HTTPRequestLog(rctx, args["id"].(ULID))
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ec.Error(ctx, err)
|
ec.Error(ctx, err)
|
||||||
@ -2366,6 +2435,41 @@ func (ec *executionContext) ___Directive_args(ctx context.Context, field graphql
|
|||||||
return ec.marshalN__InputValue2ᚕgithubᚗcomᚋ99designsᚋgqlgenᚋgraphqlᚋintrospectionᚐInputValueᚄ(ctx, field.Selections, res)
|
return ec.marshalN__InputValue2ᚕgithubᚗcomᚋ99designsᚋgqlgenᚋgraphqlᚋintrospectionᚐInputValueᚄ(ctx, field.Selections, res)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ec *executionContext) ___Directive_isRepeatable(ctx context.Context, field graphql.CollectedField, obj *introspection.Directive) (ret graphql.Marshaler) {
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
ec.Error(ctx, ec.Recover(ctx, r))
|
||||||
|
ret = graphql.Null
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
fc := &graphql.FieldContext{
|
||||||
|
Object: "__Directive",
|
||||||
|
Field: field,
|
||||||
|
Args: nil,
|
||||||
|
IsMethod: false,
|
||||||
|
IsResolver: false,
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx = graphql.WithFieldContext(ctx, fc)
|
||||||
|
resTmp, err := ec.ResolverMiddleware(ctx, func(rctx context.Context) (interface{}, error) {
|
||||||
|
ctx = rctx // use context from middleware stack in children
|
||||||
|
return obj.IsRepeatable, nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
ec.Error(ctx, err)
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
if resTmp == nil {
|
||||||
|
if !graphql.HasFieldError(ctx, fc) {
|
||||||
|
ec.Errorf(ctx, "must not be null")
|
||||||
|
}
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
res := resTmp.(bool)
|
||||||
|
fc.Result = res
|
||||||
|
return ec.marshalNBoolean2bool(ctx, field.Selections, res)
|
||||||
|
}
|
||||||
|
|
||||||
func (ec *executionContext) ___EnumValue_name(ctx context.Context, field graphql.CollectedField, obj *introspection.EnumValue) (ret graphql.Marshaler) {
|
func (ec *executionContext) ___EnumValue_name(ctx context.Context, field graphql.CollectedField, obj *introspection.EnumValue) (ret graphql.Marshaler) {
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
@ -3318,7 +3422,10 @@ func (ec *executionContext) ___Type_ofType(ctx context.Context, field graphql.Co
|
|||||||
|
|
||||||
func (ec *executionContext) unmarshalInputHttpRequestLogFilterInput(ctx context.Context, obj interface{}) (HTTPRequestLogFilterInput, error) {
|
func (ec *executionContext) unmarshalInputHttpRequestLogFilterInput(ctx context.Context, obj interface{}) (HTTPRequestLogFilterInput, error) {
|
||||||
var it HTTPRequestLogFilterInput
|
var it HTTPRequestLogFilterInput
|
||||||
var asMap = obj.(map[string]interface{})
|
asMap := map[string]interface{}{}
|
||||||
|
for k, v := range obj.(map[string]interface{}) {
|
||||||
|
asMap[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
for k, v := range asMap {
|
for k, v := range asMap {
|
||||||
switch k {
|
switch k {
|
||||||
@ -3346,7 +3453,10 @@ func (ec *executionContext) unmarshalInputHttpRequestLogFilterInput(ctx context.
|
|||||||
|
|
||||||
func (ec *executionContext) unmarshalInputScopeHeaderInput(ctx context.Context, obj interface{}) (ScopeHeaderInput, error) {
|
func (ec *executionContext) unmarshalInputScopeHeaderInput(ctx context.Context, obj interface{}) (ScopeHeaderInput, error) {
|
||||||
var it ScopeHeaderInput
|
var it ScopeHeaderInput
|
||||||
var asMap = obj.(map[string]interface{})
|
asMap := map[string]interface{}{}
|
||||||
|
for k, v := range obj.(map[string]interface{}) {
|
||||||
|
asMap[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
for k, v := range asMap {
|
for k, v := range asMap {
|
||||||
switch k {
|
switch k {
|
||||||
@ -3374,7 +3484,10 @@ func (ec *executionContext) unmarshalInputScopeHeaderInput(ctx context.Context,
|
|||||||
|
|
||||||
func (ec *executionContext) unmarshalInputScopeRuleInput(ctx context.Context, obj interface{}) (ScopeRuleInput, error) {
|
func (ec *executionContext) unmarshalInputScopeRuleInput(ctx context.Context, obj interface{}) (ScopeRuleInput, error) {
|
||||||
var it ScopeRuleInput
|
var it ScopeRuleInput
|
||||||
var asMap = obj.(map[string]interface{})
|
asMap := map[string]interface{}{}
|
||||||
|
for k, v := range obj.(map[string]interface{}) {
|
||||||
|
asMap[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
for k, v := range asMap {
|
for k, v := range asMap {
|
||||||
switch k {
|
switch k {
|
||||||
@ -3625,11 +3738,6 @@ func (ec *executionContext) _HttpResponseLog(ctx context.Context, sel ast.Select
|
|||||||
switch field.Name {
|
switch field.Name {
|
||||||
case "__typename":
|
case "__typename":
|
||||||
out.Values[i] = graphql.MarshalString("HttpResponseLog")
|
out.Values[i] = graphql.MarshalString("HttpResponseLog")
|
||||||
case "requestId":
|
|
||||||
out.Values[i] = ec._HttpResponseLog_requestId(ctx, field, obj)
|
|
||||||
if out.Values[i] == graphql.Null {
|
|
||||||
invalids++
|
|
||||||
}
|
|
||||||
case "proto":
|
case "proto":
|
||||||
out.Values[i] = ec._HttpResponseLog_proto(ctx, field, obj)
|
out.Values[i] = ec._HttpResponseLog_proto(ctx, field, obj)
|
||||||
if out.Values[i] == graphql.Null {
|
if out.Values[i] == graphql.Null {
|
||||||
@ -3678,6 +3786,8 @@ func (ec *executionContext) _Mutation(ctx context.Context, sel ast.SelectionSet)
|
|||||||
switch field.Name {
|
switch field.Name {
|
||||||
case "__typename":
|
case "__typename":
|
||||||
out.Values[i] = graphql.MarshalString("Mutation")
|
out.Values[i] = graphql.MarshalString("Mutation")
|
||||||
|
case "createProject":
|
||||||
|
out.Values[i] = ec._Mutation_createProject(ctx, field)
|
||||||
case "openProject":
|
case "openProject":
|
||||||
out.Values[i] = ec._Mutation_openProject(ctx, field)
|
out.Values[i] = ec._Mutation_openProject(ctx, field)
|
||||||
case "closeProject":
|
case "closeProject":
|
||||||
@ -3724,6 +3834,11 @@ func (ec *executionContext) _Project(ctx context.Context, sel ast.SelectionSet,
|
|||||||
switch field.Name {
|
switch field.Name {
|
||||||
case "__typename":
|
case "__typename":
|
||||||
out.Values[i] = graphql.MarshalString("Project")
|
out.Values[i] = graphql.MarshalString("Project")
|
||||||
|
case "id":
|
||||||
|
out.Values[i] = ec._Project_id(ctx, field, obj)
|
||||||
|
if out.Values[i] == graphql.Null {
|
||||||
|
invalids++
|
||||||
|
}
|
||||||
case "name":
|
case "name":
|
||||||
out.Values[i] = ec._Project_name(ctx, field, obj)
|
out.Values[i] = ec._Project_name(ctx, field, obj)
|
||||||
if out.Values[i] == graphql.Null {
|
if out.Values[i] == graphql.Null {
|
||||||
@ -3932,6 +4047,11 @@ func (ec *executionContext) ___Directive(ctx context.Context, sel ast.SelectionS
|
|||||||
if out.Values[i] == graphql.Null {
|
if out.Values[i] == graphql.Null {
|
||||||
invalids++
|
invalids++
|
||||||
}
|
}
|
||||||
|
case "isRepeatable":
|
||||||
|
out.Values[i] = ec.___Directive_isRepeatable(ctx, field, obj)
|
||||||
|
if out.Values[i] == graphql.Null {
|
||||||
|
invalids++
|
||||||
|
}
|
||||||
default:
|
default:
|
||||||
panic("unknown field " + strconv.Quote(field.Name))
|
panic("unknown field " + strconv.Quote(field.Name))
|
||||||
}
|
}
|
||||||
@ -4244,6 +4364,13 @@ func (ec *executionContext) marshalNHttpHeader2ᚕgithubᚗcomᚋdstotijnᚋhett
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4295,22 +4422,24 @@ func (ec *executionContext) marshalNHttpRequestLog2ᚕgithubᚗcomᚋdstotijnᚋ
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) unmarshalNID2int64(ctx context.Context, v interface{}) (int64, error) {
|
func (ec *executionContext) unmarshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx context.Context, v interface{}) (ULID, error) {
|
||||||
res, err := graphql.UnmarshalInt64(v)
|
var res ULID
|
||||||
|
err := res.UnmarshalGQL(v)
|
||||||
return res, graphql.ErrorOnPath(ctx, err)
|
return res, graphql.ErrorOnPath(ctx, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) marshalNID2int64(ctx context.Context, sel ast.SelectionSet, v int64) graphql.Marshaler {
|
func (ec *executionContext) marshalNID2githubᚗcomᚋdstotijnᚋhettyᚋpkgᚋapiᚐULID(ctx context.Context, sel ast.SelectionSet, v ULID) graphql.Marshaler {
|
||||||
res := graphql.MarshalInt64(v)
|
return v
|
||||||
if res == graphql.Null {
|
|
||||||
if !graphql.HasFieldError(ctx, graphql.GetFieldContext(ctx)) {
|
|
||||||
ec.Errorf(ctx, "must not be null")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return res
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ec *executionContext) unmarshalNInt2int(ctx context.Context, v interface{}) (int, error) {
|
func (ec *executionContext) unmarshalNInt2int(ctx context.Context, v interface{}) (int, error) {
|
||||||
@ -4366,6 +4495,13 @@ func (ec *executionContext) marshalNProject2ᚕgithubᚗcomᚋdstotijnᚋhetty
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4407,6 +4543,13 @@ func (ec *executionContext) marshalNScopeRule2ᚕgithubᚗcomᚋdstotijnᚋhetty
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4504,6 +4647,13 @@ func (ec *executionContext) marshalN__Directive2ᚕgithubᚗcomᚋ99designsᚋgq
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4577,6 +4727,13 @@ func (ec *executionContext) marshalN__DirectiveLocation2ᚕstringᚄ(ctx context
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4626,6 +4783,13 @@ func (ec *executionContext) marshalN__InputValue2ᚕgithubᚗcomᚋ99designsᚋg
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4667,6 +4831,13 @@ func (ec *executionContext) marshalN__Type2ᚕgithubᚗcomᚋ99designsᚋgqlgen
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4846,6 +5017,13 @@ func (ec *executionContext) marshalO__EnumValue2ᚕgithubᚗcomᚋ99designsᚋgq
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4886,6 +5064,13 @@ func (ec *executionContext) marshalO__Field2ᚕgithubᚗcomᚋ99designsᚋgqlgen
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4926,6 +5111,13 @@ func (ec *executionContext) marshalO__InputValue2ᚕgithubᚗcomᚋ99designsᚋg
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -4973,6 +5165,13 @@ func (ec *executionContext) marshalO__Type2ᚕgithubᚗcomᚋ99designsᚋgqlgen
|
|||||||
|
|
||||||
}
|
}
|
||||||
wg.Wait()
|
wg.Wait()
|
||||||
|
|
||||||
|
for _, e := range ret {
|
||||||
|
if e == graphql.Null {
|
||||||
|
return graphql.Null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
|
31
pkg/api/models.go
Normal file
31
pkg/api/models.go
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ULID ulid.ULID
|
||||||
|
|
||||||
|
func (u *ULID) UnmarshalGQL(v interface{}) (err error) {
|
||||||
|
str, ok := v.(string)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("ulid must be a string")
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := ulid.Parse(str)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to parse ULID: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
*u = ULID(id)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (u ULID) MarshalGQL(w io.Writer) {
|
||||||
|
fmt.Fprint(w, strconv.Quote(ulid.ULID(u).String()))
|
||||||
|
}
|
@ -27,7 +27,7 @@ type HTTPHeader struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type HTTPRequestLog struct {
|
type HTTPRequestLog struct {
|
||||||
ID int64 `json:"id"`
|
ID ULID `json:"id"`
|
||||||
URL string `json:"url"`
|
URL string `json:"url"`
|
||||||
Method HTTPMethod `json:"method"`
|
Method HTTPMethod `json:"method"`
|
||||||
Proto string `json:"proto"`
|
Proto string `json:"proto"`
|
||||||
@ -48,7 +48,6 @@ type HTTPRequestLogFilterInput struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type HTTPResponseLog struct {
|
type HTTPResponseLog struct {
|
||||||
RequestID int64 `json:"requestId"`
|
|
||||||
Proto string `json:"proto"`
|
Proto string `json:"proto"`
|
||||||
StatusCode int `json:"statusCode"`
|
StatusCode int `json:"statusCode"`
|
||||||
StatusReason string `json:"statusReason"`
|
StatusReason string `json:"statusReason"`
|
||||||
@ -57,6 +56,7 @@ type HTTPResponseLog struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Project struct {
|
type Project struct {
|
||||||
|
ID ULID `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
IsActive bool `json:"isActive"`
|
IsActive bool `json:"isActive"`
|
||||||
}
|
}
|
||||||
|
@ -10,6 +10,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/99designs/gqlgen/graphql"
|
"github.com/99designs/gqlgen/graphql"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
"github.com/vektah/gqlparser/v2/gqlerror"
|
"github.com/vektah/gqlparser/v2/gqlerror"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
"github.com/dstotijn/hetty/pkg/proj"
|
||||||
@ -19,9 +20,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Resolver struct {
|
type Resolver struct {
|
||||||
RequestLogService *reqlog.Service
|
|
||||||
ProjectService proj.Service
|
ProjectService proj.Service
|
||||||
ScopeService *scope.Scope
|
RequestLogService *reqlog.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
type (
|
type (
|
||||||
@ -54,8 +54,8 @@ func (r *queryResolver) HTTPRequestLogs(ctx context.Context) ([]HTTPRequestLog,
|
|||||||
return logs, nil
|
return logs, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *queryResolver) HTTPRequestLog(ctx context.Context, id int64) (*HTTPRequestLog, error) {
|
func (r *queryResolver) HTTPRequestLog(ctx context.Context, id ULID) (*HTTPRequestLog, error) {
|
||||||
log, err := r.RequestLogService.FindRequestLogByID(ctx, id)
|
log, err := r.RequestLogService.FindRequestLogByID(ctx, ulid.ULID(id))
|
||||||
if errors.Is(err, reqlog.ErrRequestNotFound) {
|
if errors.Is(err, reqlog.ErrRequestNotFound) {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
@ -70,32 +70,32 @@ func (r *queryResolver) HTTPRequestLog(ctx context.Context, id int64) (*HTTPRequ
|
|||||||
return &req, nil
|
return &req, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseRequestLog(req reqlog.Request) (HTTPRequestLog, error) {
|
func parseRequestLog(reqLog reqlog.RequestLog) (HTTPRequestLog, error) {
|
||||||
method := HTTPMethod(req.Request.Method)
|
method := HTTPMethod(reqLog.Method)
|
||||||
if method != "" && !method.IsValid() {
|
if method != "" && !method.IsValid() {
|
||||||
return HTTPRequestLog{}, fmt.Errorf("request has invalid method: %v", method)
|
return HTTPRequestLog{}, fmt.Errorf("request has invalid method: %v", method)
|
||||||
}
|
}
|
||||||
|
|
||||||
log := HTTPRequestLog{
|
log := HTTPRequestLog{
|
||||||
ID: req.ID,
|
ID: ULID(reqLog.ID),
|
||||||
Proto: req.Request.Proto,
|
Proto: reqLog.Proto,
|
||||||
Method: method,
|
Method: method,
|
||||||
Timestamp: req.Timestamp,
|
Timestamp: ulid.Time(reqLog.ID.Time()),
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Request.URL != nil {
|
if reqLog.URL != nil {
|
||||||
log.URL = req.Request.URL.String()
|
log.URL = reqLog.URL.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(req.Body) > 0 {
|
if len(reqLog.Body) > 0 {
|
||||||
reqBody := string(req.Body)
|
bodyStr := string(reqLog.Body)
|
||||||
log.Body = &reqBody
|
log.Body = &bodyStr
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Request.Header != nil {
|
if reqLog.Header != nil {
|
||||||
log.Headers = make([]HTTPHeader, 0)
|
log.Headers = make([]HTTPHeader, 0)
|
||||||
|
|
||||||
for key, values := range req.Request.Header {
|
for key, values := range reqLog.Header {
|
||||||
for _, value := range values {
|
for _, value := range values {
|
||||||
log.Headers = append(log.Headers, HTTPHeader{
|
log.Headers = append(log.Headers, HTTPHeader{
|
||||||
Key: key,
|
Key: key,
|
||||||
@ -105,27 +105,26 @@ func parseRequestLog(req reqlog.Request) (HTTPRequestLog, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Response != nil {
|
if reqLog.Response != nil {
|
||||||
log.Response = &HTTPResponseLog{
|
log.Response = &HTTPResponseLog{
|
||||||
RequestID: req.Response.RequestID,
|
Proto: reqLog.Response.Proto,
|
||||||
Proto: req.Response.Response.Proto,
|
StatusCode: reqLog.Response.StatusCode,
|
||||||
StatusCode: req.Response.Response.StatusCode,
|
|
||||||
}
|
}
|
||||||
statusReasonSubs := strings.SplitN(req.Response.Response.Status, " ", 2)
|
statusReasonSubs := strings.SplitN(reqLog.Response.Status, " ", 2)
|
||||||
|
|
||||||
if len(statusReasonSubs) == 2 {
|
if len(statusReasonSubs) == 2 {
|
||||||
log.Response.StatusReason = statusReasonSubs[1]
|
log.Response.StatusReason = statusReasonSubs[1]
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(req.Response.Body) > 0 {
|
if len(reqLog.Response.Body) > 0 {
|
||||||
resBody := string(req.Response.Body)
|
bodyStr := string(reqLog.Response.Body)
|
||||||
log.Response.Body = &resBody
|
log.Response.Body = &bodyStr
|
||||||
}
|
}
|
||||||
|
|
||||||
if req.Response.Response.Header != nil {
|
if reqLog.Response.Header != nil {
|
||||||
log.Response.Headers = make([]HTTPHeader, 0)
|
log.Response.Headers = make([]HTTPHeader, 0)
|
||||||
|
|
||||||
for key, values := range req.Response.Response.Header {
|
for key, values := range reqLog.Response.Header {
|
||||||
for _, value := range values {
|
for _, value := range values {
|
||||||
log.Response.Headers = append(log.Response.Headers, HTTPHeader{
|
log.Response.Headers = append(log.Response.Headers, HTTPHeader{
|
||||||
Key: key,
|
Key: key,
|
||||||
@ -139,8 +138,8 @@ func parseRequestLog(req reqlog.Request) (HTTPRequestLog, error) {
|
|||||||
return log, nil
|
return log, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *mutationResolver) OpenProject(ctx context.Context, name string) (*Project, error) {
|
func (r *mutationResolver) CreateProject(ctx context.Context, name string) (*Project, error) {
|
||||||
p, err := r.ProjectService.Open(ctx, name)
|
p, err := r.ProjectService.CreateProject(ctx, name)
|
||||||
if errors.Is(err, proj.ErrInvalidName) {
|
if errors.Is(err, proj.ErrInvalidName) {
|
||||||
return nil, gqlerror.Errorf("Project name must only contain alphanumeric or space chars.")
|
return nil, gqlerror.Errorf("Project name must only contain alphanumeric or space chars.")
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
@ -148,13 +147,29 @@ func (r *mutationResolver) OpenProject(ctx context.Context, name string) (*Proje
|
|||||||
}
|
}
|
||||||
|
|
||||||
return &Project{
|
return &Project{
|
||||||
|
ID: ULID(p.ID),
|
||||||
Name: p.Name,
|
Name: p.Name,
|
||||||
IsActive: p.IsActive,
|
IsActive: r.ProjectService.IsProjectActive(p.ID),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *mutationResolver) OpenProject(ctx context.Context, id ULID) (*Project, error) {
|
||||||
|
p, err := r.ProjectService.OpenProject(ctx, ulid.ULID(id))
|
||||||
|
if errors.Is(err, proj.ErrInvalidName) {
|
||||||
|
return nil, gqlerror.Errorf("Project name must only contain alphanumeric or space chars.")
|
||||||
|
} else if err != nil {
|
||||||
|
return nil, fmt.Errorf("could not open project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Project{
|
||||||
|
ID: ULID(p.ID),
|
||||||
|
Name: p.Name,
|
||||||
|
IsActive: r.ProjectService.IsProjectActive(p.ID),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *queryResolver) ActiveProject(ctx context.Context) (*Project, error) {
|
func (r *queryResolver) ActiveProject(ctx context.Context) (*Project, error) {
|
||||||
p, err := r.ProjectService.ActiveProject()
|
p, err := r.ProjectService.ActiveProject(ctx)
|
||||||
if errors.Is(err, proj.ErrNoProject) {
|
if errors.Is(err, proj.ErrNoProject) {
|
||||||
return nil, nil
|
return nil, nil
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
@ -162,13 +177,14 @@ func (r *queryResolver) ActiveProject(ctx context.Context) (*Project, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return &Project{
|
return &Project{
|
||||||
|
ID: ULID(p.ID),
|
||||||
Name: p.Name,
|
Name: p.Name,
|
||||||
IsActive: p.IsActive,
|
IsActive: r.ProjectService.IsProjectActive(p.ID),
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *queryResolver) Projects(ctx context.Context) ([]Project, error) {
|
func (r *queryResolver) Projects(ctx context.Context) ([]Project, error) {
|
||||||
p, err := r.ProjectService.Projects()
|
p, err := r.ProjectService.Projects(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("could not get projects: %w", err)
|
return nil, fmt.Errorf("could not get projects: %w", err)
|
||||||
}
|
}
|
||||||
@ -176,8 +192,9 @@ func (r *queryResolver) Projects(ctx context.Context) ([]Project, error) {
|
|||||||
projects := make([]Project, len(p))
|
projects := make([]Project, len(p))
|
||||||
for i, proj := range p {
|
for i, proj := range p {
|
||||||
projects[i] = Project{
|
projects[i] = Project{
|
||||||
|
ID: ULID(proj.ID),
|
||||||
Name: proj.Name,
|
Name: proj.Name,
|
||||||
IsActive: proj.IsActive,
|
IsActive: r.ProjectService.IsProjectActive(proj.ID),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -185,7 +202,7 @@ func (r *queryResolver) Projects(ctx context.Context) ([]Project, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *queryResolver) Scope(ctx context.Context) ([]ScopeRule, error) {
|
func (r *queryResolver) Scope(ctx context.Context) ([]ScopeRule, error) {
|
||||||
rules := r.ScopeService.Rules()
|
rules := r.ProjectService.Scope().Rules()
|
||||||
return scopeToScopeRules(rules), nil
|
return scopeToScopeRules(rules), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -200,15 +217,15 @@ func regexpToStringPtr(r *regexp.Regexp) *string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *mutationResolver) CloseProject(ctx context.Context) (*CloseProjectResult, error) {
|
func (r *mutationResolver) CloseProject(ctx context.Context) (*CloseProjectResult, error) {
|
||||||
if err := r.ProjectService.Close(); err != nil {
|
if err := r.ProjectService.CloseProject(); err != nil {
|
||||||
return nil, fmt.Errorf("could not close project: %w", err)
|
return nil, fmt.Errorf("could not close project: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &CloseProjectResult{true}, nil
|
return &CloseProjectResult{true}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *mutationResolver) DeleteProject(ctx context.Context, name string) (*DeleteProjectResult, error) {
|
func (r *mutationResolver) DeleteProject(ctx context.Context, id ULID) (*DeleteProjectResult, error) {
|
||||||
if err := r.ProjectService.Delete(name); err != nil {
|
if err := r.ProjectService.DeleteProject(ctx, ulid.ULID(id)); err != nil {
|
||||||
return nil, fmt.Errorf("could not delete project: %w", err)
|
return nil, fmt.Errorf("could not delete project: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -218,7 +235,14 @@ func (r *mutationResolver) DeleteProject(ctx context.Context, name string) (*Del
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *mutationResolver) ClearHTTPRequestLog(ctx context.Context) (*ClearHTTPRequestLogResult, error) {
|
func (r *mutationResolver) ClearHTTPRequestLog(ctx context.Context) (*ClearHTTPRequestLogResult, error) {
|
||||||
if err := r.RequestLogService.ClearRequests(ctx); err != nil {
|
project, err := r.ProjectService.ActiveProject(ctx)
|
||||||
|
if errors.Is(err, proj.ErrNoProject) {
|
||||||
|
return nil, noActiveProjectErr(ctx)
|
||||||
|
} else if err != nil {
|
||||||
|
return nil, fmt.Errorf("could not get active project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := r.RequestLogService.ClearRequests(ctx, project.ID); err != nil {
|
||||||
return nil, fmt.Errorf("could not clear request log: %w", err)
|
return nil, fmt.Errorf("could not clear request log: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -263,8 +287,9 @@ func (r *mutationResolver) SetScope(ctx context.Context, input []ScopeRuleInput)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := r.ScopeService.SetRules(ctx, rules); err != nil {
|
err := r.ProjectService.SetScopeRules(ctx, rules)
|
||||||
return nil, fmt.Errorf("could not set scope: %w", err)
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("could not set scope rules: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return scopeToScopeRules(rules), nil
|
return scopeToScopeRules(rules), nil
|
||||||
@ -283,7 +308,7 @@ func (r *mutationResolver) SetHTTPRequestLogFilter(
|
|||||||
return nil, fmt.Errorf("could not parse request log filter: %w", err)
|
return nil, fmt.Errorf("could not parse request log filter: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = r.RequestLogService.SetRequestLogFilter(ctx, filter)
|
err = r.ProjectService.SetRequestLogFindFilter(ctx, filter)
|
||||||
if errors.Is(err, proj.ErrNoProject) {
|
if errors.Is(err, proj.ErrNoProject) {
|
||||||
return nil, noActiveProjectErr(ctx)
|
return nil, noActiveProjectErr(ctx)
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
@ -333,7 +358,6 @@ func findRequestsFilterFromInput(input *HTTPRequestLogFilterInput) (filter reqlo
|
|||||||
return reqlog.FindRequestsFilter{}, fmt.Errorf("could not parse search query: %w", err)
|
return reqlog.FindRequestsFilter{}, fmt.Errorf("could not parse search query: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
filter.RawSearchExpr = *input.SearchExpression
|
|
||||||
filter.SearchExpr = expr
|
filter.SearchExpr = expr
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -350,8 +374,9 @@ func findReqFilterToHTTPReqLogFilter(findReqFilter reqlog.FindRequestsFilter) *H
|
|||||||
OnlyInScope: findReqFilter.OnlyInScope,
|
OnlyInScope: findReqFilter.OnlyInScope,
|
||||||
}
|
}
|
||||||
|
|
||||||
if findReqFilter.RawSearchExpr != "" {
|
if findReqFilter.SearchExpr != nil {
|
||||||
httpReqLogFilter.SearchExpression = &findReqFilter.RawSearchExpr
|
searchExpr := findReqFilter.SearchExpr.String()
|
||||||
|
httpReqLogFilter.SearchExpression = &searchExpr
|
||||||
}
|
}
|
||||||
|
|
||||||
return httpReqLogFilter
|
return httpReqLogFilter
|
||||||
|
@ -10,7 +10,6 @@ type HttpRequestLog {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type HttpResponseLog {
|
type HttpResponseLog {
|
||||||
requestId: ID!
|
|
||||||
proto: String!
|
proto: String!
|
||||||
statusCode: Int!
|
statusCode: Int!
|
||||||
statusReason: String!
|
statusReason: String!
|
||||||
@ -24,6 +23,7 @@ type HttpHeader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Project {
|
type Project {
|
||||||
|
id: ID!
|
||||||
name: String!
|
name: String!
|
||||||
isActive: Boolean!
|
isActive: Boolean!
|
||||||
}
|
}
|
||||||
@ -82,9 +82,10 @@ type Query {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Mutation {
|
type Mutation {
|
||||||
openProject(name: String!): Project
|
createProject(name: String!): Project
|
||||||
|
openProject(id: ID!): Project
|
||||||
closeProject: CloseProjectResult!
|
closeProject: CloseProjectResult!
|
||||||
deleteProject(name: String!): DeleteProjectResult!
|
deleteProject(id: ID!): DeleteProjectResult!
|
||||||
clearHTTPRequestLog: ClearHTTPRequestLogResult!
|
clearHTTPRequestLog: ClearHTTPRequestLogResult!
|
||||||
setScope(scope: [ScopeRuleInput!]!): [ScopeRule!]!
|
setScope(scope: [ScopeRuleInput!]!): [ScopeRule!]!
|
||||||
setHttpRequestLogFilter(
|
setHttpRequestLogFilter(
|
||||||
|
53
pkg/db/badger/badger.go
Normal file
53
pkg/db/badger/badger.go
Normal file
@ -0,0 +1,53 @@
|
|||||||
|
package badger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/dgraph-io/badger/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Key prefixes. Each prefix value should be unique.
|
||||||
|
projectPrefix = 0x00
|
||||||
|
reqLogPrefix = 0x01
|
||||||
|
resLogPrefix = 0x02
|
||||||
|
|
||||||
|
// Request log indices.
|
||||||
|
reqLogProjectIDIndex = 0x00
|
||||||
|
)
|
||||||
|
|
||||||
|
// Database is used to store and retrieve data from an underlying Badger database.
|
||||||
|
type Database struct {
|
||||||
|
badger *badger.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenDatabase opens a new Badger database.
|
||||||
|
func OpenDatabase(opts badger.Options) (*Database, error) {
|
||||||
|
db, err := badger.Open(opts)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("badger: failed to open database: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Database{badger: db}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close closes the underlying Badger database.
|
||||||
|
func (db *Database) Close() error {
|
||||||
|
return db.badger.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// DatabaseFromBadgerDB returns a Database with `db` set as the underlying
|
||||||
|
// Badger database.
|
||||||
|
func DatabaseFromBadgerDB(db *badger.DB) *Database {
|
||||||
|
return &Database{badger: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
func entryKey(prefix, index byte, value []byte) []byte {
|
||||||
|
// Key consists of: | prefix (byte) | index (byte) | value
|
||||||
|
key := make([]byte, 2+len(value))
|
||||||
|
key[0] = prefix
|
||||||
|
key[1] = index
|
||||||
|
copy(key[2:len(value)+2], value)
|
||||||
|
|
||||||
|
return key
|
||||||
|
}
|
110
pkg/db/badger/proj.go
Normal file
110
pkg/db/badger/proj.go
Normal file
@ -0,0 +1,110 @@
|
|||||||
|
package badger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/gob"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/dgraph-io/badger/v3"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/proj"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (db *Database) UpsertProject(ctx context.Context, project proj.Project) error {
|
||||||
|
buf := bytes.Buffer{}
|
||||||
|
|
||||||
|
err := gob.NewEncoder(&buf).Encode(project)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to encode project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.badger.Update(func(txn *badger.Txn) error {
|
||||||
|
return txn.Set(entryKey(projectPrefix, 0, project.ID[:]), buf.Bytes())
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) FindProjectByID(ctx context.Context, projectID ulid.ULID) (project proj.Project, err error) {
|
||||||
|
err = db.badger.View(func(txn *badger.Txn) error {
|
||||||
|
item, err := txn.Get(entryKey(projectPrefix, 0, projectID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = item.Value(func(rawProject []byte) error {
|
||||||
|
return gob.NewDecoder(bytes.NewReader(rawProject)).Decode(&project)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to retrieve or parse project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
if errors.Is(err, badger.ErrKeyNotFound) {
|
||||||
|
return proj.Project{}, proj.ErrProjectNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return proj.Project{}, fmt.Errorf("badger: failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return project, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) DeleteProject(ctx context.Context, projectID ulid.ULID) error {
|
||||||
|
err := db.ClearRequestLogs(ctx, projectID)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to delete project request logs: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.badger.Update(func(txn *badger.Txn) error {
|
||||||
|
return txn.Delete(entryKey(projectPrefix, 0, projectID[:]))
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to delete project item: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) Projects(ctx context.Context) ([]proj.Project, error) {
|
||||||
|
projects := make([]proj.Project, 0)
|
||||||
|
|
||||||
|
err := db.badger.View(func(txn *badger.Txn) error {
|
||||||
|
var rawProject []byte
|
||||||
|
prefix := entryKey(projectPrefix, 0, nil)
|
||||||
|
|
||||||
|
iterator := txn.NewIterator(badger.DefaultIteratorOptions)
|
||||||
|
defer iterator.Close()
|
||||||
|
|
||||||
|
for iterator.Seek(prefix); iterator.ValidForPrefix(prefix); iterator.Next() {
|
||||||
|
rawProject, err := iterator.Item().ValueCopy(rawProject)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to copy value: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var project proj.Project
|
||||||
|
err = gob.NewDecoder(bytes.NewReader(rawProject)).Decode(&project)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to decode project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
projects = append(projects, project)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("badger: failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return projects, nil
|
||||||
|
}
|
284
pkg/db/badger/proj_test.go
Normal file
284
pkg/db/badger/proj_test.go
Normal file
@ -0,0 +1,284 @@
|
|||||||
|
package badger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/gob"
|
||||||
|
"errors"
|
||||||
|
"math/rand"
|
||||||
|
"regexp"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
badgerdb "github.com/dgraph-io/badger/v3"
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"github.com/google/go-cmp/cmp/cmpopts"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/proj"
|
||||||
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
|
"github.com/dstotijn/hetty/pkg/search"
|
||||||
|
)
|
||||||
|
|
||||||
|
//nolint:gosec
|
||||||
|
var ulidEntropy = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||||
|
|
||||||
|
var regexpCompareOpt = cmp.Comparer(func(x, y *regexp.Regexp) bool {
|
||||||
|
switch {
|
||||||
|
case x == nil && y == nil:
|
||||||
|
return true
|
||||||
|
case x == nil || y == nil:
|
||||||
|
return false
|
||||||
|
default:
|
||||||
|
return x.String() == y.String()
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
func TestUpsertProject(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
badgerDB, err := badgerdb.Open(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
database := DatabaseFromBadgerDB(badgerDB)
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
searchExpr, err := search.ParseQuery("foo AND bar OR NOT baz")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error (expected: nil, got: %v)", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
exp := proj.Project{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
|
Name: "foobar",
|
||||||
|
Settings: proj.Settings{
|
||||||
|
ReqLogBypassOutOfScope: true,
|
||||||
|
ReqLogOnlyFindInScope: true,
|
||||||
|
ScopeRules: []scope.Rule{
|
||||||
|
{
|
||||||
|
URL: regexp.MustCompile("^https://(.*)example.com(.*)$"),
|
||||||
|
Header: scope.Header{
|
||||||
|
Key: regexp.MustCompile("^X-Foo(.*)$"),
|
||||||
|
Value: regexp.MustCompile("^foo(.*)$"),
|
||||||
|
},
|
||||||
|
Body: regexp.MustCompile("^foo(.*)"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
SearchExpr: searchExpr,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err = database.UpsertProject(context.Background(), exp)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error storing project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var rawProject []byte
|
||||||
|
|
||||||
|
err = badgerDB.View(func(txn *badgerdb.Txn) error {
|
||||||
|
item, err := txn.Get(entryKey(projectPrefix, 0, exp.ID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
rawProject, err = item.ValueCopy(nil)
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error retrieving project from database: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
got := proj.Project{}
|
||||||
|
|
||||||
|
err = gob.NewDecoder(bytes.NewReader(rawProject)).Decode(&got)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error decoding project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(exp, got, regexpCompareOpt, cmpopts.IgnoreUnexported(proj.Project{})); diff != "" {
|
||||||
|
t.Fatalf("project not equal (-exp, +got):\n%v", diff)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFindProjectByID(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
t.Run("existing project", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
badgerDB, err := badgerdb.Open(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
database := DatabaseFromBadgerDB(badgerDB)
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
exp := proj.Project{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
|
Name: "foobar",
|
||||||
|
Settings: proj.Settings{},
|
||||||
|
}
|
||||||
|
|
||||||
|
buf := bytes.Buffer{}
|
||||||
|
|
||||||
|
err = gob.NewEncoder(&buf).Encode(exp)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error encoding project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = badgerDB.Update(func(txn *badgerdb.Txn) error {
|
||||||
|
return txn.Set(entryKey(projectPrefix, 0, exp.ID[:]), buf.Bytes())
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error setting project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := database.FindProjectByID(context.Background(), exp.ID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error finding project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(exp, got, cmpopts.IgnoreUnexported(proj.Project{})); diff != "" {
|
||||||
|
t.Fatalf("project not equal (-exp, +got):\n%v", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("project not found", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
database, err := OpenDatabase(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
projectID := ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
|
||||||
|
_, err = database.FindProjectByID(context.Background(), projectID)
|
||||||
|
if !errors.Is(err, proj.ErrProjectNotFound) {
|
||||||
|
t.Fatalf("expected `proj.ErrProjectNotFound`, got: %v", err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteProject(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
badgerDB, err := badgerdb.Open(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
database := DatabaseFromBadgerDB(badgerDB)
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
// Store fixtures.
|
||||||
|
projectID := ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
reqLogID := ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
|
||||||
|
err = badgerDB.Update(func(txn *badgerdb.Txn) error {
|
||||||
|
if err := txn.Set(entryKey(projectPrefix, 0, projectID[:]), nil); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := txn.Set(entryKey(reqLogPrefix, 0, reqLogID[:]), nil); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := txn.Set(entryKey(resLogPrefix, 0, reqLogID[:]), nil); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err := txn.Set(entryKey(reqLogPrefix, reqLogProjectIDIndex, append(projectID[:], reqLogID[:]...)), nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error creating fixtures: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = database.DeleteProject(context.Background(), projectID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error deleting project: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assert project key was deleted.
|
||||||
|
err = badgerDB.View(func(txn *badgerdb.Txn) error {
|
||||||
|
_, err := txn.Get(entryKey(projectPrefix, 0, projectID[:]))
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if !errors.Is(err, badgerdb.ErrKeyNotFound) {
|
||||||
|
t.Fatalf("expected `badger.ErrKeyNotFound`, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assert request log item was deleted.
|
||||||
|
err = badgerDB.View(func(txn *badgerdb.Txn) error {
|
||||||
|
_, err := txn.Get(entryKey(reqLogPrefix, 0, reqLogID[:]))
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if !errors.Is(err, badgerdb.ErrKeyNotFound) {
|
||||||
|
t.Fatalf("expected `badger.ErrKeyNotFound`, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assert response log item was deleted.
|
||||||
|
err = badgerDB.View(func(txn *badgerdb.Txn) error {
|
||||||
|
_, err := txn.Get(entryKey(resLogPrefix, 0, reqLogID[:]))
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if !errors.Is(err, badgerdb.ErrKeyNotFound) {
|
||||||
|
t.Fatalf("expected `badger.ErrKeyNotFound`, got: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assert request log project ID index key was deleted.
|
||||||
|
err = badgerDB.View(func(txn *badgerdb.Txn) error {
|
||||||
|
_, err := txn.Get(entryKey(reqLogPrefix, reqLogProjectIDIndex, append(projectID[:], reqLogID[:]...)))
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if !errors.Is(err, badgerdb.ErrKeyNotFound) {
|
||||||
|
t.Fatalf("expected `badger.ErrKeyNotFound`, got: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestProjects(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
database, err := OpenDatabase(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
exp := []proj.Project{
|
||||||
|
{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
|
Name: "one",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now())+100, ulidEntropy),
|
||||||
|
Name: "two",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store fixtures.
|
||||||
|
for _, project := range exp {
|
||||||
|
err = database.UpsertProject(context.Background(), project)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error creating project fixture: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := database.Projects(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error finding projects: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(exp) != len(got) {
|
||||||
|
t.Fatalf("expected %v projects, got: %v", len(exp), len(got))
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(exp, got, cmpopts.IgnoreUnexported(proj.Project{})); diff != "" {
|
||||||
|
t.Fatalf("projects not equal (-exp, +got):\n%v", diff)
|
||||||
|
}
|
||||||
|
}
|
251
pkg/db/badger/reqlog.go
Normal file
251
pkg/db/badger/reqlog.go
Normal file
@ -0,0 +1,251 @@
|
|||||||
|
package badger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/gob"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/dgraph-io/badger/v3"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (db *Database) FindRequestLogs(ctx context.Context, filter reqlog.FindRequestsFilter, scope *scope.Scope) ([]reqlog.RequestLog, error) {
|
||||||
|
if filter.ProjectID.Compare(ulid.ULID{}) == 0 {
|
||||||
|
return nil, reqlog.ErrProjectIDMustBeSet
|
||||||
|
}
|
||||||
|
|
||||||
|
txn := db.badger.NewTransaction(false)
|
||||||
|
defer txn.Discard()
|
||||||
|
|
||||||
|
reqLogIDs, err := findRequestLogIDsByProjectID(txn, filter.ProjectID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("badger: failed to find request log IDs: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
reqLogs := make([]reqlog.RequestLog, 0, len(reqLogIDs))
|
||||||
|
|
||||||
|
for _, reqLogID := range reqLogIDs {
|
||||||
|
reqLog, err := getRequestLogWithResponse(txn, reqLogID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("badger: failed to get request log (id: %v): %w", reqLogID.String(), err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if filter.OnlyInScope {
|
||||||
|
if !reqLog.MatchScope(scope) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Filter by search expression.
|
||||||
|
// TODO: Once pagination is introduced, this filter logic should be done
|
||||||
|
// as items are retrieved (e.g. when using a `badger.Iterator`).
|
||||||
|
if filter.SearchExpr != nil {
|
||||||
|
match, err := reqLog.Matches(filter.SearchExpr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf(
|
||||||
|
"badger: failed to match search expression for request log (id: %v): %w",
|
||||||
|
reqLogID.String(), err,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !match {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
reqLogs = append(reqLogs, reqLog)
|
||||||
|
}
|
||||||
|
|
||||||
|
return reqLogs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getRequestLogWithResponse(txn *badger.Txn, reqLogID ulid.ULID) (reqlog.RequestLog, error) {
|
||||||
|
item, err := txn.Get(entryKey(reqLogPrefix, 0, reqLogID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return reqlog.RequestLog{}, fmt.Errorf("failed to lookup request log item: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
reqLog := reqlog.RequestLog{
|
||||||
|
ID: reqLogID,
|
||||||
|
}
|
||||||
|
|
||||||
|
err = item.Value(func(rawReqLog []byte) error {
|
||||||
|
err = gob.NewDecoder(bytes.NewReader(rawReqLog)).Decode(&reqLog)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to decode request log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return reqlog.RequestLog{}, fmt.Errorf("failed to retrieve or parse request log value: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
item, err = txn.Get(entryKey(resLogPrefix, 0, reqLogID[:]))
|
||||||
|
|
||||||
|
if errors.Is(err, badger.ErrKeyNotFound) {
|
||||||
|
return reqLog, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return reqlog.RequestLog{}, fmt.Errorf("failed to get response log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = item.Value(func(rawReslog []byte) error {
|
||||||
|
var resLog reqlog.ResponseLog
|
||||||
|
err = gob.NewDecoder(bytes.NewReader(rawReslog)).Decode(&resLog)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to decode response log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
reqLog.Response = &resLog
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return reqlog.RequestLog{}, fmt.Errorf("failed to retrieve or parse response log value: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return reqLog, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) FindRequestLogByID(ctx context.Context, reqLogID ulid.ULID) (reqLog reqlog.RequestLog, err error) {
|
||||||
|
txn := db.badger.NewTransaction(false)
|
||||||
|
defer txn.Discard()
|
||||||
|
|
||||||
|
reqLog, err = getRequestLogWithResponse(txn, reqLogID)
|
||||||
|
if err != nil {
|
||||||
|
return reqlog.RequestLog{}, fmt.Errorf("badger: failed to get request log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return reqLog, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) StoreRequestLog(ctx context.Context, reqLog reqlog.RequestLog) error {
|
||||||
|
buf := bytes.Buffer{}
|
||||||
|
|
||||||
|
err := gob.NewEncoder(&buf).Encode(reqLog)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to encode request log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
entries := []*badger.Entry{
|
||||||
|
// Request log itself.
|
||||||
|
{
|
||||||
|
Key: entryKey(reqLogPrefix, 0, reqLog.ID[:]),
|
||||||
|
Value: buf.Bytes(),
|
||||||
|
},
|
||||||
|
// Index by project ID.
|
||||||
|
{
|
||||||
|
Key: entryKey(reqLogPrefix, reqLogProjectIDIndex, append(reqLog.ProjectID[:], reqLog.ID[:]...)),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.badger.Update(func(txn *badger.Txn) error {
|
||||||
|
for i := range entries {
|
||||||
|
err := txn.SetEntry(entries[i])
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) StoreResponseLog(ctx context.Context, reqLogID ulid.ULID, resLog reqlog.ResponseLog) error {
|
||||||
|
buf := bytes.Buffer{}
|
||||||
|
|
||||||
|
err := gob.NewEncoder(&buf).Encode(resLog)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to encode response log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.badger.Update(func(txn *badger.Txn) error {
|
||||||
|
return txn.SetEntry(&badger.Entry{
|
||||||
|
Key: entryKey(resLogPrefix, 0, reqLogID[:]),
|
||||||
|
Value: buf.Bytes(),
|
||||||
|
})
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to commit transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (db *Database) ClearRequestLogs(ctx context.Context, projectID ulid.ULID) error {
|
||||||
|
// Note: this transaction is used just for reading; we use the `badger.WriteBatch`
|
||||||
|
// API to bulk delete items.
|
||||||
|
txn := db.badger.NewTransaction(false)
|
||||||
|
defer txn.Discard()
|
||||||
|
|
||||||
|
reqLogIDs, err := findRequestLogIDsByProjectID(txn, projectID)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to find request log IDs: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
writeBatch := db.badger.NewWriteBatch()
|
||||||
|
defer writeBatch.Cancel()
|
||||||
|
|
||||||
|
for _, reqLogID := range reqLogIDs {
|
||||||
|
// Delete request logs.
|
||||||
|
err := writeBatch.Delete(entryKey(reqLogPrefix, 0, reqLogID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to delete request log: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete related response log.
|
||||||
|
err = writeBatch.Delete(entryKey(resLogPrefix, 0, reqLogID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to delete request log: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := writeBatch.Flush(); err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to commit batch write: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = db.badger.DropPrefix(entryKey(reqLogPrefix, reqLogProjectIDIndex, projectID[:]))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("badger: failed to drop request log project ID index items: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func findRequestLogIDsByProjectID(txn *badger.Txn, projectID ulid.ULID) ([]ulid.ULID, error) {
|
||||||
|
reqLogIDs := make([]ulid.ULID, 0)
|
||||||
|
opts := badger.DefaultIteratorOptions
|
||||||
|
opts.PrefetchValues = false
|
||||||
|
iterator := txn.NewIterator(opts)
|
||||||
|
defer iterator.Close()
|
||||||
|
|
||||||
|
var projectIndexKey []byte
|
||||||
|
|
||||||
|
prefix := entryKey(reqLogPrefix, reqLogProjectIDIndex, projectID[:])
|
||||||
|
|
||||||
|
for iterator.Seek(prefix); iterator.ValidForPrefix(prefix); iterator.Next() {
|
||||||
|
projectIndexKey = iterator.Item().KeyCopy(projectIndexKey)
|
||||||
|
|
||||||
|
var id ulid.ULID
|
||||||
|
// The request log ID starts *after* the first 2 prefix and index bytes
|
||||||
|
// and the 16 byte project ID.
|
||||||
|
if err := id.UnmarshalBinary(projectIndexKey[18:]); err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to parse request log ID: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
reqLogIDs = append(reqLogIDs, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
return reqLogIDs, nil
|
||||||
|
}
|
121
pkg/db/badger/reqlog_test.go
Normal file
121
pkg/db/badger/reqlog_test.go
Normal file
@ -0,0 +1,121 @@
|
|||||||
|
package badger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
badgerdb "github.com/dgraph-io/badger/v3"
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestFindRequestLogs(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
t.Run("without project ID in filter", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
database, err := OpenDatabase(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
filter := reqlog.FindRequestsFilter{}
|
||||||
|
|
||||||
|
_, err = database.FindRequestLogs(context.Background(), filter, nil)
|
||||||
|
if !errors.Is(err, reqlog.ErrProjectIDMustBeSet) {
|
||||||
|
t.Fatalf("expected `reqlog.ErrProjectIDMustBeSet`, got: %v", err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("returns request logs and related response logs", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
database, err := OpenDatabase(badgerdb.DefaultOptions("").WithInMemory(true))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("failed to open badger database: %v", err)
|
||||||
|
}
|
||||||
|
defer database.Close()
|
||||||
|
|
||||||
|
projectID := ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
|
||||||
|
exp := []reqlog.RequestLog{
|
||||||
|
{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
|
ProjectID: projectID,
|
||||||
|
URL: mustParseURL(t, "https://example.com/foobar"),
|
||||||
|
Method: http.MethodPost,
|
||||||
|
Proto: "HTTP/1.1",
|
||||||
|
Header: http.Header{
|
||||||
|
"X-Foo": []string{"baz"},
|
||||||
|
},
|
||||||
|
Body: []byte("foo"),
|
||||||
|
Response: &reqlog.ResponseLog{
|
||||||
|
Proto: "HTTP/1.1",
|
||||||
|
Status: "200 OK",
|
||||||
|
StatusCode: 200,
|
||||||
|
Header: http.Header{
|
||||||
|
"X-Yolo": []string{"swag"},
|
||||||
|
},
|
||||||
|
Body: []byte("bar"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now())+100, ulidEntropy),
|
||||||
|
ProjectID: projectID,
|
||||||
|
URL: mustParseURL(t, "https://example.com/foo?bar=baz"),
|
||||||
|
Method: http.MethodGet,
|
||||||
|
Proto: "HTTP/1.1",
|
||||||
|
Header: http.Header{
|
||||||
|
"X-Foo": []string{"baz"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store fixtures.
|
||||||
|
for _, reqLog := range exp {
|
||||||
|
err = database.StoreRequestLog(context.Background(), reqLog)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error creating request log fixture: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if reqLog.Response != nil {
|
||||||
|
err = database.StoreResponseLog(context.Background(), reqLog.ID, *reqLog.Response)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error creating response log fixture: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
filter := reqlog.FindRequestsFilter{
|
||||||
|
ProjectID: projectID,
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := database.FindRequestLogs(context.Background(), filter, nil)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error finding request logs: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if diff := cmp.Diff(exp, got); diff != "" {
|
||||||
|
t.Fatalf("request logs not equal (-exp, +got):\n%v", diff)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func mustParseURL(t *testing.T, s string) *url.URL {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
u, err := url.Parse(s)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return u
|
||||||
|
}
|
@ -1,82 +0,0 @@
|
|||||||
package sqlite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"database/sql"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"net/http"
|
|
||||||
"net/url"
|
|
||||||
"strconv"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/reqlog"
|
|
||||||
)
|
|
||||||
|
|
||||||
type reqURL url.URL
|
|
||||||
|
|
||||||
type httpRequest struct {
|
|
||||||
ID int64 `db:"req_id"`
|
|
||||||
Proto string `db:"req_proto"`
|
|
||||||
URL reqURL `db:"url"`
|
|
||||||
Method string `db:"method"`
|
|
||||||
Body []byte `db:"req_body"`
|
|
||||||
Timestamp time.Time `db:"req_timestamp"`
|
|
||||||
httpResponse
|
|
||||||
}
|
|
||||||
|
|
||||||
type httpResponse struct {
|
|
||||||
ID sql.NullInt64 `db:"res_id"`
|
|
||||||
RequestID sql.NullInt64 `db:"res_req_id"`
|
|
||||||
Proto sql.NullString `db:"res_proto"`
|
|
||||||
StatusCode sql.NullInt64 `db:"status_code"`
|
|
||||||
StatusReason sql.NullString `db:"status_reason"`
|
|
||||||
Body []byte `db:"res_body"`
|
|
||||||
Timestamp sql.NullTime `db:"res_timestamp"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Value implements driver.Valuer.
|
|
||||||
func (u *reqURL) Scan(value interface{}) error {
|
|
||||||
rawURL, ok := value.(string)
|
|
||||||
if !ok {
|
|
||||||
return errors.New("sqlite: cannot scan non-string value")
|
|
||||||
}
|
|
||||||
|
|
||||||
parsed, err := url.Parse(rawURL)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not parse URL: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
*u = reqURL(*parsed)
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (dto httpRequest) toRequestLog() reqlog.Request {
|
|
||||||
u := url.URL(dto.URL)
|
|
||||||
reqLog := reqlog.Request{
|
|
||||||
ID: dto.ID,
|
|
||||||
Request: http.Request{
|
|
||||||
Proto: dto.Proto,
|
|
||||||
Method: dto.Method,
|
|
||||||
URL: &u,
|
|
||||||
},
|
|
||||||
Body: dto.Body,
|
|
||||||
Timestamp: dto.Timestamp,
|
|
||||||
}
|
|
||||||
|
|
||||||
if dto.httpResponse.ID.Valid {
|
|
||||||
reqLog.Response = &reqlog.Response{
|
|
||||||
ID: dto.httpResponse.ID.Int64,
|
|
||||||
RequestID: dto.httpResponse.RequestID.Int64,
|
|
||||||
Response: http.Response{
|
|
||||||
Status: strconv.FormatInt(dto.StatusCode.Int64, 10) + " " + dto.StatusReason.String,
|
|
||||||
StatusCode: int(dto.StatusCode.Int64),
|
|
||||||
Proto: dto.httpResponse.Proto.String,
|
|
||||||
},
|
|
||||||
Body: dto.httpResponse.Body,
|
|
||||||
Timestamp: dto.httpResponse.Timestamp.Time,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return reqLog
|
|
||||||
}
|
|
@ -1,135 +0,0 @@
|
|||||||
package sqlite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"sort"
|
|
||||||
|
|
||||||
sq "github.com/Masterminds/squirrel"
|
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/search"
|
|
||||||
)
|
|
||||||
|
|
||||||
var stringLiteralMap = map[string]string{
|
|
||||||
// http_requests
|
|
||||||
"req.id": "req.id",
|
|
||||||
"req.proto": "req.proto",
|
|
||||||
"req.url": "req.url",
|
|
||||||
"req.method": "req.method",
|
|
||||||
"req.body": "req.body",
|
|
||||||
"req.timestamp": "req.timestamp",
|
|
||||||
// http_responses
|
|
||||||
"res.id": "res.id",
|
|
||||||
"res.proto": "res.proto",
|
|
||||||
"res.statusCode": "res.status_code",
|
|
||||||
"res.statusReason": "res.status_reason",
|
|
||||||
"res.body": "res.body",
|
|
||||||
"res.timestamp": "res.timestamp",
|
|
||||||
// TODO: http_headers
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseSearchExpr(expr search.Expression) (sq.Sqlizer, error) {
|
|
||||||
switch e := expr.(type) {
|
|
||||||
case *search.PrefixExpression:
|
|
||||||
return parsePrefixExpr(e)
|
|
||||||
case *search.InfixExpression:
|
|
||||||
return parseInfixExpr(e)
|
|
||||||
case *search.StringLiteral:
|
|
||||||
return parseStringLiteral(e)
|
|
||||||
default:
|
|
||||||
return nil, fmt.Errorf("expression type (%v) not supported", expr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func parsePrefixExpr(expr *search.PrefixExpression) (sq.Sqlizer, error) {
|
|
||||||
switch expr.Operator {
|
|
||||||
case search.TokOpNot:
|
|
||||||
// TODO: Find a way to prefix an `sq.Sqlizer` with "NOT".
|
|
||||||
return nil, errors.New("not implemented")
|
|
||||||
default:
|
|
||||||
return nil, errors.New("operator is not supported")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseInfixExpr(expr *search.InfixExpression) (sq.Sqlizer, error) {
|
|
||||||
switch expr.Operator {
|
|
||||||
case search.TokOpAnd:
|
|
||||||
left, err := parseSearchExpr(expr.Left)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
right, err := parseSearchExpr(expr.Right)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return sq.And{left, right}, nil
|
|
||||||
case search.TokOpOr:
|
|
||||||
left, err := parseSearchExpr(expr.Left)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
right, err := parseSearchExpr(expr.Right)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return sq.Or{left, right}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
left, ok := expr.Left.(*search.StringLiteral)
|
|
||||||
if !ok {
|
|
||||||
return nil, errors.New("left operand must be a string literal")
|
|
||||||
}
|
|
||||||
|
|
||||||
right, ok := expr.Right.(*search.StringLiteral)
|
|
||||||
if !ok {
|
|
||||||
return nil, errors.New("right operand must be a string literal")
|
|
||||||
}
|
|
||||||
|
|
||||||
mappedLeft, ok := stringLiteralMap[left.Value]
|
|
||||||
if !ok {
|
|
||||||
return nil, fmt.Errorf("invalid string literal: %v", left)
|
|
||||||
}
|
|
||||||
|
|
||||||
switch expr.Operator {
|
|
||||||
case search.TokOpEq:
|
|
||||||
return sq.Eq{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpNotEq:
|
|
||||||
return sq.NotEq{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpGt:
|
|
||||||
return sq.Gt{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpLt:
|
|
||||||
return sq.Lt{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpGtEq:
|
|
||||||
return sq.GtOrEq{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpLtEq:
|
|
||||||
return sq.LtOrEq{mappedLeft: right.Value}, nil
|
|
||||||
case search.TokOpRe:
|
|
||||||
return sq.Expr(fmt.Sprintf("regexp(?, %v)", mappedLeft), right.Value), nil
|
|
||||||
case search.TokOpNotRe:
|
|
||||||
return sq.Expr(fmt.Sprintf("NOT regexp(?, %v)", mappedLeft), right.Value), nil
|
|
||||||
default:
|
|
||||||
return nil, errors.New("unsupported operator")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseStringLiteral(strLiteral *search.StringLiteral) (sq.Sqlizer, error) {
|
|
||||||
// Sorting is not necessary, but makes it easier to do assertions in tests.
|
|
||||||
sortedKeys := make([]string, 0, len(stringLiteralMap))
|
|
||||||
|
|
||||||
for _, v := range stringLiteralMap {
|
|
||||||
sortedKeys = append(sortedKeys, v)
|
|
||||||
}
|
|
||||||
|
|
||||||
sort.Strings(sortedKeys)
|
|
||||||
|
|
||||||
or := make(sq.Or, len(stringLiteralMap))
|
|
||||||
for i, value := range sortedKeys {
|
|
||||||
or[i] = sq.Like{value: "%" + strLiteral.Value + "%"}
|
|
||||||
}
|
|
||||||
|
|
||||||
return or, nil
|
|
||||||
}
|
|
@ -1,221 +0,0 @@
|
|||||||
package sqlite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"reflect"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
sq "github.com/Masterminds/squirrel"
|
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/search"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestParseSearchExpr(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
tests := []struct {
|
|
||||||
name string
|
|
||||||
searchExpr search.Expression
|
|
||||||
expectedSqlizer sq.Sqlizer
|
|
||||||
expectedError error
|
|
||||||
}{
|
|
||||||
{
|
|
||||||
name: "req.body = bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Eq{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body != bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpNotEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.NotEq{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body > bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpGt,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Gt{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body < bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpLt,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Lt{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body >= bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpGtEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.GtOrEq{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body <= bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpLtEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.LtOrEq{"req.body": "bar"},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body =~ bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpRe,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Expr("regexp(?, req.body)", "bar"),
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body !~ bar",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpNotRe,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Expr("NOT regexp(?, req.body)", "bar"),
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body = bar AND res.body = yolo",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpAnd,
|
|
||||||
Left: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
Right: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "res.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "yolo"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.And{
|
|
||||||
sq.Eq{"req.body": "bar"},
|
|
||||||
sq.Eq{"res.body": "yolo"},
|
|
||||||
},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body = bar AND res.body = yolo AND req.method = POST",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpAnd,
|
|
||||||
Left: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
Right: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpAnd,
|
|
||||||
Left: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "res.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "yolo"},
|
|
||||||
},
|
|
||||||
Right: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.method"},
|
|
||||||
Right: &search.StringLiteral{Value: "POST"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.And{
|
|
||||||
sq.Eq{"req.body": "bar"},
|
|
||||||
sq.And{
|
|
||||||
sq.Eq{"res.body": "yolo"},
|
|
||||||
sq.Eq{"req.method": "POST"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "req.body = bar OR res.body = yolo",
|
|
||||||
searchExpr: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpOr,
|
|
||||||
Left: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "req.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "bar"},
|
|
||||||
},
|
|
||||||
Right: &search.InfixExpression{
|
|
||||||
Operator: search.TokOpEq,
|
|
||||||
Left: &search.StringLiteral{Value: "res.body"},
|
|
||||||
Right: &search.StringLiteral{Value: "yolo"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Or{
|
|
||||||
sq.Eq{"req.body": "bar"},
|
|
||||||
sq.Eq{"res.body": "yolo"},
|
|
||||||
},
|
|
||||||
expectedError: nil,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
name: "foo",
|
|
||||||
searchExpr: &search.StringLiteral{
|
|
||||||
Value: "foo",
|
|
||||||
},
|
|
||||||
expectedSqlizer: sq.Or{
|
|
||||||
sq.Like{"req.body": "%foo%"},
|
|
||||||
sq.Like{"req.id": "%foo%"},
|
|
||||||
sq.Like{"req.method": "%foo%"},
|
|
||||||
sq.Like{"req.proto": "%foo%"},
|
|
||||||
sq.Like{"req.timestamp": "%foo%"},
|
|
||||||
sq.Like{"req.url": "%foo%"},
|
|
||||||
sq.Like{"res.body": "%foo%"},
|
|
||||||
sq.Like{"res.id": "%foo%"},
|
|
||||||
sq.Like{"res.proto": "%foo%"},
|
|
||||||
sq.Like{"res.status_code": "%foo%"},
|
|
||||||
sq.Like{"res.status_reason": "%foo%"},
|
|
||||||
sq.Like{"res.timestamp": "%foo%"},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, tt := range tests {
|
|
||||||
tt := tt
|
|
||||||
t.Run(tt.name, func(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
got, err := parseSearchExpr(tt.searchExpr)
|
|
||||||
assertError(t, tt.expectedError, err)
|
|
||||||
if !reflect.DeepEqual(tt.expectedSqlizer, got) {
|
|
||||||
t.Errorf("expected: %#v, got: %#v", tt.expectedSqlizer, got)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func assertError(t *testing.T, exp, got error) {
|
|
||||||
t.Helper()
|
|
||||||
|
|
||||||
switch {
|
|
||||||
case exp == nil && got != nil:
|
|
||||||
t.Fatalf("expected: nil, got: %v", got)
|
|
||||||
case exp != nil && got == nil:
|
|
||||||
t.Fatalf("expected: %v, got: nil", exp.Error())
|
|
||||||
case exp != nil && got != nil && exp.Error() != got.Error():
|
|
||||||
t.Fatalf("expected: %v, got: %v", exp.Error(), got.Error())
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,709 +0,0 @@
|
|||||||
package sqlite
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"database/sql"
|
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"net/http"
|
|
||||||
"net/url"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"regexp"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/99designs/gqlgen/graphql"
|
|
||||||
sq "github.com/Masterminds/squirrel"
|
|
||||||
"github.com/jmoiron/sqlx"
|
|
||||||
"github.com/mattn/go-sqlite3"
|
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
|
||||||
"github.com/dstotijn/hetty/pkg/reqlog"
|
|
||||||
"github.com/dstotijn/hetty/pkg/scope"
|
|
||||||
)
|
|
||||||
|
|
||||||
var regexpFn = func(pattern string, value interface{}) (bool, error) {
|
|
||||||
switch v := value.(type) {
|
|
||||||
case string:
|
|
||||||
return regexp.MatchString(pattern, v)
|
|
||||||
case int64:
|
|
||||||
return regexp.MatchString(pattern, fmt.Sprintf("%v", v))
|
|
||||||
case []byte:
|
|
||||||
return regexp.Match(pattern, v)
|
|
||||||
default:
|
|
||||||
return false, fmt.Errorf("unsupported type %T", v)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Client implements reqlog.Repository.
|
|
||||||
type Client struct {
|
|
||||||
db *sqlx.DB
|
|
||||||
dbPath string
|
|
||||||
activeProject string
|
|
||||||
}
|
|
||||||
|
|
||||||
type httpRequestLogsQuery struct {
|
|
||||||
requestCols []string
|
|
||||||
requestHeaderCols []string
|
|
||||||
responseHeaderCols []string
|
|
||||||
joinResponse bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
sql.Register("sqlite3_with_regexp", &sqlite3.SQLiteDriver{
|
|
||||||
ConnectHook: func(conn *sqlite3.SQLiteConn) error {
|
|
||||||
return conn.RegisterFunc("regexp", regexpFn, false)
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func New(dbPath string) (*Client, error) {
|
|
||||||
if _, err := os.Stat(dbPath); os.IsNotExist(err) {
|
|
||||||
if err := os.MkdirAll(dbPath, 0755); err != nil {
|
|
||||||
return nil, fmt.Errorf("proj: could not create project directory: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return &Client{
|
|
||||||
dbPath: dbPath,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// OpenProject opens a project database.
|
|
||||||
func (c *Client) OpenProject(name string) error {
|
|
||||||
if c.db != nil {
|
|
||||||
return errors.New("sqlite: there is already a project open")
|
|
||||||
}
|
|
||||||
|
|
||||||
opts := make(url.Values)
|
|
||||||
opts.Set("_foreign_keys", "1")
|
|
||||||
|
|
||||||
dbPath := filepath.Join(c.dbPath, name+".db")
|
|
||||||
dsn := fmt.Sprintf("file:%v?%v", dbPath, opts.Encode())
|
|
||||||
|
|
||||||
db, err := sqlx.Open("sqlite3_with_regexp", dsn)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not open database: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := db.Ping(); err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not ping database: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := prepareSchema(db); err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not prepare schema: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
c.db = db
|
|
||||||
c.activeProject = name
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) Projects() ([]proj.Project, error) {
|
|
||||||
files, err := ioutil.ReadDir(c.dbPath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not read projects directory: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
projects := make([]proj.Project, len(files))
|
|
||||||
|
|
||||||
for i, file := range files {
|
|
||||||
projName := strings.TrimSuffix(file.Name(), ".db")
|
|
||||||
projects[i] = proj.Project{
|
|
||||||
Name: projName,
|
|
||||||
IsActive: c.activeProject == projName,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return projects, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func prepareSchema(db *sqlx.DB) error {
|
|
||||||
_, err := db.Exec(`CREATE TABLE IF NOT EXISTS http_requests (
|
|
||||||
id INTEGER PRIMARY KEY,
|
|
||||||
proto TEXT,
|
|
||||||
url TEXT,
|
|
||||||
method TEXT,
|
|
||||||
body BLOB,
|
|
||||||
timestamp DATETIME
|
|
||||||
)`)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not create http_requests table: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = db.Exec(`CREATE TABLE IF NOT EXISTS http_responses (
|
|
||||||
id INTEGER PRIMARY KEY,
|
|
||||||
req_id INTEGER REFERENCES http_requests(id) ON DELETE CASCADE,
|
|
||||||
proto TEXT,
|
|
||||||
status_code INTEGER,
|
|
||||||
status_reason TEXT,
|
|
||||||
body BLOB,
|
|
||||||
timestamp DATETIME
|
|
||||||
)`)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not create http_responses table: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = db.Exec(`CREATE TABLE IF NOT EXISTS http_headers (
|
|
||||||
id INTEGER PRIMARY KEY,
|
|
||||||
req_id INTEGER REFERENCES http_requests(id) ON DELETE CASCADE,
|
|
||||||
res_id INTEGER REFERENCES http_responses(id) ON DELETE CASCADE,
|
|
||||||
key TEXT,
|
|
||||||
value TEXT
|
|
||||||
)`)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not create http_headers table: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = db.Exec(`CREATE TABLE IF NOT EXISTS settings (
|
|
||||||
module TEXT PRIMARY KEY,
|
|
||||||
settings TEXT
|
|
||||||
)`)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not create settings table: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close uses the underlying database if it's open.
|
|
||||||
func (c *Client) Close() error {
|
|
||||||
if c.db == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := c.db.Close(); err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not close database: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
c.db = nil
|
|
||||||
c.activeProject = ""
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) DeleteProject(name string) error {
|
|
||||||
if err := os.Remove(filepath.Join(c.dbPath, name+".db")); err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not remove database file: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var reqFieldToColumnMap = map[string]string{
|
|
||||||
"proto": "proto AS req_proto",
|
|
||||||
"url": "url",
|
|
||||||
"method": "method",
|
|
||||||
"body": "body AS req_body",
|
|
||||||
"timestamp": "timestamp AS req_timestamp",
|
|
||||||
}
|
|
||||||
|
|
||||||
var resFieldToColumnMap = map[string]string{
|
|
||||||
"requestId": "req_id AS res_req_id",
|
|
||||||
"proto": "proto AS res_proto",
|
|
||||||
"statusCode": "status_code",
|
|
||||||
"statusReason": "status_reason",
|
|
||||||
"body": "body AS res_body",
|
|
||||||
"timestamp": "timestamp AS res_timestamp",
|
|
||||||
}
|
|
||||||
|
|
||||||
var headerFieldToColumnMap = map[string]string{
|
|
||||||
"key": "key",
|
|
||||||
"value": "value",
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) ClearRequestLogs(ctx context.Context) error {
|
|
||||||
if c.db == nil {
|
|
||||||
return proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := c.db.Exec("DELETE FROM http_requests")
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not delete requests: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) FindRequestLogs(
|
|
||||||
ctx context.Context,
|
|
||||||
filter reqlog.FindRequestsFilter,
|
|
||||||
scope *scope.Scope,
|
|
||||||
) (reqLogs []reqlog.Request, err error) {
|
|
||||||
if c.db == nil {
|
|
||||||
return nil, proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
httpReqLogsQuery := parseHTTPRequestLogsQuery(ctx)
|
|
||||||
|
|
||||||
reqQuery := sq.
|
|
||||||
Select(httpReqLogsQuery.requestCols...).
|
|
||||||
From("http_requests req").
|
|
||||||
OrderBy("req.id DESC")
|
|
||||||
if httpReqLogsQuery.joinResponse {
|
|
||||||
reqQuery = reqQuery.LeftJoin("http_responses res ON req.id = res.req_id")
|
|
||||||
}
|
|
||||||
|
|
||||||
if filter.OnlyInScope && scope != nil {
|
|
||||||
var ruleExpr []sq.Sqlizer
|
|
||||||
|
|
||||||
for _, rule := range scope.Rules() {
|
|
||||||
if rule.URL != nil {
|
|
||||||
ruleExpr = append(ruleExpr, sq.Expr("regexp(?, req.url)", rule.URL.String()))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(ruleExpr) > 0 {
|
|
||||||
reqQuery = reqQuery.Where(sq.Or(ruleExpr))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if filter.SearchExpr != nil {
|
|
||||||
sqlizer, err := parseSearchExpr(filter.SearchExpr)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not parse search expression: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqQuery = reqQuery.Where(sqlizer)
|
|
||||||
}
|
|
||||||
|
|
||||||
sql, args, err := reqQuery.ToSql()
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not parse query: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
rows, err := c.db.QueryxContext(ctx, sql, args...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not execute query: %w", err)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
for rows.Next() {
|
|
||||||
var dto httpRequest
|
|
||||||
|
|
||||||
err = rows.StructScan(&dto)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not scan row: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLogs = append(reqLogs, dto.toRequestLog())
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := rows.Err(); err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not iterate over rows: %w", err)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
if err := c.queryHeaders(ctx, httpReqLogsQuery, reqLogs); err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not query headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return reqLogs, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) FindRequestLogByID(ctx context.Context, id int64) (reqlog.Request, error) {
|
|
||||||
if c.db == nil {
|
|
||||||
return reqlog.Request{}, proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
httpReqLogsQuery := parseHTTPRequestLogsQuery(ctx)
|
|
||||||
reqQuery := sq.
|
|
||||||
Select(httpReqLogsQuery.requestCols...).
|
|
||||||
From("http_requests req").
|
|
||||||
Where("req.id = ?")
|
|
||||||
|
|
||||||
if httpReqLogsQuery.joinResponse {
|
|
||||||
reqQuery = reqQuery.LeftJoin("http_responses res ON req.id = res.req_id")
|
|
||||||
}
|
|
||||||
|
|
||||||
reqSQL, _, err := reqQuery.ToSql()
|
|
||||||
if err != nil {
|
|
||||||
return reqlog.Request{}, fmt.Errorf("sqlite: could not parse query: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
row := c.db.QueryRowxContext(ctx, reqSQL, id)
|
|
||||||
|
|
||||||
var dto httpRequest
|
|
||||||
|
|
||||||
err = row.StructScan(&dto)
|
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
|
||||||
return reqlog.Request{}, reqlog.ErrRequestNotFound
|
|
||||||
} else if err != nil {
|
|
||||||
return reqlog.Request{}, fmt.Errorf("sqlite: could not scan row: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLog := dto.toRequestLog()
|
|
||||||
reqLogs := []reqlog.Request{reqLog}
|
|
||||||
|
|
||||||
if err := c.queryHeaders(ctx, httpReqLogsQuery, reqLogs); err != nil {
|
|
||||||
return reqlog.Request{}, fmt.Errorf("sqlite: could not query headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return reqLogs[0], nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) AddRequestLog(
|
|
||||||
ctx context.Context,
|
|
||||||
req http.Request,
|
|
||||||
body []byte,
|
|
||||||
timestamp time.Time,
|
|
||||||
) (*reqlog.Request, error) {
|
|
||||||
if c.db == nil {
|
|
||||||
return nil, proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLog := &reqlog.Request{
|
|
||||||
Request: req,
|
|
||||||
Body: body,
|
|
||||||
Timestamp: timestamp,
|
|
||||||
}
|
|
||||||
|
|
||||||
tx, err := c.db.BeginTxx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not start transaction: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
reqStmt, err := tx.PrepareContext(ctx, `INSERT INTO http_requests (
|
|
||||||
proto,
|
|
||||||
url,
|
|
||||||
method,
|
|
||||||
body,
|
|
||||||
timestamp
|
|
||||||
) VALUES (?, ?, ?, ?, ?)`)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer reqStmt.Close()
|
|
||||||
|
|
||||||
result, err := reqStmt.ExecContext(ctx,
|
|
||||||
reqLog.Request.Proto,
|
|
||||||
reqLog.Request.URL.String(),
|
|
||||||
reqLog.Request.Method,
|
|
||||||
reqLog.Body,
|
|
||||||
reqLog.Timestamp,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not execute statement: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqID, err := result.LastInsertId()
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not get last insert ID: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLog.ID = reqID
|
|
||||||
|
|
||||||
headerStmt, err := tx.PrepareContext(ctx, `INSERT INTO http_headers (
|
|
||||||
req_id,
|
|
||||||
key,
|
|
||||||
value
|
|
||||||
) VALUES (?, ?, ?)`)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer headerStmt.Close()
|
|
||||||
|
|
||||||
err = insertHeaders(ctx, headerStmt, reqID, reqLog.Request.Header)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not insert http headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := tx.Commit(); err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not commit transaction: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return reqLog, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) AddResponseLog(
|
|
||||||
ctx context.Context,
|
|
||||||
reqID int64,
|
|
||||||
res http.Response,
|
|
||||||
body []byte,
|
|
||||||
timestamp time.Time,
|
|
||||||
) (*reqlog.Response, error) {
|
|
||||||
if c.db == nil {
|
|
||||||
return nil, proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
resLog := &reqlog.Response{
|
|
||||||
RequestID: reqID,
|
|
||||||
Response: res,
|
|
||||||
Body: body,
|
|
||||||
Timestamp: timestamp,
|
|
||||||
}
|
|
||||||
|
|
||||||
tx, err := c.db.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not start transaction: %w", err)
|
|
||||||
}
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
resStmt, err := tx.PrepareContext(ctx, `INSERT INTO http_responses (
|
|
||||||
req_id,
|
|
||||||
proto,
|
|
||||||
status_code,
|
|
||||||
status_reason,
|
|
||||||
body,
|
|
||||||
timestamp
|
|
||||||
) VALUES (?, ?, ?, ?, ?, ?)`)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer resStmt.Close()
|
|
||||||
|
|
||||||
var statusReason string
|
|
||||||
if len(resLog.Response.Status) > 4 {
|
|
||||||
statusReason = resLog.Response.Status[4:]
|
|
||||||
}
|
|
||||||
|
|
||||||
result, err := resStmt.ExecContext(ctx,
|
|
||||||
resLog.RequestID,
|
|
||||||
resLog.Response.Proto,
|
|
||||||
resLog.Response.StatusCode,
|
|
||||||
statusReason,
|
|
||||||
resLog.Body,
|
|
||||||
resLog.Timestamp,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not execute statement: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
resID, err := result.LastInsertId()
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not get last insert ID: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
resLog.ID = resID
|
|
||||||
|
|
||||||
headerStmt, err := tx.PrepareContext(ctx, `INSERT INTO http_headers (
|
|
||||||
res_id,
|
|
||||||
key,
|
|
||||||
value
|
|
||||||
) VALUES (?, ?, ?)`)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer headerStmt.Close()
|
|
||||||
|
|
||||||
err = insertHeaders(ctx, headerStmt, resID, resLog.Response.Header)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not insert http headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := tx.Commit(); err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not commit transaction: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return resLog, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) UpsertSettings(ctx context.Context, module string, settings interface{}) error {
|
|
||||||
if c.db == nil {
|
|
||||||
// TODO: Fix where `ErrNoProject` lives.
|
|
||||||
return proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
jsonSettings, err := json.Marshal(settings)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not encode settings as JSON: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = c.db.ExecContext(ctx,
|
|
||||||
`INSERT INTO settings (module, settings) VALUES (?, ?)
|
|
||||||
ON CONFLICT(module) DO UPDATE SET settings = ?`, module, jsonSettings, jsonSettings)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not insert scope settings: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) FindSettingsByModule(ctx context.Context, module string, settings interface{}) error {
|
|
||||||
if c.db == nil {
|
|
||||||
return proj.ErrNoProject
|
|
||||||
}
|
|
||||||
|
|
||||||
var jsonSettings []byte
|
|
||||||
|
|
||||||
row := c.db.QueryRowContext(ctx, `SELECT settings FROM settings WHERE module = ?`, module)
|
|
||||||
|
|
||||||
err := row.Scan(&jsonSettings)
|
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
|
||||||
return proj.ErrNoSettings
|
|
||||||
} else if err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not scan row: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := json.Unmarshal(jsonSettings, &settings); err != nil {
|
|
||||||
return fmt.Errorf("sqlite: could not decode settings from JSON: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func insertHeaders(ctx context.Context, stmt *sql.Stmt, id int64, headers http.Header) error {
|
|
||||||
for key, values := range headers {
|
|
||||||
for _, value := range values {
|
|
||||||
if _, err := stmt.ExecContext(ctx, id, key, value); err != nil {
|
|
||||||
return fmt.Errorf("could not execute statement: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func findHeaders(ctx context.Context, stmt *sql.Stmt, id int64) (http.Header, error) {
|
|
||||||
headers := make(http.Header)
|
|
||||||
|
|
||||||
rows, err := stmt.QueryContext(ctx, id)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not execute query: %w", err)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
for rows.Next() {
|
|
||||||
var key, value string
|
|
||||||
|
|
||||||
err := rows.Scan(&key, &value)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not scan row: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
headers.Add(key, value)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := rows.Err(); err != nil {
|
|
||||||
return nil, fmt.Errorf("sqlite: could not iterate over rows: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return headers, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseHTTPRequestLogsQuery(ctx context.Context) httpRequestLogsQuery {
|
|
||||||
var (
|
|
||||||
joinResponse bool
|
|
||||||
reqHeaderCols, resHeaderCols []string
|
|
||||||
)
|
|
||||||
|
|
||||||
opCtx := graphql.GetOperationContext(ctx)
|
|
||||||
reqFields := graphql.CollectFieldsCtx(ctx, nil)
|
|
||||||
reqCols := []string{"req.id AS req_id", "res.id AS res_id"}
|
|
||||||
|
|
||||||
for _, reqField := range reqFields {
|
|
||||||
if col, ok := reqFieldToColumnMap[reqField.Name]; ok {
|
|
||||||
reqCols = append(reqCols, "req."+col)
|
|
||||||
}
|
|
||||||
|
|
||||||
if reqField.Name == "headers" {
|
|
||||||
headerFields := graphql.CollectFields(opCtx, reqField.Selections, nil)
|
|
||||||
for _, headerField := range headerFields {
|
|
||||||
if col, ok := headerFieldToColumnMap[headerField.Name]; ok {
|
|
||||||
reqHeaderCols = append(reqHeaderCols, col)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if reqField.Name == "response" {
|
|
||||||
joinResponse = true
|
|
||||||
resFields := graphql.CollectFields(opCtx, reqField.Selections, nil)
|
|
||||||
|
|
||||||
for _, resField := range resFields {
|
|
||||||
if resField.Name == "headers" {
|
|
||||||
reqCols = append(reqCols, "res.id AS res_id")
|
|
||||||
headerFields := graphql.CollectFields(opCtx, resField.Selections, nil)
|
|
||||||
|
|
||||||
for _, headerField := range headerFields {
|
|
||||||
if col, ok := headerFieldToColumnMap[headerField.Name]; ok {
|
|
||||||
resHeaderCols = append(resHeaderCols, col)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if col, ok := resFieldToColumnMap[resField.Name]; ok {
|
|
||||||
reqCols = append(reqCols, "res."+col)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return httpRequestLogsQuery{
|
|
||||||
requestCols: reqCols,
|
|
||||||
requestHeaderCols: reqHeaderCols,
|
|
||||||
responseHeaderCols: resHeaderCols,
|
|
||||||
joinResponse: joinResponse,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) queryHeaders(
|
|
||||||
ctx context.Context,
|
|
||||||
query httpRequestLogsQuery,
|
|
||||||
reqLogs []reqlog.Request,
|
|
||||||
) error {
|
|
||||||
if len(query.requestHeaderCols) > 0 {
|
|
||||||
reqHeadersQuery, _, err := sq.
|
|
||||||
Select(query.requestHeaderCols...).
|
|
||||||
From("http_headers").Where("req_id = ?").
|
|
||||||
ToSql()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not parse request headers query: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqHeadersStmt, err := c.db.PrepareContext(ctx, reqHeadersQuery)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer reqHeadersStmt.Close()
|
|
||||||
|
|
||||||
for i := range reqLogs {
|
|
||||||
headers, err := findHeaders(ctx, reqHeadersStmt, reqLogs[i].ID)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not query request headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLogs[i].Request.Header = headers
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(query.responseHeaderCols) > 0 {
|
|
||||||
resHeadersQuery, _, err := sq.
|
|
||||||
Select(query.responseHeaderCols...).
|
|
||||||
From("http_headers").Where("res_id = ?").
|
|
||||||
ToSql()
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not parse response headers query: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
resHeadersStmt, err := c.db.PrepareContext(ctx, resHeadersQuery)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not prepare statement: %w", err)
|
|
||||||
}
|
|
||||||
defer resHeadersStmt.Close()
|
|
||||||
|
|
||||||
for i := range reqLogs {
|
|
||||||
if reqLogs[i].Response == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
headers, err := findHeaders(ctx, resHeadersStmt, reqLogs[i].Response.ID)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("could not query response headers: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
reqLogs[i].Response.Response.Header = headers
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c *Client) IsOpen() bool {
|
|
||||||
return c.db != nil
|
|
||||||
}
|
|
222
pkg/proj/proj.go
222
pkg/proj/proj.go
@ -5,129 +5,188 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
|
"math/rand"
|
||||||
"regexp"
|
"regexp"
|
||||||
"sync"
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
|
"github.com/dstotijn/hetty/pkg/search"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
//nolint:gosec
|
||||||
|
var ulidEntropy = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||||
|
|
||||||
type (
|
type (
|
||||||
OnProjectOpenFn func(name string) error
|
OnProjectOpenFn func(projectID ulid.ULID) error
|
||||||
OnProjectCloseFn func(name string) error
|
OnProjectCloseFn func(projectID ulid.ULID) error
|
||||||
)
|
)
|
||||||
|
|
||||||
// Service is used for managing projects.
|
// Service is used for managing projects.
|
||||||
type Service interface {
|
type Service interface {
|
||||||
Open(ctx context.Context, name string) (Project, error)
|
CreateProject(ctx context.Context, name string) (Project, error)
|
||||||
Close() error
|
OpenProject(ctx context.Context, projectID ulid.ULID) (Project, error)
|
||||||
Delete(name string) error
|
CloseProject() error
|
||||||
ActiveProject() (Project, error)
|
DeleteProject(ctx context.Context, projectID ulid.ULID) error
|
||||||
Projects() ([]Project, error)
|
ActiveProject(ctx context.Context) (Project, error)
|
||||||
|
IsProjectActive(projectID ulid.ULID) bool
|
||||||
|
Projects(ctx context.Context) ([]Project, error)
|
||||||
|
Scope() *scope.Scope
|
||||||
|
SetScopeRules(ctx context.Context, rules []scope.Rule) error
|
||||||
|
SetRequestLogFindFilter(ctx context.Context, filter reqlog.FindRequestsFilter) error
|
||||||
OnProjectOpen(fn OnProjectOpenFn)
|
OnProjectOpen(fn OnProjectOpenFn)
|
||||||
OnProjectClose(fn OnProjectCloseFn)
|
OnProjectClose(fn OnProjectCloseFn)
|
||||||
}
|
}
|
||||||
|
|
||||||
type service struct {
|
type service struct {
|
||||||
repo Repository
|
repo Repository
|
||||||
activeProject string
|
reqLogSvc *reqlog.Service
|
||||||
|
scope *scope.Scope
|
||||||
|
activeProjectID ulid.ULID
|
||||||
onProjectOpenFns []OnProjectOpenFn
|
onProjectOpenFns []OnProjectOpenFn
|
||||||
onProjectCloseFns []OnProjectCloseFn
|
onProjectCloseFns []OnProjectCloseFn
|
||||||
mu sync.RWMutex
|
mu sync.RWMutex
|
||||||
}
|
}
|
||||||
|
|
||||||
type Project struct {
|
type Project struct {
|
||||||
|
ID ulid.ULID
|
||||||
Name string
|
Name string
|
||||||
IsActive bool
|
Settings Settings
|
||||||
|
|
||||||
|
isActive bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type Settings struct {
|
||||||
|
ReqLogBypassOutOfScope bool
|
||||||
|
ReqLogOnlyFindInScope bool
|
||||||
|
ScopeRules []scope.Rule
|
||||||
|
SearchExpr search.Expression
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ErrNoProject = errors.New("proj: no open project")
|
ErrProjectNotFound = errors.New("proj: project not found")
|
||||||
ErrNoSettings = errors.New("proj: settings not found")
|
ErrNoProject = errors.New("proj: no open project")
|
||||||
ErrInvalidName = errors.New("proj: invalid name, must be alphanumeric or whitespace chars")
|
ErrNoSettings = errors.New("proj: settings not found")
|
||||||
|
ErrInvalidName = errors.New("proj: invalid name, must be alphanumeric or whitespace chars")
|
||||||
)
|
)
|
||||||
|
|
||||||
var nameRegexp = regexp.MustCompile(`^[\w\d\s]+$`)
|
var nameRegexp = regexp.MustCompile(`^[\w\d\s]+$`)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Repository Repository
|
||||||
|
ReqLogService *reqlog.Service
|
||||||
|
Scope *scope.Scope
|
||||||
|
}
|
||||||
|
|
||||||
// NewService returns a new Service.
|
// NewService returns a new Service.
|
||||||
func NewService(repo Repository) (Service, error) {
|
func NewService(cfg Config) (Service, error) {
|
||||||
return &service{
|
return &service{
|
||||||
repo: repo,
|
repo: cfg.Repository,
|
||||||
|
reqLogSvc: cfg.ReqLogService,
|
||||||
|
scope: cfg.Scope,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Close closes the currently open project database (if there is one).
|
func (svc *service) CreateProject(ctx context.Context, name string) (Project, error) {
|
||||||
func (svc *service) Close() error {
|
if !nameRegexp.MatchString(name) {
|
||||||
|
return Project{}, ErrInvalidName
|
||||||
|
}
|
||||||
|
|
||||||
|
project := Project{
|
||||||
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
|
Name: name,
|
||||||
|
}
|
||||||
|
|
||||||
|
err := svc.repo.UpsertProject(ctx, project)
|
||||||
|
if err != nil {
|
||||||
|
return Project{}, fmt.Errorf("proj: could not create project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return project, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CloseProject closes the currently open project (if there is one).
|
||||||
|
func (svc *service) CloseProject() error {
|
||||||
svc.mu.Lock()
|
svc.mu.Lock()
|
||||||
defer svc.mu.Unlock()
|
defer svc.mu.Unlock()
|
||||||
|
|
||||||
closedProject := svc.activeProject
|
if svc.activeProjectID.Compare(ulid.ULID{}) == 0 {
|
||||||
|
return nil
|
||||||
if err := svc.repo.Close(); err != nil {
|
|
||||||
return fmt.Errorf("proj: could not close project: %w", err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.activeProject = ""
|
closedProjectID := svc.activeProjectID
|
||||||
|
|
||||||
svc.emitProjectClosed(closedProject)
|
svc.activeProjectID = ulid.ULID{}
|
||||||
|
svc.reqLogSvc.ActiveProjectID = ulid.ULID{}
|
||||||
|
svc.reqLogSvc.BypassOutOfScopeRequests = false
|
||||||
|
svc.reqLogSvc.FindReqsFilter = reqlog.FindRequestsFilter{}
|
||||||
|
svc.scope.SetRules(nil)
|
||||||
|
|
||||||
|
svc.emitProjectClosed(closedProjectID)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete removes a project database file from disk (if there is one).
|
// DeleteProject removes a project from the repository.
|
||||||
func (svc *service) Delete(name string) error {
|
func (svc *service) DeleteProject(ctx context.Context, projectID ulid.ULID) error {
|
||||||
if name == "" {
|
if svc.activeProjectID.Compare(projectID) == 0 {
|
||||||
return errors.New("proj: name cannot be empty")
|
return fmt.Errorf("proj: project (%v) is active", projectID.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
if svc.activeProject == name {
|
if err := svc.repo.DeleteProject(ctx, projectID); err != nil {
|
||||||
return fmt.Errorf("proj: project (%v) is active", name)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := svc.repo.DeleteProject(name); err != nil {
|
|
||||||
return fmt.Errorf("proj: could not delete project: %w", err)
|
return fmt.Errorf("proj: could not delete project: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Open opens a database identified with `name`. If a database with this
|
// OpenProject sets a project as the currently active project.
|
||||||
// identifier doesn't exist yet, it will be automatically created.
|
func (svc *service) OpenProject(ctx context.Context, projectID ulid.ULID) (Project, error) {
|
||||||
func (svc *service) Open(ctx context.Context, name string) (Project, error) {
|
|
||||||
if !nameRegexp.MatchString(name) {
|
|
||||||
return Project{}, ErrInvalidName
|
|
||||||
}
|
|
||||||
|
|
||||||
svc.mu.Lock()
|
svc.mu.Lock()
|
||||||
defer svc.mu.Unlock()
|
defer svc.mu.Unlock()
|
||||||
|
|
||||||
if err := svc.repo.Close(); err != nil {
|
project, err := svc.repo.FindProjectByID(ctx, projectID)
|
||||||
return Project{}, fmt.Errorf("proj: could not close previously open database: %w", err)
|
if err != nil {
|
||||||
|
return Project{}, fmt.Errorf("proj: failed to get project: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := svc.repo.OpenProject(name); err != nil {
|
svc.activeProjectID = project.ID
|
||||||
return Project{}, fmt.Errorf("proj: could not open database: %w", err)
|
svc.reqLogSvc.FindReqsFilter = reqlog.FindRequestsFilter{
|
||||||
|
ProjectID: project.ID,
|
||||||
|
OnlyInScope: project.Settings.ReqLogOnlyFindInScope,
|
||||||
|
SearchExpr: project.Settings.SearchExpr,
|
||||||
}
|
}
|
||||||
|
svc.reqLogSvc.BypassOutOfScopeRequests = project.Settings.ReqLogBypassOutOfScope
|
||||||
|
svc.reqLogSvc.ActiveProjectID = project.ID
|
||||||
|
|
||||||
|
svc.scope.SetRules(project.Settings.ScopeRules)
|
||||||
|
|
||||||
svc.activeProject = name
|
|
||||||
svc.emitProjectOpened()
|
svc.emitProjectOpened()
|
||||||
|
|
||||||
return Project{
|
return project, nil
|
||||||
Name: name,
|
|
||||||
IsActive: true,
|
|
||||||
}, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *service) ActiveProject() (Project, error) {
|
func (svc *service) ActiveProject(ctx context.Context) (Project, error) {
|
||||||
activeProject := svc.activeProject
|
activeProjectID := svc.activeProjectID
|
||||||
if activeProject == "" {
|
if activeProjectID.Compare(ulid.ULID{}) == 0 {
|
||||||
return Project{}, ErrNoProject
|
return Project{}, ErrNoProject
|
||||||
}
|
}
|
||||||
|
|
||||||
return Project{
|
project, err := svc.repo.FindProjectByID(ctx, activeProjectID)
|
||||||
Name: activeProject,
|
if err != nil {
|
||||||
}, nil
|
return Project{}, fmt.Errorf("proj: failed to get active project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
project.isActive = true
|
||||||
|
|
||||||
|
return project, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *service) Projects() ([]Project, error) {
|
func (svc *service) Projects(ctx context.Context) ([]Project, error) {
|
||||||
projects, err := svc.repo.Projects()
|
projects, err := svc.repo.Projects(ctx)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("proj: could not get projects: %w", err)
|
return nil, fmt.Errorf("proj: could not get projects: %w", err)
|
||||||
}
|
}
|
||||||
@ -135,6 +194,10 @@ func (svc *service) Projects() ([]Project, error) {
|
|||||||
return projects, nil
|
return projects, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (svc *service) Scope() *scope.Scope {
|
||||||
|
return svc.scope
|
||||||
|
}
|
||||||
|
|
||||||
func (svc *service) OnProjectOpen(fn OnProjectOpenFn) {
|
func (svc *service) OnProjectOpen(fn OnProjectOpenFn) {
|
||||||
svc.mu.Lock()
|
svc.mu.Lock()
|
||||||
defer svc.mu.Unlock()
|
defer svc.mu.Unlock()
|
||||||
@ -151,16 +214,59 @@ func (svc *service) OnProjectClose(fn OnProjectCloseFn) {
|
|||||||
|
|
||||||
func (svc *service) emitProjectOpened() {
|
func (svc *service) emitProjectOpened() {
|
||||||
for _, fn := range svc.onProjectOpenFns {
|
for _, fn := range svc.onProjectOpenFns {
|
||||||
if err := fn(svc.activeProject); err != nil {
|
if err := fn(svc.activeProjectID); err != nil {
|
||||||
log.Printf("[ERROR] Could not execute onProjectOpen function: %v", err)
|
log.Printf("[ERROR] Could not execute onProjectOpen function: %v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *service) emitProjectClosed(name string) {
|
func (svc *service) emitProjectClosed(projectID ulid.ULID) {
|
||||||
for _, fn := range svc.onProjectCloseFns {
|
for _, fn := range svc.onProjectCloseFns {
|
||||||
if err := fn(name); err != nil {
|
if err := fn(projectID); err != nil {
|
||||||
log.Printf("[ERROR] Could not execute onProjectClose function: %v", err)
|
log.Printf("[ERROR] Could not execute onProjectClose function: %v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (svc *service) SetScopeRules(ctx context.Context, rules []scope.Rule) error {
|
||||||
|
project, err := svc.ActiveProject(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
project.Settings.ScopeRules = rules
|
||||||
|
|
||||||
|
err = svc.repo.UpsertProject(ctx, project)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.scope.SetRules(rules)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (svc *service) SetRequestLogFindFilter(ctx context.Context, filter reqlog.FindRequestsFilter) error {
|
||||||
|
project, err := svc.ActiveProject(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
filter.ProjectID = project.ID
|
||||||
|
|
||||||
|
project.Settings.ReqLogOnlyFindInScope = filter.OnlyInScope
|
||||||
|
project.Settings.SearchExpr = filter.SearchExpr
|
||||||
|
|
||||||
|
err = svc.repo.UpsertProject(ctx, project)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("proj: failed to update project: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.reqLogSvc.FindReqsFilter = filter
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (svc *service) IsProjectActive(projectID ulid.ULID) bool {
|
||||||
|
return projectID.Compare(svc.activeProjectID) == 0
|
||||||
|
}
|
||||||
|
@ -2,13 +2,14 @@ package proj
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
|
||||||
|
"github.com/oklog/ulid"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Repository interface {
|
type Repository interface {
|
||||||
UpsertSettings(ctx context.Context, module string, settings interface{}) error
|
FindProjectByID(ctx context.Context, id ulid.ULID) (Project, error)
|
||||||
FindSettingsByModule(ctx context.Context, module string, settings interface{}) error
|
UpsertProject(ctx context.Context, project Project) error
|
||||||
OpenProject(name string) error
|
DeleteProject(ctx context.Context, id ulid.ULID) error
|
||||||
DeleteProject(name string) error
|
Projects(ctx context.Context) ([]Project, error)
|
||||||
Projects() ([]Project, error)
|
|
||||||
Close() error
|
Close() error
|
||||||
}
|
}
|
||||||
|
@ -15,7 +15,7 @@ import (
|
|||||||
|
|
||||||
type contextKey int
|
type contextKey int
|
||||||
|
|
||||||
const ReqIDKey contextKey = 0
|
const ReqLogIDKey contextKey = 0
|
||||||
|
|
||||||
// Proxy implements http.Handler and offers MITM behaviour for modifying
|
// Proxy implements http.Handler and offers MITM behaviour for modifying
|
||||||
// HTTP requests and responses.
|
// HTTP requests and responses.
|
||||||
|
@ -1,318 +0,0 @@
|
|||||||
// Code generated by moq; DO NOT EDIT.
|
|
||||||
// github.com/matryer/moq
|
|
||||||
|
|
||||||
package reqlog_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
|
||||||
"sync"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Ensure, that ProjServiceMock does implement proj.Service.
|
|
||||||
// If this is not the case, regenerate this file with moq.
|
|
||||||
var _ proj.Service = &ProjServiceMock{}
|
|
||||||
|
|
||||||
// ProjServiceMock is a mock implementation of proj.Service.
|
|
||||||
//
|
|
||||||
// func TestSomethingThatUsesService(t *testing.T) {
|
|
||||||
//
|
|
||||||
// // make and configure a mocked proj.Service
|
|
||||||
// mockedService := &ProjServiceMock{
|
|
||||||
// ActiveProjectFunc: func() (proj.Project, error) {
|
|
||||||
// panic("mock out the ActiveProject method")
|
|
||||||
// },
|
|
||||||
// CloseFunc: func() error {
|
|
||||||
// panic("mock out the Close method")
|
|
||||||
// },
|
|
||||||
// DeleteFunc: func(name string) error {
|
|
||||||
// panic("mock out the Delete method")
|
|
||||||
// },
|
|
||||||
// OnProjectCloseFunc: func(fn proj.OnProjectCloseFn) {
|
|
||||||
// panic("mock out the OnProjectClose method")
|
|
||||||
// },
|
|
||||||
// OnProjectOpenFunc: func(fn proj.OnProjectOpenFn) {
|
|
||||||
// panic("mock out the OnProjectOpen method")
|
|
||||||
// },
|
|
||||||
// OpenFunc: func(ctx context.Context, name string) (proj.Project, error) {
|
|
||||||
// panic("mock out the Open method")
|
|
||||||
// },
|
|
||||||
// ProjectsFunc: func() ([]proj.Project, error) {
|
|
||||||
// panic("mock out the Projects method")
|
|
||||||
// },
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// // use mockedService in code that requires proj.Service
|
|
||||||
// // and then make assertions.
|
|
||||||
//
|
|
||||||
// }
|
|
||||||
type ProjServiceMock struct {
|
|
||||||
// ActiveProjectFunc mocks the ActiveProject method.
|
|
||||||
ActiveProjectFunc func() (proj.Project, error)
|
|
||||||
|
|
||||||
// CloseFunc mocks the Close method.
|
|
||||||
CloseFunc func() error
|
|
||||||
|
|
||||||
// DeleteFunc mocks the Delete method.
|
|
||||||
DeleteFunc func(name string) error
|
|
||||||
|
|
||||||
// OnProjectCloseFunc mocks the OnProjectClose method.
|
|
||||||
OnProjectCloseFunc func(fn proj.OnProjectCloseFn)
|
|
||||||
|
|
||||||
// OnProjectOpenFunc mocks the OnProjectOpen method.
|
|
||||||
OnProjectOpenFunc func(fn proj.OnProjectOpenFn)
|
|
||||||
|
|
||||||
// OpenFunc mocks the Open method.
|
|
||||||
OpenFunc func(ctx context.Context, name string) (proj.Project, error)
|
|
||||||
|
|
||||||
// ProjectsFunc mocks the Projects method.
|
|
||||||
ProjectsFunc func() ([]proj.Project, error)
|
|
||||||
|
|
||||||
// calls tracks calls to the methods.
|
|
||||||
calls struct {
|
|
||||||
// ActiveProject holds details about calls to the ActiveProject method.
|
|
||||||
ActiveProject []struct {
|
|
||||||
}
|
|
||||||
// Close holds details about calls to the Close method.
|
|
||||||
Close []struct {
|
|
||||||
}
|
|
||||||
// Delete holds details about calls to the Delete method.
|
|
||||||
Delete []struct {
|
|
||||||
// Name is the name argument value.
|
|
||||||
Name string
|
|
||||||
}
|
|
||||||
// OnProjectClose holds details about calls to the OnProjectClose method.
|
|
||||||
OnProjectClose []struct {
|
|
||||||
// Fn is the fn argument value.
|
|
||||||
Fn proj.OnProjectCloseFn
|
|
||||||
}
|
|
||||||
// OnProjectOpen holds details about calls to the OnProjectOpen method.
|
|
||||||
OnProjectOpen []struct {
|
|
||||||
// Fn is the fn argument value.
|
|
||||||
Fn proj.OnProjectOpenFn
|
|
||||||
}
|
|
||||||
// Open holds details about calls to the Open method.
|
|
||||||
Open []struct {
|
|
||||||
// Ctx is the ctx argument value.
|
|
||||||
Ctx context.Context
|
|
||||||
// Name is the name argument value.
|
|
||||||
Name string
|
|
||||||
}
|
|
||||||
// Projects holds details about calls to the Projects method.
|
|
||||||
Projects []struct {
|
|
||||||
}
|
|
||||||
}
|
|
||||||
lockActiveProject sync.RWMutex
|
|
||||||
lockClose sync.RWMutex
|
|
||||||
lockDelete sync.RWMutex
|
|
||||||
lockOnProjectClose sync.RWMutex
|
|
||||||
lockOnProjectOpen sync.RWMutex
|
|
||||||
lockOpen sync.RWMutex
|
|
||||||
lockProjects sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
// ActiveProject calls ActiveProjectFunc.
|
|
||||||
func (mock *ProjServiceMock) ActiveProject() (proj.Project, error) {
|
|
||||||
if mock.ActiveProjectFunc == nil {
|
|
||||||
panic("ProjServiceMock.ActiveProjectFunc: method is nil but Service.ActiveProject was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
}{}
|
|
||||||
mock.lockActiveProject.Lock()
|
|
||||||
mock.calls.ActiveProject = append(mock.calls.ActiveProject, callInfo)
|
|
||||||
mock.lockActiveProject.Unlock()
|
|
||||||
return mock.ActiveProjectFunc()
|
|
||||||
}
|
|
||||||
|
|
||||||
// ActiveProjectCalls gets all the calls that were made to ActiveProject.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.ActiveProjectCalls())
|
|
||||||
func (mock *ProjServiceMock) ActiveProjectCalls() []struct {
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
}
|
|
||||||
mock.lockActiveProject.RLock()
|
|
||||||
calls = mock.calls.ActiveProject
|
|
||||||
mock.lockActiveProject.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// Close calls CloseFunc.
|
|
||||||
func (mock *ProjServiceMock) Close() error {
|
|
||||||
if mock.CloseFunc == nil {
|
|
||||||
panic("ProjServiceMock.CloseFunc: method is nil but Service.Close was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
}{}
|
|
||||||
mock.lockClose.Lock()
|
|
||||||
mock.calls.Close = append(mock.calls.Close, callInfo)
|
|
||||||
mock.lockClose.Unlock()
|
|
||||||
return mock.CloseFunc()
|
|
||||||
}
|
|
||||||
|
|
||||||
// CloseCalls gets all the calls that were made to Close.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.CloseCalls())
|
|
||||||
func (mock *ProjServiceMock) CloseCalls() []struct {
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
}
|
|
||||||
mock.lockClose.RLock()
|
|
||||||
calls = mock.calls.Close
|
|
||||||
mock.lockClose.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// Delete calls DeleteFunc.
|
|
||||||
func (mock *ProjServiceMock) Delete(name string) error {
|
|
||||||
if mock.DeleteFunc == nil {
|
|
||||||
panic("ProjServiceMock.DeleteFunc: method is nil but Service.Delete was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Name string
|
|
||||||
}{
|
|
||||||
Name: name,
|
|
||||||
}
|
|
||||||
mock.lockDelete.Lock()
|
|
||||||
mock.calls.Delete = append(mock.calls.Delete, callInfo)
|
|
||||||
mock.lockDelete.Unlock()
|
|
||||||
return mock.DeleteFunc(name)
|
|
||||||
}
|
|
||||||
|
|
||||||
// DeleteCalls gets all the calls that were made to Delete.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.DeleteCalls())
|
|
||||||
func (mock *ProjServiceMock) DeleteCalls() []struct {
|
|
||||||
Name string
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Name string
|
|
||||||
}
|
|
||||||
mock.lockDelete.RLock()
|
|
||||||
calls = mock.calls.Delete
|
|
||||||
mock.lockDelete.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// OnProjectClose calls OnProjectCloseFunc.
|
|
||||||
func (mock *ProjServiceMock) OnProjectClose(fn proj.OnProjectCloseFn) {
|
|
||||||
if mock.OnProjectCloseFunc == nil {
|
|
||||||
panic("ProjServiceMock.OnProjectCloseFunc: method is nil but Service.OnProjectClose was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Fn proj.OnProjectCloseFn
|
|
||||||
}{
|
|
||||||
Fn: fn,
|
|
||||||
}
|
|
||||||
mock.lockOnProjectClose.Lock()
|
|
||||||
mock.calls.OnProjectClose = append(mock.calls.OnProjectClose, callInfo)
|
|
||||||
mock.lockOnProjectClose.Unlock()
|
|
||||||
mock.OnProjectCloseFunc(fn)
|
|
||||||
}
|
|
||||||
|
|
||||||
// OnProjectCloseCalls gets all the calls that were made to OnProjectClose.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.OnProjectCloseCalls())
|
|
||||||
func (mock *ProjServiceMock) OnProjectCloseCalls() []struct {
|
|
||||||
Fn proj.OnProjectCloseFn
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Fn proj.OnProjectCloseFn
|
|
||||||
}
|
|
||||||
mock.lockOnProjectClose.RLock()
|
|
||||||
calls = mock.calls.OnProjectClose
|
|
||||||
mock.lockOnProjectClose.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// OnProjectOpen calls OnProjectOpenFunc.
|
|
||||||
func (mock *ProjServiceMock) OnProjectOpen(fn proj.OnProjectOpenFn) {
|
|
||||||
if mock.OnProjectOpenFunc == nil {
|
|
||||||
panic("ProjServiceMock.OnProjectOpenFunc: method is nil but Service.OnProjectOpen was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Fn proj.OnProjectOpenFn
|
|
||||||
}{
|
|
||||||
Fn: fn,
|
|
||||||
}
|
|
||||||
mock.lockOnProjectOpen.Lock()
|
|
||||||
mock.calls.OnProjectOpen = append(mock.calls.OnProjectOpen, callInfo)
|
|
||||||
mock.lockOnProjectOpen.Unlock()
|
|
||||||
mock.OnProjectOpenFunc(fn)
|
|
||||||
}
|
|
||||||
|
|
||||||
// OnProjectOpenCalls gets all the calls that were made to OnProjectOpen.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.OnProjectOpenCalls())
|
|
||||||
func (mock *ProjServiceMock) OnProjectOpenCalls() []struct {
|
|
||||||
Fn proj.OnProjectOpenFn
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Fn proj.OnProjectOpenFn
|
|
||||||
}
|
|
||||||
mock.lockOnProjectOpen.RLock()
|
|
||||||
calls = mock.calls.OnProjectOpen
|
|
||||||
mock.lockOnProjectOpen.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// Open calls OpenFunc.
|
|
||||||
func (mock *ProjServiceMock) Open(ctx context.Context, name string) (proj.Project, error) {
|
|
||||||
if mock.OpenFunc == nil {
|
|
||||||
panic("ProjServiceMock.OpenFunc: method is nil but Service.Open was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Name string
|
|
||||||
}{
|
|
||||||
Ctx: ctx,
|
|
||||||
Name: name,
|
|
||||||
}
|
|
||||||
mock.lockOpen.Lock()
|
|
||||||
mock.calls.Open = append(mock.calls.Open, callInfo)
|
|
||||||
mock.lockOpen.Unlock()
|
|
||||||
return mock.OpenFunc(ctx, name)
|
|
||||||
}
|
|
||||||
|
|
||||||
// OpenCalls gets all the calls that were made to Open.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.OpenCalls())
|
|
||||||
func (mock *ProjServiceMock) OpenCalls() []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Name string
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Name string
|
|
||||||
}
|
|
||||||
mock.lockOpen.RLock()
|
|
||||||
calls = mock.calls.Open
|
|
||||||
mock.lockOpen.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// Projects calls ProjectsFunc.
|
|
||||||
func (mock *ProjServiceMock) Projects() ([]proj.Project, error) {
|
|
||||||
if mock.ProjectsFunc == nil {
|
|
||||||
panic("ProjServiceMock.ProjectsFunc: method is nil but Service.Projects was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
}{}
|
|
||||||
mock.lockProjects.Lock()
|
|
||||||
mock.calls.Projects = append(mock.calls.Projects, callInfo)
|
|
||||||
mock.lockProjects.Unlock()
|
|
||||||
return mock.ProjectsFunc()
|
|
||||||
}
|
|
||||||
|
|
||||||
// ProjectsCalls gets all the calls that were made to Projects.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedService.ProjectsCalls())
|
|
||||||
func (mock *ProjServiceMock) ProjectsCalls() []struct {
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
}
|
|
||||||
mock.lockProjects.RLock()
|
|
||||||
calls = mock.calls.Projects
|
|
||||||
mock.lockProjects.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
@ -2,18 +2,16 @@ package reqlog
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"net/http"
|
|
||||||
"time"
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/scope"
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Repository interface {
|
type Repository interface {
|
||||||
FindRequestLogs(ctx context.Context, filter FindRequestsFilter, scope *scope.Scope) ([]Request, error)
|
FindRequestLogs(ctx context.Context, filter FindRequestsFilter, scope *scope.Scope) ([]RequestLog, error)
|
||||||
FindRequestLogByID(ctx context.Context, id int64) (Request, error)
|
FindRequestLogByID(ctx context.Context, id ulid.ULID) (RequestLog, error)
|
||||||
AddRequestLog(ctx context.Context, req http.Request, body []byte, timestamp time.Time) (*Request, error)
|
StoreRequestLog(ctx context.Context, reqLog RequestLog) error
|
||||||
AddResponseLog(ctx context.Context, reqID int64, res http.Response, body []byte, timestamp time.Time) (*Response, error) // nolint:lll
|
StoreResponseLog(ctx context.Context, reqLogID ulid.ULID, resLog ResponseLog) error
|
||||||
ClearRequestLogs(ctx context.Context) error
|
ClearRequestLogs(ctx context.Context, projectID ulid.ULID) error
|
||||||
UpsertSettings(ctx context.Context, module string, settings interface{}) error
|
|
||||||
FindSettingsByModule(ctx context.Context, module string, settings interface{}) error
|
|
||||||
}
|
}
|
||||||
|
@ -7,9 +7,8 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"github.com/dstotijn/hetty/pkg/reqlog"
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
"github.com/dstotijn/hetty/pkg/scope"
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
"net/http"
|
"github.com/oklog/ulid"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Ensure, that RepoMock does implement reqlog.Repository.
|
// Ensure, that RepoMock does implement reqlog.Repository.
|
||||||
@ -22,26 +21,20 @@ var _ reqlog.Repository = &RepoMock{}
|
|||||||
//
|
//
|
||||||
// // make and configure a mocked reqlog.Repository
|
// // make and configure a mocked reqlog.Repository
|
||||||
// mockedRepository := &RepoMock{
|
// mockedRepository := &RepoMock{
|
||||||
// AddRequestLogFunc: func(ctx context.Context, req http.Request, body []byte, timestamp time.Time) (*reqlog.Request, error) {
|
// ClearRequestLogsFunc: func(ctx context.Context, projectID ulid.ULID) error {
|
||||||
// panic("mock out the AddRequestLog method")
|
|
||||||
// },
|
|
||||||
// AddResponseLogFunc: func(ctx context.Context, reqID int64, res http.Response, body []byte, timestamp time.Time) (*reqlog.Response, error) {
|
|
||||||
// panic("mock out the AddResponseLog method")
|
|
||||||
// },
|
|
||||||
// ClearRequestLogsFunc: func(ctx context.Context) error {
|
|
||||||
// panic("mock out the ClearRequestLogs method")
|
// panic("mock out the ClearRequestLogs method")
|
||||||
// },
|
// },
|
||||||
// FindRequestLogByIDFunc: func(ctx context.Context, id int64) (reqlog.Request, error) {
|
// FindRequestLogByIDFunc: func(ctx context.Context, id ulid.ULID) (reqlog.RequestLog, error) {
|
||||||
// panic("mock out the FindRequestLogByID method")
|
// panic("mock out the FindRequestLogByID method")
|
||||||
// },
|
// },
|
||||||
// FindRequestLogsFunc: func(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.Request, error) {
|
// FindRequestLogsFunc: func(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.RequestLog, error) {
|
||||||
// panic("mock out the FindRequestLogs method")
|
// panic("mock out the FindRequestLogs method")
|
||||||
// },
|
// },
|
||||||
// FindSettingsByModuleFunc: func(ctx context.Context, module string, settings interface{}) error {
|
// StoreRequestLogFunc: func(ctx context.Context, reqLog reqlog.RequestLog) error {
|
||||||
// panic("mock out the FindSettingsByModule method")
|
// panic("mock out the StoreRequestLog method")
|
||||||
// },
|
// },
|
||||||
// UpsertSettingsFunc: func(ctx context.Context, module string, settings interface{}) error {
|
// StoreResponseLogFunc: func(ctx context.Context, reqLogID ulid.ULID, resLog reqlog.ResponseLog) error {
|
||||||
// panic("mock out the UpsertSettings method")
|
// panic("mock out the StoreResponseLog method")
|
||||||
// },
|
// },
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
@ -50,64 +43,36 @@ var _ reqlog.Repository = &RepoMock{}
|
|||||||
//
|
//
|
||||||
// }
|
// }
|
||||||
type RepoMock struct {
|
type RepoMock struct {
|
||||||
// AddRequestLogFunc mocks the AddRequestLog method.
|
|
||||||
AddRequestLogFunc func(ctx context.Context, req http.Request, body []byte, timestamp time.Time) (*reqlog.Request, error)
|
|
||||||
|
|
||||||
// AddResponseLogFunc mocks the AddResponseLog method.
|
|
||||||
AddResponseLogFunc func(ctx context.Context, reqID int64, res http.Response, body []byte, timestamp time.Time) (*reqlog.Response, error)
|
|
||||||
|
|
||||||
// ClearRequestLogsFunc mocks the ClearRequestLogs method.
|
// ClearRequestLogsFunc mocks the ClearRequestLogs method.
|
||||||
ClearRequestLogsFunc func(ctx context.Context) error
|
ClearRequestLogsFunc func(ctx context.Context, projectID ulid.ULID) error
|
||||||
|
|
||||||
// FindRequestLogByIDFunc mocks the FindRequestLogByID method.
|
// FindRequestLogByIDFunc mocks the FindRequestLogByID method.
|
||||||
FindRequestLogByIDFunc func(ctx context.Context, id int64) (reqlog.Request, error)
|
FindRequestLogByIDFunc func(ctx context.Context, id ulid.ULID) (reqlog.RequestLog, error)
|
||||||
|
|
||||||
// FindRequestLogsFunc mocks the FindRequestLogs method.
|
// FindRequestLogsFunc mocks the FindRequestLogs method.
|
||||||
FindRequestLogsFunc func(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.Request, error)
|
FindRequestLogsFunc func(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.RequestLog, error)
|
||||||
|
|
||||||
// FindSettingsByModuleFunc mocks the FindSettingsByModule method.
|
// StoreRequestLogFunc mocks the StoreRequestLog method.
|
||||||
FindSettingsByModuleFunc func(ctx context.Context, module string, settings interface{}) error
|
StoreRequestLogFunc func(ctx context.Context, reqLog reqlog.RequestLog) error
|
||||||
|
|
||||||
// UpsertSettingsFunc mocks the UpsertSettings method.
|
// StoreResponseLogFunc mocks the StoreResponseLog method.
|
||||||
UpsertSettingsFunc func(ctx context.Context, module string, settings interface{}) error
|
StoreResponseLogFunc func(ctx context.Context, reqLogID ulid.ULID, resLog reqlog.ResponseLog) error
|
||||||
|
|
||||||
// calls tracks calls to the methods.
|
// calls tracks calls to the methods.
|
||||||
calls struct {
|
calls struct {
|
||||||
// AddRequestLog holds details about calls to the AddRequestLog method.
|
|
||||||
AddRequestLog []struct {
|
|
||||||
// Ctx is the ctx argument value.
|
|
||||||
Ctx context.Context
|
|
||||||
// Req is the req argument value.
|
|
||||||
Req http.Request
|
|
||||||
// Body is the body argument value.
|
|
||||||
Body []byte
|
|
||||||
// Timestamp is the timestamp argument value.
|
|
||||||
Timestamp time.Time
|
|
||||||
}
|
|
||||||
// AddResponseLog holds details about calls to the AddResponseLog method.
|
|
||||||
AddResponseLog []struct {
|
|
||||||
// Ctx is the ctx argument value.
|
|
||||||
Ctx context.Context
|
|
||||||
// ReqID is the reqID argument value.
|
|
||||||
ReqID int64
|
|
||||||
// Res is the res argument value.
|
|
||||||
Res http.Response
|
|
||||||
// Body is the body argument value.
|
|
||||||
Body []byte
|
|
||||||
// Timestamp is the timestamp argument value.
|
|
||||||
Timestamp time.Time
|
|
||||||
}
|
|
||||||
// ClearRequestLogs holds details about calls to the ClearRequestLogs method.
|
// ClearRequestLogs holds details about calls to the ClearRequestLogs method.
|
||||||
ClearRequestLogs []struct {
|
ClearRequestLogs []struct {
|
||||||
// Ctx is the ctx argument value.
|
// Ctx is the ctx argument value.
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
|
// ProjectID is the projectID argument value.
|
||||||
|
ProjectID ulid.ULID
|
||||||
}
|
}
|
||||||
// FindRequestLogByID holds details about calls to the FindRequestLogByID method.
|
// FindRequestLogByID holds details about calls to the FindRequestLogByID method.
|
||||||
FindRequestLogByID []struct {
|
FindRequestLogByID []struct {
|
||||||
// Ctx is the ctx argument value.
|
// Ctx is the ctx argument value.
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
// ID is the id argument value.
|
// ID is the id argument value.
|
||||||
ID int64
|
ID ulid.ULID
|
||||||
}
|
}
|
||||||
// FindRequestLogs holds details about calls to the FindRequestLogs method.
|
// FindRequestLogs holds details about calls to the FindRequestLogs method.
|
||||||
FindRequestLogs []struct {
|
FindRequestLogs []struct {
|
||||||
@ -118,148 +83,58 @@ type RepoMock struct {
|
|||||||
// ScopeMoqParam is the scopeMoqParam argument value.
|
// ScopeMoqParam is the scopeMoqParam argument value.
|
||||||
ScopeMoqParam *scope.Scope
|
ScopeMoqParam *scope.Scope
|
||||||
}
|
}
|
||||||
// FindSettingsByModule holds details about calls to the FindSettingsByModule method.
|
// StoreRequestLog holds details about calls to the StoreRequestLog method.
|
||||||
FindSettingsByModule []struct {
|
StoreRequestLog []struct {
|
||||||
// Ctx is the ctx argument value.
|
// Ctx is the ctx argument value.
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
// Module is the module argument value.
|
// ReqLog is the reqLog argument value.
|
||||||
Module string
|
ReqLog reqlog.RequestLog
|
||||||
// Settings is the settings argument value.
|
|
||||||
Settings interface{}
|
|
||||||
}
|
}
|
||||||
// UpsertSettings holds details about calls to the UpsertSettings method.
|
// StoreResponseLog holds details about calls to the StoreResponseLog method.
|
||||||
UpsertSettings []struct {
|
StoreResponseLog []struct {
|
||||||
// Ctx is the ctx argument value.
|
// Ctx is the ctx argument value.
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
// Module is the module argument value.
|
// ReqLogID is the reqLogID argument value.
|
||||||
Module string
|
ReqLogID ulid.ULID
|
||||||
// Settings is the settings argument value.
|
// ResLog is the resLog argument value.
|
||||||
Settings interface{}
|
ResLog reqlog.ResponseLog
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
lockAddRequestLog sync.RWMutex
|
lockClearRequestLogs sync.RWMutex
|
||||||
lockAddResponseLog sync.RWMutex
|
lockFindRequestLogByID sync.RWMutex
|
||||||
lockClearRequestLogs sync.RWMutex
|
lockFindRequestLogs sync.RWMutex
|
||||||
lockFindRequestLogByID sync.RWMutex
|
lockStoreRequestLog sync.RWMutex
|
||||||
lockFindRequestLogs sync.RWMutex
|
lockStoreResponseLog sync.RWMutex
|
||||||
lockFindSettingsByModule sync.RWMutex
|
|
||||||
lockUpsertSettings sync.RWMutex
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddRequestLog calls AddRequestLogFunc.
|
|
||||||
func (mock *RepoMock) AddRequestLog(ctx context.Context, req http.Request, body []byte, timestamp time.Time) (*reqlog.Request, error) {
|
|
||||||
if mock.AddRequestLogFunc == nil {
|
|
||||||
panic("RepoMock.AddRequestLogFunc: method is nil but Repository.AddRequestLog was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Req http.Request
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
}{
|
|
||||||
Ctx: ctx,
|
|
||||||
Req: req,
|
|
||||||
Body: body,
|
|
||||||
Timestamp: timestamp,
|
|
||||||
}
|
|
||||||
mock.lockAddRequestLog.Lock()
|
|
||||||
mock.calls.AddRequestLog = append(mock.calls.AddRequestLog, callInfo)
|
|
||||||
mock.lockAddRequestLog.Unlock()
|
|
||||||
return mock.AddRequestLogFunc(ctx, req, body, timestamp)
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddRequestLogCalls gets all the calls that were made to AddRequestLog.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedRepository.AddRequestLogCalls())
|
|
||||||
func (mock *RepoMock) AddRequestLogCalls() []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Req http.Request
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
Req http.Request
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
}
|
|
||||||
mock.lockAddRequestLog.RLock()
|
|
||||||
calls = mock.calls.AddRequestLog
|
|
||||||
mock.lockAddRequestLog.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddResponseLog calls AddResponseLogFunc.
|
|
||||||
func (mock *RepoMock) AddResponseLog(ctx context.Context, reqID int64, res http.Response, body []byte, timestamp time.Time) (*reqlog.Response, error) {
|
|
||||||
if mock.AddResponseLogFunc == nil {
|
|
||||||
panic("RepoMock.AddResponseLogFunc: method is nil but Repository.AddResponseLog was just called")
|
|
||||||
}
|
|
||||||
callInfo := struct {
|
|
||||||
Ctx context.Context
|
|
||||||
ReqID int64
|
|
||||||
Res http.Response
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
}{
|
|
||||||
Ctx: ctx,
|
|
||||||
ReqID: reqID,
|
|
||||||
Res: res,
|
|
||||||
Body: body,
|
|
||||||
Timestamp: timestamp,
|
|
||||||
}
|
|
||||||
mock.lockAddResponseLog.Lock()
|
|
||||||
mock.calls.AddResponseLog = append(mock.calls.AddResponseLog, callInfo)
|
|
||||||
mock.lockAddResponseLog.Unlock()
|
|
||||||
return mock.AddResponseLogFunc(ctx, reqID, res, body, timestamp)
|
|
||||||
}
|
|
||||||
|
|
||||||
// AddResponseLogCalls gets all the calls that were made to AddResponseLog.
|
|
||||||
// Check the length with:
|
|
||||||
// len(mockedRepository.AddResponseLogCalls())
|
|
||||||
func (mock *RepoMock) AddResponseLogCalls() []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
ReqID int64
|
|
||||||
Res http.Response
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
} {
|
|
||||||
var calls []struct {
|
|
||||||
Ctx context.Context
|
|
||||||
ReqID int64
|
|
||||||
Res http.Response
|
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
|
||||||
}
|
|
||||||
mock.lockAddResponseLog.RLock()
|
|
||||||
calls = mock.calls.AddResponseLog
|
|
||||||
mock.lockAddResponseLog.RUnlock()
|
|
||||||
return calls
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ClearRequestLogs calls ClearRequestLogsFunc.
|
// ClearRequestLogs calls ClearRequestLogsFunc.
|
||||||
func (mock *RepoMock) ClearRequestLogs(ctx context.Context) error {
|
func (mock *RepoMock) ClearRequestLogs(ctx context.Context, projectID ulid.ULID) error {
|
||||||
if mock.ClearRequestLogsFunc == nil {
|
if mock.ClearRequestLogsFunc == nil {
|
||||||
panic("RepoMock.ClearRequestLogsFunc: method is nil but Repository.ClearRequestLogs was just called")
|
panic("RepoMock.ClearRequestLogsFunc: method is nil but Repository.ClearRequestLogs was just called")
|
||||||
}
|
}
|
||||||
callInfo := struct {
|
callInfo := struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
|
ProjectID ulid.ULID
|
||||||
}{
|
}{
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
|
ProjectID: projectID,
|
||||||
}
|
}
|
||||||
mock.lockClearRequestLogs.Lock()
|
mock.lockClearRequestLogs.Lock()
|
||||||
mock.calls.ClearRequestLogs = append(mock.calls.ClearRequestLogs, callInfo)
|
mock.calls.ClearRequestLogs = append(mock.calls.ClearRequestLogs, callInfo)
|
||||||
mock.lockClearRequestLogs.Unlock()
|
mock.lockClearRequestLogs.Unlock()
|
||||||
return mock.ClearRequestLogsFunc(ctx)
|
return mock.ClearRequestLogsFunc(ctx, projectID)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ClearRequestLogsCalls gets all the calls that were made to ClearRequestLogs.
|
// ClearRequestLogsCalls gets all the calls that were made to ClearRequestLogs.
|
||||||
// Check the length with:
|
// Check the length with:
|
||||||
// len(mockedRepository.ClearRequestLogsCalls())
|
// len(mockedRepository.ClearRequestLogsCalls())
|
||||||
func (mock *RepoMock) ClearRequestLogsCalls() []struct {
|
func (mock *RepoMock) ClearRequestLogsCalls() []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
|
ProjectID ulid.ULID
|
||||||
} {
|
} {
|
||||||
var calls []struct {
|
var calls []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
|
ProjectID ulid.ULID
|
||||||
}
|
}
|
||||||
mock.lockClearRequestLogs.RLock()
|
mock.lockClearRequestLogs.RLock()
|
||||||
calls = mock.calls.ClearRequestLogs
|
calls = mock.calls.ClearRequestLogs
|
||||||
@ -268,13 +143,13 @@ func (mock *RepoMock) ClearRequestLogsCalls() []struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// FindRequestLogByID calls FindRequestLogByIDFunc.
|
// FindRequestLogByID calls FindRequestLogByIDFunc.
|
||||||
func (mock *RepoMock) FindRequestLogByID(ctx context.Context, id int64) (reqlog.Request, error) {
|
func (mock *RepoMock) FindRequestLogByID(ctx context.Context, id ulid.ULID) (reqlog.RequestLog, error) {
|
||||||
if mock.FindRequestLogByIDFunc == nil {
|
if mock.FindRequestLogByIDFunc == nil {
|
||||||
panic("RepoMock.FindRequestLogByIDFunc: method is nil but Repository.FindRequestLogByID was just called")
|
panic("RepoMock.FindRequestLogByIDFunc: method is nil but Repository.FindRequestLogByID was just called")
|
||||||
}
|
}
|
||||||
callInfo := struct {
|
callInfo := struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
ID int64
|
ID ulid.ULID
|
||||||
}{
|
}{
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
ID: id,
|
ID: id,
|
||||||
@ -290,11 +165,11 @@ func (mock *RepoMock) FindRequestLogByID(ctx context.Context, id int64) (reqlog.
|
|||||||
// len(mockedRepository.FindRequestLogByIDCalls())
|
// len(mockedRepository.FindRequestLogByIDCalls())
|
||||||
func (mock *RepoMock) FindRequestLogByIDCalls() []struct {
|
func (mock *RepoMock) FindRequestLogByIDCalls() []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
ID int64
|
ID ulid.ULID
|
||||||
} {
|
} {
|
||||||
var calls []struct {
|
var calls []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
ID int64
|
ID ulid.ULID
|
||||||
}
|
}
|
||||||
mock.lockFindRequestLogByID.RLock()
|
mock.lockFindRequestLogByID.RLock()
|
||||||
calls = mock.calls.FindRequestLogByID
|
calls = mock.calls.FindRequestLogByID
|
||||||
@ -303,7 +178,7 @@ func (mock *RepoMock) FindRequestLogByIDCalls() []struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// FindRequestLogs calls FindRequestLogsFunc.
|
// FindRequestLogs calls FindRequestLogsFunc.
|
||||||
func (mock *RepoMock) FindRequestLogs(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.Request, error) {
|
func (mock *RepoMock) FindRequestLogs(ctx context.Context, filter reqlog.FindRequestsFilter, scopeMoqParam *scope.Scope) ([]reqlog.RequestLog, error) {
|
||||||
if mock.FindRequestLogsFunc == nil {
|
if mock.FindRequestLogsFunc == nil {
|
||||||
panic("RepoMock.FindRequestLogsFunc: method is nil but Repository.FindRequestLogs was just called")
|
panic("RepoMock.FindRequestLogsFunc: method is nil but Repository.FindRequestLogs was just called")
|
||||||
}
|
}
|
||||||
@ -341,80 +216,76 @@ func (mock *RepoMock) FindRequestLogsCalls() []struct {
|
|||||||
return calls
|
return calls
|
||||||
}
|
}
|
||||||
|
|
||||||
// FindSettingsByModule calls FindSettingsByModuleFunc.
|
// StoreRequestLog calls StoreRequestLogFunc.
|
||||||
func (mock *RepoMock) FindSettingsByModule(ctx context.Context, module string, settings interface{}) error {
|
func (mock *RepoMock) StoreRequestLog(ctx context.Context, reqLog reqlog.RequestLog) error {
|
||||||
if mock.FindSettingsByModuleFunc == nil {
|
if mock.StoreRequestLogFunc == nil {
|
||||||
panic("RepoMock.FindSettingsByModuleFunc: method is nil but Repository.FindSettingsByModule was just called")
|
panic("RepoMock.StoreRequestLogFunc: method is nil but Repository.StoreRequestLog was just called")
|
||||||
}
|
}
|
||||||
callInfo := struct {
|
callInfo := struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLog reqlog.RequestLog
|
||||||
Settings interface{}
|
|
||||||
}{
|
}{
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
Module: module,
|
ReqLog: reqLog,
|
||||||
Settings: settings,
|
|
||||||
}
|
}
|
||||||
mock.lockFindSettingsByModule.Lock()
|
mock.lockStoreRequestLog.Lock()
|
||||||
mock.calls.FindSettingsByModule = append(mock.calls.FindSettingsByModule, callInfo)
|
mock.calls.StoreRequestLog = append(mock.calls.StoreRequestLog, callInfo)
|
||||||
mock.lockFindSettingsByModule.Unlock()
|
mock.lockStoreRequestLog.Unlock()
|
||||||
return mock.FindSettingsByModuleFunc(ctx, module, settings)
|
return mock.StoreRequestLogFunc(ctx, reqLog)
|
||||||
}
|
}
|
||||||
|
|
||||||
// FindSettingsByModuleCalls gets all the calls that were made to FindSettingsByModule.
|
// StoreRequestLogCalls gets all the calls that were made to StoreRequestLog.
|
||||||
// Check the length with:
|
// Check the length with:
|
||||||
// len(mockedRepository.FindSettingsByModuleCalls())
|
// len(mockedRepository.StoreRequestLogCalls())
|
||||||
func (mock *RepoMock) FindSettingsByModuleCalls() []struct {
|
func (mock *RepoMock) StoreRequestLogCalls() []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLog reqlog.RequestLog
|
||||||
Settings interface{}
|
|
||||||
} {
|
} {
|
||||||
var calls []struct {
|
var calls []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLog reqlog.RequestLog
|
||||||
Settings interface{}
|
|
||||||
}
|
}
|
||||||
mock.lockFindSettingsByModule.RLock()
|
mock.lockStoreRequestLog.RLock()
|
||||||
calls = mock.calls.FindSettingsByModule
|
calls = mock.calls.StoreRequestLog
|
||||||
mock.lockFindSettingsByModule.RUnlock()
|
mock.lockStoreRequestLog.RUnlock()
|
||||||
return calls
|
return calls
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpsertSettings calls UpsertSettingsFunc.
|
// StoreResponseLog calls StoreResponseLogFunc.
|
||||||
func (mock *RepoMock) UpsertSettings(ctx context.Context, module string, settings interface{}) error {
|
func (mock *RepoMock) StoreResponseLog(ctx context.Context, reqLogID ulid.ULID, resLog reqlog.ResponseLog) error {
|
||||||
if mock.UpsertSettingsFunc == nil {
|
if mock.StoreResponseLogFunc == nil {
|
||||||
panic("RepoMock.UpsertSettingsFunc: method is nil but Repository.UpsertSettings was just called")
|
panic("RepoMock.StoreResponseLogFunc: method is nil but Repository.StoreResponseLog was just called")
|
||||||
}
|
}
|
||||||
callInfo := struct {
|
callInfo := struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLogID ulid.ULID
|
||||||
Settings interface{}
|
ResLog reqlog.ResponseLog
|
||||||
}{
|
}{
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
Module: module,
|
ReqLogID: reqLogID,
|
||||||
Settings: settings,
|
ResLog: resLog,
|
||||||
}
|
}
|
||||||
mock.lockUpsertSettings.Lock()
|
mock.lockStoreResponseLog.Lock()
|
||||||
mock.calls.UpsertSettings = append(mock.calls.UpsertSettings, callInfo)
|
mock.calls.StoreResponseLog = append(mock.calls.StoreResponseLog, callInfo)
|
||||||
mock.lockUpsertSettings.Unlock()
|
mock.lockStoreResponseLog.Unlock()
|
||||||
return mock.UpsertSettingsFunc(ctx, module, settings)
|
return mock.StoreResponseLogFunc(ctx, reqLogID, resLog)
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpsertSettingsCalls gets all the calls that were made to UpsertSettings.
|
// StoreResponseLogCalls gets all the calls that were made to StoreResponseLog.
|
||||||
// Check the length with:
|
// Check the length with:
|
||||||
// len(mockedRepository.UpsertSettingsCalls())
|
// len(mockedRepository.StoreResponseLogCalls())
|
||||||
func (mock *RepoMock) UpsertSettingsCalls() []struct {
|
func (mock *RepoMock) StoreResponseLogCalls() []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLogID ulid.ULID
|
||||||
Settings interface{}
|
ResLog reqlog.ResponseLog
|
||||||
} {
|
} {
|
||||||
var calls []struct {
|
var calls []struct {
|
||||||
Ctx context.Context
|
Ctx context.Context
|
||||||
Module string
|
ReqLogID ulid.ULID
|
||||||
Settings interface{}
|
ResLog reqlog.ResponseLog
|
||||||
}
|
}
|
||||||
mock.lockUpsertSettings.RLock()
|
mock.lockStoreResponseLog.RLock()
|
||||||
calls = mock.calls.UpsertSettings
|
calls = mock.calls.StoreResponseLog
|
||||||
mock.lockUpsertSettings.RUnlock()
|
mock.lockStoreResponseLog.RUnlock()
|
||||||
return calls
|
return calls
|
||||||
}
|
}
|
||||||
|
@ -4,15 +4,18 @@ import (
|
|||||||
"bytes"
|
"bytes"
|
||||||
"compress/gzip"
|
"compress/gzip"
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"log"
|
"log"
|
||||||
|
"math/rand"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proxy"
|
"github.com/dstotijn/hetty/pkg/proxy"
|
||||||
"github.com/dstotijn/hetty/pkg/scope"
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
"github.com/dstotijn/hetty/pkg/search"
|
"github.com/dstotijn/hetty/pkg/search"
|
||||||
@ -22,127 +25,109 @@ type contextKey int
|
|||||||
|
|
||||||
const LogBypassedKey contextKey = 0
|
const LogBypassedKey contextKey = 0
|
||||||
|
|
||||||
const moduleName = "reqlog"
|
var (
|
||||||
|
ErrRequestNotFound = errors.New("reqlog: request not found")
|
||||||
|
ErrProjectIDMustBeSet = errors.New("reqlog: project ID must be set")
|
||||||
|
)
|
||||||
|
|
||||||
var ErrRequestNotFound = errors.New("reqlog: request not found")
|
//nolint:gosec
|
||||||
|
var ulidEntropy = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||||
|
|
||||||
type Request struct {
|
type RequestLog struct {
|
||||||
ID int64
|
ID ulid.ULID
|
||||||
Request http.Request
|
ProjectID ulid.ULID
|
||||||
Body []byte
|
|
||||||
Timestamp time.Time
|
URL *url.URL
|
||||||
Response *Response
|
Method string
|
||||||
|
Proto string
|
||||||
|
Header http.Header
|
||||||
|
Body []byte
|
||||||
|
|
||||||
|
Response *ResponseLog
|
||||||
}
|
}
|
||||||
|
|
||||||
type Response struct {
|
type ResponseLog struct {
|
||||||
ID int64
|
Proto string
|
||||||
RequestID int64
|
StatusCode int
|
||||||
Response http.Response
|
Status string
|
||||||
Body []byte
|
Header http.Header
|
||||||
Timestamp time.Time
|
Body []byte
|
||||||
}
|
}
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
BypassOutOfScopeRequests bool
|
BypassOutOfScopeRequests bool
|
||||||
FindReqsFilter FindRequestsFilter
|
FindReqsFilter FindRequestsFilter
|
||||||
|
ActiveProjectID ulid.ULID
|
||||||
|
|
||||||
scope *scope.Scope
|
scope *scope.Scope
|
||||||
repo Repository
|
repo Repository
|
||||||
}
|
}
|
||||||
|
|
||||||
type FindRequestsFilter struct {
|
type FindRequestsFilter struct {
|
||||||
OnlyInScope bool
|
ProjectID ulid.ULID
|
||||||
SearchExpr search.Expression `json:"-"`
|
OnlyInScope bool
|
||||||
RawSearchExpr string
|
SearchExpr search.Expression
|
||||||
}
|
}
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
Scope *scope.Scope
|
Scope *scope.Scope
|
||||||
Repository Repository
|
Repository Repository
|
||||||
ProjectService proj.Service
|
|
||||||
BypassOutOfScopeRequests bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewService(cfg Config) *Service {
|
func NewService(cfg Config) *Service {
|
||||||
svc := &Service{
|
return &Service{
|
||||||
scope: cfg.Scope,
|
repo: cfg.Repository,
|
||||||
repo: cfg.Repository,
|
scope: cfg.Scope,
|
||||||
BypassOutOfScopeRequests: cfg.BypassOutOfScopeRequests,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
cfg.ProjectService.OnProjectOpen(func(_ string) error {
|
|
||||||
err := svc.repo.FindSettingsByModule(context.Background(), moduleName, svc)
|
|
||||||
if errors.Is(err, proj.ErrNoSettings) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("reqlog: could not load settings: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
cfg.ProjectService.OnProjectClose(func(_ string) error {
|
|
||||||
svc.BypassOutOfScopeRequests = false
|
|
||||||
svc.FindReqsFilter = FindRequestsFilter{}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
|
|
||||||
return svc
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) FindRequests(ctx context.Context) ([]Request, error) {
|
func (svc *Service) FindRequests(ctx context.Context) ([]RequestLog, error) {
|
||||||
return svc.repo.FindRequestLogs(ctx, svc.FindReqsFilter, svc.scope)
|
return svc.repo.FindRequestLogs(ctx, svc.FindReqsFilter, svc.scope)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) FindRequestLogByID(ctx context.Context, id int64) (Request, error) {
|
func (svc *Service) FindRequestLogByID(ctx context.Context, id ulid.ULID) (RequestLog, error) {
|
||||||
return svc.repo.FindRequestLogByID(ctx, id)
|
return svc.repo.FindRequestLogByID(ctx, id)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) SetRequestLogFilter(ctx context.Context, filter FindRequestsFilter) error {
|
func (svc *Service) ClearRequests(ctx context.Context, projectID ulid.ULID) error {
|
||||||
svc.FindReqsFilter = filter
|
return svc.repo.ClearRequestLogs(ctx, projectID)
|
||||||
return svc.repo.UpsertSettings(ctx, "reqlog", svc)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) ClearRequests(ctx context.Context) error {
|
func (svc *Service) storeResponse(ctx context.Context, reqLogID ulid.ULID, res *http.Response) error {
|
||||||
return svc.repo.ClearRequestLogs(ctx)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (svc *Service) addRequest(
|
|
||||||
ctx context.Context,
|
|
||||||
req http.Request,
|
|
||||||
body []byte,
|
|
||||||
timestamp time.Time,
|
|
||||||
) (*Request, error) {
|
|
||||||
return svc.repo.AddRequestLog(ctx, req, body, timestamp)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (svc *Service) addResponse(
|
|
||||||
ctx context.Context,
|
|
||||||
reqID int64,
|
|
||||||
res http.Response,
|
|
||||||
body []byte,
|
|
||||||
timestamp time.Time,
|
|
||||||
) (*Response, error) {
|
|
||||||
if res.Header.Get("Content-Encoding") == "gzip" {
|
if res.Header.Get("Content-Encoding") == "gzip" {
|
||||||
gzipReader, err := gzip.NewReader(bytes.NewBuffer(body))
|
gzipReader, err := gzip.NewReader(res.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("reqlog: could not create gzip reader: %w", err)
|
return fmt.Errorf("could not create gzip reader: %w", err)
|
||||||
}
|
}
|
||||||
defer gzipReader.Close()
|
defer gzipReader.Close()
|
||||||
|
|
||||||
body, err = ioutil.ReadAll(gzipReader)
|
buf := &bytes.Buffer{}
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("reqlog: could not read gzipped response body: %w", err)
|
if _, err := io.Copy(buf, gzipReader); err != nil {
|
||||||
|
return fmt.Errorf("could not read gzipped response body: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
res.Body = io.NopCloser(buf)
|
||||||
}
|
}
|
||||||
|
|
||||||
return svc.repo.AddResponseLog(ctx, reqID, res, body, timestamp)
|
body, err := io.ReadAll(res.Body)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("could not read body: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resLog := ResponseLog{
|
||||||
|
Proto: res.Proto,
|
||||||
|
StatusCode: res.StatusCode,
|
||||||
|
Status: res.Status,
|
||||||
|
Header: res.Header,
|
||||||
|
Body: body,
|
||||||
|
}
|
||||||
|
|
||||||
|
return svc.repo.StoreResponseLog(ctx, reqLogID, resLog)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) RequestModifier(next proxy.RequestModifyFunc) proxy.RequestModifyFunc {
|
func (svc *Service) RequestModifier(next proxy.RequestModifyFunc) proxy.RequestModifyFunc {
|
||||||
return func(req *http.Request) {
|
return func(req *http.Request) {
|
||||||
now := time.Now()
|
|
||||||
|
|
||||||
next(req)
|
next(req)
|
||||||
|
|
||||||
clone := req.Clone(req.Context())
|
clone := req.Clone(req.Context())
|
||||||
@ -160,10 +145,19 @@ func (svc *Service) RequestModifier(next proxy.RequestModifyFunc) proxy.RequestM
|
|||||||
}
|
}
|
||||||
|
|
||||||
req.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
req.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
||||||
|
clone.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bypass logging if no project is active.
|
||||||
|
if svc.ActiveProjectID.Compare(ulid.ULID{}) == 0 {
|
||||||
|
ctx := context.WithValue(req.Context(), LogBypassedKey, true)
|
||||||
|
*req = *req.WithContext(ctx)
|
||||||
|
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Bypass logging if this setting is enabled and the incoming request
|
// Bypass logging if this setting is enabled and the incoming request
|
||||||
// doens't match any rules of the scope.
|
// doesn't match any scope rules.
|
||||||
if svc.BypassOutOfScopeRequests && !svc.scope.Match(clone, body) {
|
if svc.BypassOutOfScopeRequests && !svc.scope.Match(clone, body) {
|
||||||
ctx := context.WithValue(req.Context(), LogBypassedKey, true)
|
ctx := context.WithValue(req.Context(), LogBypassedKey, true)
|
||||||
*req = *req.WithContext(ctx)
|
*req = *req.WithContext(ctx)
|
||||||
@ -171,26 +165,29 @@ func (svc *Service) RequestModifier(next proxy.RequestModifyFunc) proxy.RequestM
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
reqLog, err := svc.addRequest(req.Context(), *clone, body, now)
|
reqLog := RequestLog{
|
||||||
if errors.Is(err, proj.ErrNoProject) {
|
ID: ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy),
|
||||||
ctx := context.WithValue(req.Context(), LogBypassedKey, true)
|
ProjectID: svc.ActiveProjectID,
|
||||||
*req = *req.WithContext(ctx)
|
Method: clone.Method,
|
||||||
|
URL: clone.URL,
|
||||||
|
Proto: clone.Proto,
|
||||||
|
Header: clone.Header,
|
||||||
|
Body: body,
|
||||||
|
}
|
||||||
|
|
||||||
return
|
err := svc.repo.StoreRequestLog(req.Context(), reqLog)
|
||||||
} else if err != nil {
|
if err != nil {
|
||||||
log.Printf("[ERROR] Could not store request log: %v", err)
|
log.Printf("[ERROR] Could not store request log: %v", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx := context.WithValue(req.Context(), proxy.ReqIDKey, reqLog.ID)
|
ctx := context.WithValue(req.Context(), proxy.ReqLogIDKey, reqLog.ID)
|
||||||
*req = *req.WithContext(ctx)
|
*req = *req.WithContext(ctx)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (svc *Service) ResponseModifier(next proxy.ResponseModifyFunc) proxy.ResponseModifyFunc {
|
func (svc *Service) ResponseModifier(next proxy.ResponseModifyFunc) proxy.ResponseModifyFunc {
|
||||||
return func(res *http.Response) error {
|
return func(res *http.Response) error {
|
||||||
now := time.Now()
|
|
||||||
|
|
||||||
if err := next(res); err != nil {
|
if err := next(res); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@ -199,8 +196,8 @@ func (svc *Service) ResponseModifier(next proxy.ResponseModifyFunc) proxy.Respon
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
reqID, _ := res.Request.Context().Value(proxy.ReqIDKey).(int64)
|
reqLogID, ok := res.Request.Context().Value(proxy.ReqLogIDKey).(ulid.ULID)
|
||||||
if reqID == 0 {
|
if !ok {
|
||||||
return errors.New("reqlog: request is missing ID")
|
return errors.New("reqlog: request is missing ID")
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -213,9 +210,10 @@ func (svc *Service) ResponseModifier(next proxy.ResponseModifyFunc) proxy.Respon
|
|||||||
}
|
}
|
||||||
|
|
||||||
res.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
res.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
||||||
|
clone.Body = ioutil.NopCloser(bytes.NewBuffer(body))
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
if _, err := svc.addResponse(context.Background(), reqID, clone, body, now); err != nil {
|
if err := svc.storeResponse(context.Background(), reqLogID, &clone); err != nil {
|
||||||
log.Printf("[ERROR] Could not store response log: %v", err)
|
log.Printf("[ERROR] Could not store response log: %v", err)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
@ -223,33 +221,3 @@ func (svc *Service) ResponseModifier(next proxy.ResponseModifyFunc) proxy.Respon
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnmarshalJSON implements json.Unmarshaler.
|
|
||||||
func (f *FindRequestsFilter) UnmarshalJSON(b []byte) error {
|
|
||||||
var dto struct {
|
|
||||||
OnlyInScope bool
|
|
||||||
RawSearchExpr string
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := json.Unmarshal(b, &dto); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
filter := FindRequestsFilter{
|
|
||||||
OnlyInScope: dto.OnlyInScope,
|
|
||||||
RawSearchExpr: dto.RawSearchExpr,
|
|
||||||
}
|
|
||||||
|
|
||||||
if dto.RawSearchExpr != "" {
|
|
||||||
expr, err := search.ParseQuery(dto.RawSearchExpr)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
filter.SearchExpr = expr
|
|
||||||
}
|
|
||||||
|
|
||||||
*f = filter
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
@ -1,124 +1,43 @@
|
|||||||
package reqlog_test
|
package reqlog_test
|
||||||
|
|
||||||
//go:generate moq -out proj_mock_test.go -pkg reqlog_test ../proj Service:ProjServiceMock
|
//go:generate go run github.com/matryer/moq -out repo_mock_test.go -pkg reqlog_test . Repository:RepoMock
|
||||||
//go:generate moq -out repo_mock_test.go -pkg reqlog_test . Repository:RepoMock
|
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"io/ioutil"
|
"io"
|
||||||
|
"math/rand"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
"github.com/google/go-cmp/cmp"
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proxy"
|
"github.com/dstotijn/hetty/pkg/proxy"
|
||||||
"github.com/dstotijn/hetty/pkg/reqlog"
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
//nolint:gosec
|
||||||
|
var ulidEntropy = rand.New(rand.NewSource(time.Now().UnixNano()))
|
||||||
|
|
||||||
//nolint:paralleltest
|
//nolint:paralleltest
|
||||||
func TestNewService(t *testing.T) {
|
func TestRequestModifier(t *testing.T) {
|
||||||
projSvcMock := &ProjServiceMock{
|
|
||||||
OnProjectOpenFunc: func(fn proj.OnProjectOpenFn) {},
|
|
||||||
OnProjectCloseFunc: func(fn proj.OnProjectCloseFn) {},
|
|
||||||
}
|
|
||||||
repoMock := &RepoMock{
|
repoMock := &RepoMock{
|
||||||
FindSettingsByModuleFunc: func(_ context.Context, _ string, _ interface{}) error {
|
StoreRequestLogFunc: func(_ context.Context, _ reqlog.RequestLog) error {
|
||||||
return nil
|
return nil
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
svc := reqlog.NewService(reqlog.Config{
|
svc := reqlog.NewService(reqlog.Config{
|
||||||
ProjectService: projSvcMock,
|
Repository: repoMock,
|
||||||
Repository: repoMock,
|
Scope: &scope.Scope{},
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("registered handlers for project open and close", func(t *testing.T) {
|
|
||||||
got := len(projSvcMock.OnProjectOpenCalls())
|
|
||||||
if exp := 1; exp != got {
|
|
||||||
t.Fatalf("incorrect `proj.Service.OnProjectOpen` calls (expected: %v, got: %v)", exp, got)
|
|
||||||
}
|
|
||||||
|
|
||||||
got = len(projSvcMock.OnProjectCloseCalls())
|
|
||||||
if exp := 1; exp != got {
|
|
||||||
t.Fatalf("incorrect `proj.Service.OnProjectClose` calls (expected: %v, got: %v)", exp, got)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("calls handler when project is opened", func(t *testing.T) {
|
|
||||||
// Mock opening a project.
|
|
||||||
err := projSvcMock.OnProjectOpenCalls()[0].Fn("foobar")
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("unexpected error (expected: nil, got: %v)", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assert that settings were fetched from repository, with `svc` as the
|
|
||||||
// destination.
|
|
||||||
got := len(repoMock.FindSettingsByModuleCalls())
|
|
||||||
if exp := 1; exp != got {
|
|
||||||
t.Fatalf("incorrect `proj.Service.OnProjectOpen` calls (expected: %v, got: %v)", exp, got)
|
|
||||||
}
|
|
||||||
|
|
||||||
findSettingsByModuleCall := repoMock.FindSettingsByModuleCalls()[0]
|
|
||||||
expModule := "reqlog"
|
|
||||||
expSettings := svc
|
|
||||||
|
|
||||||
if expModule != findSettingsByModuleCall.Module {
|
|
||||||
t.Fatalf("incorrect `module` argument for `proj.Service.OnProjectOpen` (expected: %v, got: %v)",
|
|
||||||
expModule, findSettingsByModuleCall.Module)
|
|
||||||
}
|
|
||||||
|
|
||||||
if expSettings != findSettingsByModuleCall.Settings {
|
|
||||||
t.Fatalf("incorrect `settings` argument for `proj.Service.OnProjectOpen` (expected: %v, got: %v)",
|
|
||||||
expModule, findSettingsByModuleCall.Settings)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("calls handler when project is closed", func(t *testing.T) {
|
|
||||||
// Mock updating service settings.
|
|
||||||
svc.BypassOutOfScopeRequests = true
|
|
||||||
svc.FindReqsFilter = reqlog.FindRequestsFilter{OnlyInScope: true}
|
|
||||||
|
|
||||||
// Mock closing a project.
|
|
||||||
err := projSvcMock.OnProjectCloseCalls()[0].Fn("foobar")
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("unexpected error (expected: nil, got: %v)", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Assert that settings were set to defaults on project close.
|
|
||||||
expBypassOutOfScopeReqs := false
|
|
||||||
expFindReqsFilter := reqlog.FindRequestsFilter{}
|
|
||||||
|
|
||||||
if expBypassOutOfScopeReqs != svc.BypassOutOfScopeRequests {
|
|
||||||
t.Fatalf("incorrect `Service.BypassOutOfScopeRequests` value (expected: %v, got: %v)",
|
|
||||||
expBypassOutOfScopeReqs, svc.BypassOutOfScopeRequests)
|
|
||||||
}
|
|
||||||
|
|
||||||
if expFindReqsFilter != svc.FindReqsFilter {
|
|
||||||
t.Fatalf("incorrect `Service.FindReqsFilter` value (expected: %v, got: %v)",
|
|
||||||
expFindReqsFilter, svc.FindReqsFilter)
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
//nolint:paralleltest
|
|
||||||
func TestRequestModifier(t *testing.T) {
|
|
||||||
projSvcMock := &ProjServiceMock{
|
|
||||||
OnProjectOpenFunc: func(fn proj.OnProjectOpenFn) {},
|
|
||||||
OnProjectCloseFunc: func(fn proj.OnProjectCloseFn) {},
|
|
||||||
}
|
|
||||||
repoMock := &RepoMock{
|
|
||||||
AddRequestLogFunc: func(_ context.Context, _ http.Request, _ []byte, _ time.Time) (*reqlog.Request, error) {
|
|
||||||
return &reqlog.Request{}, nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
svc := reqlog.NewService(reqlog.Config{
|
|
||||||
ProjectService: projSvcMock,
|
|
||||||
Repository: repoMock,
|
|
||||||
})
|
})
|
||||||
|
svc.ActiveProjectID = ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
|
||||||
next := func(req *http.Request) {
|
next := func(req *http.Request) {
|
||||||
req.Body = ioutil.NopCloser(strings.NewReader("modified body"))
|
req.Body = io.NopCloser(strings.NewReader("modified body"))
|
||||||
}
|
}
|
||||||
reqModFn := svc.RequestModifier(next)
|
reqModFn := svc.RequestModifier(next)
|
||||||
req := httptest.NewRequest("GET", "https://example.com/", strings.NewReader("bar"))
|
req := httptest.NewRequest("GET", "https://example.com/", strings.NewReader("bar"))
|
||||||
@ -126,49 +45,54 @@ func TestRequestModifier(t *testing.T) {
|
|||||||
reqModFn(req)
|
reqModFn(req)
|
||||||
|
|
||||||
t.Run("request log was stored in repository", func(t *testing.T) {
|
t.Run("request log was stored in repository", func(t *testing.T) {
|
||||||
got := len(repoMock.AddRequestLogCalls())
|
gotCount := len(repoMock.StoreRequestLogCalls())
|
||||||
if exp := 1; exp != got {
|
if expCount := 1; expCount != gotCount {
|
||||||
t.Fatalf("incorrect `proj.Service.AddRequestLog` calls (expected: %v, got: %v)", exp, got)
|
t.Fatalf("incorrect `proj.Service.AddRequestLog` calls (expected: %v, got: %v)", expCount, gotCount)
|
||||||
}
|
}
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("ran next modifier first, before calling repository", func(t *testing.T) {
|
exp := reqlog.RequestLog{
|
||||||
got := repoMock.AddRequestLogCalls()[0].Body
|
ID: ulid.ULID{}, // Empty value
|
||||||
if exp := "modified body"; exp != string(got) {
|
ProjectID: svc.ActiveProjectID,
|
||||||
t.Fatalf("incorrect `body` argument for `Repository.AddRequestLogCalls` (expected: %v, got: %v)", exp, string(got))
|
Method: req.Method,
|
||||||
|
URL: req.URL,
|
||||||
|
Proto: req.Proto,
|
||||||
|
Header: req.Header,
|
||||||
|
Body: []byte("modified body"),
|
||||||
|
}
|
||||||
|
got := repoMock.StoreRequestLogCalls()[0].ReqLog
|
||||||
|
got.ID = ulid.ULID{} // Override to empty value so we can compare against expected value.
|
||||||
|
|
||||||
|
if diff := cmp.Diff(exp, got); diff != "" {
|
||||||
|
t.Fatalf("request log not equal (-exp, +got):\n%v", diff)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
//nolint:paralleltest
|
//nolint:paralleltest
|
||||||
func TestResponseModifier(t *testing.T) {
|
func TestResponseModifier(t *testing.T) {
|
||||||
projSvcMock := &ProjServiceMock{
|
|
||||||
OnProjectOpenFunc: func(fn proj.OnProjectOpenFn) {},
|
|
||||||
OnProjectCloseFunc: func(fn proj.OnProjectCloseFn) {},
|
|
||||||
}
|
|
||||||
repoMock := &RepoMock{
|
repoMock := &RepoMock{
|
||||||
AddResponseLogFunc: func(_ context.Context, _ int64, _ http.Response,
|
StoreResponseLogFunc: func(_ context.Context, _ ulid.ULID, _ reqlog.ResponseLog) error {
|
||||||
_ []byte, _ time.Time) (*reqlog.Response, error) {
|
return nil
|
||||||
return &reqlog.Response{}, nil
|
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
svc := reqlog.NewService(reqlog.Config{
|
svc := reqlog.NewService(reqlog.Config{
|
||||||
ProjectService: projSvcMock,
|
Repository: repoMock,
|
||||||
Repository: repoMock,
|
|
||||||
})
|
})
|
||||||
|
svc.ActiveProjectID = ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
|
||||||
next := func(res *http.Response) error {
|
next := func(res *http.Response) error {
|
||||||
res.Body = ioutil.NopCloser(strings.NewReader("modified body"))
|
res.Body = io.NopCloser(strings.NewReader("modified body"))
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
resModFn := svc.ResponseModifier(next)
|
resModFn := svc.ResponseModifier(next)
|
||||||
|
|
||||||
req := httptest.NewRequest("GET", "https://example.com/", strings.NewReader("bar"))
|
req := httptest.NewRequest("GET", "https://example.com/", strings.NewReader("bar"))
|
||||||
req = req.WithContext(context.WithValue(req.Context(), proxy.ReqIDKey, int64(42)))
|
reqLogID := ulid.MustNew(ulid.Timestamp(time.Now()), ulidEntropy)
|
||||||
|
req = req.WithContext(context.WithValue(req.Context(), proxy.ReqLogIDKey, reqLogID))
|
||||||
|
|
||||||
res := &http.Response{
|
res := &http.Response{
|
||||||
Request: req,
|
Request: req,
|
||||||
Body: ioutil.NopCloser(strings.NewReader("bar")),
|
Body: io.NopCloser(strings.NewReader("bar")),
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := resModFn(res); err != nil {
|
if err := resModFn(res); err != nil {
|
||||||
@ -178,16 +102,23 @@ func TestResponseModifier(t *testing.T) {
|
|||||||
t.Run("request log was stored in repository", func(t *testing.T) {
|
t.Run("request log was stored in repository", func(t *testing.T) {
|
||||||
// Dirty (but simple) wait for other goroutine to finish calling repository.
|
// Dirty (but simple) wait for other goroutine to finish calling repository.
|
||||||
time.Sleep(10 * time.Millisecond)
|
time.Sleep(10 * time.Millisecond)
|
||||||
got := len(repoMock.AddResponseLogCalls())
|
got := len(repoMock.StoreResponseLogCalls())
|
||||||
if exp := 1; exp != got {
|
if exp := 1; exp != got {
|
||||||
t.Fatalf("incorrect `proj.Service.AddResponseLog` calls (expected: %v, got: %v)", exp, got)
|
t.Fatalf("incorrect `proj.Service.AddResponseLog` calls (expected: %v, got: %v)", exp, got)
|
||||||
}
|
}
|
||||||
})
|
|
||||||
|
|
||||||
t.Run("ran next modifier first, before calling repository", func(t *testing.T) {
|
t.Run("ran next modifier first, before calling repository", func(t *testing.T) {
|
||||||
got := repoMock.AddResponseLogCalls()[0].Body
|
got := repoMock.StoreResponseLogCalls()[0].ResLog.Body
|
||||||
if exp := "modified body"; exp != string(got) {
|
if exp := "modified body"; exp != string(got) {
|
||||||
t.Fatalf("incorrect `body` argument for `Repository.AddResponseLogCalls` (expected: %v, got: %v)", exp, string(got))
|
t.Fatalf("incorrect `ResponseLog.Body` value (expected: %v, got: %v)", exp, string(got))
|
||||||
}
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("called repository with request log id", func(t *testing.T) {
|
||||||
|
got := repoMock.StoreResponseLogCalls()[0].ReqLogID
|
||||||
|
if exp := reqLogID; exp.Compare(got) != 0 {
|
||||||
|
t.Fatalf("incorrect `reqLogID` argument for `Repository.AddResponseLogCalls` (expected: %v, got: %v)", exp.String(), got.String())
|
||||||
|
}
|
||||||
|
})
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
235
pkg/reqlog/search.go
Normal file
235
pkg/reqlog/search.go
Normal file
@ -0,0 +1,235 @@
|
|||||||
|
package reqlog
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/oklog/ulid"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/scope"
|
||||||
|
"github.com/dstotijn/hetty/pkg/search"
|
||||||
|
)
|
||||||
|
|
||||||
|
var reqLogSearchKeyFns = map[string]func(rl RequestLog) string{
|
||||||
|
"req.id": func(rl RequestLog) string { return rl.ID.String() },
|
||||||
|
"req.proto": func(rl RequestLog) string { return rl.Proto },
|
||||||
|
"req.url": func(rl RequestLog) string {
|
||||||
|
if rl.URL == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return rl.URL.String()
|
||||||
|
},
|
||||||
|
"req.method": func(rl RequestLog) string { return rl.Method },
|
||||||
|
"req.body": func(rl RequestLog) string { return string(rl.Body) },
|
||||||
|
"req.timestamp": func(rl RequestLog) string { return ulid.Time(rl.ID.Time()).String() },
|
||||||
|
}
|
||||||
|
|
||||||
|
var resLogSearchKeyFns = map[string]func(rl ResponseLog) string{
|
||||||
|
"res.proto": func(rl ResponseLog) string { return rl.Proto },
|
||||||
|
"res.statusCode": func(rl ResponseLog) string { return strconv.Itoa(rl.StatusCode) },
|
||||||
|
"res.statusReason": func(rl ResponseLog) string { return rl.Status },
|
||||||
|
"res.body": func(rl ResponseLog) string { return string(rl.Body) },
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: Request and response headers search key functions.
|
||||||
|
|
||||||
|
// Matches returns true if the supplied search expression evaluates to true.
|
||||||
|
func (reqLog RequestLog) Matches(expr search.Expression) (bool, error) {
|
||||||
|
switch e := expr.(type) {
|
||||||
|
case search.PrefixExpression:
|
||||||
|
return reqLog.matchPrefixExpr(e)
|
||||||
|
case search.InfixExpression:
|
||||||
|
return reqLog.matchInfixExpr(e)
|
||||||
|
case search.StringLiteral:
|
||||||
|
return reqLog.matchStringLiteral(e)
|
||||||
|
default:
|
||||||
|
return false, fmt.Errorf("expression type (%T) not supported", expr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (reqLog RequestLog) matchPrefixExpr(expr search.PrefixExpression) (bool, error) {
|
||||||
|
switch expr.Operator {
|
||||||
|
case search.TokOpNot:
|
||||||
|
match, err := reqLog.Matches(expr.Right)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return !match, nil
|
||||||
|
default:
|
||||||
|
return false, errors.New("operator is not supported")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (reqLog RequestLog) matchInfixExpr(expr search.InfixExpression) (bool, error) {
|
||||||
|
switch expr.Operator {
|
||||||
|
case search.TokOpAnd:
|
||||||
|
left, err := reqLog.Matches(expr.Left)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
right, err := reqLog.Matches(expr.Right)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return left && right, nil
|
||||||
|
case search.TokOpOr:
|
||||||
|
left, err := reqLog.Matches(expr.Left)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
right, err := reqLog.Matches(expr.Right)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return left || right, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
left, ok := expr.Left.(search.StringLiteral)
|
||||||
|
if !ok {
|
||||||
|
return false, errors.New("left operand must be a string literal")
|
||||||
|
}
|
||||||
|
|
||||||
|
leftVal := reqLog.getMappedStringLiteral(left.Value)
|
||||||
|
|
||||||
|
if expr.Operator == search.TokOpRe || expr.Operator == search.TokOpNotRe {
|
||||||
|
right, ok := expr.Right.(*regexp.Regexp)
|
||||||
|
if !ok {
|
||||||
|
return false, errors.New("right operand must be a regular expression")
|
||||||
|
}
|
||||||
|
|
||||||
|
switch expr.Operator {
|
||||||
|
case search.TokOpRe:
|
||||||
|
return right.MatchString(leftVal), nil
|
||||||
|
case search.TokOpNotRe:
|
||||||
|
return !right.MatchString(leftVal), nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
right, ok := expr.Right.(search.StringLiteral)
|
||||||
|
if !ok {
|
||||||
|
return false, errors.New("right operand must be a string literal")
|
||||||
|
}
|
||||||
|
|
||||||
|
rightVal := reqLog.getMappedStringLiteral(right.Value)
|
||||||
|
|
||||||
|
switch expr.Operator {
|
||||||
|
case search.TokOpEq:
|
||||||
|
return leftVal == rightVal, nil
|
||||||
|
case search.TokOpNotEq:
|
||||||
|
return leftVal != rightVal, nil
|
||||||
|
case search.TokOpGt:
|
||||||
|
// TODO(?) attempt to parse as int.
|
||||||
|
return leftVal > rightVal, nil
|
||||||
|
case search.TokOpLt:
|
||||||
|
// TODO(?) attempt to parse as int.
|
||||||
|
return leftVal < rightVal, nil
|
||||||
|
case search.TokOpGtEq:
|
||||||
|
// TODO(?) attempt to parse as int.
|
||||||
|
return leftVal >= rightVal, nil
|
||||||
|
case search.TokOpLtEq:
|
||||||
|
// TODO(?) attempt to parse as int.
|
||||||
|
return leftVal <= rightVal, nil
|
||||||
|
default:
|
||||||
|
return false, errors.New("unsupported operator")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (reqLog RequestLog) getMappedStringLiteral(s string) string {
|
||||||
|
switch {
|
||||||
|
case strings.HasPrefix(s, "req."):
|
||||||
|
fn, ok := reqLogSearchKeyFns[s]
|
||||||
|
if ok {
|
||||||
|
return fn(reqLog)
|
||||||
|
}
|
||||||
|
case strings.HasPrefix(s, "res."):
|
||||||
|
if reqLog.Response == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
fn, ok := resLogSearchKeyFns[s]
|
||||||
|
if ok {
|
||||||
|
return fn(*reqLog.Response)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
func (reqLog RequestLog) matchStringLiteral(strLiteral search.StringLiteral) (bool, error) {
|
||||||
|
for _, fn := range reqLogSearchKeyFns {
|
||||||
|
if strings.Contains(
|
||||||
|
strings.ToLower(fn(reqLog)),
|
||||||
|
strings.ToLower(strLiteral.Value),
|
||||||
|
) {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if reqLog.Response != nil {
|
||||||
|
for _, fn := range resLogSearchKeyFns {
|
||||||
|
if strings.Contains(
|
||||||
|
strings.ToLower(fn(*reqLog.Response)),
|
||||||
|
strings.ToLower(strLiteral.Value),
|
||||||
|
) {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (reqLog RequestLog) MatchScope(s *scope.Scope) bool {
|
||||||
|
for _, rule := range s.Rules() {
|
||||||
|
if rule.URL != nil && reqLog.URL != nil {
|
||||||
|
if matches := rule.URL.MatchString(reqLog.URL.String()); matches {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for key, values := range reqLog.Header {
|
||||||
|
var keyMatches, valueMatches bool
|
||||||
|
|
||||||
|
if rule.Header.Key != nil {
|
||||||
|
if matches := rule.Header.Key.MatchString(key); matches {
|
||||||
|
keyMatches = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if rule.Header.Value != nil {
|
||||||
|
for _, value := range values {
|
||||||
|
if matches := rule.Header.Value.MatchString(value); matches {
|
||||||
|
valueMatches = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// When only key or value is set, match on whatever is set.
|
||||||
|
// When both are set, both must match.
|
||||||
|
switch {
|
||||||
|
case rule.Header.Key != nil && rule.Header.Value == nil && keyMatches:
|
||||||
|
return true
|
||||||
|
case rule.Header.Key == nil && rule.Header.Value != nil && valueMatches:
|
||||||
|
return true
|
||||||
|
case rule.Header.Key != nil && rule.Header.Value != nil && keyMatches && valueMatches:
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if rule.Body != nil {
|
||||||
|
if matches := rule.Body.Match(reqLog.Body); matches {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
203
pkg/reqlog/search_test.go
Normal file
203
pkg/reqlog/search_test.go
Normal file
@ -0,0 +1,203 @@
|
|||||||
|
package reqlog_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/dstotijn/hetty/pkg/reqlog"
|
||||||
|
"github.com/dstotijn/hetty/pkg/search"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestRequestLogMatch(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
query string
|
||||||
|
requestLog reqlog.RequestLog
|
||||||
|
expectedMatch bool
|
||||||
|
expectedError error
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "infix expression, equal operator, match",
|
||||||
|
query: "req.body = foo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, not equal operator, match",
|
||||||
|
query: "req.body != bar",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, greater than operator, match",
|
||||||
|
query: "req.body > a",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("b"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, less than operator, match",
|
||||||
|
query: "req.body < b",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("a"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, greater than or equal operator, match greater than",
|
||||||
|
query: "req.body >= a",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("b"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, greater than or equal operator, match equal",
|
||||||
|
query: "req.body >= a",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("a"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, less than or equal operator, match less than",
|
||||||
|
query: "req.body <= b",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("a"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, less than or equal operator, match equal",
|
||||||
|
query: "req.body <= b",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("b"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, regular expression operator, match",
|
||||||
|
query: `req.body =~ "^foo(.*)$"`,
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foobar"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, negate regular expression operator, match",
|
||||||
|
query: `req.body !~ "^foo(.*)$"`,
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("xoobar"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, and operator, match",
|
||||||
|
query: "req.body = bar AND res.body = yolo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("bar"),
|
||||||
|
Response: &reqlog.ResponseLog{
|
||||||
|
Body: []byte("yolo"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "infix expression, or operator, match",
|
||||||
|
query: "req.body = bar OR res.body = yolo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
Response: &reqlog.ResponseLog{
|
||||||
|
Body: []byte("yolo"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "prefix expression, not operator, match",
|
||||||
|
query: "NOT (req.body = bar)",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "string literal expression, match in request log",
|
||||||
|
query: "foo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "string literal expression, no match",
|
||||||
|
query: "foo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Body: []byte("bar"),
|
||||||
|
},
|
||||||
|
expectedMatch: false,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "string literal expression, match in response log",
|
||||||
|
query: "foo",
|
||||||
|
requestLog: reqlog.RequestLog{
|
||||||
|
Response: &reqlog.ResponseLog{
|
||||||
|
Body: []byte("foo"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
expectedMatch: true,
|
||||||
|
expectedError: nil,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
tt := tt
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
searchExpr, err := search.ParseQuery(tt.query)
|
||||||
|
assertError(t, nil, err)
|
||||||
|
|
||||||
|
got, err := tt.requestLog.Matches(searchExpr)
|
||||||
|
assertError(t, tt.expectedError, err)
|
||||||
|
|
||||||
|
if tt.expectedMatch != got {
|
||||||
|
t.Errorf("expected match result: %v, got: %v", tt.expectedMatch, got)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func assertError(t *testing.T, exp, got error) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
switch {
|
||||||
|
case exp == nil && got != nil:
|
||||||
|
t.Fatalf("expected: nil, got: %v", got)
|
||||||
|
case exp != nil && got == nil:
|
||||||
|
t.Fatalf("expected: %v, got: nil", exp.Error())
|
||||||
|
case exp != nil && got != nil && exp.Error() != got.Error():
|
||||||
|
t.Fatalf("expected: %v, got: %v", exp.Error(), got.Error())
|
||||||
|
}
|
||||||
|
}
|
@ -1,8 +0,0 @@
|
|||||||
package scope
|
|
||||||
|
|
||||||
import "context"
|
|
||||||
|
|
||||||
type Repository interface {
|
|
||||||
UpsertSettings(ctx context.Context, module string, settings interface{}) error
|
|
||||||
FindSettingsByModule(ctx context.Context, module string, settings interface{}) error
|
|
||||||
}
|
|
@ -1,24 +1,16 @@
|
|||||||
package scope
|
package scope
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/gob"
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/dstotijn/hetty/pkg/proj"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const moduleName = "scope"
|
|
||||||
|
|
||||||
type Scope struct {
|
type Scope struct {
|
||||||
rules []Rule
|
rules []Rule
|
||||||
repo Repository
|
mu sync.RWMutex
|
||||||
|
|
||||||
mu sync.RWMutex
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type Rule struct {
|
type Rule struct {
|
||||||
@ -32,30 +24,6 @@ type Header struct {
|
|||||||
Value *regexp.Regexp
|
Value *regexp.Regexp
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(repo Repository, projService proj.Service) *Scope {
|
|
||||||
s := &Scope{
|
|
||||||
repo: repo,
|
|
||||||
}
|
|
||||||
|
|
||||||
projService.OnProjectOpen(func(_ string) error {
|
|
||||||
err := s.load(context.Background())
|
|
||||||
if errors.Is(err, proj.ErrNoSettings) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("scope: could not load scope: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
projService.OnProjectClose(func(_ string) error {
|
|
||||||
s.unload()
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Scope) Rules() []Rule {
|
func (s *Scope) Rules() []Rule {
|
||||||
s.mu.RLock()
|
s.mu.RLock()
|
||||||
defer s.mu.RUnlock()
|
defer s.mu.RUnlock()
|
||||||
@ -63,41 +31,11 @@ func (s *Scope) Rules() []Rule {
|
|||||||
return s.rules
|
return s.rules
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scope) load(ctx context.Context) error {
|
func (s *Scope) SetRules(rules []Rule) {
|
||||||
s.mu.Lock()
|
s.mu.Lock()
|
||||||
defer s.mu.Unlock()
|
defer s.mu.Unlock()
|
||||||
|
|
||||||
var rules []Rule
|
|
||||||
|
|
||||||
err := s.repo.FindSettingsByModule(ctx, moduleName, &rules)
|
|
||||||
if errors.Is(err, proj.ErrNoSettings) {
|
|
||||||
return err
|
|
||||||
} else if err != nil {
|
|
||||||
return fmt.Errorf("scope: could not load scope settings: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
s.rules = rules
|
s.rules = rules
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Scope) unload() {
|
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
s.rules = nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Scope) SetRules(ctx context.Context, rules []Rule) error {
|
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
|
|
||||||
if err := s.repo.UpsertSettings(ctx, moduleName, rules); err != nil {
|
|
||||||
return fmt.Errorf("scope: cannot set rules in repository: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
s.rules = rules
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scope) Match(req *http.Request, body []byte) bool {
|
func (s *Scope) Match(req *http.Request, body []byte) bool {
|
||||||
@ -158,48 +96,54 @@ func (r Rule) Match(req *http.Request, body []byte) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// MarshalJSON implements json.Marshaler.
|
func regexpToString(r *regexp.Regexp) string {
|
||||||
func (r Rule) MarshalJSON() ([]byte, error) {
|
if r == nil {
|
||||||
type (
|
return ""
|
||||||
headerDTO struct {
|
|
||||||
Key string
|
|
||||||
Value string
|
|
||||||
}
|
|
||||||
ruleDTO struct {
|
|
||||||
URL string
|
|
||||||
Header headerDTO
|
|
||||||
Body string
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
dto := ruleDTO{
|
|
||||||
URL: regexpToString(r.URL),
|
|
||||||
Header: headerDTO{
|
|
||||||
Key: regexpToString(r.Header.Key),
|
|
||||||
Value: regexpToString(r.Header.Value),
|
|
||||||
},
|
|
||||||
Body: regexpToString(r.Body),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return json.Marshal(dto)
|
return r.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnmarshalJSON implements json.Unmarshaler.
|
func stringToRegexp(s string) (*regexp.Regexp, error) {
|
||||||
func (r *Rule) UnmarshalJSON(data []byte) error {
|
if s == "" {
|
||||||
type (
|
return nil, nil
|
||||||
headerDTO struct {
|
}
|
||||||
Key string
|
|
||||||
Value string
|
|
||||||
}
|
|
||||||
ruleDTO struct {
|
|
||||||
URL string
|
|
||||||
Header headerDTO
|
|
||||||
Body string
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
var dto ruleDTO
|
return regexp.Compile(s)
|
||||||
if err := json.Unmarshal(data, &dto); err != nil {
|
}
|
||||||
|
|
||||||
|
type ruleDTO struct {
|
||||||
|
URL string
|
||||||
|
Header struct {
|
||||||
|
Key string
|
||||||
|
Value string
|
||||||
|
}
|
||||||
|
Body string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r Rule) MarshalBinary() ([]byte, error) {
|
||||||
|
dto := ruleDTO{
|
||||||
|
URL: regexpToString(r.URL),
|
||||||
|
Body: regexpToString(r.Body),
|
||||||
|
}
|
||||||
|
dto.Header.Key = regexpToString(r.Header.Key)
|
||||||
|
dto.Header.Value = regexpToString(r.Header.Value)
|
||||||
|
|
||||||
|
buf := bytes.Buffer{}
|
||||||
|
|
||||||
|
err := gob.NewEncoder(&buf).Encode(dto)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return buf.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *Rule) UnmarshalBinary(data []byte) error {
|
||||||
|
dto := ruleDTO{}
|
||||||
|
|
||||||
|
err := gob.NewDecoder(bytes.NewReader(data)).Decode(&dto)
|
||||||
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -234,19 +178,3 @@ func (r *Rule) UnmarshalJSON(data []byte) error {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func regexpToString(r *regexp.Regexp) string {
|
|
||||||
if r == nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
return r.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
func stringToRegexp(s string) (*regexp.Regexp, error) {
|
|
||||||
if s == "" {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return regexp.Compile(s)
|
|
||||||
}
|
|
||||||
|
@ -1,6 +1,10 @@
|
|||||||
package search
|
package search
|
||||||
|
|
||||||
import "strings"
|
import (
|
||||||
|
"encoding/gob"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
type Expression interface {
|
type Expression interface {
|
||||||
String() string
|
String() string
|
||||||
@ -11,7 +15,7 @@ type PrefixExpression struct {
|
|||||||
Right Expression
|
Right Expression
|
||||||
}
|
}
|
||||||
|
|
||||||
func (pe *PrefixExpression) String() string {
|
func (pe PrefixExpression) String() string {
|
||||||
b := strings.Builder{}
|
b := strings.Builder{}
|
||||||
b.WriteString("(")
|
b.WriteString("(")
|
||||||
b.WriteString(pe.Operator.String())
|
b.WriteString(pe.Operator.String())
|
||||||
@ -28,7 +32,7 @@ type InfixExpression struct {
|
|||||||
Right Expression
|
Right Expression
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ie *InfixExpression) String() string {
|
func (ie InfixExpression) String() string {
|
||||||
b := strings.Builder{}
|
b := strings.Builder{}
|
||||||
b.WriteString("(")
|
b.WriteString("(")
|
||||||
b.WriteString(ie.Left.String())
|
b.WriteString(ie.Left.String())
|
||||||
@ -45,6 +49,32 @@ type StringLiteral struct {
|
|||||||
Value string
|
Value string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (sl *StringLiteral) String() string {
|
func (sl StringLiteral) String() string {
|
||||||
return sl.Value
|
return sl.Value
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type RegexpLiteral struct {
|
||||||
|
*regexp.Regexp
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rl RegexpLiteral) MarshalBinary() ([]byte, error) {
|
||||||
|
return []byte(rl.Regexp.String()), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (rl *RegexpLiteral) UnmarshalBinary(data []byte) error {
|
||||||
|
re, err := regexp.Compile(string(data))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
*rl = RegexpLiteral{re}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
gob.Register(PrefixExpression{})
|
||||||
|
gob.Register(InfixExpression{})
|
||||||
|
gob.Register(StringLiteral{})
|
||||||
|
gob.Register(RegexpLiteral{})
|
||||||
|
}
|
||||||
|
@ -2,6 +2,7 @@ package search
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"regexp"
|
||||||
)
|
)
|
||||||
|
|
||||||
type precedence int
|
type precedence int
|
||||||
@ -99,7 +100,7 @@ func ParseQuery(input string) (expr Expression, err error) {
|
|||||||
case expr == nil:
|
case expr == nil:
|
||||||
expr = right
|
expr = right
|
||||||
default:
|
default:
|
||||||
expr = &InfixExpression{
|
expr = InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: expr,
|
Left: expr,
|
||||||
Right: right,
|
Right: right,
|
||||||
@ -170,7 +171,7 @@ func (p *Parser) parseExpression(prec precedence) (Expression, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func parsePrefixExpression(p *Parser) (Expression, error) {
|
func parsePrefixExpression(p *Parser) (Expression, error) {
|
||||||
expr := &PrefixExpression{
|
expr := PrefixExpression{
|
||||||
Operator: p.cur.Type,
|
Operator: p.cur.Type,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -187,7 +188,7 @@ func parsePrefixExpression(p *Parser) (Expression, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func parseInfixExpression(p *Parser, left Expression) (Expression, error) {
|
func parseInfixExpression(p *Parser, left Expression) (Expression, error) {
|
||||||
expr := &InfixExpression{
|
expr := InfixExpression{
|
||||||
Operator: p.cur.Type,
|
Operator: p.cur.Type,
|
||||||
Left: left,
|
Left: left,
|
||||||
}
|
}
|
||||||
@ -200,13 +201,24 @@ func parseInfixExpression(p *Parser, left Expression) (Expression, error) {
|
|||||||
return nil, fmt.Errorf("could not parse expression for right operand: %w", err)
|
return nil, fmt.Errorf("could not parse expression for right operand: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if expr.Operator == TokOpRe || expr.Operator == TokOpNotRe {
|
||||||
|
if rightStr, ok := right.(StringLiteral); ok {
|
||||||
|
re, err := regexp.Compile(rightStr.Value)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("could not compile regular expression %q: %w", rightStr.Value, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
right = re
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
expr.Right = right
|
expr.Right = right
|
||||||
|
|
||||||
return expr, nil
|
return expr, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseStringLiteral(p *Parser) (Expression, error) {
|
func parseStringLiteral(p *Parser) (Expression, error) {
|
||||||
return &StringLiteral{Value: p.cur.Literal}, nil
|
return StringLiteral{Value: p.cur.Literal}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseGroupedExpression(p *Parser) (Expression, error) {
|
func parseGroupedExpression(p *Parser) (Expression, error) {
|
||||||
@ -227,7 +239,7 @@ func parseGroupedExpression(p *Parser) (Expression, error) {
|
|||||||
return nil, fmt.Errorf("could not parse expression: %w", err)
|
return nil, fmt.Errorf("could not parse expression: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
expr = &InfixExpression{
|
expr = InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: expr,
|
Left: expr,
|
||||||
Right: right,
|
Right: right,
|
||||||
|
@ -3,6 +3,7 @@ package search
|
|||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
"regexp"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -24,101 +25,101 @@ func TestParseQuery(t *testing.T) {
|
|||||||
{
|
{
|
||||||
name: "string literal expression",
|
name: "string literal expression",
|
||||||
input: "foobar",
|
input: "foobar",
|
||||||
expectedExpression: &StringLiteral{Value: "foobar"},
|
expectedExpression: StringLiteral{Value: "foobar"},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with equal operator",
|
name: "boolean expression with equal operator",
|
||||||
input: "foo = bar",
|
input: "foo = bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpEq,
|
Operator: TokOpEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with not equal operator",
|
name: "boolean expression with not equal operator",
|
||||||
input: "foo != bar",
|
input: "foo != bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpNotEq,
|
Operator: TokOpNotEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with greater than operator",
|
name: "boolean expression with greater than operator",
|
||||||
input: "foo > bar",
|
input: "foo > bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpGt,
|
Operator: TokOpGt,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with less than operator",
|
name: "boolean expression with less than operator",
|
||||||
input: "foo < bar",
|
input: "foo < bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpLt,
|
Operator: TokOpLt,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with greater than or equal operator",
|
name: "boolean expression with greater than or equal operator",
|
||||||
input: "foo >= bar",
|
input: "foo >= bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpGtEq,
|
Operator: TokOpGtEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with less than or equal operator",
|
name: "boolean expression with less than or equal operator",
|
||||||
input: "foo <= bar",
|
input: "foo <= bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpLtEq,
|
Operator: TokOpLtEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with regular expression operator",
|
name: "boolean expression with regular expression operator",
|
||||||
input: "foo =~ bar",
|
input: "foo =~ bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpRe,
|
Operator: TokOpRe,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: regexp.MustCompile("bar"),
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with not regular expression operator",
|
name: "boolean expression with not regular expression operator",
|
||||||
input: "foo !~ bar",
|
input: "foo !~ bar",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpNotRe,
|
Operator: TokOpNotRe,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: regexp.MustCompile("bar"),
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "boolean expression with AND, OR and NOT operators",
|
name: "boolean expression with AND, OR and NOT operators",
|
||||||
input: "foo AND bar OR NOT baz",
|
input: "foo AND bar OR NOT baz",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &InfixExpression{
|
Right: InfixExpression{
|
||||||
Operator: TokOpOr,
|
Operator: TokOpOr,
|
||||||
Left: &StringLiteral{Value: "bar"},
|
Left: StringLiteral{Value: "bar"},
|
||||||
Right: &PrefixExpression{
|
Right: PrefixExpression{
|
||||||
Operator: TokOpNot,
|
Operator: TokOpNot,
|
||||||
Right: &StringLiteral{Value: "baz"},
|
Right: StringLiteral{Value: "baz"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@ -127,16 +128,16 @@ func TestParseQuery(t *testing.T) {
|
|||||||
{
|
{
|
||||||
name: "boolean expression with nested group",
|
name: "boolean expression with nested group",
|
||||||
input: "(foo AND bar) OR NOT baz",
|
input: "(foo AND bar) OR NOT baz",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpOr,
|
Operator: TokOpOr,
|
||||||
Left: &InfixExpression{
|
Left: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
Right: &PrefixExpression{
|
Right: PrefixExpression{
|
||||||
Operator: TokOpNot,
|
Operator: TokOpNot,
|
||||||
Right: &StringLiteral{Value: "baz"},
|
Right: StringLiteral{Value: "baz"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
@ -144,59 +145,59 @@ func TestParseQuery(t *testing.T) {
|
|||||||
{
|
{
|
||||||
name: "implicit boolean expression with string literal operands",
|
name: "implicit boolean expression with string literal operands",
|
||||||
input: "foo bar baz",
|
input: "foo bar baz",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &InfixExpression{
|
Left: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
Right: &StringLiteral{Value: "baz"},
|
Right: StringLiteral{Value: "baz"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "implicit boolean expression nested in group",
|
name: "implicit boolean expression nested in group",
|
||||||
input: "(foo bar)",
|
input: "(foo bar)",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "implicit and explicit boolean expression with string literal operands",
|
name: "implicit and explicit boolean expression with string literal operands",
|
||||||
input: "foo bar OR baz yolo",
|
input: "foo bar OR baz yolo",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &InfixExpression{
|
Left: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &InfixExpression{
|
Right: InfixExpression{
|
||||||
Operator: TokOpOr,
|
Operator: TokOpOr,
|
||||||
Left: &StringLiteral{Value: "bar"},
|
Left: StringLiteral{Value: "bar"},
|
||||||
Right: &StringLiteral{Value: "baz"},
|
Right: StringLiteral{Value: "baz"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
Right: &StringLiteral{Value: "yolo"},
|
Right: StringLiteral{Value: "yolo"},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "implicit boolean expression with comparison operands",
|
name: "implicit boolean expression with comparison operands",
|
||||||
input: "foo=bar baz=~yolo",
|
input: "foo=bar baz=~yolo",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpAnd,
|
Operator: TokOpAnd,
|
||||||
Left: &InfixExpression{
|
Left: InfixExpression{
|
||||||
Operator: TokOpEq,
|
Operator: TokOpEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
Right: &InfixExpression{
|
Right: InfixExpression{
|
||||||
Operator: TokOpRe,
|
Operator: TokOpRe,
|
||||||
Left: &StringLiteral{Value: "baz"},
|
Left: StringLiteral{Value: "baz"},
|
||||||
Right: &StringLiteral{Value: "yolo"},
|
Right: regexp.MustCompile("yolo"),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
@ -204,17 +205,17 @@ func TestParseQuery(t *testing.T) {
|
|||||||
{
|
{
|
||||||
name: "eq operator takes precedence over boolean ops",
|
name: "eq operator takes precedence over boolean ops",
|
||||||
input: "foo=bar OR baz=yolo",
|
input: "foo=bar OR baz=yolo",
|
||||||
expectedExpression: &InfixExpression{
|
expectedExpression: InfixExpression{
|
||||||
Operator: TokOpOr,
|
Operator: TokOpOr,
|
||||||
Left: &InfixExpression{
|
Left: InfixExpression{
|
||||||
Operator: TokOpEq,
|
Operator: TokOpEq,
|
||||||
Left: &StringLiteral{Value: "foo"},
|
Left: StringLiteral{Value: "foo"},
|
||||||
Right: &StringLiteral{Value: "bar"},
|
Right: StringLiteral{Value: "bar"},
|
||||||
},
|
},
|
||||||
Right: &InfixExpression{
|
Right: InfixExpression{
|
||||||
Operator: TokOpEq,
|
Operator: TokOpEq,
|
||||||
Left: &StringLiteral{Value: "baz"},
|
Left: StringLiteral{Value: "baz"},
|
||||||
Right: &StringLiteral{Value: "yolo"},
|
Right: StringLiteral{Value: "yolo"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
expectedError: nil,
|
expectedError: nil,
|
||||||
|
Reference in New Issue
Block a user