ReVanced/ARSCLib
1
0
Fork 0
mirror of https://github.com/revanced/ARSCLib.git synced 2025-04-29 22:04:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

implement xml pp namespace features #18

Browse Source
This commit is contained in:
REAndroid 2023-05-02 16:41:33 +02:00
parent e8fff620f7
commit 0ec2b3fe1c
6 changed files with 1224 additions and 1105 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/reandroid/arsc/chunk/xml/ResXmlAttribute.java
View File

@ -127,7 +127,8 @@ public class ResXmlAttribute extends ValueItem implements AttributeValue, Compar
if(xmlElement == null){
return;
}
ResXmlStartNamespace startNamespace=xmlElement.getStartNamespaceByUriRef(getNamespaceReference());
ResXmlStartNamespace startNamespace =
xmlElement.getStartNamespaceByUriRef(getNamespaceReference());
if(startNamespace == null){
return;
}

14
src/main/java/com/reandroid/arsc/chunk/xml/ResXmlElement.java
View File

@ -553,6 +553,12 @@
public List<ResXmlStartNamespace> getStartNamespaceList(){
return mStartNamespaceList.getChildes();
}
public int getNamespaceCount(){
return mStartNamespaceList.size();
}
public ResXmlStartNamespace getNamespace(int index){
return mStartNamespaceList.get(index);
}
public void addStartNamespace(ResXmlStartNamespace item){
mStartNamespaceList.add(item);
}
@ -562,6 +568,14 @@
public void addEndNamespace(ResXmlEndNamespace item){
mEndNamespaceList.add(item);
}
void removeNamespace(ResXmlStartNamespace startNamespace){
if(startNamespace == null){
return;
}
startNamespace.onRemoved();
mStartNamespaceList.remove(startNamespace);
mEndNamespaceList.remove(startNamespace.getEnd());
}
ResXmlStartElement newStartElement(int lineNo){
ResXmlStartElement startElement=new ResXmlStartElement();

99
src/main/java/com/reandroid/arsc/chunk/xml/ResXmlPullParser.java
View File

@ -34,9 +34,13 @@ public class ResXmlPullParser implements XmlResourceParser {
private ResXmlDocument mDocument;
private boolean mDocumentCreatedHere;
private DocumentLoadedListener documentLoadedListener;
private boolean processNamespaces;
private boolean reportNamespaceAttrs;
public ResXmlPullParser(Decoder decoder){
this.mDecoder = decoder;
this.processNamespaces = true;
this.reportNamespaceAttrs = true;
}
public ResXmlPullParser(){
this(null);
@ -54,6 +58,7 @@ public class ResXmlPullParser implements XmlResourceParser {
public synchronized void setResXmlDocument(ResXmlDocument xmlDocument){
closeDocument();
this.mDocument = xmlDocument;
initDefaultFeatures();
initializeDecoder(xmlDocument);
xmlDocument.addEvents(mEventList);
}
@ -109,17 +114,33 @@ public class ResXmlPullParser implements XmlResourceParser {
@Override
public int getAttributeCount() {
ResXmlElement element = getCurrentElement();
if(element!=null){
return element.getAttributeCount();
}
if(element == null){
return 0;
}
int count = element.getAttributeCount();
if(reportNamespaceAttrs){
count += element.getNamespaceCount();
}
return count;
}
@Override
public String getAttributeName(int index) {
if(reportNamespaceAttrs){
int nsCount = getNamespaceCountInternal();
if(index < nsCount){
return getNamespaceAttributeName(index);
}
}
return decodeAttributeName(getResXmlAttributeAt(index));
}
@Override
public String getAttributeValue(int index) {
if(reportNamespaceAttrs){
int nsCount = getNamespaceCountInternal();
if(index < nsCount){
return getNamespaceAttributeValue(index);
}
}
return decodeAttributeValue(getResXmlAttributeAt(index));
}
@Override
@ -324,9 +345,22 @@ public class ResXmlPullParser implements XmlResourceParser {
@Override
public void setFeature(String name, boolean state) throws XmlPullParserException {
if(FEATURE_PROCESS_NAMESPACES.equals(name)) {
processNamespaces = state;
}else if(FEATURE_REPORT_NAMESPACE_ATTRIBUTES.equals(name)) {
reportNamespaceAttrs = state;
}else {
throw new XmlPullParserException("Unsupported feature: " + name);
}
}
@Override
public boolean getFeature(String name) {
if(FEATURE_PROCESS_NAMESPACES.equals(name)) {
return processNamespaces;
}else if(FEATURE_REPORT_NAMESPACE_ATTRIBUTES.equals(name)) {
return reportNamespaceAttrs;
}
return false;
}
@Override
@ -358,12 +392,15 @@ public class ResXmlPullParser implements XmlResourceParser {
}
@Override
public int getNamespaceCount(int depth) throws XmlPullParserException {
if(reportNamespaceAttrs){
return 0;
}
ResXmlElement element = getCurrentElement();
while(element!=null && element.getDepth()>depth){
element=element.getParentResXmlElement();
}
if(element!=null){
return element.getStartNamespaceList().size();
return element.getNamespaceCount();
}
return 0;
}
@ -371,7 +408,7 @@ public class ResXmlPullParser implements XmlResourceParser {
public String getNamespacePrefix(int pos) throws XmlPullParserException {
ResXmlElement element = getCurrentElement();
if(element!=null){
return element.getStartNamespaceList().get(pos).getPrefix();
return element.getNamespace(pos).getPrefix();
}
return null;
}
@ -379,7 +416,7 @@ public class ResXmlPullParser implements XmlResourceParser {
public String getNamespaceUri(int pos) throws XmlPullParserException {
ResXmlElement element = getCurrentElement();
if(element!=null){
return element.getStartNamespaceList().get(pos).getUri();
return element.getNamespace(pos).getUri();
}
return null;
}
@ -466,10 +503,13 @@ public class ResXmlPullParser implements XmlResourceParser {
if(element!=null){
return element.countResXmlNodes() == 0 && element.getAttributeCount()==0;
}
return false;
return true;
}
@Override
public String getAttributeNamespace(int index) {
if(processNamespaces){
return null;
}
ResXmlAttribute attribute = getResXmlAttributeAt(index);
if(attribute != null){
return attribute.getUri();
@ -478,6 +518,9 @@ public class ResXmlPullParser implements XmlResourceParser {
}
@Override
public String getAttributePrefix(int index) {
if(processNamespaces){
return null;
}
ResXmlAttribute attribute = getResXmlAttributeAt(index);
if(attribute != null){
return attribute.getNamePrefix();
@ -502,6 +545,9 @@ public class ResXmlPullParser implements XmlResourceParser {
name = attribute.getName();
}else {
name = mDecoder.decodeResourceName(attribute.getNameResourceID(), true);
if(processNamespaces){
name = attribute.getNamePrefix() + ":" + name;
}
}
return name;
}
@ -512,6 +558,7 @@ public class ResXmlPullParser implements XmlResourceParser {
return mDecoder.decodeAttributeValue(attribute);
}
public ResXmlAttribute getResXmlAttributeAt(int index){
index = getRealAttributeIndex(index);
ResXmlElement element = getCurrentElement();
if(element == null){
return null;
@ -545,6 +592,33 @@ public class ResXmlPullParser implements XmlResourceParser {
}
return null;
}
private int getRealAttributeIndex(int index){
if(reportNamespaceAttrs){
index = index - getNamespaceCountInternal();
}
return index;
}
private int getNamespaceCountInternal(){
ResXmlElement element = getCurrentElement();
if(element != null){
return element.getNamespaceCount();
}
return 0;
}
private String getNamespaceAttributeName(int index){
ResXmlStartNamespace namespace = getCurrentElement()
.getNamespace(index);
String prefix = namespace.getPrefix();
if(processNamespaces){
prefix = "xmlns:" + prefix;
}
return prefix;
}
private String getNamespaceAttributeValue(int index){
ResXmlStartNamespace namespace = getCurrentElement()
.getNamespace(index);
return namespace.getUri();
}
@Override
public int getEventType() throws XmlPullParserException {
return mEventList.getType();
@ -552,6 +626,10 @@ public class ResXmlPullParser implements XmlResourceParser {
@Override
public int next() throws XmlPullParserException, IOException {
mEventList.next();
int type = mEventList.getType();
if(type == START_TAG){
onStartTag();
}
return mEventList.getType();
}
@Override
@ -629,6 +707,13 @@ public class ResXmlPullParser implements XmlResourceParser {
this.mDocumentCreatedHere = true;
}
}
private void initDefaultFeatures(){
processNamespaces = true;
reportNamespaceAttrs = true;
}
private void onStartTag(){
}
public static interface DocumentLoadedListener{
public ResXmlDocument onDocumentLoaded(ResXmlDocument resXmlDocument);

1
src/main/java/com/reandroid/arsc/chunk/xml/ResXmlStartNamespace.java
View File

@ -24,6 +24,7 @@
public class ResXmlStartNamespace extends ResXmlNamespace {
private final Set<ResXmlAttribute> mReferencedAttributes;
public ResXmlStartNamespace() {
super(ChunkType.XML_START_NAMESPACE);
this.mReferencedAttributes = new HashSet<>();

22
src/main/java/com/reandroid/xml/XmlParserToSerializer.java
View File

@ -16,6 +16,7 @@
package com.reandroid.xml;
import android.content.res.XmlResourceParser;
import org.xmlpull.v1.XmlPullParser;
import org.xmlpull.v1.XmlPullParserException;
import org.xmlpull.v1.XmlSerializer;
@ -24,10 +25,18 @@ import java.io.IOException;
public class XmlParserToSerializer {
private final XmlSerializer serializer;
private final XmlResourceParser parser;
private boolean enableIndent;
public XmlParserToSerializer(XmlResourceParser parser, XmlSerializer serializer){
this.parser = parser;
this.serializer = serializer;
this.enableIndent = true;
}
public void setEnableIndent(boolean enableIndent) {
this.enableIndent = enableIndent;
}
public void write() throws IOException, XmlPullParserException {
XmlResourceParser parser = this.parser;
int event = parser.next();
@ -71,17 +80,24 @@ public class XmlParserToSerializer {
private void onStartTag() throws IOException, XmlPullParserException {
XmlResourceParser parser = this.parser;
XmlSerializer serializer = this.serializer;
serializer.setFeature("http://xmlpull.org/v1/doc/features.html#indent-output", true);
boolean processNamespace = parser.getFeature(XmlPullParser.FEATURE_PROCESS_NAMESPACES);
boolean reportNamespaceAttrs = parser.getFeature(XmlPullParser.FEATURE_REPORT_NAMESPACE_ATTRIBUTES);
serializer.setFeature(FEATURE_INDENT_OUTPUT, enableIndent);
if(!reportNamespaceAttrs){
int nsCount = parser.getNamespaceCount(parser.getDepth());
for(int i=0; i<nsCount; i++){
String prefix = parser.getNamespacePrefix(i);
String namespace = parser.getNamespaceUri(i);
serializer.setPrefix(prefix, namespace);
}
}
serializer.startTag(parser.getNamespace(), parser.getName());
int attrCount = parser.getAttributeCount();
for(int i=0; i<attrCount; i++){
serializer.attribute(parser.getAttributeNamespace(i),
String namespace = processNamespace ?
parser.getAttributeNamespace(i) : null;
serializer.attribute(namespace,
parser.getAttributeName(i),
parser.getAttributeValue(i));
}
@ -98,4 +114,6 @@ public class XmlParserToSerializer {
private void onEndDocument() throws IOException{
serializer.endDocument();
}
private static final String FEATURE_INDENT_OUTPUT = "http://xmlpull.org/v1/doc/features.html#indent-output";
}