Skip to content

Commit 8d0419b

Browse files
committed
Add OOS scope check box
1 parent bd12fb6 commit 8d0419b

File tree

3 files changed

+17
-14
lines changed

3 files changed

+17
-14
lines changed

src/main/kotlin/ReflectListener.kt

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,12 @@ data class ReflectedResponse(
2727
class ReflectListener(private val reflectPanel: ReflectPanel) : HttpSenderListener {
2828

2929
override fun onHttpResponseReceive(msg: HttpMessage?, initiator: Int, sender: HttpSender?) {
30-
msg?.let {
31-
if (msg.isInScope) {
30+
msg?.let { req ->
31+
if (req.isInScope || reflectPanel.scopeCheckbox.isSelected) {
3232
val params = mutableSetOf<HtmlParameter>()
33-
params.addAll(msg.urlParams)
34-
params.addAll(msg.formParams)
35-
val responseBody = msg.responseBody.toString()
33+
params.addAll(req.urlParams)
34+
params.addAll(req.formParams)
35+
val responseBody = req.responseBody.toString()
3636
val reflected = params.asSequence().filter { it.value.length >= 4 }
3737
.filter { responseBody.contains(it.value, ignoreCase = true) }.toSet()
3838
if (reflected.isNotEmpty()) {
@@ -41,17 +41,17 @@ class ReflectListener(private val reflectPanel: ReflectPanel) : HttpSenderListen
4141
val dateTime = now.format(dateFormatter) ?: ""
4242
val parameters = reflected.joinToString { "${it.name}=${it.value}" }
4343
val reqRes = ReflectedResponse(
44-
msg = msg,
44+
msg = req,
4545
dateTime = dateTime,
46-
host = msg.requestHeader?.uri?.host ?: "",
47-
url = msg.requestHeader?.uri,
48-
path = msg.requestHeader.uri.path,
49-
method = msg.requestHeader?.method ?: "",
50-
statusCode = msg.responseHeader.statusCode,
46+
host = req.requestHeader?.uri?.host ?: "",
47+
url = req.requestHeader?.uri,
48+
path = req.requestHeader.uri.path,
49+
method = req.requestHeader?.method ?: "",
50+
statusCode = req.responseHeader.statusCode,
5151
title = getTitle(responseBody),
52-
length = msg.responseHeader.contentLength,
53-
mimeType = msg.responseHeader.getHeaderValues(HttpHeader.CONTENT_TYPE).toString(),
54-
protocol = msg.requestHeader.uri.scheme,
52+
length = req.responseHeader.contentLength,
53+
mimeType = req.responseHeader.getHeaderValues(HttpHeader.CONTENT_TYPE).toString(),
54+
protocol = req.requestHeader.uri.scheme,
5555
parameters = parameters
5656
)
5757
reflectPanel.addReflection(reqRes)

src/main/kotlin/ReflectOptions.kt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ class ReflectOptions(
1010
private val loadPanel = JPanel(FlowLayout(FlowLayout.RIGHT))
1111
private val searchBar = JTextField("", 20)
1212
private val searchPanel = JPanel(FlowLayout(FlowLayout.LEFT))
13+
val scopeCheckBox = JCheckBox("Collect out of scope requests")
1314

1415
init {
1516
val clearButton = JButton("Clear Reflections")
@@ -25,6 +26,7 @@ class ReflectOptions(
2526
searchPanel.add(searchButton)
2627
searchPanel.add(resetButton)
2728
loadPanel.add(clearButton)
29+
searchPanel.add(scopeCheckBox)
2830
panel.leftComponent = searchPanel
2931
panel.rightComponent = loadPanel
3032
panel.dividerSize = 0

src/main/kotlin/ReflectPanel.kt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ class ReflectPanel : AbstractPanel() {
1616
val table = JTable(model)
1717
val reflections = model.reflections
1818
private val reflectionOptions = ReflectOptions(this)
19+
val scopeCheckbox = reflectionOptions.scopeCheckBox
1920
private val panel = JSplitPane(JSplitPane.VERTICAL_SPLIT)
2021

2122
init {

0 commit comments

Comments
 (0)